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

logs: crash if rag_template would be wrong

Browse Source
This commit is contained in:
thiswillbeyourgithub 2024-09-12 15:19:24 +02:00
parent 209e246e6f
commit adf26789b8
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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