Peter De-Ath il y a 2 mois
Parent
commit
7e3be60f0e
1 fichiers modifiés avec 11 ajouts et 3 suppressions
  1. 11 3
      backend/open_webui/utils/middleware.py

+ 11 - 3
backend/open_webui/utils/middleware.py

@@ -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