admin_add_server_exception_update_fix

Signed-off-by: Satya <[email protected]>

Author: TS0713

mcpgateway/admin.py

@@ -2293,12 +2293,9 @@ async def admin_add_gateway(request: Request, db: Session = Depends(get_db), use
         # Convert KeyError to ValidationError-like response
         return JSONResponse(content={"message": f"Missing required field: {e}", "success": False}, status_code=422)
 
-    except ValueError as ex:
+    except ValidationError as ex:
         # --- Getting only the custom message from the ValueError ---
-        error_ctx = []
-        logger.info(f" ALL -> {ex.errors()}")
-        for err in ex.errors():
-            error_ctx.append(str(err["ctx"]["error"]))
+        error_ctx = [str(err["ctx"]["error"]) for err in ex.errors()]
         return JSONResponse(content={"success": False, "message": "; ".join(error_ctx)}, status_code=422)
 
     try:

mcpgateway/schemas.py

@@ -1572,6 +1572,16 @@ class PromptInvocation(BaseModelWithConfigDict):
 
 # --- Transport Type ---
 class TransportType(str, Enum):
+    """
+    Enumeration of supported transport mechanisms for communication between components.
+
+    Attributes:
+        SSE (str): Server-Sent Events transport.
+        HTTP (str): Standard HTTP-based transport.
+        STDIO (str): Standard input/output transport.
+        STREAMABLEHTTP (str): HTTP transport with streaming.
+    """
+
     SSE = "SSE"
     HTTP = "HTTP"
     STDIO = "STDIO"
@@ -1690,6 +1700,24 @@ def create_auth_value(cls, v, info):
     @field_validator("transport")
     @classmethod
     def validate_transport(cls, v: str) -> str:
+        """
+        Validates that the given transport value is one of the supported TransportType values.
+
+        Args:
+            v (str): The transport value to validate.
+
+        Returns:
+            str: The validated transport value if it is valid.
+
+        Raises:
+            ValueError: If the provided value is not a valid transport type.
+
+        Valid transport types are defined in the TransportType enum:
+            - SSE
+            - HTTP
+            - STDIO
+            - STREAMABLEHTTP
+        """
         if v not in [t.value for t in TransportType]:
             raise ValueError(f"Invalid transport type: {v}. Must be one of: {', '.join([t.value for t in TransportType])}")
         return v