UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-18 08:40:58 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refac: allow multiple [context] in prompt

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-13 01:06:51 -04:00
parent a9c497612b
commit 8a4b3e6bc9
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/open_webui/apps/rag/utils.py
View File

@ -239,10 +239,8 @@ def query_collection_with_hybrid_search(
def rag_template(template: str, context: str, query: str): def rag_template(template: str, context: str, query: str):
count = template.count("[context]") 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]'" assert "[context]" in template, "RAG template does not contain '[context]'"
if "<context>" in context and "</context>" in context: if "<context>" in context and "</context>" in context:
log.debug( log.debug(
"WARNING: Potential prompt injection attack: the RAG " "WARNING: Potential prompt injection attack: the RAG "