58 lines
1.9 KiB
TypeScript
58 lines
1.9 KiB
TypeScript
import { NextResponse } from "next/server";
|
|
import { prisma, resolvePersona } from "@/lib/db";
|
|
import { getActiveAccountProfile } from "@/lib/account-context";
|
|
import { getOrCreateSettings } from "@/lib/user-settings";
|
|
import { refineResearchMap, type RefineChatMessage } from "@/lib/ai/refine-research-map";
|
|
import { parseProviderApiKeys } from "@/lib/ai/keys";
|
|
import type { ResearchMap } from "@/lib/types/research";
|
|
import { trackAiTask } from "@/lib/jobs/track";
|
|
|
|
export const maxDuration = 120;
|
|
|
|
export async function POST(request: Request) {
|
|
try {
|
|
const body = (await request.json()) as {
|
|
topicId?: string;
|
|
researchMap?: ResearchMap;
|
|
message?: string;
|
|
history?: RefineChatMessage[];
|
|
};
|
|
|
|
const { topicId, researchMap, message, history } = body;
|
|
|
|
if (!topicId || !researchMap || !message?.trim()) {
|
|
return NextResponse.json(
|
|
{ error: "缺少 topicId、researchMap 或 message" },
|
|
{ status: 400 }
|
|
);
|
|
}
|
|
|
|
const topic = await prisma.topic.findUnique({ where: { id: topicId } });
|
|
if (!topic) {
|
|
return NextResponse.json({ error: "找不到主題" }, { status: 404 });
|
|
}
|
|
|
|
const settings = await getOrCreateSettings();
|
|
const account = await getActiveAccountProfile();
|
|
const apiKeys = parseProviderApiKeys(settings.providerApiKeys);
|
|
|
|
const result = await trackAiTask("微調研究地圖", () => refineResearchMap({
|
|
label: topic.label,
|
|
query: topic.query,
|
|
brief: topic.brief,
|
|
persona: resolvePersona(settings, account),
|
|
currentMap: researchMap,
|
|
message: message.trim(),
|
|
history: history ?? [],
|
|
aiProvider: settings.researchAiProvider ?? settings.aiProvider,
|
|
aiModel: settings.researchAiModel ?? settings.aiModel,
|
|
apiKeys,
|
|
}));
|
|
|
|
return NextResponse.json(result);
|
|
} catch (error) {
|
|
const msg = error instanceof Error ? error.message : "微調失敗";
|
|
return NextResponse.json({ error: msg }, { status: 500 });
|
|
}
|
|
}
|