From ddd6660638490c6f10804ff4edf0df7e7b7e9e6a Mon Sep 17 00:00:00 2001
From: prestonrasmussen <prasmuss15@gmail.com>
Date: Sun, 7 Sep 2025 22:53:42 -0400
Subject: [PATCH] regression tested

---
 graphiti_core/models/edges/edge_db_queries.py | 20 ++++++++++---------
 graphiti_core/models/nodes/node_db_queries.py | 12 ++++++-----
 2 files changed, 18 insertions(+), 14 deletions(-)

diff --git a/graphiti_core/models/edges/edge_db_queries.py b/graphiti_core/models/edges/edge_db_queries.py
index d2ce2183..92e294fb 100644
--- a/graphiti_core/models/edges/edge_db_queries.py
+++ b/graphiti_core/models/edges/edge_db_queries.py
@@ -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
diff --git a/graphiti_core/models/nodes/node_db_queries.py b/graphiti_core/models/nodes/node_db_queries.py
index b62c0689..0972dac4 100644
--- a/graphiti_core/models/nodes/node_db_queries.py
+++ b/graphiti_core/models/nodes/node_db_queries.py
@@ -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