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

fix: EMBEDDING_MODEL env config

Browse Source 
solve the problem that EMBEDDING_MODEL config did not work
Close #264
This commit is contained in:
aries_ckt 2023-06-22 23:33:27 +08:00
parent 3b17fa4ead
commit 683ed09e10
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/server/webserver.py
View File

@ -25,7 +25,7 @@ from pilot.configs.config import Config
from pilot.configs.model_config import ( from pilot.configs.model_config import (
DATASETS_DIR, DATASETS_DIR,
KNOWLEDGE_UPLOAD_ROOT_PATH, KNOWLEDGE_UPLOAD_ROOT_PATH,
LOGDIR, LOGDIR, LLM_MODEL_CONFIG,
) )
from pilot.conversation import ( from pilot.conversation import (
@ -631,7 +631,7 @@ def knowledge_embedding_store(vs_id, files):
) )
knowledge_embedding_client = KnowledgeEmbedding( knowledge_embedding_client = KnowledgeEmbedding(
file_path=os.path.join(KNOWLEDGE_UPLOAD_ROOT_PATH, vs_id, filename), 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_config={
"vector_store_name": vector_store_name["vs_name"], "vector_store_name": vector_store_name["vs_name"],
"vector_store_path": KNOWLEDGE_UPLOAD_ROOT_PATH, "vector_store_path": KNOWLEDGE_UPLOAD_ROOT_PATH,