Commit bcf79c83 authored by Self Denial's avatar Self Denial
Browse files

Format fixes

parent 3b66aa55
......@@ -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:
......
......@@ -34,7 +34,7 @@
};
let embeddingModel = {
embedding_model: '',
embedding_model: ''
};
const scanHandler = async () => {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment