chore: update Azure OpenAI API version to 2024-08-01-preview (#25)

kflili · lili-ms · web-flow · commit 49f2ae2bd98b · 2025-01-16T17:31:20.000-08:00
- Update Azure OpenAI API version in client creation
- Update corresponding unit test
- Verified working with both unit tests and real queries

Co-authored-by: Li Li &lt;lili5@microsoft.com&gt;
diff --git 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
@@ -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":

Original file line number	Diff line number	Diff line change
`@@ -151,7 +151,7 @@ def test_create_azure_client(self, mock_azure):`
`151`	`151`	`client = create_llm_client("azure")`
`152`	`152`	`mock_azure.assert_called_once_with(`
`153`	`153`	`api_key='test-azure-key',`
`154`		`- api_version="2024-02-15-preview",`
	`154`	`+ api_version="2024-08-01-preview",`
`155`	`155`	`azure_endpoint="https://msopenai.openai.azure.com"`
`156`	`156`	`)`
`157`	`157`	`self.assertEqual(client, self.mock_azure_client)`
Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ def create_llm_client(provider="openai"):`
`57`	`57`	`raise ValueError("AZURE_OPENAI_API_KEY not found in environment variables")`
`58`	`58`	`return AzureOpenAI(`
`59`	`59`	`api_key=api_key,`
`60`		`- api_version="2024-02-15-preview",`
	`60`	`+ api_version="2024-08-01-preview",`
`61`	`61`	`azure_endpoint="https://msopenai.openai.azure.com"`
`62`	`62`	`)`
`63`	`63`	`elif provider == "deepseek":`