[azopenai] Fixing failing tests (Azure#23955)

Author: glecaros

sdk/ai/azopenai/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "go",
   "TagPrefix": "go/ai/azopenai",
-  "Tag": "go/ai/azopenai_cb43c9a227"
+  "Tag": "go/ai/azopenai_ce36c649fa"
 }

sdk/ai/azopenai/client_chat_completions_test.go

@@ -108,7 +108,18 @@ func TestClient_GetChatCompletions(t *testing.T) {
 			resp.ChatCompletions.Choices[i].ContentFilterResults = nil
 		}
 
+		// Exclude Usage field from comparison as they can change
+		expectedUsage := expected.Usage
+		actualUsage := resp.ChatCompletions.Usage
+		expected.Usage = nil
+		resp.ChatCompletions.Usage = nil
+
 		require.Equal(t, expected, resp.ChatCompletions)
+
+		// Compare the usage numbers, but allow for a small margin of error
+		require.InDelta(t, *expectedUsage.CompletionTokens, *actualUsage.CompletionTokens, 5)
+		require.InDelta(t, *expectedUsage.PromptTokens, *actualUsage.PromptTokens, 5)
+		require.InDelta(t, *expectedUsage.TotalTokens, *actualUsage.TotalTokens, 5)
 	}
 
 	t.Run("AzureOpenAI", func(t *testing.T) {

sdk/ai/azopenai/client_shared_test.go

@@ -169,7 +169,7 @@ var azureOpenAI, openAI = func() (testVars, testVars) {
 			},
 			Vision: endpointWithModel{
 				Endpoint: ifAzure(azure, servers.SWECentral, servers.OpenAI),
-				Model:    ifAzure(azure, "gpt-4-vision-preview", "gpt-4-vision-preview"),
+				Model:    ifAzure(azure, "gpt-4o", "gpt-4o"),
 			},
 			Whisper: endpointWithModel{
 				Endpoint: ifAzure(azure, servers.USNorthCentral, servers.OpenAI),

sdk/ai/azopenai/example_client_getchatcompletions_vision_test.go

@@ -17,7 +17,7 @@ import (
 
 func ExampleClient_GetChatCompletions_vision() {
 	azureOpenAIKey := os.Getenv("AOAI_VISION_API_KEY")
-	modelDeployment := os.Getenv("AOAI_VISION_MODEL") // ex: gpt-4-vision-preview"
+	modelDeployment := os.Getenv("AOAI_VISION_MODEL") // ex: gpt-4o"
 
 	// Ex: "https://<your-azure-openai-host>.openai.azure.com"
 	azureOpenAIEndpoint := os.Getenv("AOAI_VISION_ENDPOINT")