gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update memgraph integration

Browse source
This commit is contained in:
matea16 2025-06-17 15:29:33 +02:00
parent ce9173c598
commit 5ede210f12
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
cognee/infrastructure/databases/graph/memgraph/memgraph_adapter.py
View file

@ -376,12 +376,22 @@ class MemgraphAdapter(GraphDBInterface):
The result of the edge addition operation, including relationship details. The result of the edge addition operation, including relationship details.
""" """
exists = await asyncio.gather(
self.has_node(str(from_node)),
self.has_node(str(to_node))
)
if not all(exists):
return None
serialized_properties = self.serialize_properties(edge_properties or {}) serialized_properties = self.serialize_properties(edge_properties or {})
query = dedent( query = dedent(
f"""\ f"""\
MATCH (from_node {{id: $from_node}}), MATCH (from_node {{id: $from_node}}),
(to_node {{id: $to_node}}) (to_node {{id: $to_node}})
WHERE from_node IS NOT NULL AND to_node IS NOT NULL
MERGE (from_node)-[r:{relationship_name}]->(to_node) MERGE (from_node)-[r:{relationship_name}]->(to_node)
ON CREATE SET r += $properties, r.updated_at = timestamp() ON CREATE SET r += $properties, r.updated_at = timestamp()
ON MATCH SET r += $properties, r.updated_at = timestamp() ON MATCH SET r += $properties, r.updated_at = timestamp()