Additional return codes durig failure in register gateway

Signed-off-by: Madhav Kandukuri <[email protected]>

Author: madhav165

mcpgateway/admin.py

@@ -48,7 +48,7 @@
     ToolRead,
     ToolUpdate,
 )
-from mcpgateway.services.gateway_service import GatewayService, GatewayConnectionError
+from mcpgateway.services.gateway_service import GatewayConnectionError, GatewayService
 from mcpgateway.services.prompt_service import PromptService
 from mcpgateway.services.resource_service import ResourceService
 from mcpgateway.services.root_service import RootService
@@ -769,6 +769,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):
+            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:
+            return RedirectResponse(f"{root_path}/admin#gateways", status_code=500)
 
 
 @admin_router.post("/gateways/{gateway_id}/edit")

mcpgateway/main.py

@@ -1510,6 +1510,12 @@ async def register_gateway(
     except Exception as ex:
         if isinstance(ex, GatewayConnectionError):
             return JSONResponse(content={"message": "Unable to connect to gateway"}, status_code=502)
+        elif isinstance(ex, ValueError):
+            return JSONResponse(content={"message": "Unable to process input"}, status_code=400)
+        elif isinstance(ex, RuntimeError):
+            return JSONResponse(content={"message": "Error during execution"}, status_code=500)
+        else:
+            return JSONResponse(content={"message": "Unexpected error"}, status_code=500)
 
 
 @gateway_router.get("/{gateway_id}", response_model=GatewayRead)