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

fix: tools not working

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-26 12:21:05 -07:00
parent 565ea92693
commit badbe4ea06
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
backend/open_webui/main.py
View File

@ -439,10 +439,17 @@ async def chat_completion_tools_handler(
tool_function_params = result.get("parameters", {}) tool_function_params = result.get("parameters", {})
try: try:
required_params = (
tools[tool_function_name]
.get("spec", {})
.get("parameters", {})
.get("required", [])
)
tool_function = tools[tool_function_name]["callable"] tool_function = tools[tool_function_name]["callable"]
sig = inspect.signature(tool_function)
tool_function_params = { tool_function_params = {
k: v for k, v in tool_function_params.items() if k in sig.parameters k: v
for k, v in tool_function_params.items()
if k in required_params
} }
tool_output = await tool_function(**tool_function_params) tool_output = await tool_function(**tool_function_params)