fix(chroma): wrap update() ids/embeddings/metadatas in lists (#5757)

2026-06-24 10:54:15 +05:30
parent c2862831db
commit b611f69381
2 changed files with 30 additions and 2 deletions
@@ -185,7 +185,11 @@ class ChromaDB(VectorStoreBase):
            vector (Optional[List[float]], optional): Updated vector. Defaults to None.
            payload (Optional[Dict], optional): Updated payload. Defaults to None.
        """
-        self.collection.update(ids=vector_id, embeddings=vector, metadatas=payload)
+        self.collection.update(
+            ids=[vector_id],
+            embeddings=[vector] if vector is not None else None,
+            metadatas=[payload] if payload is not None else None,
+        )

    def get(self, vector_id: str) -> Optional[OutputData]:
        """