UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-27 17:51:31 +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:38:46 -06:00
parent e570a98bf7
commit 56dc7c5640
2 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1912,7 +1912,8 @@ async def process_chat_response(
tool_result_files = []
if isinstance(tool_result, list):
for item in tool_result:
if item.startswith("data:"):
# check if string
if isinstance(item, str) and item.startswith("data:"):
tool_result_files.append(item)
tool_result.remove(item)

18
backend/open_webui/utils/tools.py
View File

@ -69,14 +69,18 @@ def get_tools(
elif auth_type == "session":
token = request.state.token.credentials
async def tool_function(**args):
return await execute_tool_server(
token=token,
url=tool_server_data["url"],
name=function_name,
params=args,
server_data=tool_server_data,
)
callable = get_async_tool_function_and_apply_extra_params(
execute_tool_server,
{
"token": token,
"url": tool_server_data["url"],
"name": function_name,
"server_data": tool_server_data,
},
tool_function,
{},
)
tool_dict = {