Linting fixes

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

Author: madhav165

mcpgateway/config.py

@@ -41,7 +41,7 @@
 import jq
 from jsonpath_ng.ext import parse
 from jsonpath_ng.jsonpath import JSONPath
-from pydantic import Field, field_validator
+from pydantic import field_validator
 from pydantic_settings import BaseSettings, NoDecode, SettingsConfigDict
 
 

mcpgateway/db.py

@@ -105,12 +105,14 @@
     connect_args=connect_args,
 )
 
+
 # ---------------------------------------------------------------------------
 # 6. Function to return UTC timestamp
 # ---------------------------------------------------------------------------
 def utc_now():
     return datetime.now(timezone.utc)
 
+
 # Session factory
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 

mcpgateway/schemas.py

@@ -35,16 +35,7 @@
 from mcpgateway.utils.services_auth import decode_auth, encode_auth
 
 # Third-Party
-from pydantic import (
-    AnyHttpUrl,
-    BaseModel,
-    Field,
-    model_validator,
-    ConfigDict,
-    field_serializer,
-    field_validator,
-    ValidationInfo
-)
+from pydantic import AnyHttpUrl, BaseModel, ConfigDict, Field, field_serializer, field_validator, model_validator, ValidationInfo
 
 logger = logging.getLogger(__name__)
 
@@ -546,7 +537,6 @@ class PromptArgument(BaseModelWithConfigDict):
     description: Optional[str] = Field(None, description="Argument description")
     required: bool = Field(default=False, description="Whether argument is required")
 
-
     model_config: ConfigDict = ConfigDict(
         **{
             # start with every key from the base
@@ -555,9 +545,9 @@ class PromptArgument(BaseModelWithConfigDict):
             "json_schema_extra": {
                 **BaseModelWithConfigDict.model_config.get("json_schema_extra", {}),
                 "example": {
-                    "name":        "language",
+                    "name": "language",
                     "description": "Programming language",
-                    "required":    True,
+                    "required": True,
                 },
             },
         }
@@ -702,7 +692,7 @@ def _process_auth_fields(info: ValidationInfo) -> Optional[Dict[str, Any]]:
         This method is called based on the selected auth_type.
 
         Args:
-            values: Dict containing auth fields
+            info: ValidationInfo containing auth fields
 
         Returns:
             Dict with encoded auth
@@ -792,7 +782,7 @@ def create_auth_value(cls, v, info):
 
         Args:
             v: Input URL
-            values: Dict containing auth_type
+            info: ValidationInfo containing auth_type
 
         Returns:
             str: Auth value or URL

mcpgateway/translate.py

@@ -382,12 +382,15 @@ async def pump_sse_to_stdio():
 
         await asyncio.gather(read_stdout(), pump_sse_to_stdio())
 
+
 def start_stdio(cmd, port, log_level, cors):
     return asyncio.run(_run_stdio_to_sse(cmd, port, log_level, cors))
 
+
 def start_sse(url, bearer, log_level):
     return asyncio.run(_run_sse_to_stdio(url, bearer, log_level))
 
+
 def main(argv: Optional[Sequence[str]] | None = None) -> None:  # entry-point
     args = _parse_args(argv or sys.argv[1:])
     logging.basicConfig(

mcpgateway/types.py

@@ -22,7 +22,7 @@
 from typing import Any, Dict, List, Literal, Optional, Union
 
 # Third-Party
-from pydantic import AnyHttpUrl, AnyUrl, BaseModel, Field, ConfigDict
+from pydantic import AnyHttpUrl, AnyUrl, BaseModel, ConfigDict, Field
 
 
 class Role(str, Enum):
@@ -282,6 +282,7 @@ class InitializeResult(BaseModel):
         populate_by_name=True,
     )
 
+
 # Message types
 class Message(BaseModel):
     """A message in a conversation.
@@ -464,6 +465,7 @@ class ListResourceTemplatesResult(BaseModel):
         populate_by_name=True,
     )
 
+
 # Root types
 class FileUrl(AnyUrl):
     """A specialized URL type for local file-scheme resources.

mcpgateway/version.py

@@ -253,7 +253,7 @@ def _build_payload(
     """
     db_ver, db_ok = _database_version()
     return {
-        "timestamp": datetime.now(timezone.utc).isoformat().replace('+00:00', 'Z'),
+        "timestamp": datetime.now(timezone.utc).isoformat().replace("+00:00", "Z"),
         "host": HOSTNAME,
         "uptime_seconds": int(time.time() - START_TIME),
         "app": {

tests/conftest.py

@@ -13,14 +13,14 @@
 from unittest.mock import AsyncMock, patch
 
 # First-Party
+from mcpgateway import translate
 from mcpgateway.config import Settings
 from mcpgateway.db import Base
 
 # Third-Party
 import pytest
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
-from mcpgateway import translate
 
 
 @pytest.fixture(scope="session")
@@ -98,6 +98,7 @@ def mock_websocket():
     mock.close = AsyncMock()
     return mock
 
+
 # @pytest.fixture(scope="session", autouse=True)
 # def _patch_stdio_first():
 #     """
@@ -106,4 +107,4 @@ def mock_websocket():
 #     """
 #     import mcpgateway.translate as translate
 #     translate._run_stdio_to_sse = AsyncMock(return_value=None)
-#     translate._run_sse_to_stdio = AsyncMock(return_value=None)
+#     translate._run_sse_to_stdio = AsyncMock(return_value=None)

tests/unit/mcpgateway/test_translate.py

@@ -42,12 +42,14 @@
 import types
 from typing import Sequence
 from unittest.mock import AsyncMock, Mock
-# import inspect
 
 # Third-Party
 from fastapi.testclient import TestClient
 import pytest
 
+# import inspect
+
+
 # ---------------------------------------------------------------------------#
 # Pytest fixtures                                                            #
 # ---------------------------------------------------------------------------#
@@ -253,7 +255,11 @@ async def test_fastapi_message_endpoint_invalid_json(translate):
     app = translate._build_fastapi(ps, stdio)
     client = TestClient(app)
 
-    response = client.post("/message", content="invalid json", headers={"content-type": "application/json"},)
+    response = client.post(
+        "/message",
+        content="invalid json",
+        headers={"content-type": "application/json"},
+    )
     assert response.status_code == 400
     assert "Invalid JSON payload" in response.text