Skip to content

Commit 5811021

Browse files
committed
fix more integration tests
1 parent ea59ce1 commit 5811021

File tree

1 file changed

+18
-8
lines changed

1 file changed

+18
-8
lines changed

backend/tests/integration/tests/llm_provider/test_llm_provider.py

Lines changed: 18 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,11 @@ def test_create_llm_provider(
179179
(
180180
(
181181
"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+
],
183187
),
184188
[
185189
ModelConfigurationUpsertRequest(
@@ -188,7 +192,7 @@ def test_create_llm_provider(
188192
],
189193
(
190194
"gpt-4",
191-
[ModelConfigurationUpsertRequest(name="gpt-4")],
195+
[ModelConfigurationUpsertRequest(name="gpt-4", is_visible=True)],
192196
),
193197
[ModelConfigurationUpsertRequest(name="gpt-4", is_visible=True)],
194198
),
@@ -198,19 +202,25 @@ def test_create_llm_provider(
198202
(
199203
"gpt-4",
200204
[
201-
ModelConfigurationUpsertRequest(name="gpt-4"),
205+
ModelConfigurationUpsertRequest(name="gpt-4", is_visible=True),
202206
ModelConfigurationUpsertRequest(
203-
name="gpt-4o", max_input_tokens=4096
207+
name="gpt-4o", is_visible=True, max_input_tokens=4096
204208
),
205209
],
206210
),
207211
[
208212
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+
),
210216
],
211217
(
212218
"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+
],
214224
),
215225
[
216226
ModelConfigurationUpsertRequest(
@@ -325,8 +335,8 @@ def test_update_model_configurations(
325335
(
326336
"gpt-4",
327337
[
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),
330340
],
331341
),
332342
],

0 commit comments

Comments
 (0)