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

bug: Chroma needs diff client as per settings (#342)

Browse Source
This commit is contained in:
Taranjeet Singh
2023-07-20 18:17:01 -07:00
committed by GitHub
parent 40643663cb
commit fa63a16591
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
embedchain/vectordb/chroma_db.py
View File

@@ -18,15 +18,20 @@ class ChromaDB(BaseVectorDB):
if host and port: if host and port:
logging.info(f"Connecting to ChromaDB server: {host}:{port}") logging.info(f"Connecting to ChromaDB server: {host}:{port}")
self.settings = Settings(chroma_server_host=host, chroma_server_http_port=port) self.settings = Settings(chroma_server_host=host, chroma_server_http_port=port)
self.client = chromadb.HttpClient(self.settings)
else: else:
if db_dir is None: if db_dir is None:
db_dir = "db" db_dir = "db"
self.settings = Settings(persist_directory=db_dir, anonymized_telemetry=False, allow_reset=True) self.settings = Settings(anonymized_telemetry=False, allow_reset=True)
self.client = chromadb.PersistentClient(
path=db_dir,
settings=self.settings,
)
super().__init__() super().__init__()
def _get_or_create_db(self): def _get_or_create_db(self):
"""Get or create the database.""" """Get or create the database."""
return chromadb.Client(self.settings) return self.client
def _get_or_create_collection(self): def _get_or_create_collection(self):
"""Get or create the collection.""" """Get or create the collection."""