tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

Merge pull request #1868 from Yanyutin753/rag-dev

Browse source 
📌 fixed a bug where RAG would not reply after not reading the file correctly
This commit is contained in:
Timothy Jaeryang Baek 2024-04-30 11:40:47 -07:00 committed by GitHub
commit 3f0fae1d10
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
backend/apps/rag/utils.py
View file

@ -321,8 +321,12 @@ def rag_messages(
context_string = "" context_string = ""
for context in relevant_contexts: for context in relevant_contexts:
items = context["documents"][0] try:
context_string += "\n\n".join(items) if "documents" in context:
items = [item for item in context["documents"][0] if item is not None]
context_string += "\n\n".join(items)
except Exception as e:
log.exception(e)
context_string = context_string.strip() context_string = context_string.strip()
ra_content = rag_template( ra_content = rag_template(