diff --git a/tests/test_azure_openai_provider.py b/tests/test_azure_openai_provider.py index df2cdc3..680ddf4 100644 --- a/tests/test_azure_openai_provider.py +++ b/tests/test_azure_openai_provider.py @@ -291,7 +291,7 @@ async def test_chat_connection_error(): result = await provider.chat(messages) assert isinstance(result, LLMResponse) - assert "Error calling Azure OpenAI: Connection failed" in result.content + assert "Error calling Azure OpenAI: Exception('Connection failed')" in result.content assert result.finish_reason == "error"