Skip to content

Commit 85b7e1d

Browse files
update test models to 4.1 nano
1 parent 88d37a5 commit 85b7e1d

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

tests/llm/test_chat_api.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,12 @@
2424
@pytest.mark.skipif(not os.getenv("AZURE_OPENAI_API_KEY"), reason="Skipping as Azure API key not set")
2525
def test_api_model_args_azure():
2626
model_args = AzureModelArgs(
27-
model_name="gpt-5-nano",
28-
deployment_name="gpt-5-nano",
27+
model_name="gpt-4.1-nano",
28+
deployment_name="gpt-4.1-nano",
2929
max_total_tokens=8192,
3030
max_input_tokens=8192 - 512,
3131
max_new_tokens=512,
32-
temperature=1,
32+
temperature=1e-1,
3333
)
3434
model = model_args.make_model()
3535

tests/llm/test_tracking.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def test_openai_chat_model():
146146
not AZURE_OPENAI_API_KEY_AVAILABLE, reason="Azure OpenAI API key is not available"
147147
)
148148
def test_azure_chat_model():
149-
chat_model = AzureChatModel(model_name="gpt-4.1-nano", deployment_name="gpt-4.1-nano", temperature=1)
149+
chat_model = AzureChatModel(model_name="gpt-4.1-nano", deployment_name="gpt-4.1-nano")
150150
assert chat_model.input_cost > 0
151151
assert chat_model.output_cost > 0
152152

0 commit comments

Comments
 (0)