Improved import sorting in key files

Author: tanzilahmed0

backend/models/response_schemas.py

@@ -1,50 +1,41 @@
-from typing import Any, Optional, Generic, TypeVar, List, Dict
-from pydantic import BaseModel
 from datetime import datetime
 from enum import Enum
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 
-T = TypeVar("T")
+from pydantic import BaseModel
 
+T = TypeVar("T")
 
 class ApiResponse(BaseModel, Generic[T]):
     """Standard API response format matching the frontend contract"""
-
     success: bool
     data: Optional[T] = None
     error: Optional[str] = None
     message: Optional[str] = None
 
-
 class HealthStatus(BaseModel):
     """Health check status model"""
-
     status: str
     service: str
     version: str
     timestamp: str
     checks: dict
 
-
 class HealthChecks(BaseModel):
     """Individual health checks"""
-
     database: bool
     redis: bool
     storage: bool
     llm_service: bool
 
-
 class ValidationError(BaseModel):
     """Validation error details"""
-
     field: str
     message: str
     code: str
 
-
 class ApiError(BaseModel):
     """API error response format"""
-
     error: str
     message: str
     code: int
@@ -59,7 +50,6 @@ class ApiError(BaseModel):
 
 class User(BaseModel):
     """User model"""
-
     id: str
     email: str
     name: str
@@ -70,13 +60,11 @@ class User(BaseModel):
 
 class LoginRequest(BaseModel):
     """Google OAuth login request"""
-
     google_token: str
 
 
 class AuthResponse(BaseModel):
     """Authentication response"""
-
     user: User
     access_token: str
     refresh_token: str
@@ -85,7 +73,6 @@ class AuthResponse(BaseModel):
 
 class RefreshTokenRequest(BaseModel):
     """Refresh token request"""
-
     refresh_token: str
 
 
@@ -96,7 +83,6 @@ class RefreshTokenRequest(BaseModel):
 
 class ProjectStatus(str, Enum):
     """Project status enum"""
-
     UPLOADING = "uploading"
     PROCESSING = "processing"
     READY = "ready"
@@ -105,7 +91,6 @@ class ProjectStatus(str, Enum):
 
 class ColumnMetadata(BaseModel):
     """Column metadata model"""
-
     name: str
     type: str
     nullable: bool
@@ -115,7 +100,6 @@ class ColumnMetadata(BaseModel):
 
 class Project(BaseModel):
     """Project model"""
-
     id: str
     user_id: str
     name: str
@@ -132,29 +116,25 @@ class Project(BaseModel):
 
 class CreateProjectRequest(BaseModel):
     """Create project request"""
-
     name: str
     description: Optional[str] = None
 
 
 class CreateProjectResponse(BaseModel):
     """Create project response"""
-
     project: Project
     upload_url: str
     upload_fields: Dict[str, str]
 
 
 class PaginationParams(BaseModel):
     """Pagination parameters"""
-
     page: int = 1
     limit: int = 20
 
 
 class PaginatedResponse(BaseModel, Generic[T]):
     """Paginated response"""
-
     items: List[T]
     total: int
     page: int
@@ -164,7 +144,6 @@ class PaginatedResponse(BaseModel, Generic[T]):
 
 class UploadStatusResponse(BaseModel):
     """Upload status response"""
-
     project_id: str
     status: str
     progress: int
@@ -178,7 +157,6 @@ class UploadStatusResponse(BaseModel):
 
 class ChatMessage(BaseModel):
     """Chat message model"""
-
     id: str
     project_id: str
     user_id: str
@@ -190,14 +168,12 @@ class ChatMessage(BaseModel):
 
 class SendMessageRequest(BaseModel):
     """Send message request"""
-
     message: str
     context: Optional[List[str]] = None
 
 
 class QueryResult(BaseModel):
     """Query result model"""
-
     id: str
     query: str
     sql_query: str
@@ -210,14 +186,12 @@ class QueryResult(BaseModel):
 
 class SendMessageResponse(BaseModel):
     """Send message response"""
-
     message: ChatMessage
     result: QueryResult
 
 
 class CSVPreview(BaseModel):
     """CSV preview model"""
-
     columns: List[str]
     sample_data: List[List[Any]]
     total_rows: int
@@ -226,7 +200,6 @@ class CSVPreview(BaseModel):
 
 class QuerySuggestion(BaseModel):
     """Query suggestion model"""
-
     id: str
     text: str
     category: str

backend/tests/test_mock_endpoints.py

@@ -1,9 +1,10 @@
-import pytest
-from fastapi.testclient import TestClient
-import jwt
 import uuid
 from datetime import datetime, timedelta
 
+import jwt
+import pytest
+from fastapi.testclient import TestClient
+
 from main import app
 
 client = TestClient(app)