Fix chroma get_all method (#1701)

2024-08-14 15:04:56 +05:30
parent a461091ba5
commit aba5bb052d
2 changed files with 3 additions and 4 deletions
--- a/mem0/vector_stores/chroma.py
+++ b/mem0/vector_stores/chroma.py
@@ -209,6 +209,5 @@ class ChromaDB(VectorStoreBase):
        Returns:
            List[OutputData]: List of vectors.
        """
-        array = [[0 for _ in range(1536)] for _ in range(1536)]
-        results = self.collection.query(query_embeddings=array, where=filters, n_results=limit)
-        return self._parse_output(results)
+        results = self.collection.get(where=filters, limit=limit)
+        return [self._parse_output(results)]