55 lines
2.0 KiB
Go
55 lines
2.0 KiB
Go
package spark
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
)
|
||
|
||
var BlessingStyles = map[string]string{
|
||
"classic": "古风典雅",
|
||
"modern": "现代文风",
|
||
"funny": "搞怪文风",
|
||
"roast": "损友文风",
|
||
}
|
||
|
||
func NormalizeStyle(style string) (string, bool) {
|
||
s := strings.TrimSpace(style)
|
||
_, ok := BlessingStyles[s]
|
||
return s, ok
|
||
}
|
||
|
||
func (c *Client) GenerateBlessing(style string) (string, error) {
|
||
styleKey, ok := NormalizeStyle(style)
|
||
if !ok {
|
||
return "", fmt.Errorf("无效的祝福风格")
|
||
}
|
||
label := BlessingStyles[styleKey]
|
||
|
||
styleGuide := map[string]string{
|
||
"classic": "文言雅致、含蓄温润；可点到意象（花月、琴瑟、白首），忌生僻堆砌与整段文言文。",
|
||
"modern": "口语真诚、干净利落；像朋友当面说的话，不鸡汤、不口号堆叠。",
|
||
"funny": "俏皮轻松、会心一笑；可玩梗但保持善意，忌低俗与尴尬烂梗。",
|
||
"roast": "损友口吻：先轻怼一句再落真心祝福；尺度友好，不人身攻击、不揭隐私。",
|
||
}
|
||
|
||
sys := `你是婚礼短祝福文案助手。
|
||
只输出一条可直接填写的中文祝福正文。
|
||
禁止：标题、序号、引号包裹、解释说明、表情符号、英文、话题标签。
|
||
禁止：出现任何真实姓名、网名、称呼占位（如某某、X先生/女士）、新郎新娘具体姓氏。
|
||
称呼统一用「你们」「新人」等泛称即可。
|
||
文案要有意境：有画面感与余韵，可用轻巧意象（光、风、茶、路、灯火、四季等）点染，忌空泛套话与意象堆砌。`
|
||
|
||
user := fmt.Sprintf(`请写一条送给婚礼新人的短祝福。
|
||
风格：%s
|
||
写法：%s
|
||
意境：先落一两处可感的画面或氛围，再落到真诚祝福，读完有余味。
|
||
篇幅：28～48个汉字，一两句说完。
|
||
内容：祝福婚姻美满、相守与日常，语气自然，适合弹幕或回执留言。
|
||
请直接输出祝福正文。`, label, styleGuide[styleKey])
|
||
|
||
return c.Chat([]Message{
|
||
{Role: "system", Content: sys},
|
||
{Role: "user", Content: user},
|
||
})
|
||
}
|