From aba5bb052dc54149b04f61a2bd702dd4d02e185b Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Wed, 14 Aug 2024 15:04:56 +0530 Subject: [PATCH] Fix chroma `get_all` method (#1701) --- mem0/vector_stores/chroma.py | 5 ++--- pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/mem0/vector_stores/chroma.py b/mem0/vector_stores/chroma.py index 10e978a5..c06a0bda 100644 --- 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)] diff --git a/pyproject.toml b/pyproject.toml index 38e46bce..4bb5c5d0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "mem0ai" -version = "0.0.18" +version = "0.0.19" description = "Long-term memory for AI Agents" authors = ["Mem0 "] exclude = [