style: apply linting fixes across codebase

Apply automatic linting fixes from Black and Ruff:
- Import statement ordering
- Remove unused imports
- Code formatting fixes
- Type annotation improvements

Author: Sakeeb91

app/agent/engine.py

@@ -9,9 +9,9 @@
 import asyncio
 import time
 import uuid
+from collections.abc import Callable
 from dataclasses import dataclass
 from typing import Any
-from collections.abc import Callable
 
 from smolagents import CodeAgent
 from smolagents.agents import LogLevel

app/agent/model_backends.py

@@ -3,7 +3,8 @@
 """
 
 import asyncio
-from typing import Any, Coroutine, TypeVar
+from collections.abc import Coroutine
+from typing import Any, TypeVar
 
 from smolagents import InferenceClientModel
 from smolagents.models import ChatMessage, MessageRole, Model, get_clean_message_list

app/agent/tools.py

@@ -13,8 +13,8 @@
 import asyncio
 import re
 import time
-from collections.abc import Callable
-from typing import Any, Coroutine, TypeVar
+from collections.abc import Callable, Coroutine
+from typing import Any, TypeVar
 
 from smolagents import tool
 
@@ -88,8 +88,8 @@ def sql_executor(query: str) -> str:
 
         {schema_placeholder}
         """
-        from db.executor import SafeQueryExecutor
         from app.monitoring.tracing import get_tracing_manager
+        from db.executor import SafeQueryExecutor
 
         logger.info(
             "sql_executor_tool_called",
@@ -317,8 +317,8 @@ def schema_inspector(table_name: str) -> str:
         Returns:
             Detailed schema information for the table, or error if not found.
         """
-        from db.schema import SchemaIntrospector
         from app.monitoring.tracing import get_tracing_manager
+        from db.schema import SchemaIntrospector
 
         logger.info(
             "schema_inspector_tool_called",
@@ -444,9 +444,9 @@ def sql_generator(question: str) -> str:
         Returns:
             Generated SQL query or error message.
         """
+        from app.monitoring.tracing import get_tracing_manager
         from models.inference import InferenceEngine
         from models.prompts import build_prompt
-        from app.monitoring.tracing import get_tracing_manager
 
         logger.info(
             "sql_generator_tool_called",

app/explanations/visualizers.py

@@ -6,8 +6,6 @@
 visualizations.
 """
 
-from typing import Any
-
 from app.explanations.models import (
     ClauseBreakdown,
     ExecutionPlanNode,

app/routes_examples.py

@@ -17,7 +17,7 @@
     validate_natural_language_query,
 )
 from app.security.input_validation import validate_sql_query
-from db.examples import ExampleRecord, ExampleSearchResult, get_example_store
+from db.examples import ExampleRecord, get_example_store
 
 logger = get_logger(__name__)
 

app/routes_models.py

@@ -11,7 +11,6 @@
 from app.security import limiter, require_auth, require_mutation_scope
 from models.versioning import (
     ModelVersion,
-    ModelVersionManager,
     ModelVersionStatus,
     get_model_version_manager,
 )

app/security/auth.py

@@ -8,10 +8,10 @@
 """
 
 import secrets
-from collections.abc import Awaitable, Callable
+from collections.abc import Awaitable, Callable, Iterable
 from dataclasses import dataclass
 from datetime import datetime, timedelta
-from typing import Any, Iterable, Literal
+from typing import Any, Literal
 
 from fastapi import Depends, Security
 from fastapi.security import APIKeyHeader, HTTPAuthorizationCredentials, HTTPBearer

app/security/rate_limiting.py

@@ -7,7 +7,8 @@
 - Rate limit exceeded handling
 """
 
-from typing import Any, Callable, cast
+from collections.abc import Callable
+from typing import Any, cast
 
 from fastapi import FastAPI, Request
 from slowapi import Limiter, _rate_limit_exceeded_handler

app/text2sql_engine.py

@@ -24,14 +24,14 @@
 from enum import Enum
 from typing import Any
 
-from app.config import FewShotSettings, get_settings
 from app.cache import (
     cache_prompt,
     cache_schema,
     generate_prompt_hash,
     get_cached_prompt,
     get_cached_schema,
 )
+from app.config import FewShotSettings, get_settings
 from app.exceptions import (
     AgentMaxStepsExceededException,
     CircuitBreakerOpenException,

db/__init__.py

@@ -57,12 +57,26 @@
     get_async_driver,
     get_dialect_adapter,
 )
+from db.examples import (
+    ExampleRecord,
+    ExampleSearchResult,
+    ExampleStore,
+    get_example_store,
+    reset_example_store,
+)
 from db.executor import (
     QueryResult,
     QueryValidator,
     SafeQueryExecutor,
     sanitize_identifier,
 )
+from db.feedback import (
+    FeedbackRecord,
+    FeedbackStatus,
+    FeedbackStore,
+    get_feedback_store,
+    reset_feedback_store,
+)
 from db.registry import (
     DatabaseConfig,
     DatabaseHealth,
@@ -81,20 +95,6 @@
     TableInfo,
     get_sample_data,
 )
-from db.examples import (
-    ExampleRecord,
-    ExampleSearchResult,
-    ExampleStore,
-    get_example_store,
-    reset_example_store,
-)
-from db.feedback import (
-    FeedbackRecord,
-    FeedbackStatus,
-    FeedbackStore,
-    get_feedback_store,
-    reset_feedback_store,
-)
 
 __all__ = [
     # Connection management