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]>
79 lines
2.3 KiB
Go
79 lines
2.3 KiB
Go
package openai //nolint:testpackage // testing private field
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"errors"
|
|
"testing"
|
|
|
|
utils "github.com/hanzoai/go-openai/internal"
|
|
"github.com/hanzoai/go-openai/internal/test"
|
|
"github.com/hanzoai/go-openai/internal/test/checks"
|
|
)
|
|
|
|
var errTestUnmarshalerFailed = errors.New("test unmarshaler failed")
|
|
|
|
type failingUnMarshaller struct{}
|
|
|
|
func (*failingUnMarshaller) Unmarshal(_ []byte, _ any) error {
|
|
return errTestUnmarshalerFailed
|
|
}
|
|
|
|
func TestStreamReaderReturnsUnmarshalerErrors(t *testing.T) {
|
|
stream := &streamReader[ChatCompletionStreamResponse]{
|
|
errAccumulator: utils.NewErrorAccumulator(),
|
|
unmarshaler: &failingUnMarshaller{},
|
|
}
|
|
|
|
respErr := stream.unmarshalError()
|
|
if respErr != nil {
|
|
t.Fatalf("Did not return nil with empty buffer: %v", respErr)
|
|
}
|
|
|
|
err := stream.errAccumulator.Write([]byte("{"))
|
|
if err != nil {
|
|
t.Fatalf("%+v", err)
|
|
}
|
|
|
|
respErr = stream.unmarshalError()
|
|
if respErr != nil {
|
|
t.Fatalf("Did not return nil when unmarshaler failed: %v", respErr)
|
|
}
|
|
}
|
|
|
|
func TestStreamReaderReturnsErrTooManyEmptyStreamMessages(t *testing.T) {
|
|
stream := &streamReader[ChatCompletionStreamResponse]{
|
|
emptyMessagesLimit: 3,
|
|
reader: bufio.NewReader(bytes.NewReader([]byte("\n\n\n\n"))),
|
|
errAccumulator: utils.NewErrorAccumulator(),
|
|
unmarshaler: &utils.JSONUnmarshaler{},
|
|
}
|
|
_, err := stream.Recv()
|
|
checks.ErrorIs(t, err, ErrTooManyEmptyStreamMessages, "Did not return error when recv failed", err.Error())
|
|
}
|
|
|
|
func TestStreamReaderReturnsErrTestErrorAccumulatorWriteFailed(t *testing.T) {
|
|
stream := &streamReader[ChatCompletionStreamResponse]{
|
|
reader: bufio.NewReader(bytes.NewReader([]byte("\n"))),
|
|
errAccumulator: &utils.DefaultErrorAccumulator{
|
|
Buffer: &test.FailingErrorBuffer{},
|
|
},
|
|
unmarshaler: &utils.JSONUnmarshaler{},
|
|
}
|
|
_, err := stream.Recv()
|
|
checks.ErrorIs(t, err, test.ErrTestErrorAccumulatorWriteFailed, "Did not return error when write failed", err.Error())
|
|
}
|
|
|
|
func TestStreamReaderRecvRaw(t *testing.T) {
|
|
stream := &streamReader[ChatCompletionStreamResponse]{
|
|
reader: bufio.NewReader(bytes.NewReader([]byte("data: {\"key\": \"value\"}\n"))),
|
|
}
|
|
rawLine, err := stream.RecvRaw()
|
|
if err != nil {
|
|
t.Fatalf("Did not return raw line: %v", err)
|
|
}
|
|
if !bytes.Equal(rawLine, []byte("{\"key\": \"value\"}")) {
|
|
t.Fatalf("Did not return raw line: %v", string(rawLine))
|
|
}
|
|
}
|