gabrielaltay commited on
Commit
ce293a6
·
1 Parent(s): 9ee8cbe

restrict models

Browse files
Files changed (1) hide show
  1. src/legisqa_local/config/models.py +12 -12
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 = {