mirror of
https://github.com/hanzo-go/openai.git
synced 2026-09-24 00:31:39 +00:00
zeekayandHanzo Dev
88762272c3
The fork kept the upstream module path (module github.com/sashabaranov/ go-openai), so every consumer could only reach it through a "replace github.com/sashabaranov/go-openai => github.com/hanzoai/go-openai" directive — and Go replace directives do NOT propagate to consumers, so each downstream repo had to copy the same replace or fail to compile against fork-only fields (e.g. ReasoningContent). That is the opposite of one-and- only-one-way. Declare the fork's OWN module path so it can be required directly and inherited transitively. Package name stays 'openai' (import identifier unchanged); only the module path + internal self-imports (/internal, /jsonschema) are renamed. Upstream issue-URL comments are left intact as attribution. Builds + vets clean. Co-authored-by: Hanzo Dev <[email protected]>
110 lines
3.0 KiB
Go
110 lines
3.0 KiB
Go
package openai //nolint:testpackage // testing private field
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/go-openai/internal/test"
|
|
"github.com/hanzoai/go-openai/internal/test/checks"
|
|
)
|
|
|
|
func TestAudioWithFailingFormBuilder(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "fake.mp3")
|
|
test.CreateTestFile(t, path)
|
|
|
|
req := AudioRequest{
|
|
FilePath: path,
|
|
Prompt: "test",
|
|
Temperature: 0.5,
|
|
Language: "en",
|
|
Format: AudioResponseFormatSRT,
|
|
TimestampGranularities: []TranscriptionTimestampGranularity{
|
|
TranscriptionTimestampGranularitySegment,
|
|
TranscriptionTimestampGranularityWord,
|
|
},
|
|
}
|
|
|
|
mockFailedErr := fmt.Errorf("mock form builder fail")
|
|
mockBuilder := &mockFormBuilder{}
|
|
|
|
mockBuilder.mockCreateFormFile = func(string, *os.File) error {
|
|
return mockFailedErr
|
|
}
|
|
err := audioMultipartForm(req, mockBuilder)
|
|
checks.ErrorIs(t, err, mockFailedErr, "audioMultipartForm should return error if form builder fails")
|
|
|
|
mockBuilder.mockCreateFormFile = func(string, *os.File) error {
|
|
return nil
|
|
}
|
|
|
|
var failForField string
|
|
mockBuilder.mockWriteField = func(fieldname, _ string) error {
|
|
if fieldname == failForField {
|
|
return mockFailedErr
|
|
}
|
|
return nil
|
|
}
|
|
|
|
failOn := []string{"model", "prompt", "temperature", "language", "response_format", "timestamp_granularities[]"}
|
|
for _, failingField := range failOn {
|
|
failForField = failingField
|
|
mockFailedErr = fmt.Errorf("mock form builder fail on field %s", failingField)
|
|
|
|
err = audioMultipartForm(req, mockBuilder)
|
|
checks.ErrorIs(t, err, mockFailedErr, "audioMultipartForm should return error if form builder fails")
|
|
}
|
|
}
|
|
|
|
func TestCreateFileField(t *testing.T) {
|
|
t.Run("createFileField failing file", func(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "fake.mp3")
|
|
test.CreateTestFile(t, path)
|
|
|
|
req := AudioRequest{
|
|
FilePath: path,
|
|
}
|
|
|
|
mockFailedErr := fmt.Errorf("mock form builder fail")
|
|
mockBuilder := &mockFormBuilder{
|
|
mockCreateFormFile: func(string, *os.File) error {
|
|
return mockFailedErr
|
|
},
|
|
}
|
|
|
|
err := createFileField(req, mockBuilder)
|
|
checks.ErrorIs(t, err, mockFailedErr, "createFileField using a file should return error if form builder fails")
|
|
})
|
|
|
|
t.Run("createFileField failing reader", func(t *testing.T) {
|
|
req := AudioRequest{
|
|
FilePath: "test.wav",
|
|
Reader: bytes.NewBuffer([]byte(`wav test contents`)),
|
|
}
|
|
|
|
mockFailedErr := fmt.Errorf("mock form builder fail")
|
|
mockBuilder := &mockFormBuilder{
|
|
mockCreateFormFileReader: func(string, io.Reader, string) error {
|
|
return mockFailedErr
|
|
},
|
|
}
|
|
|
|
err := createFileField(req, mockBuilder)
|
|
checks.ErrorIs(t, err, mockFailedErr, "createFileField using a reader should return error if form builder fails")
|
|
})
|
|
|
|
t.Run("createFileField failing open", func(t *testing.T) {
|
|
req := AudioRequest{
|
|
FilePath: "non_existing_file.wav",
|
|
}
|
|
|
|
mockBuilder := &mockFormBuilder{}
|
|
|
|
err := createFileField(req, mockBuilder)
|
|
checks.HasError(t, err, "createFileField using file should return error when open file fails")
|
|
})
|
|
}
|