diff --git a/backend/open_webui/apps/rag/utils.py b/backend/open_webui/apps/rag/utils.py index 22f34c6a9..efa22841d 100644 --- a/backend/open_webui/apps/rag/utils.py +++ b/backend/open_webui/apps/rag/utils.py @@ -239,10 +239,8 @@ def query_collection_with_hybrid_search( def rag_template(template: str, context: str, query: str): count = template.count("[context]") - assert ( - count == 1 - ), f"RAG template contains an unexpected number of '[context]' : {count}" assert "[context]" in template, "RAG template does not contain '[context]'" + if "" in context and "" in context: log.debug( "WARNING: Potential prompt injection attack: the RAG "