Przeglądaj źródła

fix: updatepost_response_handler nativ tools handler to use parameters from properties fixing defaults always used

Peter De-Ath 2 miesięcy temu
rodzic
commit
3b39807377
1 zmienionych plików z 2 dodań i 4 usunięć
  1. 2 4
      backend/open_webui/utils/middleware.py

+ 2 - 4
backend/open_webui/utils/middleware.py

@@ -1759,14 +1759,12 @@ async def process_chat_response(
                             spec = tool.get("spec", {})
                             spec = tool.get("spec", {})
 
 
                             try:
                             try:
-                                required_params = spec.get("parameters", {}).get(
-                                    "required", []
-                                )
+                                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
                                     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