105 lines
3.2 KiB
Go
105 lines
3.2 KiB
Go
|
package placement
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"regexp"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type ResearchMap struct {
|
|||
|
|
AudienceSummary string `json:"audienceSummary"`
|
|||
|
|
ContentGoal string `json:"contentGoal"`
|
|||
|
|
Questions []string `json:"questions"`
|
|||
|
|
Pillars []string `json:"pillars"`
|
|||
|
|
Exclusions []string `json:"exclusions"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type ResearchMapInput struct {
|
|||
|
|
Label string
|
|||
|
|
SeedQuery string
|
|||
|
|
Brief string
|
|||
|
|
ProductContext string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var researchMapFenceRE = regexp.MustCompile("(?s)```(?:json)?\\s*([\\s\\S]*?)```")
|
|||
|
|
|
|||
|
|
func BuildResearchMapSystemPrompt() string {
|
|||
|
|
return strings.TrimSpace(`你是 Threads（脆）產品置入研究顧問。目標是幫品牌找到「近期發文、作者有需求、現在留言還來得及自然推薦產品」的貼文。
|
|||
|
|
|
|||
|
|
規則：
|
|||
|
|
1. questions 與 pillars 會直接拿去 Threads 搜尋——每句 5～20 字，像真人求助
|
|||
|
|
2. questions 至少 5 個；pillars 至少 4 個；exclusions 至少 4 個
|
|||
|
|
3. contentGoal 要寫：找到近期發文且可自然留言置入的貼文
|
|||
|
|
4. 全部繁體中文，貼近台灣 Threads
|
|||
|
|
5. 只回傳一個 JSON：audienceSummary, contentGoal, questions, pillars, exclusions`)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func BuildResearchMapUserPrompt(in ResearchMapInput) string {
|
|||
|
|
var b strings.Builder
|
|||
|
|
b.WriteString("【主題名稱】")
|
|||
|
|
b.WriteString(strings.TrimSpace(in.Label))
|
|||
|
|
b.WriteString("\n【種子關鍵字】")
|
|||
|
|
b.WriteString(strings.TrimSpace(in.SeedQuery))
|
|||
|
|
b.WriteString("\n【這個主題想做什么】\n")
|
|||
|
|
b.WriteString(strings.TrimSpace(in.Brief))
|
|||
|
|
b.WriteString("\n【產品置入】\n")
|
|||
|
|
product := FormatProductContextForPrompt(in.ProductContext)
|
|||
|
|
if product == "" {
|
|||
|
|
product = "（尚未填寫）"
|
|||
|
|
}
|
|||
|
|
b.WriteString(product)
|
|||
|
|
b.WriteString("\n\n請產出研究地圖 JSON。")
|
|||
|
|
return b.String()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func ParseResearchMapOutput(raw string) (ResearchMap, error) {
|
|||
|
|
payload, err := extractResearchJSONObject(raw)
|
|||
|
|
if err != nil {
|
|||
|
|
return ResearchMap{}, err
|
|||
|
|
}
|
|||
|
|
var out ResearchMap
|
|||
|
|
if err := json.Unmarshal(payload, &out); err != nil {
|
|||
|
|
return ResearchMap{}, fmt.Errorf("parse research map json: %w", err)
|
|||
|
|
}
|
|||
|
|
out.AudienceSummary = strings.TrimSpace(out.AudienceSummary)
|
|||
|
|
out.ContentGoal = strings.TrimSpace(out.ContentGoal)
|
|||
|
|
out.Questions = cleanStringList(out.Questions)
|
|||
|
|
out.Pillars = cleanStringList(out.Pillars)
|
|||
|
|
out.Exclusions = cleanStringList(out.Exclusions)
|
|||
|
|
if out.AudienceSummary == "" && len(out.Questions) == 0 {
|
|||
|
|
return ResearchMap{}, fmt.Errorf("research map missing audience or questions")
|
|||
|
|
}
|
|||
|
|
return out, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func cleanStringList(items []string) []string {
|
|||
|
|
out := make([]string, 0, len(items))
|
|||
|
|
seen := map[string]struct{}{}
|
|||
|
|
for _, item := range items {
|
|||
|
|
item = strings.TrimSpace(item)
|
|||
|
|
if item == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if _, ok := seen[item]; ok {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
seen[item] = struct{}{}
|
|||
|
|
out = append(out, item)
|
|||
|
|
}
|
|||
|
|
return out
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func extractResearchJSONObject(raw string) ([]byte, error) {
|
|||
|
|
raw = strings.TrimSpace(raw)
|
|||
|
|
if m := researchMapFenceRE.FindStringSubmatch(raw); len(m) == 2 {
|
|||
|
|
raw = strings.TrimSpace(m[1])
|
|||
|
|
}
|
|||
|
|
start := strings.Index(raw, "{")
|
|||
|
|
end := strings.LastIndex(raw, "}")
|
|||
|
|
if start < 0 || end <= start {
|
|||
|
|
return nil, fmt.Errorf("research map output missing json object")
|
|||
|
|
}
|
|||
|
|
return []byte(raw[start : end+1]), nil
|
|||
|
|
}
|