diff --git a/backend/apps/rag/utils.py b/backend/apps/rag/utils.py index 249048b3e..4222609f1 100644 --- a/backend/apps/rag/utils.py +++ b/backend/apps/rag/utils.py @@ -191,7 +191,9 @@ def rag_messages(docs, messages, template, k, embedding_function): return messages -def embedding_model_get_path(embedding_model: str, update_embedding_model: bool = False): +def embedding_model_get_path( + embedding_model: str, update_embedding_model: bool = False +): # Construct huggingface_hub kwargs with local_files_only to return the snapshot path cache_dir = os.getenv("SENTENCE_TRANSFORMERS_HOME") local_files_only = not update_embedding_model @@ -206,7 +208,11 @@ def embedding_model_get_path(embedding_model: str, update_embedding_model: bool log.debug(f"local_files_only: {local_files_only}") # Inspiration from upstream sentence_transformers - if (os.path.exists(embedding_model) or ("\\" in embedding_model or embedding_model.count("/") > 1) and local_files_only): + if ( + os.path.exists(embedding_model) + or ("\\" in embedding_model or embedding_model.count("/") > 1) + and local_files_only + ): # If fully qualified path exists, return input, else set repo_id return embedding_model elif "/" not in embedding_model: diff --git a/src/lib/components/documents/Settings/General.svelte b/src/lib/components/documents/Settings/General.svelte index c20f20422..02b813dea 100644 --- a/src/lib/components/documents/Settings/General.svelte +++ b/src/lib/components/documents/Settings/General.svelte @@ -34,7 +34,7 @@ }; let embeddingModel = { - embedding_model: '', + embedding_model: '' }; const scanHandler = async () => {