src/legisqa_local/config/models.py

"""Model configurations for different LLM providers"""

CONGRESS_NUMBERS = [113, 114, 115, 116, 117, 118, 119]
SPONSOR_PARTIES = ["D", "R", "L", "I"]

OPENAI_CHAT_MODELS = {
    "gpt-5-nano": {"cost": {"pmi": 0.05, "pmo": 0.40}},
#    "gpt-5-mini": {"cost": {"pmi": 0.25, "pmo": 2.00}},
#    "gpt-5": {"cost": {"pmi": 1.25, "pmo": 10.0}},
#    "gpt-4o-mini": {"cost": {"pmi": 0.15, "pmo": 0.60}},
#    "gpt-4o": {"cost": {"pmi": 2.50, "pmo": 10.0}},
}

ANTHROPIC_CHAT_MODELS = {
    "claude-3-5-haiku-20241022": {"cost": {"pmi": 0.80, "pmo": 4.00}},
#    "claude-sonnet-4-20250514": {"cost": {"pmi": 3.0, "pmo": 15.0}},
#    "claude-opus-4-1-20250805": {"cost": {"pmi": 15.0, "pmo": 75.0}},
}

TOGETHER_CHAT_MODELS = {
    "openai/gpt-oss-20b": {"cost": {"pmi": 0.05, "pmo": 0.20}},
    "meta-llama/Llama-3.3-70B-Instruct-Turbo-Free": {"cost": {"pmi": 0.00, "pmo": 0.00}},
#    "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo": {"cost": {"pmi": 0.18, "pmo": 0.18}},
#    "meta-llama/Llama-3.3-70B-Instruct-Turbo": {"cost": {"pmi": 0.88, "pmo": 0.88}},
#    "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo": {"cost": {"pmi": 3.50, "pmo": 3.50}},
    "Qwen/Qwen3-235B-A22B-Thinking-2507": {"cost": {"pmi": 0.65, "pmo": 3.00}},
#    "moonshotai/Kimi-K2-Instruct": {"cost": {"pmi": 1.00, "pmo": 3.00}},
}

GOOGLE_CHAT_MODELS = {
    "gemini-2.5-flash-lite": {"cost": {"pmi": 0.10, "pmo": 0.40}},            
#    "gemini-2.5-flash": {"cost": {"pmi": 0.30, "pmo": 2.50}},
#    "gemini-2.5-pro": {"cost": {"pmi": 1.25, "pmo": 10.0}},
}

PROVIDER_MODELS = {
    "OpenAI": OPENAI_CHAT_MODELS,
    "Anthropic": ANTHROPIC_CHAT_MODELS,
    "Together": TOGETHER_CHAT_MODELS,
    "Google": GOOGLE_CHAT_MODELS,
}