mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-28 08:08:32 +00:00
IanShaw027
a04ce49016
官方目录价：<200k 为 $2 / 缓存读取 $0.50 / $6，≥200k 全部 2 倍。 原先没有独立价卡，/models 宣称可配 reasoning 但请求路径会剥掉 effort。 - 目录与别名接入 grok-4.6 / grok-4.6-latest，默认文本模型仍为 grok-4.5 - 独立回退价卡含缓存读取价与 200k 长上下文倍率 - 保留 reasoning_effort；Chat→Responses 的 cache/vision 桥接同样识别 4.6
75 lines
2.8 KiB
Go
75 lines
2.8 KiB
Go
package xai
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDefaultModelMappingExcludesCrossClientWildcards(t *testing.T) {
|
|
original := RuntimeModelMappingOptions()
|
|
t.Cleanup(func() { SetRuntimeModelMappingOptions(original) })
|
|
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, DefaultTextModel, mapping["grok-build-latest"])
|
|
require.Equal(t, DefaultImagineImageQualityModel, mapping["grok-imagine-edit"])
|
|
require.Equal(t, DefaultImagineVideo15LegacyModel, 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"))
|
|
require.Equal(t, "grok-imagine-video-2", CanonicalImagineVideoModel("grok-imagine-video-2"))
|
|
}
|
|
|
|
func TestIsGrokModelID(t *testing.T) {
|
|
t.Parallel()
|
|
require.True(t, IsGrokModelID("grok-4.5"))
|
|
require.True(t, IsGrokModelID("grok-4.6"))
|
|
require.True(t, IsGrokModelID("x-ai/grok-4.3"))
|
|
require.False(t, IsGrokModelID("gpt-5"))
|
|
require.False(t, IsGrokModelID("claude-sonnet-4"))
|
|
}
|
|
|
|
func TestDefaultModelsIncludesGrok46(t *testing.T) {
|
|
t.Parallel()
|
|
ids := DefaultModelIDs()
|
|
require.Contains(t, ids, "grok-4.6")
|
|
require.Equal(t, "grok-4.6", ResolveGrokTextResponsesModelID("grok-4.6"))
|
|
require.Equal(t, "grok-4.6", ResolveGrokTextResponsesModelID("grok-4.6-latest"))
|
|
}
|
|
|
|
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"))
|
|
}
|