Skip to content

Commit 74bf446

Browse files
committed
fix: assert vector native typing on batch write
1 parent a4a4dc0 commit 74bf446

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

  • integrations/falkor_db/src/haystack_integrations/document_stores/falkor_db

integrations/falkor_db/src/haystack_integrations/document_stores/falkor_db/document_store.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -362,6 +362,9 @@ def _write_batch(self, documents: list[Document], policy: DuplicatePolicy) -> in
362362
MERGE (d:{self._node_label} {{id: doc.id}})
363363
ON CREATE SET d += doc
364364
ON MATCH SET d += doc
365+
FOREACH (_ IN CASE WHEN doc.{self._embedding_field} IS NOT NULL THEN [1] ELSE [] END |
366+
SET d.{self._embedding_field} = vecf32(doc.{self._embedding_field})
367+
)
365368
RETURN count(d) AS n
366369
"""
367370
else:
@@ -371,6 +374,9 @@ def _write_batch(self, documents: list[Document], policy: DuplicatePolicy) -> in
371374
UNWIND $docs AS doc
372375
MERGE (d:{self._node_label} {{id: doc.id}})
373376
ON CREATE SET d += doc
377+
FOREACH (_ IN CASE WHEN doc.{self._embedding_field} IS NOT NULL THEN [1] ELSE [] END |
378+
SET d.{self._embedding_field} = vecf32(doc.{self._embedding_field})
379+
)
374380
RETURN count(d) AS n
375381
"""
376382

0 commit comments

Comments
 (0)