diff --git a/tests/unit/llms/test_llm_azure_openai.py b/tests/unit/llms/test_llm_azure_openai.py index 1b2db3cb976b25ae1d0be89a9c3cedceb51f7aac..793091957df7fdd89792aebec43a6fe87763e02b 100644 --- a/tests/unit/llms/test_llm_azure_openai.py +++ b/tests/unit/llms/test_llm_azure_openai.py @@ -13,9 +13,7 @@ def azure_openai_llm(mocker): class TestOpenAILLM: def test_azure_openai_llm_init_with_api_key(self, azure_openai_llm): assert azure_openai_llm.client is not None, "Client should be initialized" - assert ( - azure_openai_llm.name == "gpt-4o" - ), "Default name not set correctly" + assert azure_openai_llm.name == "gpt-4o", "Default name not set correctly" def test_azure_openai_llm_init_success(self, mocker): mocker.patch("os.getenv", return_value="fake-api-key")