|
|
|
@ -87,23 +87,26 @@ class Chroma(VectorStore):
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
if client is not None:
|
|
|
|
|
self._client_settings = client_settings
|
|
|
|
|
self._client = client
|
|
|
|
|
self._persist_directory = persist_directory
|
|
|
|
|
else:
|
|
|
|
|
if client_settings:
|
|
|
|
|
self._client_settings = client_settings
|
|
|
|
|
_client_settings = client_settings
|
|
|
|
|
elif persist_directory:
|
|
|
|
|
_client_settings = chromadb.config.Settings(
|
|
|
|
|
chroma_db_impl="duckdb+parquet",
|
|
|
|
|
persist_directory=persist_directory,
|
|
|
|
|
)
|
|
|
|
|
else:
|
|
|
|
|
self._client_settings = chromadb.config.Settings()
|
|
|
|
|
if persist_directory is not None:
|
|
|
|
|
self._client_settings = chromadb.config.Settings(
|
|
|
|
|
chroma_db_impl="duckdb+parquet",
|
|
|
|
|
persist_directory=persist_directory,
|
|
|
|
|
)
|
|
|
|
|
self._client = chromadb.Client(self._client_settings)
|
|
|
|
|
_client_settings = chromadb.config.Settings()
|
|
|
|
|
self._client_settings = _client_settings
|
|
|
|
|
self._client = chromadb.Client(_client_settings)
|
|
|
|
|
self._persist_directory = (
|
|
|
|
|
_client_settings.persist_directory or persist_directory
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
self._embedding_function = embedding_function
|
|
|
|
|
self._persist_directory = (
|
|
|
|
|
self._client_settings.persist_directory or persist_directory
|
|
|
|
|
)
|
|
|
|
|
self._collection = self._client.get_or_create_collection(
|
|
|
|
|
name=collection_name,
|
|
|
|
|
embedding_function=self._embedding_function.embed_documents
|
|
|
|
|