diff --git a/tests/test_llm_api.py b/tests/test_llm_api.py index 61d08e2..def977a 100644 --- a/tests/test_llm_api.py +++ b/tests/test_llm_api.py @@ -151,7 +151,7 @@ def test_create_azure_client(self, mock_azure): client = create_llm_client("azure") mock_azure.assert_called_once_with( api_key='test-azure-key', - api_version="2024-02-15-preview", + api_version="2024-08-01-preview", azure_endpoint="https://msopenai.openai.azure.com" ) self.assertEqual(client, self.mock_azure_client) diff --git a/tools/llm_api.py b/tools/llm_api.py index 235746c..8578c80 100644 --- a/tools/llm_api.py +++ b/tools/llm_api.py @@ -57,7 +57,7 @@ def create_llm_client(provider="openai"): raise ValueError("AZURE_OPENAI_API_KEY not found in environment variables") return AzureOpenAI( api_key=api_key, - api_version="2024-02-15-preview", + api_version="2024-08-01-preview", azure_endpoint="https://msopenai.openai.azure.com" ) elif provider == "deepseek":