浏览代码

refac: allow multiple [context] in prompt

Timothy J. Baek 7 月之前
父节点
当前提交
8a4b3e6bc9
共有 1 个文件被更改,包括 1 次插入3 次删除
  1. 1 3
      backend/open_webui/apps/rag/utils.py

+ 1 - 3
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 "<context>" in context and "</context>" in context:
         log.debug(
             "WARNING: Potential prompt injection attack: the RAG "