From f2f839138445a9c66eccf5517c5ce6da761cbd96 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 27 Sep 2025 10:17:16 -0700 Subject: [PATCH] test azure openai fix --- tests/llm_translation/test_azure_openai.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/llm_translation/test_azure_openai.py b/tests/llm_translation/test_azure_openai.py index 5dab7e6d68..33947e25a6 100644 --- a/tests/llm_translation/test_azure_openai.py +++ b/tests/llm_translation/test_azure_openai.py @@ -339,7 +339,7 @@ def test_azure_gpt_4o_with_tool_call_and_response_format(api_version): with patch.object(client.chat.completions.with_raw_response, "create") as mock_post: response = litellm.completion( - model="azure/gpt-4o-new-test", + model="azure/gpt-4.1-nano", messages=[ { "role": "system", @@ -474,7 +474,7 @@ def test_azure_max_retries_0( try: completion( - model="azure/gpt-4o-new-test", + model="azure/gpt-4.1-nano", messages=[{"role": "user", "content": "Hello world"}], max_retries=max_retries, stream=stream, @@ -502,7 +502,7 @@ async def test_async_azure_max_retries_0( try: await acompletion( - model="azure/gpt-4o-new-test", + model="azure/gpt-4.1-nano", messages=[{"role": "user", "content": "Hello world"}], max_retries=max_retries, stream=stream, @@ -565,7 +565,7 @@ async def test_azure_embedding_max_retries_0( from litellm import aembedding, embedding args = { - "model": "azure/azure-embedding-model", + "model": "text-embedding-ada-002", "input": "Hello world", "max_retries": max_retries, } @@ -598,7 +598,7 @@ def test_azure_safety_result(): litellm._turn_on_debug() response = completion( - model="azure/gpt-4o-new-test", + model="azure/gpt-4.1-nano", messages=[{"role": "user", "content": "Hello world"}], ) print(f"response: {response}")