浏览代码

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

thiswillbeyourgithub 7 月之前
父节点
当前提交
b4ad64586a
共有 1 个文件被更改,包括 5 次插入0 次删除
  1. 5 0
      backend/open_webui/main.py

+ 5 - 0
backend/open_webui/main.py

@@ -588,6 +588,11 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
             prompt = get_last_user_message(body["messages"])
             if prompt is None:
                 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
             # TODO: replace with add_or_update_system_message
             if model["owned_by"] == "ollama":