75 lines
2.5 KiB
TypeScript
75 lines
2.5 KiB
TypeScript
import { z } from "zod";
|
||
import type { ProviderApiKeys } from "./keys";
|
||
import { generateStructuredObject } from "./generate-structured";
|
||
import { getModel } from "./provider";
|
||
import { HASHTAG_WRITING_RULES } from "./hashtag-rules";
|
||
import { buildSystemPrompt } from "./prompts";
|
||
import { THREADS_MAX_CHARS } from "@/lib/utils";
|
||
|
||
export type OptimizeMode = "polish" | "hook" | "shorter" | "engaging" | "custom";
|
||
|
||
const MODE_INSTRUCTIONS: Record<Exclude<OptimizeMode, "custom">, string> = {
|
||
polish: "整體潤飾：改善語氣流暢度、可讀性與說服力，保留核心觀點。",
|
||
hook: "強化開頭 hook：讓前 1–2 句更有吸引力，讓人想停下來看。",
|
||
shorter: "精簡篇幅：刪除贅字，保留重點，盡量壓到更短但不失完整。",
|
||
engaging: "提升互動感：加入引發留言的問句、觀點或行動呼籲，但不過度推銷。",
|
||
};
|
||
|
||
const optimizeSchema = z.object({
|
||
text: z.string().max(THREADS_MAX_CHARS),
|
||
summary: z.string(),
|
||
});
|
||
|
||
export interface OptimizeInput {
|
||
text: string;
|
||
mode?: OptimizeMode;
|
||
instruction?: string | null;
|
||
persona?: string | null;
|
||
angle?: string | null;
|
||
aiProvider: string;
|
||
aiModel: string;
|
||
apiKeys?: ProviderApiKeys;
|
||
}
|
||
|
||
export async function optimizePost(input: OptimizeInput) {
|
||
const mode = input.mode ?? "polish";
|
||
const modeInstruction =
|
||
mode === "custom"
|
||
? input.instruction?.trim() || "依照使用者指示優化貼文。"
|
||
: MODE_INSTRUCTIONS[mode];
|
||
|
||
const model = getModel(input.aiProvider, input.aiModel, input.apiKeys ?? {});
|
||
|
||
const object = await generateStructuredObject({
|
||
model,
|
||
provider: input.aiProvider,
|
||
modelId: input.aiModel,
|
||
schema: optimizeSchema,
|
||
system: `${buildSystemPrompt(input.persona)}
|
||
|
||
你現在的任務是「優化既有 Threads 貼文草稿」，不是從零重寫。
|
||
- 保留原文核心觀點與事實，不要捏造
|
||
- 優化後必須更符合上方人設，尤其語氣與代表句範例的語感
|
||
- 保留或補上結尾的 1～3 個 #話題標籤，不要優化掉
|
||
${HASHTAG_WRITING_RULES}
|
||
- 輸出繁體中文台灣用語
|
||
- 字數必須 ≤ ${THREADS_MAX_CHARS}
|
||
- summary 用 1–2 句話說明你做了什麼調整`,
|
||
prompt: `請優化以下 Threads 貼文草稿。
|
||
|
||
優化方向：${modeInstruction}
|
||
${input.angle ? `切入角度：${input.angle}` : ""}
|
||
|
||
原文：
|
||
"""
|
||
${input.text}
|
||
"""
|
||
|
||
請輸出優化後的貼文（text）與調整說明（summary）。`,
|
||
});
|
||
|
||
return {
|
||
text: object.text.slice(0, THREADS_MAX_CHARS),
|
||
summary: object.summary,
|
||
};
|
||
} |