logs: crash if rag_template would be wrong

2024-09-12 15:19:24 +02:00 · 2024-09-12 15:19:24 +02:00 · adf26789b8
parent 209e246e6f
commit adf26789b8
1 changed files with 5 additions and 0 deletions
--- a/backend/open_webui/apps/rag/utils.py
+++ b/backend/open_webui/apps/rag/utils.py
@ -192,6 +192,11 @@ 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]'"
    template = template.replace("[context]", context)
    template = template.replace("[query]", query)
    return template