Spaces:
Sleeping
Sleeping
Commit
·
ce293a6
1
Parent(s):
9ee8cbe
restrict models
Browse files
src/legisqa_local/config/models.py
CHANGED
|
@@ -5,32 +5,32 @@ SPONSOR_PARTIES = ["D", "R", "L", "I"]
|
|
| 5 |
|
| 6 |
OPENAI_CHAT_MODELS = {
|
| 7 |
"gpt-5-nano": {"cost": {"pmi": 0.05, "pmo": 0.40}},
|
| 8 |
-
"gpt-5-mini": {"cost": {"pmi": 0.25, "pmo": 2.00}},
|
| 9 |
-
"gpt-5": {"cost": {"pmi": 1.25, "pmo": 10.0}},
|
| 10 |
-
"gpt-4o-mini": {"cost": {"pmi": 0.15, "pmo": 0.60}},
|
| 11 |
-
"gpt-4o": {"cost": {"pmi": 2.50, "pmo": 10.0}},
|
| 12 |
}
|
| 13 |
|
| 14 |
ANTHROPIC_CHAT_MODELS = {
|
| 15 |
"claude-3-5-haiku-20241022": {"cost": {"pmi": 0.80, "pmo": 4.00}},
|
| 16 |
-
"claude-sonnet-4-20250514": {"cost": {"pmi": 3.0, "pmo": 15.0}},
|
| 17 |
-
"claude-opus-4-1-20250805": {"cost": {"pmi": 15.0, "pmo": 75.0}},
|
| 18 |
}
|
| 19 |
|
| 20 |
TOGETHER_CHAT_MODELS = {
|
| 21 |
"openai/gpt-oss-20b": {"cost": {"pmi": 0.05, "pmo": 0.20}},
|
| 22 |
"meta-llama/Llama-3.3-70B-Instruct-Turbo-Free": {"cost": {"pmi": 0.00, "pmo": 0.00}},
|
| 23 |
-
"meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo": {"cost": {"pmi": 0.18, "pmo": 0.18}},
|
| 24 |
-
"meta-llama/Llama-3.3-70B-Instruct-Turbo": {"cost": {"pmi": 0.88, "pmo": 0.88}},
|
| 25 |
-
"meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo": {"cost": {"pmi": 3.50, "pmo": 3.50}},
|
| 26 |
"Qwen/Qwen3-235B-A22B-Thinking-2507": {"cost": {"pmi": 0.65, "pmo": 3.00}},
|
| 27 |
-
"moonshotai/Kimi-K2-Instruct": {"cost": {"pmi": 1.00, "pmo": 3.00}},
|
| 28 |
}
|
| 29 |
|
| 30 |
GOOGLE_CHAT_MODELS = {
|
| 31 |
"gemini-2.5-flash-lite": {"cost": {"pmi": 0.10, "pmo": 0.40}},
|
| 32 |
-
"gemini-2.5-flash": {"cost": {"pmi": 0.30, "pmo": 2.50}},
|
| 33 |
-
"gemini-2.5-pro": {"cost": {"pmi": 1.25, "pmo": 10.0}},
|
| 34 |
}
|
| 35 |
|
| 36 |
PROVIDER_MODELS = {
|
|
|
|
| 5 |
|
| 6 |
OPENAI_CHAT_MODELS = {
|
| 7 |
"gpt-5-nano": {"cost": {"pmi": 0.05, "pmo": 0.40}},
|
| 8 |
+
# "gpt-5-mini": {"cost": {"pmi": 0.25, "pmo": 2.00}},
|
| 9 |
+
# "gpt-5": {"cost": {"pmi": 1.25, "pmo": 10.0}},
|
| 10 |
+
# "gpt-4o-mini": {"cost": {"pmi": 0.15, "pmo": 0.60}},
|
| 11 |
+
# "gpt-4o": {"cost": {"pmi": 2.50, "pmo": 10.0}},
|
| 12 |
}
|
| 13 |
|
| 14 |
ANTHROPIC_CHAT_MODELS = {
|
| 15 |
"claude-3-5-haiku-20241022": {"cost": {"pmi": 0.80, "pmo": 4.00}},
|
| 16 |
+
# "claude-sonnet-4-20250514": {"cost": {"pmi": 3.0, "pmo": 15.0}},
|
| 17 |
+
# "claude-opus-4-1-20250805": {"cost": {"pmi": 15.0, "pmo": 75.0}},
|
| 18 |
}
|
| 19 |
|
| 20 |
TOGETHER_CHAT_MODELS = {
|
| 21 |
"openai/gpt-oss-20b": {"cost": {"pmi": 0.05, "pmo": 0.20}},
|
| 22 |
"meta-llama/Llama-3.3-70B-Instruct-Turbo-Free": {"cost": {"pmi": 0.00, "pmo": 0.00}},
|
| 23 |
+
# "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo": {"cost": {"pmi": 0.18, "pmo": 0.18}},
|
| 24 |
+
# "meta-llama/Llama-3.3-70B-Instruct-Turbo": {"cost": {"pmi": 0.88, "pmo": 0.88}},
|
| 25 |
+
# "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo": {"cost": {"pmi": 3.50, "pmo": 3.50}},
|
| 26 |
"Qwen/Qwen3-235B-A22B-Thinking-2507": {"cost": {"pmi": 0.65, "pmo": 3.00}},
|
| 27 |
+
# "moonshotai/Kimi-K2-Instruct": {"cost": {"pmi": 1.00, "pmo": 3.00}},
|
| 28 |
}
|
| 29 |
|
| 30 |
GOOGLE_CHAT_MODELS = {
|
| 31 |
"gemini-2.5-flash-lite": {"cost": {"pmi": 0.10, "pmo": 0.40}},
|
| 32 |
+
# "gemini-2.5-flash": {"cost": {"pmi": 0.30, "pmo": 2.50}},
|
| 33 |
+
# "gemini-2.5-pro": {"cost": {"pmi": 1.25, "pmo": 10.0}},
|
| 34 |
}
|
| 35 |
|
| 36 |
PROVIDER_MODELS = {
|