李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
72 lines
2.4 KiB
Go
72 lines
2.4 KiB
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/xai"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNormalizeOpenAIMessagesDispatchModelConfig(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cfg := normalizeOpenAIMessagesDispatchModelConfig(OpenAIMessagesDispatchModelConfig{
|
|
OpusMappedModel: " gpt-5.4-high ",
|
|
SonnetMappedModel: "gpt-5.3-codex",
|
|
HaikuMappedModel: " gpt-5.4-mini-medium ",
|
|
ExactModelMappings: map[string]string{
|
|
" claude-sonnet-4-5-20250929 ": " gpt-5.2-high ",
|
|
"": "gpt-5.4",
|
|
"claude-opus-4-6": " ",
|
|
},
|
|
})
|
|
|
|
require.Equal(t, "gpt-5.4", cfg.OpusMappedModel)
|
|
require.Equal(t, "gpt-5.3-codex", cfg.SonnetMappedModel)
|
|
require.Equal(t, "gpt-5.4-mini", cfg.HaikuMappedModel)
|
|
require.Equal(t, map[string]string{
|
|
"claude-sonnet-4-5-20250929": "gpt-5.2",
|
|
}, cfg.ExactModelMappings)
|
|
}
|
|
|
|
func TestGroupResolveMessagesDispatchModel_GrokRequiresCrossClientMapping(t *testing.T) {
|
|
original := xai.RuntimeModelMappingOptions()
|
|
t.Cleanup(func() { xai.SetRuntimeModelMappingOptions(original) })
|
|
group := &Group{Platform: PlatformGrok}
|
|
|
|
xai.SetRuntimeModelMappingOptions(xai.ModelMappingOptions{})
|
|
require.Empty(t, group.ResolveMessagesDispatchModel("claude-sonnet-4-5"))
|
|
|
|
xai.SetRuntimeModelMappingOptions(xai.ModelMappingOptions{
|
|
DefaultText: "grok-build-0.1",
|
|
EnableCrossClientMap: true,
|
|
})
|
|
require.Equal(t, "grok-build-0.1", group.ResolveMessagesDispatchModel("claude-sonnet-4-5"))
|
|
require.Equal(t, "grok-build-0.1", group.ResolveMessagesDispatchModel("claude-opus-4-6"))
|
|
require.Equal(t, "grok-build-0.1", group.ResolveMessagesDispatchModel("claude-haiku-4-5"))
|
|
require.Empty(t, group.ResolveMessagesDispatchModel("grok"))
|
|
require.Empty(t, group.ResolveMessagesDispatchModel("gpt-5.3-codex"))
|
|
}
|
|
|
|
func TestSanitizeGroupMessagesDispatchFields_ClearsNonOpenAIPlatform(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
group := &Group{
|
|
Platform: PlatformAnthropic,
|
|
AllowMessagesDispatch: true,
|
|
DefaultMappedModel: "gpt-5.6-sol",
|
|
MessagesDispatchModelConfig: OpenAIMessagesDispatchModelConfig{
|
|
SonnetMappedModel: "gpt-5.3-codex",
|
|
ExactModelMappings: map[string]string{
|
|
"claude-fable-5": "gpt-5.6-sol",
|
|
},
|
|
},
|
|
}
|
|
|
|
sanitizeGroupMessagesDispatchFields(group)
|
|
|
|
require.False(t, group.AllowMessagesDispatch)
|
|
require.Empty(t, group.DefaultMappedModel)
|
|
require.Equal(t, OpenAIMessagesDispatchModelConfig{}, group.MessagesDispatchModelConfig)
|
|
}
|