UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-10 15:45:45 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-04-05 05:55:56 -06:00
parent ae484e8edf
commit 807b208418
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/open_webui/utils/middleware.py
View File

@ -221,6 +221,14 @@ async def chat_completion_tools_handler(
except Exception as e:
tool_result = str(e)
tool_result_files = []
if isinstance(tool_result, list):
for item in tool_result:
# check if string
if isinstance(item, str) and item.startswith("data:"):
tool_result_files.append(item)
tool_result.remove(item)
if isinstance(tool_result, dict) or isinstance(tool_result, list):
tool_result = json.dumps(tool_result, indent=2)
@ -240,7 +248,7 @@ async def chat_completion_tools_handler(
else f"{tool_function_name}"
),
},
"document": [tool_result],
"document": [tool_result, *tool_result_files],
"metadata": [
{
"source": (
@ -256,7 +264,7 @@ async def chat_completion_tools_handler(
sources.append(
{
"source": {},
"document": [tool_result],
"document": [tool_result, *tool_result_files],
"metadata": [
{
"source": (