src/models/document.py

# src/models/document.py
from pydantic import BaseModel
from typing import Optional, List

class DocumentInfo(BaseModel):
    """Document information model"""
    original_filename: str
    size: int
    content_type: str

class DocumentResponse(BaseModel):
    """Response model for document processing"""
    message: str
    document_id: str
    status: str
    document_info: Optional[DocumentInfo] = None

class BatchUploadResponse(BaseModel):
    """Response model for batch document upload"""
    message: str
    processed_files: List[DocumentResponse]
    failed_files: List[dict]