Spaces:
Sleeping
Sleeping
File size: 1,695 Bytes
ac2020e ce293a6 ac2020e ce293a6 ac2020e ce293a6 ac2020e ce293a6 ac2020e ce293a6 ac2020e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | """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,
}
|