From afeed82a6ac3bb003a9e66055fe84192d35021ad Mon Sep 17 00:00:00 2001
From: Aidan Petti <apetti1920@aol.com>
Date: Tue, 25 Nov 2025 07:59:29 -0700
Subject: [PATCH] bug fixes

---
 graphiti_core/models/nodes/node_db_queries.py | 2 +-
 graphiti_core/nodes.py                        | 2 +-
 graphiti_core/search/search_utils.py          | 8 ++++----
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/graphiti_core/models/nodes/node_db_queries.py b/graphiti_core/models/nodes/node_db_queries.py
index 34e3d8b8..5f879811 100644
--- a/graphiti_core/models/nodes/node_db_queries.py
+++ b/graphiti_core/models/nodes/node_db_queries.py
@@ -122,7 +122,7 @@ EPISODIC_NODE_RETURN_NEPTUNE = """
     e.group_id AS group_id,
     e.source_description AS source_description,
     e.source AS source,
-    split(e.entity_edges, ",") AS entity_edges
+    split(e.entity_edges, "|") AS entity_edges
 """
 
 
diff --git a/graphiti_core/nodes.py b/graphiti_core/nodes.py
index cd3d003d..2f31fe96 100644
--- a/graphiti_core/nodes.py
+++ b/graphiti_core/nodes.py
@@ -595,7 +595,7 @@ class CommunityNode(Node):
     async def save(self, driver: GraphDriver):
         if driver.provider == GraphProvider.NEPTUNE:
             await driver.save_to_aoss(  # pyright: ignore reportAttributeAccessIssue
-                'communities',
+                'community_name',
                 [{'name': self.name, 'uuid': self.uuid, 'group_id': self.group_id}],
             )
         result = await driver.execute_query(
diff --git a/graphiti_core/search/search_utils.py b/graphiti_core/search/search_utils.py
index 4c0e84fa..22e7d665 100644
--- a/graphiti_core/search/search_utils.py
+++ b/graphiti_core/search/search_utils.py
@@ -1075,7 +1075,7 @@ async def community_similarity_search(
                         comm.name AS name,
                         comm.created_at AS created_at,
                         comm.summary AS summary,
-                        comm.name_embedding AS name_embedding
+                        [x IN split(comm.name_embedding, ",") | toFloat(x)] AS name_embedding
                     ORDER BY i.score DESC
                     LIMIT $limit
                 """
@@ -1887,7 +1887,7 @@ async def get_embeddings_for_nodes(
         WHERE n.uuid IN $node_uuids
         RETURN DISTINCT
             n.uuid AS uuid,
-            split(n.name_embedding, ",") AS name_embedding
+            [x IN split(n.name_embedding, ",") | toFloat(x)] AS name_embedding
         """
     else:
         query = """
@@ -1922,7 +1922,7 @@ async def get_embeddings_for_communities(
         WHERE c.uuid IN $community_uuids
         RETURN DISTINCT
             c.uuid AS uuid,
-            split(c.name_embedding, ",") AS name_embedding
+            [x IN split(c.name_embedding, ",") | toFloat(x)] AS name_embedding
         """
     else:
         query = """
@@ -1959,7 +1959,7 @@ async def get_embeddings_for_edges(
         WHERE e.uuid IN $edge_uuids
         RETURN DISTINCT
             e.uuid AS uuid,
-            split(e.fact_embedding, ",") AS fact_embedding
+            [x IN split(e.fact_embedding, ",") | toFloat(x)] AS fact_embedding
         """
     else:
         match_query = """