mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-28 00:07:33 +00:00
IanShaw027
74249b8fed
补齐 Grok/Imagine 官方模型与别名，并支持运行时默认文本模型。 默认 model_mapping 仅含 Grok 家族；新增 grok_default_text_model 与 grok_cross_client_model_map_enabled，仅在显式开启时将 gpt/claude/codex 等客户端模型名映射到默认文本模型，避免非 Grok 请求被静默改成 grok-4.5。
63 lines
2.3 KiB
Go
63 lines
2.3 KiB
Go
package xai
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDefaultModelMappingExcludesCrossClientWildcards(t *testing.T) {
|
|
t.Parallel()
|
|
SetRuntimeModelMappingOptions(ModelMappingOptions{})
|
|
mapping := DefaultModelMapping()
|
|
|
|
require.Equal(t, "grok-4.5", mapping["grok"])
|
|
require.Equal(t, "grok-4.5", mapping["grok-latest"])
|
|
require.Equal(t, "grok-build-0.1", mapping["grok-build"])
|
|
require.Equal(t, "grok-build-0.1", mapping["grok-build-latest"])
|
|
require.Equal(t, DefaultImagineVideo15Model, mapping["grok-imagine-video-1.5"])
|
|
require.Equal(t, DefaultImagineVideo15Model, mapping["grok-imagine-video-1.5-preview"])
|
|
require.Equal(t, "grok-4.5", mapping["xai/grok"])
|
|
|
|
// Cross-vendor wildcards must stay opt-in.
|
|
_, hasGPT := mapping["gpt-*"]
|
|
_, hasClaude := mapping["claude-*"]
|
|
require.False(t, hasGPT)
|
|
require.False(t, hasClaude)
|
|
}
|
|
|
|
func TestModelMappingWithOptionsCrossClient(t *testing.T) {
|
|
t.Parallel()
|
|
mapping := ModelMappingWithOptions(ModelMappingOptions{
|
|
DefaultText: "grok-4.3",
|
|
EnableCrossClientMap: true,
|
|
})
|
|
require.Equal(t, "grok-4.3", mapping["grok"])
|
|
require.Equal(t, "grok-4.3", mapping["gpt-*"])
|
|
require.Equal(t, "grok-4.3", mapping["claude-*"])
|
|
require.Equal(t, "grok-4.3", mapping["codex-*"])
|
|
}
|
|
|
|
func TestCanonicalImagineVideoModel(t *testing.T) {
|
|
t.Parallel()
|
|
require.Equal(t, DefaultImagineVideoModel, CanonicalImagineVideoModel("grok-imagine-video"))
|
|
require.Equal(t, DefaultImagineVideo15Model, CanonicalImagineVideoModel("grok-imagine-video-1.5"))
|
|
require.Equal(t, DefaultImagineVideo15Model, CanonicalImagineVideoModel("grok-imagine-video-1.5-preview"))
|
|
require.Equal(t, DefaultImagineVideo15Model, CanonicalImagineVideoModel("xai/grok-video-1.5"))
|
|
}
|
|
|
|
func TestIsGrokModelID(t *testing.T) {
|
|
t.Parallel()
|
|
require.True(t, IsGrokModelID("grok-4.5"))
|
|
require.True(t, IsGrokModelID("x-ai/grok-4.3"))
|
|
require.False(t, IsGrokModelID("gpt-5"))
|
|
require.False(t, IsGrokModelID("claude-sonnet-4"))
|
|
}
|
|
|
|
func TestResolveGrokTextResponsesModelID(t *testing.T) {
|
|
t.Parallel()
|
|
require.Equal(t, "grok-4.5", ResolveGrokTextResponsesModelID(""))
|
|
require.Equal(t, "grok-4.3", ResolveGrokTextResponsesModelID("grok", "grok-4.3"))
|
|
require.Equal(t, "grok-4.20-multi-agent-0309", ResolveGrokTextResponsesModelID("grok-4.20-multi-agent"))
|
|
}
|