|
@@ -18,15 +18,20 @@ class ChromaDB(BaseVectorDB):
|
|
|
if host and port:
|
|
|
logging.info(f"Connecting to ChromaDB server: {host}:{port}")
|
|
|
self.settings = Settings(chroma_server_host=host, chroma_server_http_port=port)
|
|
|
+ self.client = chromadb.HttpClient(self.settings)
|
|
|
else:
|
|
|
if db_dir is None:
|
|
|
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__()
|
|
|
|
|
|
def _get_or_create_db(self):
|
|
|
"""Get or create the database."""
|
|
|
- return chromadb.Client(self.settings)
|
|
|
+ return self.client
|
|
|
|
|
|
def _get_or_create_collection(self):
|
|
|
"""Get or create the collection."""
|