mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-24 23:43:46 +08:00
40 lines
1.3 KiB
Go
40 lines
1.3 KiB
Go
package service
|
|
|
|
import "testing"
|
|
|
|
import "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_GrokMapsClaudeFamilyToGrok(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
group := &Group{Platform: PlatformGrok}
|
|
|
|
require.Equal(t, "grok-4.3", group.ResolveMessagesDispatchModel("claude-sonnet-4-5"))
|
|
require.Equal(t, "grok-4.3", group.ResolveMessagesDispatchModel("claude-opus-4-6"))
|
|
require.Equal(t, "grok-4.3", group.ResolveMessagesDispatchModel("claude-haiku-4-5"))
|
|
require.Empty(t, group.ResolveMessagesDispatchModel("grok"))
|
|
require.Empty(t, group.ResolveMessagesDispatchModel("gpt-5.3-codex"))
|
|
}
|