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

formating

Browse Source
This commit is contained in:
Peter De-Ath 2025-03-07 20:37:22 +00:00
parent 3b39807377
commit 7e3be60f0e
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -190,10 +190,14 @@ async def chat_completion_tools_handler(
try: try:
spec = tools[tool_function_name].get("spec", {}) spec = tools[tool_function_name].get("spec", {})
allowed_params = spec.get("parameters", {}).get("properties", {}).keys() allowed_params = (
spec.get("parameters", {}).get("properties", {}).keys()
)
tool_function = tools[tool_function_name]["callable"] tool_function = tools[tool_function_name]["callable"]
tool_function_params = { tool_function_params = {
k: v for k, v in tool_function_params.items() if k in allowed_params k: v
for k, v in tool_function_params.items()
if k in allowed_params
} }
tool_output = await tool_function(**tool_function_params) tool_output = await tool_function(**tool_function_params)
@ -1759,7 +1763,11 @@ async def process_chat_response(
spec = tool.get("spec", {}) spec = tool.get("spec", {})
try: try:
allowed_params = spec.get("parameters", {}).get("properties", {}).keys() allowed_params = (
spec.get("parameters", {})
.get("properties", {})
.keys()
)
tool_function = tool["callable"] tool_function = tool["callable"]
tool_function_params = { tool_function_params = {
k: v k: v