ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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": (