mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-23 20:26:15 +00:00
Merge e4495f2090
into 441cb55377
This commit is contained in:
commit
7576a53ffd
@ -114,9 +114,7 @@ class ChromaStore(VectorStoreBase):
|
|||||||
except ImportError:
|
except ImportError:
|
||||||
raise ImportError("Please install chroma package first.")
|
raise ImportError("Please install chroma package first.")
|
||||||
chroma_vector_config = vector_store_config.to_dict()
|
chroma_vector_config = vector_store_config.to_dict()
|
||||||
chroma_path = chroma_vector_config.get(
|
chroma_path = chroma_vector_config.get("persist_path") or os.path.join(PILOT_PATH, "data")
|
||||||
"persist_path", os.path.join(PILOT_PATH, "data")
|
|
||||||
)
|
|
||||||
self.persist_dir = os.path.join(resolve_root_path(chroma_path) + "/chromadb")
|
self.persist_dir = os.path.join(resolve_root_path(chroma_path) + "/chromadb")
|
||||||
self.embeddings = embedding_fn
|
self.embeddings = embedding_fn
|
||||||
if not self.embeddings:
|
if not self.embeddings:
|
||||||
|
Loading…
Reference in New Issue
Block a user