OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions 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: except Exception as e:
tool_result = str(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): if isinstance(tool_result, dict) or isinstance(tool_result, list):
tool_result = json.dumps(tool_result, indent=2) tool_result = json.dumps(tool_result, indent=2)
@ -240,7 +248,7 @@ async def chat_completion_tools_handler(
else f"{tool_function_name}" else f"{tool_function_name}"
), ),
}, },
"document": [tool_result], "document": [tool_result, *tool_result_files],
"metadata": [ "metadata": [
{ {
"source": ( "source": (
@ -256,7 +264,7 @@ async def chat_completion_tools_handler(
sources.append( sources.append(
{ {
"source": {}, "source": {},
"document": [tool_result], "document": [tool_result, *tool_result_files],
"metadata": [ "metadata": [
{ {
"source": ( "source": (