yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-12-25 22:02:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

As Graph edges should be treated undirectional, fix incorrect upsert_edge method in MongoDBGraph

Browse Source
This commit is contained in:
Ken Chen 2025-06-28 21:03:54 +08:00
parent 5116d61eaa
commit 4a953d6829
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
lightrag/kg/mongo_impl.py
View File

@ -663,7 +663,7 @@ class MongoGraphStorage(BaseGraphStorage):
self, source_node_id: str, target_node_id: str, edge_data: dict[str, str]
) -> None:
"""
Upsert an edge from source_node_id -> target_node_id with optional 'relation'.
Upsert an edge between source_node_id and target_node_id with optional 'relation'.
If an edge with the same target exists, we remove it and re-insert with updated data.
"""
# Ensure source node exists
@ -675,8 +675,22 @@ class MongoGraphStorage(BaseGraphStorage):
GRAPH_FIELD_SEP
)
edge_data["source_node_id"] = source_node_id
edge_data["target_node_id"] = target_node_id
await self.edge_collection.update_one(
{"source_node_id": source_node_id, "target_node_id": target_node_id},
{
"$or": [
{
"source_node_id": source_node_id,
"target_node_id": target_node_id,
},
{
"source_node_id": target_node_id,
"target_node_id": source_node_id,
},
]
},
update_doc,
upsert=True,
)