from typing import Optional, List, Dict, Any
from pydantic import BaseModel, Field


class QueryRequest(BaseModel):
    instance_id: int
    database: Optional[str] = None
    sql: str = Field(..., min_length=1)


class QueryResult(BaseModel):
    columns: List[str]
    rows: List[Dict[str, Any]]
    total: int
    elapsed_ms: int
    truncated: bool = False