Format fixes

This commit is contained in:
Self Denial 2024-04-04 12:02:48 -06:00
parent 3b66aa55c0
commit bcf79c8366
2 changed files with 9 additions and 3 deletions

View File

@ -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:

View File

@ -34,7 +34,7 @@
};
let embeddingModel = {
embedding_model: '',
embedding_model: ''
};
const scanHandler = async () => {