OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-24 23:04:36 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge pull request #11384 from Peter-De-Ath/hotfix/tool-calls-params

Browse Source 
fix: bug where default params in tools calls always used
This commit is contained in:
Timothy Jaeryang Baek 2025-03-07 19:58:05 -04:00 committed by GitHub
commit 88303093c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -189,17 +189,15 @@ async def chat_completion_tools_handler(
tool_function_params = tool_call.get("parameters", {}) tool_function_params = tool_call.get("parameters", {})
try: try:
required_params = ( spec = tools[tool_function_name].get("spec", {})
tools[tool_function_name] allowed_params = (
.get("spec", {}) spec.get("parameters", {}).get("properties", {}).keys()
.get("parameters", {})
.get("required", [])
) )
tool_function = tools[tool_function_name]["callable"] tool_function = tools[tool_function_name]["callable"]
tool_function_params = { tool_function_params = {
k: v k: v
for k, v in tool_function_params.items() for k, v in tool_function_params.items()
if k in required_params if k in allowed_params
} }
tool_output = await tool_function(**tool_function_params) tool_output = await tool_function(**tool_function_params)
@ -1765,14 +1763,16 @@ async def process_chat_response(
spec = tool.get("spec", {}) spec = tool.get("spec", {})
try: try:
required_params = spec.get("parameters", {}).get( allowed_params = (
"required", [] spec.get("parameters", {})
.get("properties", {})
.keys()
) )
tool_function = tool["callable"] tool_function = tool["callable"]
tool_function_params = { tool_function_params = {
k: v k: v
for k, v in tool_function_params.items() for k, v in tool_function_params.items()
if k in required_params if k in allowed_params
} }
tool_result = await tool_function( tool_result = await tool_function(
**tool_function_params **tool_function_params