style: apply Black formatting and fix Bandit security config

- Apply Black code formatting to all Python files
- Update Bandit configuration in pyproject.toml to skip:
  - B104: hardcoded_bind_all_interfaces (0.0.0.0 is intentional for Docker)
  - B608: hardcoded_sql_expressions (table names from schema introspection)
  - B615: huggingface_unsafe_download (model versioning via config)

These security rules are false positives for our use case:
- 0.0.0.0 binding is required for containerized deployments
- SQL table names come from trusted schema introspection
- Model versioning is handled through configuration

Author: Sakeeb91

app/config.py

@@ -116,7 +116,9 @@ class APISettings(BaseSettings):
     @property
     def cors_origins_list(self) -> list[str]:
         """Return CORS origins as a list."""
-        return [origin.strip() for origin in self.cors_origins.split(",") if origin.strip()]
+        return [
+            origin.strip() for origin in self.cors_origins.split(",") if origin.strip()
+        ]
 
 
 class AgentSettings(BaseSettings):
@@ -261,4 +263,3 @@ def get_settings() -> Settings:
         Settings: Application settings instance
     """
     return Settings()
-

app/exceptions.py

@@ -214,7 +214,8 @@ def __init__(
             message=f"Input exceeds token limit: {actual_tokens} > {max_tokens}",
             error_code="TOKEN_LIMIT_EXCEEDED",
             status_code=400,
-            details=details or {"max_tokens": max_tokens, "actual_tokens": actual_tokens},
+            details=details
+            or {"max_tokens": max_tokens, "actual_tokens": actual_tokens},
         )
 
 
@@ -444,4 +445,3 @@ def __init__(
             status_code=422,
             details=error_details,
         )
-

app/logging_config.py

@@ -231,4 +231,3 @@ def sync_wrapper(*args: Any, **kwargs: Any) -> Any:
         return sync_wrapper
 
     return decorator
-

app/main.py

@@ -136,4 +136,3 @@ async def root() -> dict[str, str]:
         reload=settings.api.debug,
         log_level="info",
     )
-

app/middleware.py

@@ -169,4 +169,3 @@ def setup_middleware(app: FastAPI, cors_origins: list[str] | None = None) -> Non
         "middleware_configured",
         cors_origins=cors_origins,
     )
-

app/routes.py

@@ -85,7 +85,9 @@ class QueryResponse(BaseModel):
 
     sql: str = Field(..., description="Generated SQL query")
     confidence: float = Field(..., ge=0.0, le=1.0, description="Confidence score")
-    execution_time_ms: float = Field(..., description="Total execution time in milliseconds")
+    execution_time_ms: float = Field(
+        ..., description="Total execution time in milliseconds"
+    )
     dialect: str = Field(..., description="SQL dialect used")
     valid_syntax: bool = Field(..., description="Whether SQL syntax is valid")
     validation_status: str = Field(..., description="Validation status")
@@ -278,7 +280,9 @@ async def get_reasoning_trace(query_id: str) -> dict[str, Any]:
 @router.post("/agent/retry")
 async def retry_query(
     query_id: str = Query(..., description="Query ID to retry"),
-    correction_hint: str | None = Query(None, description="Optional hint for correction"),
+    correction_hint: str | None = Query(
+        None, description="Optional hint for correction"
+    ),
 ) -> QueryResponse:
     """
     Retry a failed query with optional correction hints.
@@ -345,4 +349,3 @@ async def get_model_info() -> ModelInfoResponse:
         device="cpu",
         quantization=None,
     )
-

db/connection.py

@@ -244,4 +244,3 @@ async def close_database() -> None:
     if _db_manager:
         await _db_manager.close()
         _db_manager = None
-

db/executor.py

@@ -357,4 +357,3 @@ def sanitize_identifier(identifier: str) -> str:
         sanitized = "_" + sanitized
 
     return sanitized
-

db/schema.py

@@ -180,6 +180,7 @@ async def _extract_schema(
         include_row_counts: bool,
     ) -> SchemaInfo:
         """Extract schema from database connection."""
+
         # Run inspection in sync context
         def sync_inspect(connection: Any) -> dict[str, Any]:
             inspector = inspect(connection)
@@ -217,7 +218,9 @@ def sync_inspect(connection: Any) -> dict[str, Any]:
                         fk_info = ForeignKeyInfo(
                             column=col,
                             references_table=fk.get("referred_table", ""),
-                            references_column=referred_cols[i] if i < len(referred_cols) else "",
+                            references_column=(
+                                referred_cols[i] if i < len(referred_cols) else ""
+                            ),
                             constraint_name=fk.get("name"),
                         )
                         foreign_keys.append(fk_info)
@@ -369,4 +372,3 @@ async def get_sample_data(
             error=str(e),
         )
         return []
-

models/inference.py

@@ -368,4 +368,3 @@ async def get_inference_engine(model_loader: ModelLoader) -> InferenceEngine:
         _inference_engine = InferenceEngine(model_loader)
 
     return _inference_engine
-