diff --git a/graphiti_core/models/nodes/node_db_queries.py b/graphiti_core/models/nodes/node_db_queries.py index fdcf48f1..cc28d1de 100644 --- a/graphiti_core/models/nodes/node_db_queries.py +++ b/graphiti_core/models/nodes/node_db_queries.py @@ -111,7 +111,7 @@ def get_community_node_save_query(provider: GraphProvider) -> str: if provider == GraphProvider.FALKORDB: return """ MERGE (n:Community {uuid: $uuid}) - SET n = {uuid: $uuid, name: $name, group_id: $group_id, summary: $summary, created_at: $created_at, name_embedding: $name_embedding} + SET n = {uuid: $uuid, name: $name, group_id: $group_id, summary: $summary, created_at: $created_at, name_embedding: vecf32($name_embedding)} RETURN n.uuid AS uuid """ @@ -130,4 +130,6 @@ COMMUNITY_NODE_RETURN = """ n.group_id AS group_id, n.summary AS summary, n.created_at AS created_at -""" +""" + + diff --git a/graphiti_core/utils/maintenance/community_operations.py b/graphiti_core/utils/maintenance/community_operations.py index a48000c9..bdcb0e1d 100644 --- a/graphiti_core/utils/maintenance/community_operations.py +++ b/graphiti_core/utils/maintenance/community_operations.py @@ -33,7 +33,8 @@ async def get_community_clusters( if group_ids is None: group_id_values, _, _ = await driver.execute_query( """ - MATCH (n:Entity WHERE n.group_id IS NOT NULL) + MATCH (n:Entity) + WHERE n.group_id IS NOT NULL RETURN collect(DISTINCT n.group_id) AS group_ids """,