dosubot[bot]
commented
7 months ago 🤖
Hello @JinSeoung-Oh! Good to see you again. I appreciate your continued interest in LlamaIndex.
Sure, you can add a new relationship to a persisted KnowledgeGraph using the upsert_triplet method. This method is part of the Neo4jGraphStore class in LlamaIndex. Here's an example of how you can use it:
graph_store = Neo4jGraphStore(username="your_username", password="your_password", url="your_neo4j_url")
graph_store.upsert_triplet("subject_node_id", "relationship_type", "object_node_id")In this example, replace "subject_node_id", "relationship_type", and "object_node_id" with the IDs of the subject and object nodes and the type of the relationship you want to add, respectively.
The upsert_triplet method constructs a Cypher query to add the new relationship and executes it in a Neo4j session. The MERGE keyword in the query ensures that the nodes and the relationship are created if they do not exist, or matched if they do exist.
Here's the code that defines the upsert_triplet method:
def upsert_triplet(self, subj: str, rel: str, obj: str) -> None:
"""Add triplet."""
query = """
MERGE (n1:`%s` {id:$subj})
MERGE (n2:`%s` {id:$obj})
MERGE (n1)-[:`%s`]->(n2)
"""
prepared_statement = query % (
self.node_label,
self.node_label,
rel.replace(" ", "_").upper(),
)
with self._driver.session(database=self._database) as session:
session.run(prepared_statement, {"subj": subj, "obj": obj})You can find this code in the llama_index/graph_stores/neo4j.py file.
I hope this helps! If you have any more questions, feel free to ask.
Best, Dosu
Sources
- [Question]: How can I add new [node-relationship-node] in Graph Store with llama_index
- Adding documents to a Composible graph after creating the graph
- llama_index/graph_stores/neo4j.py
This response is meant to be useful and save you time. It is not meant to be a precise solution, but rather a starting point for your own research.
Help me be more useful! Please leave a 👍 if this is helpful and 👎 if it is irrelevant.
If you want to continue the conversation, start your reply with @dosu-bot.
JinSeoung-Oh
Question Validation
Question
Hi, on this time, I want to add new relationship to a persisted KnowledgeGraph It is easy to find about VectorDB, but about KnowledgeGraph is so hard.
I already know how to add new relationship to knowledgeGraphIndex class, but I cannot find how to add new relationship to a persisted KG
To load persisted KG, I used this code
storage_context = StorageContext.from_defaults( docstore = SimpleDocumentStore.from_persist_dir(persist_dir=os.getcwd() + "/kg/graph/test"), vector_store = SimpleVectorStore.from_persist_dir(persist_dir=os.getcwd() + "/kg/graph/test"), index_store = SimpleIndexStore.from_persist_dir(persist_dir=os.getcwd() + "/kg/graph/test"))
from llama_index import load_index_from_storage, load_indices_from_storage, load_graph_from_storage
index = load_index_from_storage(storage_context) indices = load_indices_from_storage(storage_context) graph = load_graph_from_storage(storage_context)
Can I apply upsert method on index? or graph?
Thanks!