apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-25 00:16:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1
trustgraph/trustgraphETL.py

719 lines
24 KiB
Python
Raw Normal View History

Trustgraph, first drop of code
2024-07-10 17:04:24 +01:00
# import boto3
import json
import re
import rdflib
import time
from langchain_community.document_loaders import PyPDFLoader
from langchain_community.llms import Ollama
from mistralai.client import MistralClient
from mistralai.models.chat_completion import ChatMessage
llm = Ollama(
base_url="http://monster:11434",
# model="deepseek-v2"
# model="phi3:14b"
model="gemma2"
)
api_key = open("mistral-ai.key").read().strip()
client = MistralClient(api_key=api_key)
model = "open-mixtral-8x7b"
# from llama_index.core import Document
# from llama_index.core.node_parser import SentenceSplitter
# from unisim import TextSim
# bedrock = boto3.client(service_name='bedrock-runtime', region_name="us-west-2")
exampleXML = f"""## RDF/XML Examples:
```xml
<?xml version="1.0"?>
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
xmlns:dc="http://purl.org/dc/elements/1.1/"
xmlns:ex="http://example.org/stuff/1.0/">
<rdf:Description rdf:about="http://www.w3.org/TR/rdf-syntax-grammar"
dc:title="RDF1.1 XML Syntax">
<ex:editor>
<rdf:Description ex:fullName="Dave Beckett">
<ex:homePage rdf:resource="http://purl.org/net/dajobe/" />
</rdf:Description>
</ex:editor>
</rdf:Description>
</rdf:RDF>
```
```xml
<?xml version="1.0" encoding="utf-8"?>
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
xmlns:dc="http://purl.org/dc/elements/1.1/">
<rdf:Description rdf:about="http://www.w3.org/TR/rdf-syntax-grammar">
<dc:title>RDF 1.1 XML Syntax</dc:title>
<dc:title xml:lang="en">RDF 1.1 XML Syntax</dc:title>
<dc:title xml:lang="en-US">RDF 1.1 XML Syntax</dc:title>
</rdf:Description>
<rdf:Description rdf:about="http://example.org/buecher/baum" xml:lang="de">
<dc:title>Der Baum</dc:title>
<dc:description>Das Buch ist außergewöhnlich</dc:description>
<dc:title xml:lang="en">The Tree</dc:title>
</rdf:Description>
</rdf:RDF>
```
```xml
<?xml version="1.0"?>
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
xmlns:ex="http://example.org/stuff/1.0/">
<rdf:Description rdf:about="http://example.org/item01">
<ex:prop rdf:parseType="Literal" xmlns:a="http://example.org/a#">
<a:Box required="true">
<a:widget size="10" />
<a:grommit id="23" />
</a:Box>
</ex:prop>
</rdf:Description>
</rdf:RDF>
```
## RDF/XML Instructions:
1. All valid RDF/XML shall written with the following first line: <?xml version="1.0"?>
2. The subject of a RDF triple must be a URI, Uniform Resource Identifier.
3. The predicate of a RDF triple must be a URI, Uniform Resource Identier.
4. All URIs must use the following base URL: http://example.org/
5. Predicates should be written using knowledge schema using the SKOS, Simple Knowledge Organization System, approach.
6. The object of a RDF triple can be a URI or a value.
7. If the object of a RDF triple is a value, the object should have a XSD type.
"""
exampleTurtle = """Knowledge is written between <<<>>>>. That knowledge will be rewritten as RDF Turtle.
<<<Fred lives with Hope. Fred is a cat. Fred has 4 legs.>>>
Here's the RDF Turtle for the knowledge
```turtle
@prefix rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#> .
@prefix animal: <http://example.org/animal/> .
@prefix prop: <http://example.org/property/> .
@prefix type: <http://example.org/type/> .
@prefix rdfs: <http://www.w3.org/2000/01/rdf-schema#> .
animal:fred
prop:has-legs 4 ;
prop:lives-with animal:hope ;
a type:cat ;
rdfs:label "Fred" .
animal:hope
prop:has-legs 4 ;
prop:lives-with animal:fred ;
a type:cat ;
rdfs:label "Hope" .
prop:has-legs
a rdfs:Property ;
rdfs:label "has legs" .
prop:lives-with
a rdfs:Property ;
rdfs:label "lives with" .
type:cat
a rdfs:Class ;
rdfs:label "cat" .
```
# RDF Turtle Rules
1. The object of a RDF triple can be a URI or a value.
2. If the object of a RDF triple is a value, the object should have a XSD type.
3. All triples must have a subject, predicate, and object.
4. A subject shall always be followed by a predicate and object.
5. When a subject has multiple sets of predicates and objects, use a semicolon character to separate each set of predicates and objects.
6. When a subject and predicate have multiple objects, use a comma character to separate the set of objects.
7. Use the 'a' character in place of the predicate 'is'.
8. Use @prefix rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#> .
9. Use @prefix rdfs: <http://www.w3.org/2000/01/rdf-schema#> .
10. Use @prefix xsd: <http://www.w3.org/2001/XMLSchema#> .
"""
# Build the prompt for Article style extraction
def scholar(chunk):
promptscholar = f"""[INST]You are a scholar. You have profound knowledge of the provided text. Your task is to read the provided text and write an article to fully explain the provided text.
# Provided Text:
{chunk}
# Instructions:
## Focus on Concepts
The article should focus on concepts in the provided text. The article should be factually accurate. Do not write any concepts not found in the provided text. Do not speculate.
## Completeness
The article should capture all topics the reader will need to understand the concepts found in the provided text. Describe all terms, definitions, entities, people, events, concepts, conceptual relationships, and any other topics necessary for the reader to understand the concepts of the provided text.
[/INST]
"""
return promptscholar
# Build the prompt for the custom RDF/XML schema
def schema(article):
promptschema = f"""[INST]You are a scholar. You have profound understanding of knowledge schemas that are parsed into RDF knowledge graphs. Your task is to read the provided article and write a complete RDF/XML document for the provided article. You will respond only with the RDF/XML document. Do not provide explanations. Obey all RDF/XML instructions.
# Provided Article:
{article}
# Example RDF/XML and Instructions:
{exampleXML}
[/INST]
"""
return promptschema
# Build the prompt for the SKOS style RDF/XML schema
def skos(article):
promptconcepts = f"""[INST]You are a scholar. You have profound understanding of knowledge schemas that are parsed into RDF knowledge graphs. Your task is to read the provided article and write a complete RDF/XML document for the provided article that uses SKOS, Simple Knowledge Organization System, to structure the conceptual relationships in the article. You will respond only with the RDF/XML document. Do not provide explanations. Obey all RDF/XML instructions.
# Provided Article:
{article}
# RDF/XML Example and Instructions:
{exampleXML}
[/INST]
"""
return promptconcepts
# Add the outputs from the first pass to stabilize the build process
def example_follow(x1, x2):
exampleXML = f"""## RDF/XML Examples:
```xml
<?xml version="1.0"?>
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
xmlns:dc="http://purl.org/dc/elements/1.1/"
xmlns:ex="http://example.org/stuff/1.0/">
<rdf:Description rdf:about="http://www.w3.org/TR/rdf-syntax-grammar"
dc:title="RDF1.1 XML Syntax">
<ex:editor>
<rdf:Description ex:fullName="Dave Beckett">
<ex:homePage rdf:resource="http://purl.org/net/dajobe/" />
</rdf:Description>
</ex:editor>
</rdf:Description>
</rdf:RDF>
```
```xml
<?xml version="1.0" encoding="utf-8"?>
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
xmlns:dc="http://purl.org/dc/elements/1.1/">
<rdf:Description rdf:about="http://www.w3.org/TR/rdf-syntax-grammar">
<dc:title>RDF 1.1 XML Syntax</dc:title>
<dc:title xml:lang="en">RDF 1.1 XML Syntax</dc:title>
<dc:title xml:lang="en-US">RDF 1.1 XML Syntax</dc:title>
</rdf:Description>
<rdf:Description rdf:about="http://example.org/buecher/baum" xml:lang="de">
<dc:title>Der Baum</dc:title>
<dc:description>Das Buch ist außergewöhnlich</dc:description>
<dc:title xml:lang="en">The Tree</dc:title>
</rdf:Description>
</rdf:RDF>
```
```xml
<?xml version="1.0"?>
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
xmlns:ex="http://example.org/stuff/1.0/">
<rdf:Description rdf:about="http://example.org/item01">
<ex:prop rdf:parseType="Literal" xmlns:a="http://example.org/a#">
<a:Box required="true">
<a:widget size="10" />
<a:grommit id="23" />
</a:Box>
</ex:prop>
</rdf:Description>
</rdf:RDF>
```
```xml
{x1}
```
```xml
{x2}
```
## RDF/XML Instructions:
1. All valid RDF/XML shall written with the following first line: <?xml version="1.0"?>
2. The subject of a RDF triple must be a URI, Uniform Resource Identifier.
3. The predicate of a RDF triple must be a URI, Uniform Resource Identier.
4. All URIs must use the following base URL: http://example.org/
5. Predicates should be written using knowledge schema using the SKOS, Simple Knowledge Organization System, approach.
6. The object of a RDF triple can be a URI or a value.
7. If the object of a RDF triple is a value, the object should have a XSD type.
"""
return exampleXML
# Build the follow prompt for the custom RDF/XML schema
def schema_follow(article, example_follow):
promptschema = f"""[INST]You are a scholar. You have profound understanding of knowledge schemas that are parsed into RDF knowledge graphs. Your task is to read the provided article and write a complete RDF/XML document for the provided article. You will respond only with the RDF/XML document. Do not provide explanations. Obey all RDF/XML instructions.
# Provided Article:
{article}
# Example RDF/XML and Instructions:
{example_follow}
[/INST]
"""
return promptschema
# Build the follow prompt for the SKOS style RDF/XML schema
def skos_follow(article, example_follow):
promptconcepts = f"""[INST]You are a scholar. You have profound understanding of knowledge schemas that are parsed into RDF knowledge graphs. Your task is to read the provided article and write a complete RDF/XML document for the provided article that uses SKOS, Simple Knowledge Organization System, to structure the conceptual relationships in the article. You will respond only with the RDF/XML document. Do not provide explanations. Obey all RDF/XML instructions.
# Provided Article:
{article}
# RDF/XML Example and Instructions:
{example_follow}
[/INST]
"""
return promptconcepts
def p1(chunk):
promptcustom = f"""[INST]You write knowledge schemas that are parsed into RDF knowledge graphs. You read the provided text and develop a knwoledge schema to represent the concepts. Your task is to read the provided text and write a RDF Turtle schema for the concepts found in the provided text. You will respond only with the RDF Turtle schema. Do not provide explanations. Obey all of the provided RDF Turtle rules.
# RDF Turtle Example and Rules
{exampleTurtle}
# Provided Text
{chunk}
[/INST]
"""
return promptcustom
def p2(chunk):
promptconcepts = f"""[INST]You write knowledge schemas that are parsed into RDF knowledge graphs. You read the provided text and develop a knwoledge schema to represent the conceptual relationships in the text. Your task is to read the provided text and write a RDF Turtle schema for the concepts found in the provided text using SKOS, Simple Knowledge Organization System. You will respond only with the RDF Turtle schema. Do not provide explanations. Obey all of the provided RDF Turtle rules.
# Valid RDF Turtle Example and Rules
{exampleTurtle}
# Provided Text
{chunk}
[/INST]
"""
return promptconcepts
def summaryprompt(chunk):
promptsummary = f"""[INST]You are an analyst. You read complex text and write a detailed summary. Your task is to read the provided text and write a thorough summary that captures all of the key points.
# Provided text:
{chunk}
# Instructions:
## Summarize:
In clear and concise language, summarize the key points in the provided text.
## Key Points:
It is important to capture unique terms and their definitions, entities, people, events, concepts, conceptual relationships, and relationships between entities and people.
## Requirements:
Include all key points in the summary. Do not omit any facts that might be considered a key point.
[/INST]
"""
return promptsummary
def extractpdf(path):
loader = PyPDFLoader(path)
pages = loader.load_and_split()
pdf = f""
for page in pages:
content = page.page_content
pdf += "".join(content)
return pdf
def chunktext(splitpdf, chunk_size, chunk_overlap):
node_parser = SentenceSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
nodes = node_parser.get_nodes_from_documents([Document(text=splitpdf)], show_progress=False)
textlist = []
for node in nodes:
textnode = node.text
textlist.append(textnode)
return textlist
def OLDcallmixtral(prompt):
chat_response = client.chat(
model=model,
messages=[
ChatMessage(
role="user", content=prompt
)
]
)
print(chat_response)
resp = chat_response.choices[0].message.content
print(resp)
return resp
def callmixtral(prompt):
print(prompt)
resp = llm.invoke(prompt)
print(resp)
return resp
def extract_xml(text):
cleantext = replace_ampersand(text)
pattern = r'<\?xml version="1\.0"\?>.*?</rdf:RDF>'
match = re.search(pattern, cleantext, re.DOTALL)
if match:
return match.group()
else:
return None
def replace_ampersand(text):
return text.replace('&', 'and')
def convert_summary(dataset):
num_passes = len(dataset)
print("Beginning Converting to Summaries")
start_time = time.time() # Start the timer
summaries = []
for i in range(5):
print(f"Begin Pass {i+1}")
promptchunk = dataset[i]
prompt = summaryprompt(promptchunk)
output = callmixtral(prompt)
summaries.append(output)
print(f"Finished Pass {i+1}")
end_time = time.time() # End the timer
total_time = end_time - start_time # Calculate the total time
print(f"Total time taken: {total_time:.2f} seconds")
return summaries
def extract_articles(dataset):
num_passes = len(dataset)
print("Beginning Converting to Articles")
start_time = time.time() # Start the timer
articles = []
# Only Doing 5 chunks for now...
for i in range(5):
print(f"Begin Pass {i+1}")
promptchunk = dataset[i]
prompt = scholar(promptchunk) #Build the prompt
output = callmixtral(prompt)
articles.append(output)
print(f"Finished Pass {i+1}")
end_time = time.time() # End the timer
total_time = end_time - start_time # Calculate the total time
print(f"Total time taken: {total_time:.2f} seconds")
return articles
def full_extract(dataset):
g = rdflib.Graph()
num_passes = len(dataset)
print(f"Initializing Graph and beginning extraction:")
start_time = time.time() # Start the timer
for i in range(num_passes):
print(f"Begin Pass {i+1}")
output1 = ""
output2 = ""
promptchunk = dataset[i]
prompt1 = p1(promptchunk)
prompt2 = p2(promptchunk)
output1 = callmixtral(prompt1)
# Extract text between ### characters from output1
match1 = re.search(r'###(.*?)###', output1, re.DOTALL)
if match1:
data1 = match1.group()
g.parse(data=data1, format="turtle")
print(f"Complete first half of Pass {i+1} from ###")
else:
g.parse(data=output1, format="turtle")
print(f"Complete first half of Pass {i+1} raw")
output2 = callmixtral(prompt2)
# Extract text between ### characters from output2
match2 = re.search(r'###(.*?)###', output2, re.DOTALL)
if match2:
data2 = match2.group()
g.parse(data=data2, format="turtle")
print(f"Completed Pass {i+1} from ###")
else:
g.parse(data=output2, format="turtle")
print(f"Complete Pass {i+1} raw")
running_length = len(g)
print(f"Pass {i+1} Graph Length: {running_length}")
end_time = time.time() # End the timer
total_time = end_time - start_time # Calculate the total time
print("Holy shit! This actually worked?!?!?!")
print(f"Total time taken: {total_time:.2f} seconds")
return g
def firstpass_build(dataset):
firstpass = []
out1 = ""
out2 = ""
chunk = dataset[0]
p1 = schema(chunk)
p2 = skos(chunk)
out1 = callmixtral(p1)
out2 = callmixtral(p2)
xml1 = extract_xml(out1)
xml2 = extract_xml(out2)
firstpass.append(xml1)
firstpass.append(xml2)
return firstpass
def build_graph(dataset):
start_time = time.time() # Start the timer
#Build the first set of outputs
firstset = []
firstset = firstpass_build(dataset)
example1 = firstset[0]
example2 = firstset[1]
#Build the new set of examples using first pass outputs
newexamples = example_follow(example1, example2)
print(f"Initializing Graph:")
g = rdflib.Graph()
# Parse the outputs from the first pass function
g.parse(data=firstset[0], format="xml")
g.parse(data=firstset[1], format="xml")
print(f"Parsed First Pass, Graph Length: {len(g)}")
num_passes = len(dataset)
for i in range(1, num_passes, 1):
output1 = ""
output2 = ""
promptchunk = dataset[i]
prompt1 = schema_follow(promptchunk, newexamples) #Build the schema follow prompt
prompt2 = skos_follow(promptchunk, newexamples) #Build the SKOS follow prompt
# Get schema style output
output1 = callmixtral(prompt1)
# Extract text between ```xml``` characters from output1
match1 = extract_xml(output1)
if match1:
data1 = match1
g.parse(data=data1, format="xml")
# Get SKOS style output
output2 = callmixtral(prompt2)
# Extract text between ```xml``` characters from output2
match2 = extract_xml(output2)
if match2:
data2 = match2
g.parse(data=data2, format="xml")
running_length = len(g)
print(f"Parsed Pass {i+1}, Graph Length: {running_length}")
end_time = time.time() # End the timer
total_time = end_time - start_time # Calculate the total time
outputpath = f"../Columbia_graph_extract.ttl"
g.serialize(destination=outputpath, format="turtle")
print("Holy shit! This actually worked?!?!?!")
print(f"Total time taken: {total_time:.2f} seconds")
return
# Graph Building with some error handling and return the graph
def build_graph_robust(dataset):
start_time = time.time() # Start the timer
#Build the first set of outputs
firstset = []
firstset = firstpass_build(dataset)
example1 = firstset[0]
example2 = firstset[1]
#Build the new set of examples using first pass outputs
newexamples = example_follow(example1, example2)
print(f"Initializing Graph:")
g = rdflib.Graph()
# Parse the outputs from the first pass function
try:
g.parse(data=firstset[0], format="xml")
g.parse(data=firstset[1], format="xml")
print(f"Parsed First Pass, Graph Length: {len(g)}")
except Exception as e:
print(f"Error parsing first pass outputs: {str(e)}")
return None
num_passes = len(dataset)
i = 1
while i < num_passes:
output1 = ""
output2 = ""
promptchunk = dataset[i]
prompt1 = schema_follow(promptchunk, newexamples) #Build the schema follow prompt
prompt2 = skos_follow(promptchunk, newexamples) #Build the SKOS follow prompt
# Get schema style output
output1 = callmixtral(prompt1)
# Extract text between ```xml``` characters from output1
match1 = extract_xml(output1)
if match1:
data1 = match1
try:
g.parse(data=data1, format="xml")
except Exception as e:
print(f"Error parsing schema output in pass {i+1}: {str(e)}\n\nTrying Again...")
continue
# Get SKOS style output
output2 = callmixtral(prompt2)
# Extract text between ```xml``` characters from output2
match2 = extract_xml(output2)
if match2:
data2 = match2
try:
g.parse(data=data2, format="xml")
except Exception as e:
print(f"Error parsing SKOS output in pass {i+1}: {str(e)}\n\nTrying Again...")
continue
running_length = len(g)
print(f"Parsed Pass {i+1}, Graph Length: {running_length}")
i += 1
end_time = time.time() # End the timer
total_time = end_time - start_time # Calculate the total time
# outputpath = f"./Columbia_graph_extract.ttl"
# g.serialize(destination=outputpath, format="turtle")
print("Holy shit! This actually worked?!?!?!")
print(f"Total time taken: {total_time:.2f} seconds")
return g
# Get a list of subjects from graph as URIs
def get_subject_list(graph):
query = """
SELECT DISTINCT ?subject
WHERE {
?subject ?predicate ?object .
}
"""
subjects = graph.query(query)
subjectlist = []
for row in subjects:
text = f"{row.subject}"
subjectlist.append(text)
return subjectlist
# Function for removing URI artifacts to return clean text
def clean_resource(resource):
# Remove the URL prefixes
cleaned_resource = resource.replace("http://example.org/", "")
cleaned_resource = cleaned_resource.replace("http://www.w3.org/2004/02/skos/core#", "")
cleaned_resource = cleaned_resource.replace("http://www.w3.org/1999/02/22-rdf-syntax-ns#", "")
cleaned_resource = cleaned_resource.replace("http://www.w3.org/2001/XMLSchema#", "")
cleaned_resource = cleaned_resource.replace("http://purl.org/dc/elements/1.1/", "")
# Replace special symbols with a space
cleaned_resource = re.sub(r'[_#/]', ' ', cleaned_resource)
return cleaned_resource
# Clean the entire subject list
def clean_subject_list(subjectlist):
cleaned_subjects = []
for subject in subjectlist:
clean = f"{clean_resource(subject)}"
cleaned_subjects.append(clean)
return cleaned_subjects
# Dump all the triples from the Graph
def get_triples_list(graph):
triples_list = list(graph.triples((None, None, None)))
semantic_list = []
for subject, predicate, object in triples_list:
triple = f"{subject} {predicate} {object}"
semantic_list.append(triple)
return semantic_list
#Clean the entire triples list
def clean_triples_list(tripleslist):
cleaned_triples = []
for triple in tripleslist:
cleaned_triple = clean_resource(triple)
cleaned_triples.append(cleaned_triple)
return cleaned_triples
# Build UniSim instance for graph subjects
def build_subject_Sim(graph):
subject_sim = TextSim()
subject_sim.reset_index() # Make sure it's empty
subject_sim = TextSim(
store_data=True, # set to False for large datasets to save memory
index_type="exact", # set to "approx" for large datasets to use ANN search
batch_size=128, # increasing batch_size on GPU may be faster
use_accelerator=True, # uses GPU if available, otherwise uses CPU
)
subjects = get_subject_list(graph) # Get the subjects
clean_subjects = clean_subject_list(subjects) # Clean the subjects
idx = subject_sim.add(clean_subjects) # Build sim
return subject_sim
# Build UniSim instance for graph triples
def build_triples_Sim(graph):
triples_sim = TextSim()
triples_sim.reset_index() # Make sure it's empty
triples_sim = TextSim(
store_data=True, # set to False for large datasets to save memory
index_type="exact", # set to "approx" for large datasets to use ANN search
batch_size=128, # increasing batch_size on GPU may be faster
use_accelerator=True, # uses GPU if available, otherwise uses CPU
)
triples = get_triples_list(graph) # Get the triples
clean_triples = clean_triples_list(triples) # Clean the triples
idx = triples_sim.add(clean_triples) # Build sim
return triples_sim
Reference in a new issue Copy permalink