Timothy J. Baek 1 year ago
parent
commit
cb2158a794
1 changed files with 1 additions and 0 deletions
  1. 1 0
      backend/apps/rag/main.py

+ 1 - 0
backend/apps/rag/main.py

@@ -167,6 +167,7 @@ async def update_embedding_model(
         )
         )
 
 
     if sentence_transformer_ef:
     if sentence_transformer_ef:
+        app.state.RAG_EMBEDDING_MODEL = form_data.embedding_model
         app.state.RAG_EMBEDDING_MODEL_PATH = embedding_model_path
         app.state.RAG_EMBEDDING_MODEL_PATH = embedding_model_path
         app.state.sentence_transformer_ef = sentence_transformer_ef
         app.state.sentence_transformer_ef = sentence_transformer_ef