fix: EMBEDDING_MODEL env config (#265)

1.solve the problem that EMBEDDING_MODEL config did not work Close #264
eosphoros-ai · Jun 22, 2023 · 879f6ae · 879f6ae
2 parents 3b17fa4 + cc422a2
commit 879f6ae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pilot/server/webserver.py b/pilot/server/webserver.py
@@ -26,6 +26,7 @@
     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,