Merge pull request #196 from IBM/cleanup-isort

Cleanup isort and add import headers

Author: crivetimihai

.isort.cfg

@@ -1,11 +1,14 @@
 # -*- coding: utf-8 -*-
+# Standard
 from logging.config import fileConfig
 
-from sqlalchemy import engine_from_config, pool
-
+# First-Party
 from alembic import context
 from mcpgateway.config import settings
 
+# Third-Party
+from sqlalchemy import engine_from_config, pool
+
 # from mcpgateway.db import get_metadata
 # target_metadata = get_metadata()
 
@@ -18,6 +21,7 @@
 if config.config_file_name is not None:
     fileConfig(config.config_file_name)
 
+# First-Party
 # add your model's MetaData object here
 # for 'autogenerate' support
 # from myapp import mymodel

alembic/env.py

@@ -6,16 +6,19 @@
 Create Date: 2025-06-26 21:29:59.117140
 
 """
-import uuid
+# Standard
 from typing import Sequence, Union
+import uuid
 
-import sqlalchemy as sa
-from sqlalchemy.orm import Session
-
+# First-Party
 from alembic import op
 from mcpgateway.config import settings
 from mcpgateway.utils.create_slug import slugify
 
+# Third-Party
+import sqlalchemy as sa
+from sqlalchemy.orm import Session
+
 # revision identifiers, used by Alembic.
 revision: str = 'b77ca9d2de7e'
 down_revision: Union[str, Sequence[str], None] = None

alembic/versions/b77ca9d2de7e_uuid_pk_and_slug_refactor.py

@@ -6,12 +6,15 @@
 Create Date: 2025-06-27 21:45:35.099713
 
 """
+# Standard
 from typing import Sequence, Union
 
-import sqlalchemy as sa
-
+# First-Party
 from alembic import op
 
+# Third-Party
+import sqlalchemy as sa
+
 # revision identifiers, used by Alembic.
 revision: str = 'e4fc04d1a442'
 down_revision: Union[str, Sequence[str], None] = 'b77ca9d2de7e'

alembic/versions/e4fc04d1a442_add_annotations_to_tables.py

@@ -14,6 +14,7 @@
 Reference: https://stackoverflow.com/questions/10855197/frequent-worker-timeout
 """
 
+# First-Party
 # Import Pydantic Settings singleton
 from mcpgateway.config import settings
 

gunicorn.config.py

@@ -17,14 +17,12 @@
 underlying data.
 """
 
+# Standard
 import json
 import logging
 from typing import Any, Dict, List, Union
 
-from fastapi import APIRouter, Depends, HTTPException, Request
-from fastapi.responses import HTMLResponse, JSONResponse, RedirectResponse
-from sqlalchemy.orm import Session
-
+# First-Party
 from mcpgateway.config import settings
 from mcpgateway.db import get_db
 from mcpgateway.schemas import (
@@ -61,6 +59,11 @@
 from mcpgateway.utils.create_jwt_token import get_jwt_token
 from mcpgateway.utils.verify_credentials import require_auth, require_basic_auth
 
+# Third-Party
+from fastapi import APIRouter, Depends, HTTPException, Request
+from fastapi.responses import HTMLResponse, JSONResponse, RedirectResponse
+from sqlalchemy.orm import Session
+
 # Initialize services
 server_service = ServerService()
 tool_service = ToolService()

mcpgateway/admin.py

@@ -12,10 +12,11 @@
 - Thread-safe operations
 """
 
+# Standard
 import asyncio
+from dataclasses import dataclass
 import logging
 import time
-from dataclasses import dataclass
 from typing import Any, Dict, Optional
 
 logger = logging.getLogger(__name__)

mcpgateway/cache/resource_cache.py

@@ -9,35 +9,40 @@
 using Redis or SQLAlchemy as optional backends for shared state between workers.
 """
 
+# Standard
 import asyncio
 import json
 import logging
 import time
 from typing import Any, Dict, Optional
 
-import httpx
-from fastapi import HTTPException, status
-
+# First-Party
 from mcpgateway.config import settings
-from mcpgateway.db import SessionMessageRecord, SessionRecord, get_db
+from mcpgateway.db import get_db, SessionMessageRecord, SessionRecord
 from mcpgateway.services import PromptService, ResourceService, ToolService
 from mcpgateway.transports import SSETransport
 from mcpgateway.types import Implementation, InitializeResult, ServerCapabilities
 
+# Third-Party
+from fastapi import HTTPException, status
+import httpx
+
 logger = logging.getLogger(__name__)
 
 tool_service = ToolService()
 resource_service = ResourceService()
 prompt_service = PromptService()
 
 try:
+    # Third-Party
     from redis.asyncio import Redis
 
     REDIS_AVAILABLE = True
 except ImportError:
     REDIS_AVAILABLE = False
 
 try:
+    # Third-Party
     from sqlalchemy import func
 
     SQLALCHEMY_AVAILABLE = True

mcpgateway/cache/session_registry.py

@@ -33,16 +33,20 @@
 ```
 """
 
+# Future
 from __future__ import annotations
 
+# Standard
 import os
 import sys
 from typing import List
 
-import uvicorn
-
+# First-Party
 from mcpgateway import __version__
 
+# Third-Party
+import uvicorn
+
 # ---------------------------------------------------------------------------
 # Configuration defaults (overridable via environment variables)
 # ---------------------------------------------------------------------------

mcpgateway/cli.py

@@ -29,14 +29,16 @@
 - HEALTH_CHECK_INTERVAL: Gateway health check interval (default: 60)
 """
 
-import json
+# Standard
 from functools import lru_cache
 from importlib.resources import files
+import json
 from pathlib import Path
 from typing import Annotated, Any, Dict, List, Optional, Set, Union
 
-import jq
+# Third-Party
 from fastapi import HTTPException
+import jq
 from jsonpath_ng.ext import parse
 from jsonpath_ng.jsonpath import JSONPath
 from pydantic import Field, field_validator