51 lines
1.7 KiB
TypeScript
51 lines
1.7 KiB
TypeScript
import { NextRequest, NextResponse } from 'next/server';
|
|
import { query } from '@/lib/db';
|
|
import { generate } from '@/lib/ai/client';
|
|
import { prompts } from '@/lib/ai/prompts';
|
|
import { ToolName, Session } from '@/lib/types';
|
|
|
|
export async function POST(
|
|
request: NextRequest,
|
|
{ params }: { params: { tool: string } }
|
|
) {
|
|
try {
|
|
const tool = params.tool as ToolName;
|
|
const { session_id, improvements } = await request.json();
|
|
|
|
const [session] = await query<Session>(
|
|
'SELECT * FROM sessions WHERE id = $1',
|
|
[session_id]
|
|
);
|
|
if (!session) {
|
|
return NextResponse.json({ error: 'Session not found' }, { status: 404 });
|
|
}
|
|
if (!session.is_paid) {
|
|
return NextResponse.json({ error: 'Payment required' }, { status: 402 });
|
|
}
|
|
if (!session.output) {
|
|
return NextResponse.json({ error: 'No output to finalize' }, { status: 400 });
|
|
}
|
|
|
|
const toolPrompts = prompts[tool];
|
|
const systemPrompt = 'You are a professional NGO grant writing assistant producing final documents.';
|
|
const userPrompt = toolPrompts.finalize(session.output, improvements || 'No additional improvements.');
|
|
|
|
const result = await generate(systemPrompt, userPrompt);
|
|
|
|
await query(
|
|
`UPDATE sessions SET output = $1, updated_at = NOW() WHERE id = $2`,
|
|
[result.text, session_id]
|
|
);
|
|
|
|
await query(
|
|
`INSERT INTO usage (session_id, tool, action, tokens_used) VALUES ($1, $2, $3, $4)`,
|
|
[session_id, tool, 'finalize', result.tokens]
|
|
);
|
|
|
|
return NextResponse.json({ finalized: result.text });
|
|
} catch (error) {
|
|
console.error('Finalize error:', error);
|
|
return NextResponse.json({ error: 'Failed to finalize' }, { status: 500 });
|
|
}
|
|
}
|