fix: wrap embeddings with vecf32() in FalkorDB single save paths (#991)

Fixes #972. Entity and edge single save operations now properly convert
embeddings to vecf32 type, matching bulk save behavior and preventing
type mismatch errors during vector similarity searches.
This commit is contained in:
Naseem Ali 2025-10-12 19:42:52 +03:00 committed by GitHub
parent b7358e52eb
commit a8ec45b1bd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 0 deletions

View file

@ -68,6 +68,7 @@ def get_entity_edge_save_query(provider: GraphProvider, has_aoss: bool = False)
MATCH (target:Entity {uuid: $edge_data.target_uuid})
MERGE (source)-[e:RELATES_TO {uuid: $edge_data.uuid}]->(target)
SET e = $edge_data
SET e.fact_embedding = vecf32($edge_data.fact_embedding)
RETURN e.uuid AS uuid
"""
case GraphProvider.NEPTUNE:

View file

@ -133,6 +133,7 @@ def get_entity_node_save_query(provider: GraphProvider, labels: str, has_aoss: b
MERGE (n:Entity {{uuid: $entity_data.uuid}})
SET n:{labels}
SET n = $entity_data
SET n.name_embedding = vecf32($entity_data.name_embedding)
RETURN n.uuid AS uuid
"""
case GraphProvider.KUZU: