feat: add integration tests for gpt-4o-audio-preview

Author: WqyJh

api_integration_test.go

@@ -108,6 +108,38 @@ func TestAPI(t *testing.T) {
 		},
 	)
 	checks.NoError(t, err, "CreateChatCompletion (with functions) returned error")
+
+	response, err := c.CreateChatCompletion(
+		ctx,
+		openai.ChatCompletionRequest{
+			Model: openai.GPT4oAudioPreview,
+			Messages: []openai.ChatCompletionMessage{
+				{
+					Role:    openai.ChatMessageRoleUser,
+					Content: "hi",
+				},
+			},
+			Audio: &openai.AudioOutput{
+				Voice:  openai.AudioVoiceAlloy,
+				Format: openai.AudioFormatPCM16,
+			},
+			Modalities: []openai.Modality{openai.ModalityText, openai.ModalityAudio},
+		},
+	)
+	checks.NoError(t, err, "CreateChatCompletion (with audio) returned error")
+	if response.Choices[0].Message.Audio == nil {
+		t.Fatal("Audio response is nil")
+	}
+	if len(response.Choices[0].Message.Audio.Data) == 0 {
+		t.Fatal("Audio response data is empty")
+	}
+	if response.Choices[0].Message.Audio.Transcript == "" {
+		t.Fatal("Audio response transcript is empty")
+	}
+	if response.Usage.PromptTokens == 0 || response.Usage.CompletionTokens == 0 || response.Usage.TotalTokens == 0 {
+		t.Fatal("Usage is zero")
+	}
+	t.Logf("Usage: %+v", response.Usage)
 }
 
 func TestCompletionStream(t *testing.T) {
@@ -145,6 +177,60 @@ func TestCompletionStream(t *testing.T) {
 	}
 }
 
+func TestChatCompletionStream(t *testing.T) {
+	apiToken := os.Getenv("OPENAI_TOKEN")
+	if apiToken == "" {
+		t.Skip("Skipping testing against production OpenAI API. Set OPENAI_TOKEN environment variable to enable it.")
+	}
+
+	c := openai.NewClient(apiToken)
+	ctx := context.Background()
+
+	stream, err := c.CreateChatCompletionStream(ctx, openai.ChatCompletionRequest{
+		Model: openai.GPT4oAudioPreview,
+		Messages: []openai.ChatCompletionMessage{
+			{
+				Role:    openai.ChatMessageRoleUser,
+				Content: "hi",
+			},
+		},
+		Audio: &openai.AudioOutput{
+			Voice:  openai.AudioVoiceAlloy,
+			Format: openai.AudioFormatPCM16,
+		},
+		Modalities: []openai.Modality{openai.ModalityText, openai.ModalityAudio},
+		StreamOptions: &openai.StreamOptions{
+			IncludeUsage: true,
+		},
+	})
+	checks.NoError(t, err, "CreateCompletionStream returned error")
+	defer stream.Close()
+
+	var usage *openai.Usage
+	counter := 0
+	for {
+		response, err := stream.Recv()
+		if err != nil {
+			if errors.Is(err, io.EOF) {
+				break
+			}
+			t.Errorf("Stream error: %v", err)
+		} else {
+			counter++
+		}
+		if response.Usage != nil {
+			usage = response.Usage
+			t.Logf("Usage: %+v", usage)
+		}
+	}
+	if counter == 0 {
+		t.Error("Stream did not return any responses")
+	}
+	if usage == nil {
+		t.Error("Usage is nil")
+	}
+}
+
 func TestAPIError(t *testing.T) {
 	apiToken := os.Getenv("OPENAI_TOKEN")
 	if apiToken == "" {

internal/test/checks/checks.go

@@ -73,7 +73,7 @@ type tHelper interface {
 // Pointer variable equality is determined based on the equality of the
 // referenced values (as opposed to the memory addresses). Function equality
 // cannot be determined and will always fail.
-func Equal(t TestingT, expected, actual interface{}) bool {
+func Equal(t TestingT, expected, actual any) bool {
 	if h, ok := t.(tHelper); ok {
 		h.Helper()
 	}
@@ -97,7 +97,7 @@ func JSONEq(t TestingT, expected string, actual string) bool {
 	if h, ok := t.(tHelper); ok {
 		h.Helper()
 	}
-	var expectedJSONAsInterface, actualJSONAsInterface interface{}
+	var expectedJSONAsInterface, actualJSONAsInterface any
 
 	if err := json.Unmarshal([]byte(expected), &expectedJSONAsInterface); err != nil {
 		t.Fatalf("Expected value ('%s') is not valid json.\nJSON parsing error: '%s'", expected, err.Error())
@@ -112,7 +112,7 @@ func JSONEq(t TestingT, expected string, actual string) bool {
 
 // validateEqualArgs checks whether provided arguments can be safely used in the
 // Equal/NotEqual functions.
-func validateEqualArgs(expected, actual interface{}) error {
+func validateEqualArgs(expected, actual any) error {
 	if expected == nil && actual == nil {
 		return nil
 	}
@@ -123,7 +123,7 @@ func validateEqualArgs(expected, actual interface{}) error {
 	return nil
 }
 
-func isFunction(arg interface{}) bool {
+func isFunction(arg any) bool {
 	if arg == nil {
 		return false
 	}
@@ -137,7 +137,7 @@ func isFunction(arg interface{}) bool {
 // ObjectsAreEqual determines if two objects are considered equal.
 //
 // This function does no assertion of any kind.
-func ObjectsAreEqual(expected, actual interface{}) bool {
+func ObjectsAreEqual(expected, actual any) bool {
 	if expected == nil || actual == nil {
 		return expected == actual
 	}