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

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

@ -110,7 +110,8 @@ def get_entity_edge_save_query(provider: GraphProvider, has_aoss: bool = False)
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
) )
+ """ + """
@ -173,7 +174,8 @@ def get_entity_edge_save_bulk_query(provider: GraphProvider, has_aoss: bool = Fa
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

6
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
@ -242,7 +243,8 @@ def get_entity_node_save_bulk_query(
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