MODEL_GEMINI = "gemini-2.5-flash"
MODEL_OLMOCR = "olmOCR-2-7B-1025-FP8"


MODELS_MAP = {
    # GPT-5.2 family
    "gpt-5.2": {"input": 1.75, "output": 14.00, "backend": "openai"},
    "gpt-5.2-chat-latest": {"input": 1.75, "output": 14.00, "backend": "openai"},
    "gpt-5.2-pro": {"input": 21.00, "output": 168.00, "backend": "openai"},

    # GPT-5.1 / GPT-5 family
    "gpt-5.1": {"input": 1.25, "output": 10.00, "backend": "openai"},
    "gpt-5": {"input": 1.25, "output": 10.00, "backend": "openai"},
    "gpt-5-mini": {"input": 0.25, "output": 2.00, "backend": "openai"},
    "gpt-5-nano": {"input": 0.05, "output": 0.40, "backend": "openai"},

    # GPT-4.1 family
    "gpt-4.1": {"input": 2.00, "output": 8.00, "backend": "openai"},
    "gpt-4.1-mini": {"input": 0.40, "output": 1.60, "backend": "openai"},
    "gpt-4.1-nano": {"input": 0.10, "output": 0.40, "backend": "openai"},
    # Other backends (mock rates)
    MODEL_GEMINI: {"input": 0.30, "output": 2.50, "backend": "gemini"},
    "gemini-3-pro-preview": {"input": 2.00, "output": 12.00, "backend": "gemini"},
    MODEL_OLMOCR: {"input": 1.35, "output": 0.30, "backend": "olmocr"},
    "gemini-2.5-pro": {"input": 1.25, "output": 10.00, "backend": "gemini"},
    "default": {"input": 2.50, "output": 10.00, "backend": "openai"},
}