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

Memgraph fix (#1062)

Browse source 
<!-- .github/pull_request_template.md -->

## Description
Updated get_edges function so the output id matches internal ID

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.

---------

Co-authored-by: github-actions[bot] <github-actions@users.noreply.github.com>
Co-authored-by: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
This commit is contained in:
Matea Pesic 2025-07-08 20:00:11 +02:00 • committed by GitHub
parent 3a6e973c35
commit c600eb7a56
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -155,7 +155,7 @@ class MemgraphAdapter(GraphDBInterface):
MERGE (node {id: $node_id}) MERGE (node {id: $node_id})
ON CREATE SET node:$node_label, node += $properties, node.updated_at = timestamp() ON CREATE SET node:$node_label, node += $properties, node.updated_at = timestamp()
ON MATCH SET node:$node_label, node += $properties, node.updated_at = timestamp() ON MATCH SET node:$node_label, node += $properties, node.updated_at = timestamp()
RETURN ID(node) AS internal_id,node.id AS nodeId RETURN ID(node) AS internal_id, node.id AS nodeId
""" """
params = { params = {
@ -845,8 +845,8 @@ class MemgraphAdapter(GraphDBInterface):
result = await self.query(query) result = await self.query(query)
edges = [ edges = [
( (
record["properties"]["source_node_id"], record["source"],
record["properties"]["target_node_id"], record["target"],
record["type"], record["type"],
record["properties"], record["properties"],
) )