Przeglądaj źródła

More format fixes

Self Denial 1 rok temu
rodzic
commit
075fbedb02
2 zmienionych plików z 12 dodań i 5 usunięć
  1. 11 5
      backend/apps/rag/main.py
  2. 1 0
      backend/apps/rag/utils.py

+ 11 - 5
backend/apps/rag/main.py

@@ -141,17 +141,21 @@ async def update_embedding_model(
     app.state.RAG_EMBEDDING_MODEL = form_data.embedding_model
 
     log.debug(f"form_data.embedding_model: {form_data.embedding_model}")
-    log.info(f"Updating embedding model: {app.state.RAG_EMBEDDING_MODEL} to {form_data.embedding_model}")
+    log.info(
+        f"Updating embedding model: {app.state.RAG_EMBEDDING_MODEL} to {form_data.embedding_model}"
+    )
 
     try:
-        app.state.RAG_EMBEDDING_MODEL_PATH = embedding_model_get_path(app.state.RAG_EMBEDDING_MODEL, True)
+        app.state.RAG_EMBEDDING_MODEL_PATH = embedding_model_get_path(
+            app.state.RAG_EMBEDDING_MODEL, True
+        )
         app.state.sentence_transformer_ef = (
             embedding_functions.SentenceTransformerEmbeddingFunction(
                 model_name=app.state.RAG_EMBEDDING_MODEL_PATH,
                 device=RAG_EMBEDDING_MODEL_DEVICE_TYPE,
             )
         )
-    except Exception as e: 
+    except Exception as e:
         log.exception(f"Problem updating embedding model: {e}")
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
@@ -159,9 +163,11 @@ async def update_embedding_model(
         )
 
     if app.state.RAG_EMBEDDING_MODEL_PATH == old_model_path:
-      status = False
+        status = False
 
-    log.debug(f"app.state.RAG_EMBEDDING_MODEL_PATH: {app.state.RAG_EMBEDDING_MODEL_PATH}")
+    log.debug(
+        f"app.state.RAG_EMBEDDING_MODEL_PATH: {app.state.RAG_EMBEDDING_MODEL_PATH}"
+    )
     log.debug(f"old_model_path: {old_model_path}")
     log.debug(f"status: {status}")
 

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

@@ -191,6 +191,7 @@ def rag_messages(docs, messages, template, k, embedding_function):
 
     return messages
 
+
 def embedding_model_get_path(
     embedding_model: str, update_embedding_model: bool = False
 ):