tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

Format fixes

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

10
backend/apps/rag/utils.py
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: