feat: add seedance media params and model visibility

2026-05-27 13:02:36 +08:00
parent 871bedc072
commit 4bffb55695
25 changed files with 873 additions and 140 deletions
@@ -60,3 +60,41 @@ func TestBuildModelChannelURLNormalizesArkPlanTaskPath(t *testing.T) {
 		t.Fatalf("BuildModelChannelURL = %q, want %q", got, want)
 	}
 }
+
+func TestNormalizeSettingsPublishesEnabledChannelModelsAndRepairsDefaults(t *testing.T) {
+	settings := normalizeSettings(model.Settings{
+		Public: model.PublicSetting{
+			ModelChannel: model.PublicModelChannelSetting{
+				AvailableModels:   []string{"grok-imagine-video", "disabled-model"},
+				DefaultModel:      "grok-imagine-video",
+				DefaultTextModel:  "missing-text",
+				DefaultImageModel: "missing-image",
+				DefaultVideoModel: "missing-video",
+			},
+		},
+		Private: model.PrivateSetting{
+			Channels: []model.ModelChannel{
+				{Enabled: true, Models: []string{"gpt-5.5", "doubao-seedream-5.0-lite", "doubao-seedance-2.0-fast", "gpt-5.5"}},
+				{Enabled: false, Models: []string{"disabled-model"}},
+			},
+		},
+	})
+
+	channel := settings.Public.ModelChannel
+	wantModels := []string{"gpt-5.5", "doubao-seedream-5.0-lite", "doubao-seedance-2.0-fast"}
+	if !reflect.DeepEqual(channel.AvailableModels, wantModels) {
+		t.Fatalf("available models = %#v, want %#v", channel.AvailableModels, wantModels)
+	}
+	if channel.DefaultModel != "gpt-5.5" {
+		t.Fatalf("default model = %q, want text model", channel.DefaultModel)
+	}
+	if channel.DefaultTextModel != "gpt-5.5" {
+		t.Fatalf("default text model = %q, want text model", channel.DefaultTextModel)
+	}
+	if channel.DefaultImageModel != "doubao-seedream-5.0-lite" {
+		t.Fatalf("default image model = %q, want seedream", channel.DefaultImageModel)
+	}
+	if channel.DefaultVideoModel != "doubao-seedance-2.0-fast" {
+		t.Fatalf("default video model = %q, want seedance", channel.DefaultVideoModel)
+	}
+}