tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages 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
Show stats 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 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 # Construct huggingface_hub kwargs with local_files_only to return the snapshot path
cache_dir = os.getenv("SENTENCE_TRANSFORMERS_HOME") cache_dir = os.getenv("SENTENCE_TRANSFORMERS_HOME")
local_files_only = not update_embedding_model 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}") log.debug(f"local_files_only: {local_files_only}")
# Inspiration from upstream sentence_transformers # 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 # If fully qualified path exists, return input, else set repo_id
return embedding_model return embedding_model
elif "/" not in embedding_model: elif "/" not in embedding_model:

2
src/lib/components/documents/Settings/General.svelte
View file

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