33 lines
1.0 KiB
Go
33 lines
1.0 KiB
Go
package seed
|
|
|
|
import "testing"
|
|
|
|
func TestDefaultsMatchConfirmedPrompts(t *testing.T) {
|
|
defaults, err := LoadDefaults()
|
|
if err != nil {
|
|
t.Fatalf("LoadDefaults() error = %v", err)
|
|
}
|
|
want := map[string]string{
|
|
"mvp0-chat-default": "{{.UserPrompt}}",
|
|
"mvp0-images-edits-default": "用户要求：\n{{.UserPrompt}}\n\n图片说明：\n- 第 1 张图片是需要处理的主图。\n- 其余图片仅作为参考。\n- 仅按用户明确要求进行修改，不添加用户未要求的风格、文字、商品属性或场景。",
|
|
}
|
|
seen := make(map[string]bool)
|
|
for _, prompt := range defaults.Prompts {
|
|
if seen[prompt.Key] {
|
|
t.Fatalf("duplicate prompt key %q", prompt.Key)
|
|
}
|
|
seen[prompt.Key] = true
|
|
if prompt.Template != want[prompt.Key] {
|
|
t.Errorf("prompt %q does not match confirmed text", prompt.Key)
|
|
}
|
|
if prompt.Capability == "" {
|
|
t.Errorf("prompt %q has no capability", prompt.Key)
|
|
}
|
|
}
|
|
for _, model := range defaults.Models {
|
|
if model.Capability == "" {
|
|
t.Errorf("model %q has no capability", model.ModelID)
|
|
}
|
|
}
|
|
}
|