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

regression tested

Browse source
This commit is contained in:
prestonrasmussen 2025-09-07 22:53:42 -04:00
parent 84092a706e
commit ddd6660638
2 changed files with 18 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
graphiti_core/models/edges/edge_db_queries.py
View file

@ -107,10 +107,11 @@ def get_entity_edge_save_query(provider: GraphProvider, has_aoss: bool = False)
return (
(
"""
MATCH (source:Entity {uuid: $edge_data.source_uuid})
MATCH (target:Entity {uuid: $edge_data.target_uuid})
MERGE (source)-[e:RELATES_TO {uuid: $edge_data.uuid}]->(target)
SET e = $edge_data"""
MATCH (source:Entity {uuid: $edge_data.source_uuid})
MATCH (target:Entity {uuid: $edge_data.target_uuid})
MERGE (source)-[e:RELATES_TO {uuid: $edge_data.uuid}]->(target)
SET e = $edge_data
"""
+ save_embedding_query
)
+ """
@ -169,11 +170,12 @@ def get_entity_edge_save_bulk_query(provider: GraphProvider, has_aoss: bool = Fa
)
return (
"""
UNWIND $entity_edges AS edge
MATCH (source:Entity {uuid: edge.source_node_uuid})
MATCH (target:Entity {uuid: edge.target_node_uuid})
MERGE (source)-[e:RELATES_TO {uuid: edge.uuid}]->(target)
SET e = edge"""
UNWIND $entity_edges AS edge
MATCH (source:Entity {uuid: edge.source_node_uuid})
MATCH (target:Entity {uuid: edge.target_node_uuid})
MERGE (source)-[e:RELATES_TO {uuid: edge.uuid}]->(target)
SET e = edge
"""
+ save_embedding_query
+ """
RETURN edge.uuid AS uuid

12
graphiti_core/models/nodes/node_db_queries.py
View file

@ -170,7 +170,8 @@ def get_entity_node_save_query(provider: GraphProvider, labels: str, has_aoss: b
f"""
MERGE (n:Entity {{uuid: $entity_data.uuid}})
SET n:{labels}
SET n = $entity_data"""
SET n = $entity_data
"""
+ save_embedding_query
+ """
RETURN n.uuid AS uuid
@ -239,10 +240,11 @@ def get_entity_node_save_bulk_query(
)
return (
"""
UNWIND $nodes AS node
MERGE (n:Entity {uuid: node.uuid})
SET n:$(node.labels)
SET n = node"""
UNWIND $nodes AS node
MERGE (n:Entity {uuid: node.uuid})
SET n:$(node.labels)
SET n = node
"""
+ save_embedding_query
+ """
RETURN n.uuid AS uuid