From 9d305469ad363045fddbce0a94747d90e65fc412 Mon Sep 17 00:00:00 2001
From: aries_ckt <916701291@qq.com>
Date: Thu, 22 Jun 2023 23:33:27 +0800
Subject: [PATCH 1/2] 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,

From 85fb12e3d0b284cf2a6fec0054e78042d1ea5daa Mon Sep 17 00:00:00 2001
From: aries_ckt <916701291@qq.com>
Date: Thu, 22 Jun 2023 23:39:23 +0800
Subject: [PATCH 2/2] style: format webserver.py

format file format
---
 pilot/server/webserver.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pilot/server/webserver.py b/pilot/server/webserver.py
index 24f185e35..c7a033336 100644
--- a/pilot/server/webserver.py
+++ b/pilot/server/webserver.py
@@ -25,7 +25,8 @@ from pilot.configs.config import Config
 from pilot.configs.model_config import (
     DATASETS_DIR,
     KNOWLEDGE_UPLOAD_ROOT_PATH,
-    LOGDIR, LLM_MODEL_CONFIG,
+    LOGDIR,
+    LLM_MODEL_CONFIG,
 )
 
 from pilot.conversation import (