@@ -179,7 +179,11 @@ def test_create_llm_provider(
179
179
(
180
180
(
181
181
"gpt-4" ,
182
- [ModelConfigurationUpsertRequest (name = "gpt-4" , max_input_tokens = 4096 )],
182
+ [
183
+ ModelConfigurationUpsertRequest (
184
+ name = "gpt-4" , is_visible = True , max_input_tokens = 4096
185
+ )
186
+ ],
183
187
),
184
188
[
185
189
ModelConfigurationUpsertRequest (
@@ -188,7 +192,7 @@ def test_create_llm_provider(
188
192
],
189
193
(
190
194
"gpt-4" ,
191
- [ModelConfigurationUpsertRequest (name = "gpt-4" )],
195
+ [ModelConfigurationUpsertRequest (name = "gpt-4" , is_visible = True )],
192
196
),
193
197
[ModelConfigurationUpsertRequest (name = "gpt-4" , is_visible = True )],
194
198
),
@@ -198,19 +202,25 @@ def test_create_llm_provider(
198
202
(
199
203
"gpt-4" ,
200
204
[
201
- ModelConfigurationUpsertRequest (name = "gpt-4" ),
205
+ ModelConfigurationUpsertRequest (name = "gpt-4" , is_visible = True ),
202
206
ModelConfigurationUpsertRequest (
203
- name = "gpt-4o" , max_input_tokens = 4096
207
+ name = "gpt-4o" , is_visible = True , max_input_tokens = 4096
204
208
),
205
209
],
206
210
),
207
211
[
208
212
ModelConfigurationUpsertRequest (name = "gpt-4" , is_visible = True ),
209
- ModelConfigurationUpsertRequest (name = "gpt-4o" , max_input_tokens = 4096 ),
213
+ ModelConfigurationUpsertRequest (
214
+ name = "gpt-4o" , is_visible = True , max_input_tokens = 4096
215
+ ),
210
216
],
211
217
(
212
218
"gpt-4" ,
213
- [ModelConfigurationUpsertRequest (name = "gpt-4" , max_input_tokens = 4096 )],
219
+ [
220
+ ModelConfigurationUpsertRequest (
221
+ name = "gpt-4" , is_visible = True , max_input_tokens = 4096
222
+ )
223
+ ],
214
224
),
215
225
[
216
226
ModelConfigurationUpsertRequest (
@@ -325,8 +335,8 @@ def test_update_model_configurations(
325
335
(
326
336
"gpt-4" ,
327
337
[
328
- ModelConfigurationUpsertRequest (name = "gpt-4o" ),
329
- ModelConfigurationUpsertRequest (name = "gpt-4" ),
338
+ ModelConfigurationUpsertRequest (name = "gpt-4o" , is_visible = True ),
339
+ ModelConfigurationUpsertRequest (name = "gpt-4" , is_visible = True ),
330
340
],
331
341
),
332
342
],
0 commit comments