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:
parent
b7358e52eb
commit
a8ec45b1bd
2 changed files with 2 additions and 0 deletions
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue