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

fix: EMBEDDING_MODEL env config (#265)

Browse Source 
1.solve the problem that EMBEDDING_MODEL config did not work 
Close #264
This commit is contained in:
magic.chen 2023-06-22 23:46:07 +08:00 committed by GitHub
commit 72c53b6287
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/server/webserver.py
View File

@ -26,6 +26,7 @@ from pilot.configs.model_config import (
DATASETS_DIR,
KNOWLEDGE_UPLOAD_ROOT_PATH,
LOGDIR,
LLM_MODEL_CONFIG,
)
from pilot.conversation import (
@ -631,7 +632,7 @@ def knowledge_embedding_store(vs_id, files):
)
knowledge_embedding_client = KnowledgeEmbedding(
file_path=os.path.join(KNOWLEDGE_UPLOAD_ROOT_PATH, vs_id, filename),
model_name=CFG.EMBEDDING_MODEL,
model_name=LLM_MODEL_CONFIG[CFG.EMBEDDING_MODEL],
vector_store_config={
"vector_store_name": vector_store_name["vs_name"],
"vector_store_path": KNOWLEDGE_UPLOAD_ROOT_PATH,