wehub-resource-sync
126 lines
3.6 KiB
Go
126 lines
3.6 KiB
Go
package openai
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
"testing"
|
|
|
|
"reasonix/internal/provider"
|
|
)
|
|
|
|
func newClient(t *testing.T, baseURL, effort string) *client {
|
|
t.Helper()
|
|
extra := map[string]any{}
|
|
if effort != "" {
|
|
extra["effort"] = effort
|
|
}
|
|
p, err := New(provider.Config{Name: "p", BaseURL: baseURL, Model: "m", APIKey: "k", Extra: extra})
|
|
if err != nil {
|
|
t.Fatalf("New(%q, effort=%q): %v", baseURL, effort, err)
|
|
}
|
|
return p.(*client)
|
|
}
|
|
|
|
func TestEffortNormalization(t *testing.T) {
|
|
const mimo = "https://api.xiaomimimo.com/v1"
|
|
const deepseek = "https://api.deepseek.com/v1"
|
|
|
|
tests := []struct {
|
|
base, effort, want string
|
|
}{
|
|
{mimo, "max", "high"}, // DeepSeek-ism clamped to the OpenAI ceiling — MiMo 400s on "max"
|
|
{mimo, "high", "high"},
|
|
{mimo, "medium", "medium"},
|
|
{mimo, "low", "low"},
|
|
{mimo, "MAX", "high"}, // case-insensitive
|
|
{mimo, "auto", ""}, // UI/config auto means omit provider-specific effort
|
|
{mimo, "", ""}, // unset stays omitted
|
|
{deepseek, "max", "max"},
|
|
{deepseek, "high", "high"},
|
|
{deepseek, "auto", "high"},
|
|
{deepseek, "", "high"}, // DeepSeek default depth
|
|
}
|
|
for _, tc := range tests {
|
|
if got := newClient(t, tc.base, tc.effort).effort; got != tc.want {
|
|
t.Errorf("base=%s effort=%q: got %q, want %q", tc.base, tc.effort, got, tc.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestEffortInvalidRejected(t *testing.T) {
|
|
_, err := New(provider.Config{
|
|
Name: "p", BaseURL: "https://api.xiaomimimo.com/v1", Model: "m", APIKey: "k",
|
|
Extra: map[string]any{"effort": "turbo"},
|
|
})
|
|
if err == nil || !strings.Contains(err.Error(), "low, medium, or high") {
|
|
t.Fatalf("expected a low/medium/high validation error, got: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestReasoningProtocolOverridesEndpointHeuristic(t *testing.T) {
|
|
p, err := New(provider.Config{
|
|
Name: "deepseek-proxy",
|
|
BaseURL: "https://proxy.example.com/v1",
|
|
Model: "deepseek-v4-flash",
|
|
APIKey: "k",
|
|
Extra: map[string]any{"reasoning_protocol": "deepseek"},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("New deepseek protocol: %v", err)
|
|
}
|
|
c := p.(*client)
|
|
if !c.deepseek || c.effort != "high" {
|
|
t.Fatalf("deepseek=%v effort=%q, want true/high", c.deepseek, c.effort)
|
|
}
|
|
|
|
p, err = New(provider.Config{
|
|
Name: "deepseek-direct",
|
|
BaseURL: "https://api.deepseek.com/v1",
|
|
Model: "deepseek-v4-flash",
|
|
APIKey: "k",
|
|
Extra: map[string]any{"reasoning_protocol": "none", "effort": "max"},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("New none protocol: %v", err)
|
|
}
|
|
c = p.(*client)
|
|
if c.deepseek || c.effort != "" {
|
|
t.Fatalf("deepseek=%v effort=%q, want false/empty", c.deepseek, c.effort)
|
|
}
|
|
}
|
|
|
|
func TestLongCatThinkingUsesThinkingField(t *testing.T) {
|
|
p, err := New(provider.Config{
|
|
Name: "longcat",
|
|
BaseURL: "https://api.longcat.chat/openai/v1",
|
|
Model: "LongCat-2.0",
|
|
APIKey: "k",
|
|
Extra: map[string]any{"effort": "disabled", "thinking": "enabled"},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("New longcat: %v", err)
|
|
}
|
|
c := p.(*client)
|
|
if !c.longcat || c.effort != "disabled" {
|
|
t.Fatalf("longcat=%v effort=%q, want true/disabled", c.longcat, c.effort)
|
|
}
|
|
req := c.buildRequest(provider.Request{
|
|
Messages: []provider.Message{{Role: provider.RoleUser, Content: "hi"}},
|
|
})
|
|
b, err := json.Marshal(req)
|
|
if err != nil {
|
|
t.Fatalf("marshal request: %v", err)
|
|
}
|
|
var body map[string]any
|
|
if err := json.Unmarshal(b, &body); err != nil {
|
|
t.Fatalf("decode request: %v", err)
|
|
}
|
|
thinking, _ := body["thinking"].(map[string]any)
|
|
if thinking["type"] != "disabled" {
|
|
t.Fatalf("thinking = %#v, want disabled", body["thinking"])
|
|
}
|
|
if _, ok := body["reasoning_effort"]; ok {
|
|
t.Fatalf("LongCat request must omit reasoning_effort: %s", b)
|
|
}
|
|
}
|