Update Azure OpenAI to 2024-04-01-preview. (Azure#22798)

Update Azure OpenAI to 2024-04-01-preview.
* Fixing some more common throttled areas.

Author: glecaros

sdk/ai/azopenai/CHANGELOG.md

@@ -1,17 +1,15 @@
 # Release History
 
-## 0.5.2 (Unreleased)
+## 0.5.2 (2024-05-07)
 
 ### Features Added
 
-### Breaking Changes
+- Updating to the `2024-04-01-preview` API version for Azure OpenAI. (PR#22798)
 
 ### Bugs Fixed
 
 - EventReader can now handle chunks of text larger than 64k. Thank you @ChrisTrenkamp for finding the issue and suggesting a fix. (PR#22703)
 
-### Other Changes
-
 ## 0.5.1 (2024-04-02)
 
 ### Features Added

sdk/ai/azopenai/assets.json

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

sdk/ai/azopenai/client.go

@@ -160,7 +160,6 @@ func TestClient_GetAudioTranslation(t *testing.T) {
 
 				args := newTranslationOptions(azopenai.AudioTranslationFormatVerboseJSON, model, "testdata/sampledata_audiofiles_myVoiceIsMyPassportVerifyMe01.mp3")
 				transcriptResp, err := client.GetAudioTranslation(context.Background(), args, nil)
-				require.NoError(t, err)
 				customRequireNoError(t, err, true)
 
 				require.NotEmpty(t, *transcriptResp.Text)

sdk/ai/azopenai/client_audio_test.go

@@ -62,8 +62,7 @@ func TestClient_GetChatCompletions_AzureOpenAI_ContentFilter_WithResponse(t *tes
 		Temperature:    to.Ptr(float32(0.0)),
 		DeploymentName: &azureOpenAI.ChatCompletionsRAI.Model,
 	}, nil)
-
-	require.NoError(t, err)
+	customRequireNoError(t, err, true)
 
 	require.Equal(t, safeContentFilter, resp.ChatCompletions.Choices[0].ContentFilterResults)
 }

sdk/ai/azopenai/client_rai_test.go

@@ -450,12 +450,12 @@ func customRequireNoError(t *testing.T, err error, throttlingAllowed bool) {
 
 	if throttlingAllowed {
 		if respErr := (*azcore.ResponseError)(nil); errors.As(err, &respErr) && respErr.StatusCode == http.StatusTooManyRequests {
-			t.Skip("Skipping test because of throttling")
+			t.Skip("Skipping test because of throttling (http.StatusTooManyRequests)")
 			return
 		}
 
 		if errors.Is(err, context.DeadlineExceeded) {
-			t.Skip("Skipping test because of throttling")
+			t.Skip("Skipping test because of throttling (DeadlineExceeded)")
 			return
 		}
 	}

sdk/ai/azopenai/client_shared_test.go

@@ -25,12 +25,12 @@ func TestImageGeneration_AzureOpenAI(t *testing.T) {
 	}
 
 	client := newTestClient(t, azureOpenAI.DallE.Endpoint)
-	testImageGeneration(t, client, azureOpenAI.DallE.Model, azopenai.ImageGenerationResponseFormatURL)
+	testImageGeneration(t, client, azureOpenAI.DallE.Model, azopenai.ImageGenerationResponseFormatURL, true)
 }
 
 func TestImageGeneration_OpenAI(t *testing.T) {
 	client := newTestClient(t, openAI.DallE.Endpoint)
-	testImageGeneration(t, client, openAI.DallE.Model, azopenai.ImageGenerationResponseFormatURL)
+	testImageGeneration(t, client, openAI.DallE.Model, azopenai.ImageGenerationResponseFormatURL, false)
 }
 
 func TestImageGeneration_AzureOpenAI_WithError(t *testing.T) {
@@ -57,11 +57,11 @@ func TestImageGeneration_OpenAI_Base64(t *testing.T) {
 	}
 
 	client := newTestClient(t, openAI.DallE.Endpoint)
-	testImageGeneration(t, client, openAI.DallE.Model, azopenai.ImageGenerationResponseFormatBase64)
+	testImageGeneration(t, client, openAI.DallE.Model, azopenai.ImageGenerationResponseFormatBase64, false)
 }
 
-func testImageGeneration(t *testing.T, client *azopenai.Client, model string, responseFormat azopenai.ImageGenerationResponseFormat) {
-	ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
+func testImageGeneration(t *testing.T, client *azopenai.Client, model string, responseFormat azopenai.ImageGenerationResponseFormat, azure bool) {
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Minute)
 	defer cancel()
 
 	resp, err := client.GetImageGenerations(ctx, azopenai.ImageGenerationOptions{
@@ -72,7 +72,7 @@ func testImageGeneration(t *testing.T, client *azopenai.Client, model string, re
 		ResponseFormat: &responseFormat,
 		DeploymentName: &model,
 	}, nil)
-	require.NoError(t, err)
+	customRequireNoError(t, err, azure)
 
 	if recording.GetRecordMode() == recording.LiveMode {
 		switch responseFormat {