UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-07 00:59:52 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: only append if distances are available

Browse Source
This commit is contained in:
Jannik Streidl 2024-10-07 21:13:13 +02:00
parent b105efa05f
commit 9fc813cfa6
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/apps/retrieval/utils.py
View File

@ -393,14 +393,14 @@ def get_rag_context(
) )
if "metadatas" in context: if "metadatas" in context:
citations.append( citation = {
{
"source": context["file"], "source": context["file"],
"document": context["documents"][0], "document": context["documents"][0],
"metadata": context["metadatas"][0], "metadata": context["metadatas"][0],
"distances": context["distances"][0]
} }
) if "distances" in context and context["distances"]:
citation["distances"] = context["distances"][0]
citations.append(citation)
except Exception as e: except Exception as e:
log.exception(e) log.exception(e)