Skip to content

Commit faca9c4

Browse files
authored
Update known models on Cerebras and Heroku (#3375)
1 parent 86b645f commit faca9c4

File tree

2 files changed

+38
-19
lines changed

2 files changed

+38
-19
lines changed

pydantic_ai_slim/pydantic_ai/models/__init__.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -126,18 +126,16 @@
126126
'cerebras:gpt-oss-120b',
127127
'cerebras:llama3.1-8b',
128128
'cerebras:llama-3.3-70b',
129-
'cerebras:llama-4-scout-17b-16e-instruct',
130-
'cerebras:llama-4-maverick-17b-128e-instruct',
131129
'cerebras:qwen-3-235b-a22b-instruct-2507',
132130
'cerebras:qwen-3-32b',
133-
'cerebras:qwen-3-coder-480b',
134131
'cerebras:qwen-3-235b-a22b-thinking-2507',
135132
'cohere:c4ai-aya-expanse-32b',
136133
'cohere:c4ai-aya-expanse-8b',
137134
'cohere:command-nightly',
138135
'cohere:command-r-08-2024',
139136
'cohere:command-r-plus-08-2024',
140137
'cohere:command-r7b-12-2024',
138+
'cerebras:zai-glm-4.6',
141139
'deepseek:deepseek-chat',
142140
'deepseek:deepseek-reasoner',
143141
'google-gla:gemini-2.0-flash',
@@ -189,11 +187,15 @@
189187
'groq:llama-3.2-3b-preview',
190188
'groq:llama-3.2-11b-vision-preview',
191189
'groq:llama-3.2-90b-vision-preview',
190+
'heroku:amazon-rerank-1-0',
192191
'heroku:claude-3-5-haiku',
193192
'heroku:claude-3-5-sonnet-latest',
194193
'heroku:claude-3-7-sonnet',
195-
'heroku:claude-4-sonnet',
196194
'heroku:claude-3-haiku',
195+
'heroku:claude-4-5-haiku',
196+
'heroku:claude-4-5-sonnet',
197+
'heroku:claude-4-sonnet',
198+
'heroku:cohere-rerank-3-5',
197199
'heroku:gpt-oss-120b',
198200
'heroku:nova-lite',
199201
'heroku:nova-pro',

tests/models/cassettes/test_model_names/test_known_model_names.yaml

Lines changed: 32 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ interactions:
1313
response:
1414
headers:
1515
content-length:
16-
- '760'
16+
- '1080'
1717
content-security-policy:
1818
- default-src 'none'; frame-ancestors 'none'
1919
content-type:
@@ -25,6 +25,12 @@ interactions:
2525
strict-transport-security:
2626
- max-age=63072000
2727
parsed_body:
28+
- model_id: amazon-rerank-1-0
29+
regions:
30+
- eu
31+
- us
32+
type:
33+
- text-to-text
2834
- model_id: claude-3-5-haiku
2935
regions:
3036
- us
@@ -46,6 +52,18 @@ interactions:
4652
- eu
4753
type:
4854
- text-to-text
55+
- model_id: claude-4-5-haiku
56+
regions:
57+
- eu
58+
- us
59+
type:
60+
- text-to-text
61+
- model_id: claude-4-5-sonnet
62+
regions:
63+
- eu
64+
- us
65+
type:
66+
- text-to-text
4967
- model_id: claude-4-sonnet
5068
regions:
5169
- eu
@@ -58,8 +76,15 @@ interactions:
5876
- us
5977
type:
6078
- text-to-embedding
79+
- model_id: cohere-rerank-3-5
80+
regions:
81+
- eu
82+
- us
83+
type:
84+
- text-to-text
6185
- model_id: gpt-oss-120b
6286
regions:
87+
- eu
6388
- us
6489
type:
6590
- text-to-text
@@ -99,7 +124,7 @@ interactions:
99124
alt-svc:
100125
- h3=":443"; ma=86400
101126
content-length:
102-
- '762'
127+
- '570'
103128
content-type:
104129
- application/json
105130
referrer-policy:
@@ -108,40 +133,32 @@ interactions:
108133
- max-age=3600; includeSubDomains
109134
parsed_body:
110135
data:
111-
- created: 0
112-
id: llama-4-scout-17b-16e-instruct
113-
object: model
114-
owned_by: Cerebras
115136
- created: 0
116137
id: qwen-3-32b
117138
object: model
118139
owned_by: Cerebras
119-
- created: 0
120-
id: llama-3.3-70b
121-
object: model
122-
owned_by: Cerebras
123140
- created: 0
124141
id: qwen-3-235b-a22b-instruct-2507
125142
object: model
126143
owned_by: Cerebras
127144
- created: 0
128-
id: llama-4-maverick-17b-128e-instruct
145+
id: gpt-oss-120b
129146
object: model
130147
owned_by: Cerebras
131148
- created: 0
132-
id: qwen-3-coder-480b
149+
id: zai-glm-4.6
133150
object: model
134151
owned_by: Cerebras
135152
- created: 0
136-
id: gpt-oss-120b
153+
id: llama3.1-8b
137154
object: model
138155
owned_by: Cerebras
139156
- created: 0
140-
id: qwen-3-235b-a22b-thinking-2507
157+
id: llama-3.3-70b
141158
object: model
142159
owned_by: Cerebras
143160
- created: 0
144-
id: llama3.1-8b
161+
id: qwen-3-235b-a22b-thinking-2507
145162
object: model
146163
owned_by: Cerebras
147164
object: list

0 commit comments

Comments
 (0)