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]>
44 lines
1.4 KiB
Go
44 lines
1.4 KiB
Go
package test
|
|
|
|
import (
|
|
"github.com/hanzoai/go-openai/internal/test/checks"
|
|
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
// CreateTestFile creates a fake file with "hello" as the content.
|
|
func CreateTestFile(t *testing.T, path string) {
|
|
file, err := os.Create(path)
|
|
checks.NoError(t, err, "failed to create file")
|
|
|
|
if _, err = file.WriteString("hello"); err != nil {
|
|
t.Fatalf("failed to write to file %v", err)
|
|
}
|
|
file.Close()
|
|
}
|
|
|
|
// TokenRoundTripper is a struct that implements the RoundTripper
|
|
// interface, specifically to handle the authentication token by adding a token
|
|
// to the request header. We need this because the API requires that each
|
|
// request include a valid API token in the headers for authentication and
|
|
// authorization.
|
|
type TokenRoundTripper struct {
|
|
Token string
|
|
Fallback http.RoundTripper
|
|
}
|
|
|
|
// RoundTrip takes an *http.Request as input and returns an
|
|
// *http.Response and an error.
|
|
//
|
|
// It is expected to use the provided request to create a connection to an HTTP
|
|
// server and return the response, or an error if one occurred. The returned
|
|
// Response should have its Body closed. If the RoundTrip method returns an
|
|
// error, the Client's Get, Head, Post, and PostForm methods return the same
|
|
// error.
|
|
func (t *TokenRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
req.Header.Set("Authorization", "Bearer "+t.Token)
|
|
return t.Fallback.RoundTrip(req)
|
|
}
|