gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding neo4j

Browse Source
This commit is contained in:
Ken Wiltshire
2024-10-25 17:45:59 -04:00
parent 4ef705b13f
commit 6e216abee6
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lightrag/storage.py
View File

@@ -395,7 +395,17 @@ class Neo4JStorage(BaseGraphStorage):
# edge_degree # edge_degree
# from neo4j import GraphDatabase # from neo4j import GraphDatabase
async def edge_degree(self, src_id: str, tgt_id: str) -> int:
entity_name__label_source = src_id
entity_name_label_target = tgt_id
with graph_db.session() as session:
result = session.run(
"""MATCH (n1:{node_label1})-[r]-(n2:{node_label2})
RETURN count(r) AS degree"""
.format(node_label1=node_label1, node_label2=node_label2)
)
record = result.single()
return record["degree"]
# driver = GraphDatabase.driver("bolt://localhost:7687", auth=("neo4j", "password")) # driver = GraphDatabase.driver("bolt://localhost:7687", auth=("neo4j", "password"))
# #