Merge pull request #1868 from Yanyutin753/rag-dev

📌 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

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:
if "documents" in context:
items = [item for item in context["documents"][0] if item is not None]
context_string += "\n\n".join(items) 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(