110 lines
3.7 KiB
TypeScript
110 lines
3.7 KiB
TypeScript
|
import { z } from "zod";
|
|||
|
|
import type { ProviderApiKeys } from "./keys";
|
|||
|
|
import { generateStructuredObject } from "./generate-structured";
|
|||
|
|
import { getModel } from "./provider";
|
|||
|
|
|
|||
|
|
const filterSchema = z.object({
|
|||
|
|
items: z.array(
|
|||
|
|
z.object({
|
|||
|
|
id: z.string(),
|
|||
|
|
relevant: z.boolean(),
|
|||
|
|
score: z.number().min(0).max(1),
|
|||
|
|
reason: z.string(),
|
|||
|
|
})
|
|||
|
|
),
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
export interface DiscoverFilterItem {
|
|||
|
|
id: string;
|
|||
|
|
text: string;
|
|||
|
|
username?: string;
|
|||
|
|
source: string;
|
|||
|
|
tags?: string[];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export interface DiscoverFilterResult {
|
|||
|
|
relevant: boolean;
|
|||
|
|
score: number;
|
|||
|
|
reason: string;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export async function filterDiscoverItemsWithAi(input: {
|
|||
|
|
label: string;
|
|||
|
|
query: string;
|
|||
|
|
brief?: string | null;
|
|||
|
|
exclusions?: string[];
|
|||
|
|
pillars?: string[];
|
|||
|
|
requiredConcepts?: string[];
|
|||
|
|
items: DiscoverFilterItem[];
|
|||
|
|
aiProvider: string;
|
|||
|
|
aiModel: string;
|
|||
|
|
apiKeys?: ProviderApiKeys;
|
|||
|
|
}): Promise<Map<string, DiscoverFilterResult>> {
|
|||
|
|
const fallback = new Map<string, DiscoverFilterResult>();
|
|||
|
|
for (const item of input.items) {
|
|||
|
|
fallback.set(item.id, { relevant: true, score: 0.55, reason: "規則通過（AI 未審核）" });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (input.items.length === 0) return fallback;
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
const model = getModel(input.aiProvider, input.aiModel, input.apiKeys ?? {});
|
|||
|
|
const listBlock = input.items
|
|||
|
|
.map((item) => {
|
|||
|
|
const tags = item.tags?.length ? `\n標籤：${item.tags.join("、")}` : "";
|
|||
|
|
const user = item.username ? `\n帳號：@${item.username}` : "";
|
|||
|
|
return `[${item.id}]（來源：${item.source}）${user}${tags}\n內容：${item.text.slice(0, 280)}`;
|
|||
|
|
})
|
|||
|
|
.join("\n\n");
|
|||
|
|
|
|||
|
|
const result = await generateStructuredObject({
|
|||
|
|
model,
|
|||
|
|
provider: input.aiProvider,
|
|||
|
|
modelId: input.aiModel,
|
|||
|
|
schema: filterSchema,
|
|||
|
|
system: `你是 Threads 主題研究助理。任務：判斷搜尋結果是否與「指定主題」真正相關，用於找相似創作者帳號。
|
|||
|
|
|
|||
|
|
審核要嚴格、扣緊主題：
|
|||
|
|
- 複合主題要同時符合核心概念，不能只命中其中一個字
|
|||
|
|
- 易混淆要剔除：例如「寵物洗毛精」vs「洗碗精」、「寵物」vs「人類掉髮／洗髮」
|
|||
|
|
- 太寬泛的泛用詞（只有「寵物」「洗毛精」其中一項而無法確認是目標情境）→ relevant=false
|
|||
|
|
- 明顯不同產品類別、不同受眾、不同痛點 → relevant=false
|
|||
|
|
- 只有稍微相關但無法當相似帳號參考 → score ≤ 0.4
|
|||
|
|
|
|||
|
|
每筆都要輸出 id、relevant、score（0-1）、reason（繁體中文一句話）。`,
|
|||
|
|
prompt: `【主題】${input.label}
|
|||
|
|
【核心查詢】${input.query}
|
|||
|
|
${input.brief ? `【受眾簡述】${input.brief}` : ""}
|
|||
|
|
${input.pillars?.length ? `【內容支柱】${input.pillars.join("、")}` : ""}
|
|||
|
|
${input.requiredConcepts?.length ? `【必備概念（需同時符合）】${input.requiredConcepts.join(" + ")}` : ""}
|
|||
|
|
${input.exclusions?.length ? `【明確排除】${input.exclusions.join("、")}` : ""}
|
|||
|
|
|
|||
|
|
請審核以下 ${input.items.length} 筆搜尋結果：
|
|||
|
|
|
|||
|
|
${listBlock}`,
|
|||
|
|
jsonPromptSuffix:
|
|||
|
|
'\n\n只回傳 JSON：{"items":[{"id":"...","relevant":true,"score":0.8,"reason":"..."}]}',
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
const mapped = new Map<string, DiscoverFilterResult>();
|
|||
|
|
for (const row of result.items) {
|
|||
|
|
mapped.set(row.id, {
|
|||
|
|
relevant: row.relevant,
|
|||
|
|
score: row.score,
|
|||
|
|
reason: row.reason.slice(0, 120),
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for (const item of input.items) {
|
|||
|
|
if (!mapped.has(item.id)) {
|
|||
|
|
mapped.set(item.id, { relevant: false, score: 0, reason: "AI 未回傳審核結果" });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return mapped;
|
|||
|
|
} catch (error) {
|
|||
|
|
console.warn("[filter-discover-relevance] AI filter failed, using rules only:", error);
|
|||
|
|
return fallback;
|
|||
|
|
}
|
|||
|
|
}
|