From 683ed09e107db3bb4f04fc93918fb3e695c65fea Mon Sep 17 00:00:00 2001
From: aries_ckt <916701291@qq.com>
Date: Thu, 22 Jun 2023 23:33:27 +0800
Subject: [PATCH] fix: EMBEDDING_MODEL env config solve the problem that
 EMBEDDING_MODEL config did not work Close #264

---
 pilot/server/webserver.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pilot/server/webserver.py b/pilot/server/webserver.py
index 08fe75f5a..24f185e35 100644
--- a/pilot/server/webserver.py
+++ b/pilot/server/webserver.py
@@ -25,7 +25,7 @@ from pilot.configs.config import Config
 from pilot.configs.model_config import (
     DATASETS_DIR,
     KNOWLEDGE_UPLOAD_ROOT_PATH,
-    LOGDIR,
+    LOGDIR, LLM_MODEL_CONFIG,
 )
 
 from pilot.conversation import (
@@ -631,7 +631,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,