Add skip logic if OpenAI is overloaded.

Azure · Sep 26, 2023 · 722171e · 722171e
1 parent 3f49132
commit 722171e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/sdk/ai/azopenai/client_chat_completions_test.go b/sdk/ai/azopenai/client_chat_completions_test.go
@@ -10,6 +10,7 @@ import (
 	"context"
 	"errors"
 	"io"
+	"net/http"
 	"os"
 	"testing"
 
@@ -89,6 +90,11 @@ func testGetChatCompletions(t *testing.T, client *azopenai.Client, tv testVars)
 	}
 
 	resp, err := client.GetChatCompletions(context.Background(), newTestChatCompletionOptions(tv), nil)
+
+	if respErr := (*azcore.ResponseError)(nil); errors.As(err, &respErr) && respErr.StatusCode == http.StatusTooManyRequests {
+		t.Skipf("OpenAI resource overloaded, skipping this test")
+	}
+
 	require.NoError(t, err)
 
 	if tv.Endpoint.Azure {
@@ -111,6 +117,11 @@ func testGetChatCompletions(t *testing.T, client *azopenai.Client, tv testVars)
 
 func testGetChatCompletionsStream(t *testing.T, client *azopenai.Client, tv testVars) {
 	streamResp, err := client.GetChatCompletionsStream(context.Background(), newTestChatCompletionOptions(tv), nil)
+
+	if respErr := (*azcore.ResponseError)(nil); errors.As(err, &respErr) && respErr.StatusCode == http.StatusTooManyRequests {
+		t.Skipf("OpenAI resource overloaded, skipping this test")
+	}
+
 	require.NoError(t, err)
 
 	// the data comes back differently for streaming