From 075fbedb02c256843afd496db707c30be557d3e8 Mon Sep 17 00:00:00 2001
From: Self Denial <selfdenial@pm.me>
Date: Thu, 4 Apr 2024 12:07:42 -0600
Subject: [PATCH] More format fixes

---
 backend/apps/rag/main.py  | 16 +++++++++++-----
 backend/apps/rag/utils.py |  1 +
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/backend/apps/rag/main.py b/backend/apps/rag/main.py
index 3d12084c..7e384447 100644
--- a/backend/apps/rag/main.py
+++ b/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}")
 
diff --git a/backend/apps/rag/utils.py b/backend/apps/rag/utils.py
index 4222609f..866759d9 100644
--- a/backend/apps/rag/utils.py
+++ b/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
 ):