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 ( return (
( (
""" """
MATCH (source:Entity {uuid: $edge_data.source_uuid}) MATCH (source:Entity {uuid: $edge_data.source_uuid})
MATCH (target:Entity {uuid: $edge_data.target_uuid}) MATCH (target:Entity {uuid: $edge_data.target_uuid})
MERGE (source)-[e:RELATES_TO {uuid: $edge_data.uuid}]->(target) MERGE (source)-[e:RELATES_TO {uuid: $edge_data.uuid}]->(target)
SET e = $edge_data""" SET e = $edge_data
"""
+ save_embedding_query + save_embedding_query
) )
+ """ + """
@ -169,11 +170,12 @@ def get_entity_edge_save_bulk_query(provider: GraphProvider, has_aoss: bool = Fa
) )
return ( return (
""" """
UNWIND $entity_edges AS edge UNWIND $entity_edges AS edge
MATCH (source:Entity {uuid: edge.source_node_uuid}) MATCH (source:Entity {uuid: edge.source_node_uuid})
MATCH (target:Entity {uuid: edge.target_node_uuid}) MATCH (target:Entity {uuid: edge.target_node_uuid})
MERGE (source)-[e:RELATES_TO {uuid: edge.uuid}]->(target) MERGE (source)-[e:RELATES_TO {uuid: edge.uuid}]->(target)
SET e = edge""" SET e = edge
"""
+ save_embedding_query + save_embedding_query
+ """ + """
RETURN edge.uuid AS uuid 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""" f"""
MERGE (n:Entity {{uuid: $entity_data.uuid}}) MERGE (n:Entity {{uuid: $entity_data.uuid}})
SET n:{labels} SET n:{labels}
SET n = $entity_data""" SET n = $entity_data
"""
+ save_embedding_query + save_embedding_query
+ """ + """
RETURN n.uuid AS uuid RETURN n.uuid AS uuid
@ -239,10 +240,11 @@ def get_entity_node_save_bulk_query(
) )
return ( return (
""" """
UNWIND $nodes AS node UNWIND $nodes AS node
MERGE (n:Entity {uuid: node.uuid}) MERGE (n:Entity {uuid: node.uuid})
SET n:$(node.labels) SET n:$(node.labels)
SET n = node""" SET n = node
"""
+ save_embedding_query + save_embedding_query
+ """ + """
RETURN n.uuid AS uuid RETURN n.uuid AS uuid