244 lines
8.0 KiB
TypeScript
244 lines
8.0 KiB
TypeScript
|
import { generateObject, generateText } from "ai";
|
|||
|
|
import { z } from "zod";
|
|||
|
|
import type { ProviderApiKeys } from "./keys";
|
|||
|
|
import { withAgentSystem } from "./agent";
|
|||
|
|
import { buildPersonaPromptBlock } from "./persona";
|
|||
|
|
import { getModel } from "./provider";
|
|||
|
|
import { extractJsonFromText } from "./coerce-research-map";
|
|||
|
|
import {
|
|||
|
|
getOpenCodeGenerationSettings,
|
|||
|
|
prefersOpenCodeTextFirst,
|
|||
|
|
} from "./opencode-go-settings";
|
|||
|
|
import {
|
|||
|
|
coerceMatrixRaw,
|
|||
|
|
formatMatrixError,
|
|||
|
|
matrixLooseSchema,
|
|||
|
|
} from "./coerce-matrix";
|
|||
|
|
import { HASHTAG_USER_REMINDER, HASHTAG_WRITING_RULES } from "./hashtag-rules";
|
|||
|
|
import { sanitizePromptText, THREADS_MAX_CHARS } from "@/lib/utils";
|
|||
|
|
import type { MatrixRow, ResearchMap } from "@/lib/types/research";
|
|||
|
|
|
|||
|
|
const matrixSchema = z.object({
|
|||
|
|
rows: z.array(
|
|||
|
|
z.object({
|
|||
|
|
sortOrder: z.number().int().min(1),
|
|||
|
|
searchTag: z.string(),
|
|||
|
|
angle: z.string(),
|
|||
|
|
hook: z.string(),
|
|||
|
|
text: z.string().max(THREADS_MAX_CHARS),
|
|||
|
|
referenceNotes: z.string(),
|
|||
|
|
sourcePermalinks: z.array(z.string()),
|
|||
|
|
rationale: z.string(),
|
|||
|
|
})
|
|||
|
|
),
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
/** 非 OpenCode Go 但 structured output 不穩的模型 */
|
|||
|
|
const TEXT_FIRST_MODELS = new Set(["grok-3", "grok-3-fast"]);
|
|||
|
|
|
|||
|
|
export interface GenerateMatrixInput {
|
|||
|
|
topicLabel: string;
|
|||
|
|
query: string;
|
|||
|
|
brief?: string | null;
|
|||
|
|
persona?: string | null;
|
|||
|
|
researchMap?: ResearchMap | null;
|
|||
|
|
aiProvider: string;
|
|||
|
|
aiModel: string;
|
|||
|
|
apiKeys?: ProviderApiKeys;
|
|||
|
|
count: number;
|
|||
|
|
posts: Array<{
|
|||
|
|
text: string;
|
|||
|
|
authorName?: string | null;
|
|||
|
|
permalink?: string | null;
|
|||
|
|
searchTag?: string | null;
|
|||
|
|
likeCount?: number | null;
|
|||
|
|
replyCount?: number | null;
|
|||
|
|
qualityReason?: string | null;
|
|||
|
|
replies?: Array<{ text: string; authorName?: string | null; likeCount?: number | null }>;
|
|||
|
|
}>;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function buildSystemPrompt(persona?: string | null) {
|
|||
|
|
return withAgentSystem(`你是 Threads 內容企劃師。根據篩選後的優質素材，產出「內容矩陣」——一週可發的貼文企劃表。
|
|||
|
|
|
|||
|
|
${buildPersonaPromptBlock(persona)}
|
|||
|
|
|
|||
|
|
規則：
|
|||
|
|
- 每篇必須原創，不可抄襲參考貼文
|
|||
|
|
- hook 是一句吸引人的開頭（可獨立於正文），風格符合人設開場習慣
|
|||
|
|
- text 是完整 Threads 貼文（≤500字，含 #標籤），語感要像創作者本人親筆
|
|||
|
|
${HASHTAG_WRITING_RULES}
|
|||
|
|
- referenceNotes 是從參考素材摘出的 2-3 個重點（簡短條列感，用換行分隔）
|
|||
|
|
- 每篇對應一個 searchTag（主題標籤）
|
|||
|
|
- sortOrder 從 1 開始遞增
|
|||
|
|
- 繁體中文台灣用語
|
|||
|
|
- 知識型貼文：不寫未查證的醫療／數據斷言，rationale 標註需網路查證的關鍵句`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function buildUserPrompt(input: GenerateMatrixInput, researchBlock: string, materials: string) {
|
|||
|
|
return `主題：${sanitizePromptText(input.topicLabel)}
|
|||
|
|
種子關鍵字：${sanitizePromptText(input.query)}
|
|||
|
|
${input.brief ? `Brief：${sanitizePromptText(input.brief)}` : ""}
|
|||
|
|
${researchBlock}
|
|||
|
|
|
|||
|
|
優質參考素材：
|
|||
|
|
${materials}
|
|||
|
|
|
|||
|
|
請產出 ${input.count} 篇內容矩陣。每篇涵蓋不同切角，避免重複。
|
|||
|
|
寫作提醒：矩陣裡的 text 必須像創作者親筆，有代表句範例時語感要向範例靠攏。${HASHTAG_USER_REMINDER}`;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function buildJsonPromptSuffix(count: number) {
|
|||
|
|
return `
|
|||
|
|
|
|||
|
|
請只回傳 JSON（不要 markdown），格式：
|
|||
|
|
{
|
|||
|
|
"rows": [
|
|||
|
|
{
|
|||
|
|
"sortOrder": 1,
|
|||
|
|
"searchTag": "標籤",
|
|||
|
|
"angle": "切角說明",
|
|||
|
|
"hook": "開頭一句",
|
|||
|
|
"text": "完整貼文（結尾含 1～3 個 #話題標籤）",
|
|||
|
|
"referenceNotes": "參考重點（多行以換行分隔）",
|
|||
|
|
"sourcePermalinks": ["https://..."],
|
|||
|
|
"rationale": "為什麼這篇有效"
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
請產出 ${count} 篇，sortOrder 從 1 遞增，每篇 text 不超過 ${THREADS_MAX_CHARS} 字。`;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function normalizeRows(rows: MatrixRow[]): MatrixRow[] {
|
|||
|
|
return rows
|
|||
|
|
.sort((a, b) => a.sortOrder - b.sortOrder)
|
|||
|
|
.map((row) => ({
|
|||
|
|
...row,
|
|||
|
|
text: row.text.slice(0, THREADS_MAX_CHARS),
|
|||
|
|
}));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function generateWithText(
|
|||
|
|
model: ReturnType<typeof getModel>,
|
|||
|
|
system: string,
|
|||
|
|
prompt: string,
|
|||
|
|
fallback: { query: string; count: number },
|
|||
|
|
provider: string,
|
|||
|
|
modelId: string
|
|||
|
|
) {
|
|||
|
|
const settings = getOpenCodeGenerationSettings(provider, modelId);
|
|||
|
|
const { text } = await generateText({
|
|||
|
|
model,
|
|||
|
|
system: sanitizePromptText(`${system}${buildJsonPromptSuffix(fallback.count)}`),
|
|||
|
|
prompt: sanitizePromptText(prompt),
|
|||
|
|
...settings,
|
|||
|
|
});
|
|||
|
|
const parsed = extractJsonFromText(text);
|
|||
|
|
matrixLooseSchema.parse(parsed);
|
|||
|
|
return coerceMatrixRaw(parsed, fallback);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function generateWithObject(
|
|||
|
|
model: ReturnType<typeof getModel>,
|
|||
|
|
system: string,
|
|||
|
|
prompt: string,
|
|||
|
|
fallback: { query: string; count: number },
|
|||
|
|
provider: string,
|
|||
|
|
modelId: string
|
|||
|
|
) {
|
|||
|
|
const settings = getOpenCodeGenerationSettings(provider, modelId);
|
|||
|
|
const { object } = await generateObject({
|
|||
|
|
model,
|
|||
|
|
schema: matrixSchema,
|
|||
|
|
system: sanitizePromptText(system),
|
|||
|
|
prompt: sanitizePromptText(`${prompt}${buildJsonPromptSuffix(fallback.count)}`),
|
|||
|
|
...settings,
|
|||
|
|
});
|
|||
|
|
return normalizeRows(object.rows);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export async function generateContentMatrix(input: GenerateMatrixInput): Promise<MatrixRow[]> {
|
|||
|
|
const model = getModel(input.aiProvider, input.aiModel, input.apiKeys ?? {});
|
|||
|
|
|
|||
|
|
const researchBlock = input.researchMap
|
|||
|
|
? `
|
|||
|
|
受眾：${sanitizePromptText(input.researchMap.audienceSummary)}
|
|||
|
|
內容目標：${sanitizePromptText(input.researchMap.contentGoal)}
|
|||
|
|
想回答的問題：${input.researchMap.questions.map(sanitizePromptText).join("、")}
|
|||
|
|
內容支柱：${input.researchMap.pillars.map(sanitizePromptText).join("、")}
|
|||
|
|
`
|
|||
|
|
: "";
|
|||
|
|
|
|||
|
|
const materials = input.posts
|
|||
|
|
.map((post, i) => {
|
|||
|
|
const repliesBlock =
|
|||
|
|
post.replies && post.replies.length > 0
|
|||
|
|
? `\n 留言：${post.replies
|
|||
|
|
.slice(0, 3)
|
|||
|
|
.map(
|
|||
|
|
(r) =>
|
|||
|
|
`@${sanitizePromptText(r.authorName) || "匿名"}：${sanitizePromptText(r.text).slice(0, 80)}`
|
|||
|
|
)
|
|||
|
|
.join(" | ")}`
|
|||
|
|
: "";
|
|||
|
|
|
|||
|
|
return `${i + 1}. [${sanitizePromptText(post.searchTag) || sanitizePromptText(input.query)}] @${sanitizePromptText(post.authorName) || "匿名"}（${post.likeCount ?? 0}讚）
|
|||
|
|
${sanitizePromptText(post.text).slice(0, 300)}
|
|||
|
|
連結：${sanitizePromptText(post.permalink) || "無"}${repliesBlock}${post.qualityReason ? `\n 品質說明：${sanitizePromptText(post.qualityReason)}` : ""}`;
|
|||
|
|
})
|
|||
|
|
.join("\n\n");
|
|||
|
|
|
|||
|
|
const system = buildSystemPrompt(input.persona);
|
|||
|
|
const prompt = buildUserPrompt(input, researchBlock, materials);
|
|||
|
|
const fallback = { query: input.query, count: input.count };
|
|||
|
|
|
|||
|
|
const preferText =
|
|||
|
|
prefersOpenCodeTextFirst(input.aiProvider, input.aiModel) ||
|
|||
|
|
TEXT_FIRST_MODELS.has(input.aiModel);
|
|||
|
|
|
|||
|
|
let rows: MatrixRow[] | null = null;
|
|||
|
|
let lastError: unknown;
|
|||
|
|
|
|||
|
|
const attempts: Array<() => Promise<MatrixRow[]>> = preferText
|
|||
|
|
? [
|
|||
|
|
() => generateWithText(model, system, prompt, fallback, input.aiProvider, input.aiModel),
|
|||
|
|
() => generateWithObject(model, system, prompt, fallback, input.aiProvider, input.aiModel),
|
|||
|
|
() =>
|
|||
|
|
generateWithText(
|
|||
|
|
model,
|
|||
|
|
system,
|
|||
|
|
`${prompt}\n\n上次格式不完整，請務必補齊 ${input.count} 篇 rows。`,
|
|||
|
|
fallback,
|
|||
|
|
input.aiProvider,
|
|||
|
|
input.aiModel
|
|||
|
|
),
|
|||
|
|
]
|
|||
|
|
: [
|
|||
|
|
() => generateWithObject(model, system, prompt, fallback, input.aiProvider, input.aiModel),
|
|||
|
|
() => generateWithText(model, system, prompt, fallback, input.aiProvider, input.aiModel),
|
|||
|
|
() =>
|
|||
|
|
generateWithText(
|
|||
|
|
model,
|
|||
|
|
system,
|
|||
|
|
`${prompt}\n\n上次格式不完整，請務必補齊 ${input.count} 篇 rows。`,
|
|||
|
|
fallback,
|
|||
|
|
input.aiProvider,
|
|||
|
|
input.aiModel
|
|||
|
|
),
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
for (const attempt of attempts) {
|
|||
|
|
try {
|
|||
|
|
rows = await attempt();
|
|||
|
|
break;
|
|||
|
|
} catch (error) {
|
|||
|
|
lastError = error;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!rows) {
|
|||
|
|
throw new Error(formatMatrixError(lastError));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return normalizeRows(rows);
|
|||
|
|
}
|