110 lines
3.5 KiB
Go
110 lines
3.5 KiB
Go
package formula
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"strings"
|
||
|
||
domai "haixun-backend/internal/model/ai/domain/usecase"
|
||
cfdom "haixun-backend/internal/model/content_formula/domain/usecase"
|
||
)
|
||
|
||
type GenerateInput struct {
|
||
Topic string
|
||
Brief string
|
||
PersonaBlock string
|
||
ResearchNotes string
|
||
Formula cfdom.FormulaSummary
|
||
}
|
||
|
||
type GeneratedDraft struct {
|
||
Text string `json:"text"`
|
||
Hook string `json:"hook"`
|
||
Angle string `json:"angle"`
|
||
Rationale string `json:"rationale"`
|
||
StructureNotes string `json:"structure_notes"`
|
||
}
|
||
|
||
func BuildGenerateSystemPrompt() string {
|
||
return strings.TrimSpace(`你是 Threads 文案寫手。依「爆款公式」與「人設」為指定主題寫一篇新貼文。
|
||
|
||
規則：
|
||
- 用台灣繁體中文，符合人設語氣。
|
||
- 套用公式結構，但內容必須圍繞使用者主題，不可抄襲範例貼文。
|
||
- 只回傳 JSON：text, hook, angle, rationale, structure_notes（繁體中文）。`)
|
||
}
|
||
|
||
func BuildGenerateUserPrompt(in GenerateInput) string {
|
||
var b strings.Builder
|
||
b.WriteString("【主題】\n")
|
||
b.WriteString(strings.TrimSpace(in.Topic))
|
||
if brief := strings.TrimSpace(in.Brief); brief != "" {
|
||
b.WriteString("\n補充：")
|
||
b.WriteString(brief)
|
||
}
|
||
b.WriteString("\n\n【人設】\n")
|
||
b.WriteString(strings.TrimSpace(in.PersonaBlock))
|
||
b.WriteString("\n\n【公式】\n")
|
||
b.WriteString(strings.TrimSpace(in.Formula.Formula))
|
||
if tpl := strings.TrimSpace(in.Formula.PostTemplate); tpl != "" {
|
||
b.WriteString("\n\n【發文模板】\n")
|
||
b.WriteString(tpl)
|
||
}
|
||
if hook := strings.TrimSpace(in.Formula.HookPattern); hook != "" {
|
||
b.WriteString("\n\n【Hook 模式】\n")
|
||
b.WriteString(hook)
|
||
}
|
||
if structure := strings.TrimSpace(in.Formula.Structure); structure != "" {
|
||
b.WriteString("\n\n【結構節奏】\n")
|
||
b.WriteString(structure)
|
||
}
|
||
if len(in.Formula.ReplicationTips) > 0 {
|
||
b.WriteString("\n\n【複製技巧】\n")
|
||
b.WriteString(strings.Join(in.Formula.ReplicationTips, "\n"))
|
||
}
|
||
if len(in.Formula.Avoid) > 0 {
|
||
b.WriteString("\n\n【避免】\n")
|
||
b.WriteString(strings.Join(in.Formula.Avoid, "\n"))
|
||
}
|
||
if notes := strings.TrimSpace(in.ResearchNotes); notes != "" {
|
||
b.WriteString("\n\n【周邊知識參考】\n")
|
||
b.WriteString(notes)
|
||
}
|
||
if src := strings.TrimSpace(in.Formula.SourcePostText); src != "" {
|
||
b.WriteString("\n\n【參考原文（僅學結構，勿抄內容）】\n")
|
||
b.WriteString(src)
|
||
}
|
||
b.WriteString("\n\n請產出一篇可直接發布的 Threads 貼文 JSON。")
|
||
return b.String()
|
||
}
|
||
|
||
func GenerateDraft(ctx context.Context, ai domai.UseCase, req domai.GenerateRequest, in GenerateInput) (*GeneratedDraft, error) {
|
||
temp := 0.75
|
||
req.System = BuildGenerateSystemPrompt()
|
||
req.Messages = []domai.Message{{Role: "user", Content: BuildGenerateUserPrompt(in)}}
|
||
req.Temperature = &temp
|
||
out, err := ai.GenerateText(ctx, req)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return ParseGeneratedDraft(out.Text)
|
||
}
|
||
|
||
func ParseGeneratedDraft(raw string) (*GeneratedDraft, error) {
|
||
raw = strings.TrimSpace(raw)
|
||
start := strings.Index(raw, "{")
|
||
end := strings.LastIndex(raw, "}")
|
||
if start < 0 || end <= start {
|
||
return nil, fmt.Errorf("generate output missing json object")
|
||
}
|
||
var draft GeneratedDraft
|
||
if err := json.Unmarshal([]byte(raw[start:end+1]), &draft); err != nil {
|
||
return nil, fmt.Errorf("parse generated draft: %w", err)
|
||
}
|
||
draft.Text = strings.TrimSpace(draft.Text)
|
||
if draft.Text == "" {
|
||
return nil, fmt.Errorf("generated draft missing text")
|
||
}
|
||
return &draft, nil
|
||
} |