from pydantic import BaseModel

class AgentConfig:
    """Configuration settings for the advanced research agent."""
    
    # Model settings
    WRITER_MODEL = "gemini-2.5-flash-lite-preview-06-17"
    PLANNER_MODEL = "gemini-2.5-flash-lite-preview-06-17"
    
    # Research settings - Deep search parameters
    INITIAL_SEARCH_RESULTS = 8  # For initial planning phase
    DEEP_DIVE_SEARCH_RESULTS = 12  # Per query in each section
    MAX_QUERIES_PER_SECTION = 4  # Limit queries to avoid rate limits
    
    # RAG settings - Document processing
    CHUNKS_TO_RETRIEVE = 50  # Initial retrieval before re-ranking
    CHUNKS_TO_USE_FOR_WRITING = 15  # Final chunks after re-ranking
    CHUNK_SIZE = 1000  # Size of text chunks
    CHUNK_OVERLAP = 200  # Overlap between chunks
    
    # LLM settings - Temperature controls
    WRITER_TEMPERATURE = 0.4  # Balanced creativity for writing
    PLANNER_TEMPERATURE = 0.2  # Low for structured planning
    VERIFICATION_TEMPERATURE = 0.1  # Very low for fact-checking
    
    # Quality settings
    ENABLE_VERIFICATION = True  # Enable fact-checking step
    ENABLE_CITATIONS = True  # Include source citations
    MIN_SOURCES_PER_SECTION = 3  # Minimum sources per section
    
    # Rate limiting
    SEARCH_DELAY = 0.5  # Seconds between searches
    SECTION_DELAY = 1.0  # Seconds between sections