7466119867
* feat(image): 图片参数 - 统一插件声明配置与生成合同 * docs(image): 图片参数 - 说明声明合同与验证边界 * fix(plugins): 图片声明 - 保留内联素材支持并同步制品文档 * fix(image): 参数合同 - 对齐审批意图和尺寸预检调用方 * fix(image): 调用链 - 保留公开意图并接通独立分辨率 * fix(image): 图片参数 - 补齐分辨率接线与目录价格投影
68 lines
3.0 KiB
Go
68 lines
3.0 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestImageParametersPluginRequestPreservesValuesAndModelScope(t *testing.T) {
|
|
adapter := officialPackageAdapter(t, "kacang-midjourney.yingce-plugin", "kacang-midjourney")
|
|
if len(adapter.Metadata().ImageParameters) == 0 {
|
|
t.Fatal("packaged plugin is missing image parameter declarations")
|
|
}
|
|
request := GenerationRequest{Capability: CapabilityImage, Model: "mj-v8.2", Prompt: "portrait", ImageParameters: map[string]any{
|
|
"size": "1:1", "count": 1, "seed": 0, "raw": false, "chaos": 0, "stylize": 450, "mjQuality": 0.5,
|
|
}}
|
|
spec, err := adapter.BuildCreate(context.Background(), RequestContext{Request: request})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
body := manifestTestBody(t, spec)
|
|
for name, expected := range map[string]any{"size": "1:1", "seed": float64(0), "raw": false, "chaos": float64(0), "stylize": float64(450), "quality": 0.5} {
|
|
if body[name] != expected {
|
|
t.Fatalf("%s = %#v, want %#v", name, body[name], expected)
|
|
}
|
|
}
|
|
request.Model = "Midjourney v8.2 高速"
|
|
if _, err := adapter.BuildCreate(context.Background(), RequestContext{Request: request}); err == nil {
|
|
t.Fatal("fast model accepted unsupported stable-only parameters")
|
|
}
|
|
request.Model = "mj-v8.2"
|
|
request.ImageParameters["stylize"] = 1001
|
|
if _, err := adapter.BuildCreate(context.Background(), RequestContext{Request: request}); err == nil {
|
|
t.Fatal("out of range value accepted")
|
|
}
|
|
}
|
|
|
|
func TestImageParametersCannotExpandPluginConstraintsOrBindings(t *testing.T) {
|
|
minimum, maximum := 0.0, 100.0
|
|
definitions := []ImageParameter{{Name: "amount", Label: "Amount", Type: "number", Min: &minimum, Max: &maximum, Binding: "providerOptions.example.amount"}}
|
|
expanded := 101.0
|
|
if _, err := ResolveImageParameters(definitions, []ImageParameter{{Name: "amount", Max: &expanded}}, "model"); err == nil {
|
|
t.Fatal("channel expanded plugin bounds")
|
|
}
|
|
resolved, err := ResolveImageParameters(definitions, []ImageParameter{{Name: "amount", Binding: "providerOptions.evil.amount", Pricing: true, Default: 0}}, "model")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if resolved[0].Binding != definitions[0].Binding || resolved[0].Pricing {
|
|
t.Fatalf("channel replaced trusted metadata: %#v", resolved[0])
|
|
}
|
|
if _, err := ResolveImageParameters(definitions, []ImageParameter{{Name: "invented"}}, "model"); err == nil {
|
|
t.Fatal("undeclared field accepted")
|
|
}
|
|
}
|
|
|
|
func TestImageParametersGeminiResolutionIsIndependentFromStaleQuality(t *testing.T) {
|
|
adapter := officialPackageAdapter(t, "google-gemini-image.yingce-plugin", "gemini-image")
|
|
request := GenerationRequest{Capability: CapabilityImage, Model: "image-model", Prompt: "portrait", ImageParameters: map[string]any{"imageResolution": "2k", "seed": 0}}
|
|
spec, err := adapter.BuildCreate(context.Background(), RequestContext{Request: request})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
config := manifestTestBody(t, spec)["generationConfig"].(map[string]any)
|
|
if config["imageConfig"].(map[string]any)["imageSize"] != "2K" || config["seed"] != float64(0) {
|
|
t.Fatalf("mapped request: %#v", config)
|
|
}
|
|
}
|