5f70f13da3
- Remove better-sqlite3, add pg driver - Convert schema to PostgreSQL types (serial, timestamp, boolean, jsonb) - Generate fresh PostgreSQL migrations - Update database connection layer with pg.Pool - Fix all API routes: remove JSON.parse/stringify, use native timestamps and booleans - Update drizzle.config.ts and .env.example for PostgreSQL
89 lines
2.4 KiB
TypeScript
89 lines
2.4 KiB
TypeScript
import { NextRequest, NextResponse } from 'next/server';
|
|
import { db } from '@/lib/db';
|
|
import { spanAnnotations } from '@/lib/db/schema';
|
|
import { eq, desc } from 'drizzle-orm';
|
|
|
|
// GET - List all span annotations for a chart
|
|
export async function GET(request: NextRequest) {
|
|
try {
|
|
const { searchParams } = request.nextUrl;
|
|
const chartId = searchParams.get('chartId');
|
|
|
|
if (!chartId) {
|
|
return NextResponse.json(
|
|
{ error: 'chartId parameter is required' },
|
|
{ status: 400 }
|
|
);
|
|
}
|
|
|
|
const spans = await db
|
|
.select()
|
|
.from(spanAnnotations)
|
|
.where(eq(spanAnnotations.chart_id, parseInt(chartId)))
|
|
.orderBy(desc(spanAnnotations.start_time));
|
|
|
|
return NextResponse.json(spans);
|
|
} catch (error) {
|
|
console.error('Error fetching span annotations:', error);
|
|
return NextResponse.json(
|
|
{ error: 'Failed to fetch span annotations' },
|
|
{ status: 500 }
|
|
);
|
|
}
|
|
}
|
|
|
|
// POST - Create new span annotation
|
|
export async function POST(request: NextRequest) {
|
|
try {
|
|
const body = await request.json();
|
|
const {
|
|
chart_id,
|
|
start_time,
|
|
end_time,
|
|
label,
|
|
confidence,
|
|
outcome,
|
|
notes,
|
|
sub_spans,
|
|
color,
|
|
source,
|
|
model_prediction,
|
|
} = body;
|
|
|
|
if (!chart_id || start_time === undefined || end_time === undefined || !label) {
|
|
return NextResponse.json(
|
|
{ error: 'chart_id, start_time, end_time, and label are required' },
|
|
{ status: 400 }
|
|
);
|
|
}
|
|
|
|
// Ensure start_time <= end_time (swap if needed)
|
|
const actualStartTime = Math.min(start_time, end_time);
|
|
const actualEndTime = Math.max(start_time, end_time);
|
|
|
|
const result = await db
|
|
.insert(spanAnnotations)
|
|
.values({
|
|
chart_id,
|
|
start_time: actualStartTime,
|
|
end_time: actualEndTime,
|
|
label,
|
|
confidence: confidence || null,
|
|
outcome: outcome || null,
|
|
notes: notes || null,
|
|
sub_spans: sub_spans || null,
|
|
color: color || '#2196F3',
|
|
source: source || 'human', // 'human', 'model', or 'human_correction'
|
|
model_prediction: model_prediction || null,
|
|
})
|
|
.returning();
|
|
|
|
return NextResponse.json(result[0], { status: 201 });
|
|
} catch (error: any) {
|
|
console.error('Error creating span annotation:', error);
|
|
return NextResponse.json(
|
|
{ error: 'Failed to create span annotation' },
|
|
{ status: 500 }
|
|
);
|
|
}
|
|
}
|