feat: add domestic model provider presets

This commit is contained in:
Codex Local
2026-08-30 19:55:17 +08:00
parent 522785eda8
commit dcf0fc0d7f
13 changed files with 314 additions and 28 deletions
+27
View File
@@ -2,6 +2,33 @@ package contentmodel
import "testing"
func TestValidateInputAcceptsSupportedDomesticProviders(t *testing.T) {
providers := []string{"ark", "dashscope", "deepseek", "moonshot", "zhipu", "qianfan", "hunyuan", "minimax", "baichuan", "siliconflow"}
for _, provider := range providers {
t.Run(provider, func(t *testing.T) {
input := CreateInput{
TenantID: "10000000-0000-0000-0000-000000000001", DisplayName: provider,
Provider: provider, BaseURL: "https://model.example.com/v1", ModelID: "summary-model",
APIKey: "test-api-key", Temperature: 0.2, MaxTokens: 1200,
}
if err := validateInput(input); err != nil {
t.Fatalf("supported provider was rejected: %v", err)
}
})
}
}
func TestValidateInputRejectsUnknownProvider(t *testing.T) {
input := CreateInput{
TenantID: "10000000-0000-0000-0000-000000000001", DisplayName: "unknown",
Provider: "unknown", BaseURL: "https://model.example.com/v1", ModelID: "summary-model",
APIKey: "test-api-key", Temperature: 0.2, MaxTokens: 1200,
}
if err := validateInput(input); err == nil {
t.Fatal("unknown provider was accepted")
}
}
func TestValidateInputAllowsComposeOllamaEndpoint(t *testing.T) {
input := CreateInput{
TenantID: "10000000-0000-0000-0000-000000000001", DisplayName: "本地 Qwen",