From ae484e8edf9f77e021ab501861b0d21a565a663c Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sat, 5 Apr 2025 05:45:32 -0600 Subject: [PATCH] refac --- backend/open_webui/utils/tools.py | 26 ++++++++++++++++++-------- 1 file changed, 18 insertions(+), 8 deletions(-) diff --git a/backend/open_webui/utils/tools.py b/backend/open_webui/utils/tools.py index 4746450f5..60311a690 100644 --- a/backend/open_webui/utils/tools.py +++ b/backend/open_webui/utils/tools.py @@ -69,14 +69,24 @@ 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, - ) + def make_tool_function(function_name, token, tool_server_data): + async def tool_function(**kwargs): + print( + f"Executing tool function {function_name} with params: {kwargs}" + ) + return await execute_tool_server( + token=token, + url=tool_server_data["url"], + name=function_name, + params=kwargs, + server_data=tool_server_data, + ) + + return tool_function + + tool_function = make_tool_function( + function_name, token, tool_server_data + ) callable = get_async_tool_function_and_apply_extra_params( tool_function,