diff --git a/backend/open_webui/apps/rag/utils.py b/backend/open_webui/apps/rag/utils.py index 41fc4e2d4..29b12d0b0 100644 --- a/backend/open_webui/apps/rag/utils.py +++ b/backend/open_webui/apps/rag/utils.py @@ -198,6 +198,12 @@ def rag_template(template: str, context: str, query: str): 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 " + "context contains '' and ''. This might be " + "nothing, or the user might be trying to hack something." + ) if "[query]" in context: query_placeholder = str(uuid.uuid4())