github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-07 19:34:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into tt_dev

Browse Source
This commit is contained in:
csunny 2023-10-09 19:07:33 +08:00
commit 32ae362eac
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"