linting

Author: kevalmahajan

mcpgateway/admin.py

@@ -755,7 +755,7 @@ async def admin_add_gateway(request: Request, db: Session = Depends(get_db), use
         name=form["name"],
         url=form["url"],
         description=form.get("description"),
-        transport=form.get("transport","SSE"),
+        transport=form.get("transport", "SSE"),
         auth_type=form.get("auth_type", ""),
         auth_username=form.get("auth_username", ""),
         auth_password=form.get("auth_password", ""),
@@ -770,12 +770,12 @@ async def admin_add_gateway(request: Request, db: Session = Depends(get_db), use
     except Exception as ex:
         if isinstance(ex, GatewayConnectionError):
             return RedirectResponse(f"{root_path}/admin#gateways", status_code=502)
-        elif isinstance(ex, ValueError):
+        if isinstance(ex, ValueError):
             return RedirectResponse(f"{root_path}/admin#gateways", status_code=400)
-        elif isinstance(ex, RuntimeError):
-            return RedirectResponse(f"{root_path}/admin#gateways", status_code=500)
-        else:
+        if isinstance(ex, RuntimeError):
             return RedirectResponse(f"{root_path}/admin#gateways", status_code=500)
+        
+        return RedirectResponse(f"{root_path}/admin#gateways", status_code=500)
 
 
 @admin_router.post("/gateways/{gateway_id}/edit")
@@ -807,7 +807,7 @@ async def admin_edit_gateway(
         name=form["name"],
         url=form["url"],
         description=form.get("description"),
-        transport=form.get("transport","SSE"),
+        transport=form.get("transport", "SSE"),
         auth_type=form.get("auth_type", None),
         auth_username=form.get("auth_username", None),
         auth_password=form.get("auth_password", None),

mcpgateway/services/gateway_service.py

@@ -576,8 +576,8 @@ async def check_health_of_gateways(self, gateways: List[DbGateway]) -> bool:
 
                     # Perform the GET and raise on 4xx/5xx
                     if (gateway.transport).lower() == "sse":
-                        timeout = httpx.Timeout(settings.health_check_timeout) 
-                        async with client.stream("GET", gateway.url, headers=headers,timeout=timeout) as response:
+                        timeout = httpx.Timeout(settings.health_check_timeout)
+                        async with client.stream("GET", gateway.url, headers=headers, timeout=timeout) as response:
                             # This will raise immediately if status is 4xx/5xx
                             response.raise_for_status()
                     elif (gateway.transport).lower() == "streamablehttp":
@@ -640,7 +640,7 @@ async def subscribe_events(self) -> AsyncGenerator[Dict[str, Any], None]:
         finally:
             self._event_subscribers.remove(queue)
 
-    async def _initialize_gateway(self, url: str, authentication: Optional[Dict[str, str]] = None, transport: str= "sse") -> Any:
+    async def _initialize_gateway(self, url: str, authentication: Optional[Dict[str, str]] = None, transport: str = "SSE") -> Any:
         """Initialize connection to a gateway and retrieve its capabilities.
 
         Args:

mcpgateway/transports/streamablehttp_transport.py

@@ -25,8 +25,8 @@
 from typing import List, Union
 from uuid import uuid4
 
-from mcp import types
 from fastapi.security.utils import get_authorization_scheme_param
+from mcp import types
 from mcp.server.lowlevel import Server
 from mcp.server.streamable_http import (
     EventCallback,