lint

Signed-off-by: Mihai Criveti <crivetimihai@gmail.com>

Author: crivetimihai

mcpgateway/services/gateway_service.py

@@ -3547,10 +3547,7 @@ def _validate_tools(self, tools: list[dict[str, Any]], context: str = "default")
                 validation_errors.append(error_msg)
             except ValueError as e:
                 if "JSON structure exceeds maximum depth" in str(e):
-                    error_msg = (
-                        f"Tool '{tool_name}' schema too deeply nested. "
-                        f"Current depth limit: {settings.validation_max_json_depth}"
-                    )
+                    error_msg = f"Tool '{tool_name}' schema too deeply nested. " f"Current depth limit: {settings.validation_max_json_depth}"
                     logger.error(error_msg)
                     logger.warning("Consider increasing VALIDATION_MAX_JSON_DEPTH environment variable")
                 else:
@@ -3563,23 +3560,14 @@ def _validate_tools(self, tools: list[dict[str, Any]], context: str = "default")
                 validation_errors.append(error_msg)
 
         if validation_errors:
-            logger.warning(
-                f"Tool validation completed with {len(validation_errors)} error(s). "
-                f"Successfully validated {len(valid_tools)} tool(s)."
-            )
+            logger.warning(f"Tool validation completed with {len(validation_errors)} error(s). " f"Successfully validated {len(valid_tools)} tool(s).")
             for err in validation_errors[:3]:
                 logger.debug(f"Validation error: {err}")
 
         if not valid_tools and validation_errors:
             if context == "oauth":
-                raise OAuthToolValidationError(
-                    f"OAuth tool fetch failed: all {len(tools)} tools failed validation. "
-                    f"First error: {validation_errors[0][:200]}"
-                )
-            raise GatewayConnectionError(
-                f"Failed to fetch tools: All {len(tools)} tools failed validation. "
-                f"First error: {validation_errors[0][:200]}"
-            )
+                raise OAuthToolValidationError(f"OAuth tool fetch failed: all {len(tools)} tools failed validation. " f"First error: {validation_errors[0][:200]}")
+            raise GatewayConnectionError(f"Failed to fetch tools: All {len(tools)} tools failed validation. " f"First error: {validation_errors[0][:200]}")
 
         return valid_tools
 
@@ -3610,9 +3598,7 @@ async def _connect_to_sse_server_without_validation(self, server_url: str, authe
 
                     response = await session.list_tools()
                     tools = response.tools
-                    tools = [
-                        tool.model_dump(by_alias=True, exclude_none=True, exclude_unset=True) for tool in tools
-                    ]
+                    tools = [tool.model_dump(by_alias=True, exclude_none=True, exclude_unset=True) for tool in tools]
 
                     tools = self._validate_tools(tools, context="oauth")
                     if tools:
@@ -3755,9 +3741,7 @@ def get_httpx_client_factory(
 
                 response = await session.list_tools()
                 tools = response.tools
-                tools = [
-                    tool.model_dump(by_alias=True, exclude_none=True, exclude_unset=True) for tool in tools
-                ]
+                tools = [tool.model_dump(by_alias=True, exclude_none=True, exclude_unset=True) for tool in tools]
 
                 tools = self._validate_tools(tools)
                 if tools:
@@ -3897,9 +3881,7 @@ def get_httpx_client_factory(
 
                 response = await session.list_tools()
                 tools = response.tools
-                tools = [
-                    tool.model_dump(by_alias=True, exclude_none=True, exclude_unset=True) for tool in tools
-                ]
+                tools = [tool.model_dump(by_alias=True, exclude_none=True, exclude_unset=True) for tool in tools]
 
                 tools = self._validate_tools(tools)
                 for tool in tools:

plugins/tools_telemetry_exporter/__init__.py

@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 """Tools Telemetry Exporter Plugin - exports tool invocation telemetry to OpenTelemetry."""
 
 from plugins.tools_telemetry_exporter.telemetry_exporter import ToolsTelemetryExporterPlugin