UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-24 21:13:59 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: add check that the context for RAG is not empty if the threshold is 0

Browse Source
This commit is contained in:
thiswillbeyourgithub 2024-09-12 16:36:37 +02:00
parent 9661fee554
commit b4ad64586a
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/open_webui/main.py
View File

@ -588,6 +588,11 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
prompt = get_last_user_message(body["messages"]) prompt = get_last_user_message(body["messages"])
if prompt is None: if prompt is None:
raise Exception("No user message found") raise Exception("No user message found")
if rag_app.state.config.RELEVANCE_THRESHOLD == 0:
assert context_string.strip(), (
"With a 0 relevancy threshold for RAG, the context cannot "
"be empty"
)
# Workaround for Ollama 2.0+ system prompt issue # Workaround for Ollama 2.0+ system prompt issue
# TODO: replace with add_or_update_system_message # TODO: replace with add_or_update_system_message
if model["owned_by"] == "ollama": if model["owned_by"] == "ollama":