Timothy J. Baek il y a 1 an
Parent
commit
cb2158a794
1 fichiers modifiés avec 1 ajouts et 0 suppressions
  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:
+        app.state.RAG_EMBEDDING_MODEL = form_data.embedding_model
         app.state.RAG_EMBEDDING_MODEL_PATH = embedding_model_path
         app.state.sentence_transformer_ef = sentence_transformer_ef