feat: add domestic model provider presets
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user