Skip to content

Commit 68682ef

Browse files
committed
fix tests
1 parent 1313e7f commit 68682ef

File tree

2 files changed

+13
-22
lines changed

2 files changed

+13
-22
lines changed

tests/models/cassettes/test_model_names/test_known_model_names.yaml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -134,31 +134,31 @@ interactions:
134134
parsed_body:
135135
data:
136136
- created: 0
137-
id: qwen-3-32b
137+
id: gpt-oss-120b
138138
object: model
139139
owned_by: Cerebras
140140
- created: 0
141-
id: qwen-3-235b-a22b-instruct-2507
141+
id: qwen-3-235b-a22b-thinking-2507
142142
object: model
143143
owned_by: Cerebras
144144
- created: 0
145-
id: gpt-oss-120b
145+
id: llama-3.3-70b
146146
object: model
147147
owned_by: Cerebras
148148
- created: 0
149-
id: zai-glm-4.6
149+
id: qwen-3-235b-a22b-instruct-2507
150150
object: model
151151
owned_by: Cerebras
152152
- created: 0
153-
id: llama3.1-8b
153+
id: zai-glm-4.6
154154
object: model
155155
owned_by: Cerebras
156156
- created: 0
157-
id: llama-3.3-70b
157+
id: llama3.1-8b
158158
object: model
159159
owned_by: Cerebras
160160
- created: 0
161-
id: qwen-3-235b-a22b-thinking-2507
161+
id: qwen-3-32b
162162
object: model
163163
owned_by: Cerebras
164164
object: list

tests/models/test_model.py

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -29,30 +29,21 @@
2929
TEST_CASES = [
3030
pytest.param(
3131
{'PYDANTIC_AI_GATEWAY_API_KEY': 'gateway-api-key'},
32-
'gateway/openai:gpt-5',
32+
'gateway/chat:gpt-5',
3333
'gpt-5',
3434
'openai',
3535
'openai',
3636
OpenAIChatModel,
37-
id='gateway/openai:gpt-5',
37+
id='gateway/chat:gpt-5',
3838
),
3939
pytest.param(
4040
{'PYDANTIC_AI_GATEWAY_API_KEY': 'gateway-api-key'},
41-
'gateway/openai-chat:gpt-5',
42-
'gpt-5',
43-
'openai',
44-
'openai',
45-
OpenAIChatModel,
46-
id='gateway/openai-chat:gpt-5',
47-
),
48-
pytest.param(
49-
{'PYDANTIC_AI_GATEWAY_API_KEY': 'gateway-api-key'},
50-
'gateway/openai-responses:gpt-5',
41+
'gateway/responses:gpt-5',
5142
'gpt-5',
5243
'openai',
5344
'openai',
5445
OpenAIResponsesModel,
55-
id='gateway/openai-responses:gpt-5',
46+
id='gateway/responses:gpt-5',
5647
),
5748
pytest.param(
5849
{'PYDANTIC_AI_GATEWAY_API_KEY': 'gateway-api-key'},
@@ -65,12 +56,12 @@
6556
),
6657
pytest.param(
6758
{'PYDANTIC_AI_GATEWAY_API_KEY': 'gateway-api-key'},
68-
'gateway/google-vertex:gemini-1.5-flash',
59+
'gateway/gemini:gemini-1.5-flash',
6960
'gemini-1.5-flash',
7061
'google-vertex',
7162
'google',
7263
GoogleModel,
73-
id='gateway/google-vertex:gemini-1.5-flash',
64+
id='gateway/gemini:gemini-1.5-flash',
7465
),
7566
pytest.param(
7667
{'PYDANTIC_AI_GATEWAY_API_KEY': 'gateway-api-key'},

0 commit comments

Comments
 (0)