klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Bug Fix] Fix issue with chromadb not working with 0.6.0 and onwards (#2279)

Browse Source
This commit is contained in:
Deshraj Yadav
2025-02-28 23:39:59 -08:00
committed by GitHub
parent 4318663697
commit 32ebdaef2f
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mem0/vector_stores/chroma.py
View File

@@ -105,10 +105,10 @@ class ChromaDB(VectorStoreBase):
chromadb.Collection: The created or retrieved collection.
"""
# Skip creating collection if already exists
collections = self.list_cols()
for collection in collections:
if collection.name == name:
logging.debug(f"Collection {name} already exists. Skipping creation.")
collection_info = self.client.get_collection(name=name)
if collection_info:
logging.debug(f"Collection {name} already exists. Skipping creation.")
return collection_info
collection = self.client.get_or_create_collection(
name=name,