from pydantic import BaseModel
from typing import Optional
from datetime import datetime


class DocumentUpload(BaseModel):
    """Schema for document upload."""
    filename: str


class DocumentResponse(BaseModel):
    """Schema for document response."""
    id: str
    user_id: str
    filename: str
    file_type: Optional[str]
    file_size: Optional[int]
    created_at: datetime
    
    class Config:
        from_attributes = True


class DocumentListResponse(BaseModel):
    """Schema for document list response."""
    documents: list[DocumentResponse]
    total: int