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]>
103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package openai_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hanzoai/go-openai"
|
|
)
|
|
|
|
func TestGetAzureDeploymentByModel(t *testing.T) {
|
|
cases := []struct {
|
|
Model string
|
|
AzureModelMapperFunc func(model string) string
|
|
Expect string
|
|
}{
|
|
{
|
|
Model: "gpt-3.5-turbo",
|
|
Expect: "gpt-35-turbo",
|
|
},
|
|
{
|
|
Model: "gpt-3.5-turbo-0301",
|
|
Expect: "gpt-35-turbo-0301",
|
|
},
|
|
{
|
|
Model: "text-embedding-ada-002",
|
|
Expect: "text-embedding-ada-002",
|
|
},
|
|
{
|
|
Model: "",
|
|
Expect: "",
|
|
},
|
|
{
|
|
Model: "models",
|
|
Expect: "models",
|
|
},
|
|
{
|
|
Model: "gpt-3.5-turbo",
|
|
Expect: "my-gpt35",
|
|
AzureModelMapperFunc: func(model string) string {
|
|
modelmapper := map[string]string{
|
|
"gpt-3.5-turbo": "my-gpt35",
|
|
}
|
|
if val, ok := modelmapper[model]; ok {
|
|
return val
|
|
}
|
|
return model
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.Model, func(t *testing.T) {
|
|
conf := openai.DefaultAzureConfig("", "https://test.openai.azure.com/")
|
|
if c.AzureModelMapperFunc != nil {
|
|
conf.AzureModelMapperFunc = c.AzureModelMapperFunc
|
|
}
|
|
actual := conf.GetAzureDeploymentByModel(c.Model)
|
|
if actual != c.Expect {
|
|
t.Errorf("Expected %s, got %s", c.Expect, actual)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDefaultAnthropicConfig(t *testing.T) {
|
|
apiKey := "test-key"
|
|
baseURL := "https://api.anthropic.com/v1"
|
|
|
|
config := openai.DefaultAnthropicConfig(apiKey, baseURL)
|
|
|
|
if config.APIType != openai.APITypeAnthropic {
|
|
t.Errorf("Expected APIType to be %v, got %v", openai.APITypeAnthropic, config.APIType)
|
|
}
|
|
|
|
if config.APIVersion != openai.AnthropicAPIVersion {
|
|
t.Errorf("Expected APIVersion to be 2023-06-01, got %v", config.APIVersion)
|
|
}
|
|
|
|
if config.BaseURL != baseURL {
|
|
t.Errorf("Expected BaseURL to be %v, got %v", baseURL, config.BaseURL)
|
|
}
|
|
|
|
if config.EmptyMessagesLimit != 300 {
|
|
t.Errorf("Expected EmptyMessagesLimit to be 300, got %v", config.EmptyMessagesLimit)
|
|
}
|
|
}
|
|
|
|
func TestDefaultAnthropicConfigWithEmptyValues(t *testing.T) {
|
|
config := openai.DefaultAnthropicConfig("", "")
|
|
|
|
if config.APIType != openai.APITypeAnthropic {
|
|
t.Errorf("Expected APIType to be %v, got %v", openai.APITypeAnthropic, config.APIType)
|
|
}
|
|
|
|
if config.APIVersion != openai.AnthropicAPIVersion {
|
|
t.Errorf("Expected APIVersion to be %s, got %v", openai.AnthropicAPIVersion, config.APIVersion)
|
|
}
|
|
|
|
expectedBaseURL := "https://api.anthropic.com/v1"
|
|
if config.BaseURL != expectedBaseURL {
|
|
t.Errorf("Expected BaseURL to be %v, got %v", expectedBaseURL, config.BaseURL)
|
|
}
|
|
}
|