github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-13 22:15:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:chromadb persist_path compatible with old version

Browse Source
This commit is contained in:
aries_ckt 2023-10-09 15:01:42 +05:00
parent 4aa825d708
commit 813e2260a6
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/vector_store/chroma_store.py
View File

@ -17,7 +17,8 @@ class ChromaStore(VectorStoreBase):
self.ctx = ctx self.ctx = ctx
chroma_path = ctx.get( chroma_path = ctx.get(
"CHROMA_PERSIST_PATH", os.getenv("CHROMA_PERSIST_PATH", os.getcwd()) "CHROMA_PERSIST_PATH",
os.path.join(os.path.dirname(os.path.dirname(__file__)), "data"),
) )
self.persist_dir = os.path.join( self.persist_dir = os.path.join(
chroma_path, ctx["vector_store_name"] + ".vectordb" chroma_path, ctx["vector_store_name"] + ".vectordb"