85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package matrix
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
|
||
libprompt "haixun-backend/internal/library/prompt"
|
||
)
|
||
|
||
type CopyGenerateInput struct {
|
||
Count int
|
||
TopicLabel string
|
||
TopicBrief string
|
||
ResearchMap string
|
||
SelectedTags []string
|
||
ViralSamples string
|
||
PersonaBlock string
|
||
}
|
||
|
||
func BuildCopyUserPrompt(in CopyGenerateInput) (string, error) {
|
||
count := in.Count
|
||
if count <= 0 {
|
||
count = 5
|
||
}
|
||
if count > 12 {
|
||
count = 12
|
||
}
|
||
return libprompt.MatrixCopyUser(map[string]string{
|
||
"count": fmt.Sprintf("%d", count),
|
||
"topic_label": strings.TrimSpace(in.TopicLabel),
|
||
"topic_brief": strings.TrimSpace(in.TopicBrief),
|
||
"research_map_block": strings.TrimSpace(in.ResearchMap),
|
||
"selected_tags_block": formatTagList(in.SelectedTags),
|
||
"viral_samples_block": strings.TrimSpace(in.ViralSamples),
|
||
"persona_block": strings.TrimSpace(in.PersonaBlock),
|
||
})
|
||
}
|
||
|
||
func formatTagList(tags []string) string {
|
||
if len(tags) == 0 {
|
||
return "（尚未選擇）"
|
||
}
|
||
lines := make([]string, 0, len(tags))
|
||
for _, tag := range tags {
|
||
tag = strings.TrimSpace(tag)
|
||
if tag == "" {
|
||
continue
|
||
}
|
||
lines = append(lines, "- "+tag)
|
||
}
|
||
if len(lines) == 0 {
|
||
return "（尚未選擇）"
|
||
}
|
||
return strings.Join(lines, "\n")
|
||
}
|
||
|
||
func FormatCopyResearchMapBlock(audience, goal string, questions, pillars, exclusions []string) string {
|
||
var b strings.Builder
|
||
if audience = strings.TrimSpace(audience); audience != "" {
|
||
b.WriteString("受眾：")
|
||
b.WriteString(audience)
|
||
b.WriteString("\n")
|
||
}
|
||
if goal = strings.TrimSpace(goal); goal != "" {
|
||
b.WriteString("內容目標：")
|
||
b.WriteString(goal)
|
||
b.WriteString("\n")
|
||
}
|
||
if len(pillars) > 0 {
|
||
b.WriteString("支柱：")
|
||
b.WriteString(strings.Join(pillars, "、"))
|
||
b.WriteString("\n")
|
||
}
|
||
if len(questions) > 0 {
|
||
b.WriteString("受眾問題：")
|
||
b.WriteString(strings.Join(questions, "；"))
|
||
b.WriteString("\n")
|
||
}
|
||
if len(exclusions) > 0 {
|
||
b.WriteString("排除：")
|
||
b.WriteString(strings.Join(exclusions, "；"))
|
||
}
|
||
return strings.TrimSpace(b.String())
|
||
}
|