若干更新迭代
This commit is contained in:
1 parent
deb9cc5bcf
commit
4ac786aafb
64 files changed
+3251
-291
No files matched your search
@@ -432,6 +432,38 @@ func TestProviderVideoModelsKeepSeedance20And25Available(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestProviderVideoModelsIncludeWan30AndScopedReferenceLimits(t *testing.T) {
|
||||
targets := providerVideoModels(func(string) string { return "" })
|
||||
if target := targets["wan3.0-video"]; target.Provider != "bailian" || target.Model != "wan3.0-video" {
|
||||
t.Fatalf("Wan 3.0 target=%#v", target)
|
||||
}
|
||||
if target := targets[providers.MinimaxH3Model]; target.Provider != "minimax" {
|
||||
t.Fatalf("MiniMax target=%#v", target)
|
||||
}
|
||||
if got := bailianVideoCapabilityLimitsForModel("wan3.0-video")["inputImages"].(map[string]int)["max"]; got != 10 {
|
||||
t.Fatalf("Wan 3.0 max images=%d", got)
|
||||
}
|
||||
if got := minimaxVideoCapabilityLimits()["inputImages"].(map[string]int)["max"]; got != 9 {
|
||||
t.Fatalf("MiniMax max images=%d", got)
|
||||
}
|
||||
if got := seedanceCapabilityLimits("doubao-seedance-2-0-260128")["materials"].(map[string]int)["images"]; got != 9 {
|
||||
t.Fatalf("Seedance 2.0 max images=%d", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBailianCapabilityLimitsUseItsConfiguredModelWhenAnotherEngineIsDefault(t *testing.T) {
|
||||
values := map[string]string{"VIDEO_GENERATE_ENGINE": "seedance", "BAILIAN_VIDEO_MODEL": "wan3.0-video"}
|
||||
capabilities, err := capabilitySummary(func(name string) string { return values[name] })(context.Background())
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
bailian := capabilities.([]any)[2].(map[string]any)
|
||||
limits := bailian["limits"].(map[string]any)
|
||||
if bailian["reqKey"] != "wan3.0-video" || limits["inputImages"].(map[string]int)["max"] != 10 || limits["durationSeconds"].(map[string]int)["max"] != 30 {
|
||||
t.Fatalf("Bailian capability=%#v", bailian)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCapabilitySummaryMatchesConfiguredDefaultVideoEngine(t *testing.T) {
|
||||
for _, test := range []struct {
|
||||
name, engine, wantEngine, wantProvider, wantModel string
|
||||
|
||||
Reference in new issue
Block a user