github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-10 20:52:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:issue2484

Browse Source
This commit is contained in:
aries_ckt 2025-03-18 18:58:43 +08:00
parent 88bbd695d0
commit 43763945a8
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
packages/dbgpt-ext/src/dbgpt_ext/storage/vector_store/chroma_store.py
View File

@ -14,6 +14,7 @@ from dbgpt.storage.vector_store.base import (
VectorStoreConfig, VectorStoreConfig,
) )
from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters from dbgpt.storage.vector_store.filters import FilterOperator, MetadataFilters
from dbgpt.util import string_utils
from dbgpt.util.i18n_utils import _ from dbgpt.util.i18n_utils import _
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -110,6 +111,11 @@ class ChromaStore(VectorStoreBase):
self.embeddings = embedding_fn self.embeddings = embedding_fn
if not self.embeddings: if not self.embeddings:
raise ValueError("Embeddings is None") raise ValueError("Embeddings is None")
self._collection_name = name
if string_utils.contains_chinese(name):
bytes_str = self._collection_name.encode("utf-8")
hex_str = bytes_str.hex()
self._collection_name = hex_str
chroma_settings = Settings( chroma_settings = Settings(
# chroma_db_impl="duckdb+parquet", => deprecated configuration of Chroma # chroma_db_impl="duckdb+parquet", => deprecated configuration of Chroma
persist_directory=self.persist_dir, persist_directory=self.persist_dir,
@ -121,9 +127,9 @@ class ChromaStore(VectorStoreBase):
path=self.persist_dir, settings=chroma_settings path=self.persist_dir, settings=chroma_settings
) )
collection_metadata = collection_metadata or {"hnsw:space": "cosine"} collection_metadata = collection_metadata or {"hnsw:space": "cosine"}
self._collection_name = name
self._collection = self._chroma_client.get_or_create_collection( self._collection = self._chroma_client.get_or_create_collection(
name=name, name=self._collection_name,
embedding_function=None, embedding_function=None,
metadata=collection_metadata, metadata=collection_metadata,
) )