import os
from typing import Optional
from pydantic import BaseModel
from dotenv import load_dotenv

load_dotenv()

class Config(BaseModel):
    # OpenRouter API configuration
    OPENROUTER_API_KEY: Optional[str] = os.getenv("OPENROUTER_API_KEY")
    OPENROUTER_BASE_URL: str = "https://openrouter.ai/api/v1"
    
    # Model configuration - using free models when possible
    EXTRACTION_MODEL: str = "qwen/qwen3-32b"  # Free model
    BACKUP_MODEL: str = "qwen/qwen3-32b"  # Backup free model
    
    # Processing configuration
    CHUNK_SIZE: int = 2000
    CHUNK_OVERLAP: int = 200
    MAX_FILE_SIZE_MB: int = 10
    
    # Graph configuration
    MAX_ENTITIES: int = 100
    MAX_RELATIONSHIPS: int = 200
    ENTITY_IMPORTANCE_THRESHOLD: float = 0.3
    
    class Config:
        env_file = ".env"