OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-22 18:07:17 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge 4f982e244f into aef0ad2d10

Browse Source
This commit is contained in:
Olivier Lacroix 2025-06-22 00:27:14 +08:00 committed by GitHub
commit 551841e557
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 26 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -55,16 +55,34 @@ def get_async_tool_function_and_apply_extra_params(
extra_params = {k: v for k, v in extra_params.items() if k in sig.parameters} extra_params = {k: v for k, v in extra_params.items() if k in sig.parameters}
partial_func = partial(function, **extra_params) partial_func = partial(function, **extra_params)
# Remove the 'frozen' keyword arguments from the signature
# python-genai uses the signature to infer the tool properties for native function calling
parameters = []
for name, parameter in sig.parameters.items():
# Exclude keyword arguments that are frozen
if name in extra_params:
continue
# Keep remaining parameters
parameters.append(parameter)
new_sig = inspect.Signature(
parameters=parameters, return_annotation=sig.return_annotation
)
if inspect.iscoroutinefunction(function): if inspect.iscoroutinefunction(function):
update_wrapper(partial_func, function) # wrap the functools.partial as python-genai has trouble with it
return partial_func # https://github.com/googleapis/python-genai/issues/907
async def new_function(*args, **kwargs):
return await partial_func(*args, **kwargs)
else: else:
# Make it a coroutine function # Make it a coroutine function when it is not already
async def new_function(*args, **kwargs): async def new_function(*args, **kwargs):
return partial_func(*args, **kwargs) return partial_func(*args, **kwargs)
update_wrapper(new_function, function) update_wrapper(new_function, function)
return new_function new_function.__signature__ = new_sig
return new_function
def get_tools( def get_tools(
@ -286,15 +304,15 @@ def convert_function_to_pydantic_model(func: Callable) -> type[BaseModel]:
field_defs = {} field_defs = {}
for name, param in parameters.items(): for name, param in parameters.items():
type_hint = type_hints.get(name, Any) type_hint = type_hints.get(name, Any)
default_value = param.default if param.default is not param.empty else ... default_value = param.default if param.default is not param.empty else ...
param_description = function_param_descriptions.get(name, None) param_description = function_param_descriptions.get(name, None)
if param_description: if param_description:
field_defs[name] = type_hint, Field( field_defs[name] = (
default_value, description=param_description type_hint,
Field(default_value, description=param_description),
) )
else: else:
field_defs[name] = type_hint, default_value field_defs[name] = type_hint, default_value