93 lines
2.7 KiB
TypeScript
93 lines
2.7 KiB
TypeScript
import { NextRequest, NextResponse } from 'next/server';
|
|
import { db } from '@/lib/db';
|
|
import { spanAnnotations } from '@/lib/db/schema';
|
|
import { eq } from 'drizzle-orm';
|
|
|
|
// PATCH - Update span annotation
|
|
export async function PATCH(
|
|
request: NextRequest,
|
|
{ params }: { params: Promise<{ id: string }> }
|
|
) {
|
|
try {
|
|
const { id } = await params;
|
|
const body = await request.json();
|
|
|
|
const { label, confidence, outcome, notes, sub_spans } = body;
|
|
|
|
// Check if the span exists
|
|
const existing = await db
|
|
.select()
|
|
.from(spanAnnotations)
|
|
.where(eq(spanAnnotations.id, parseInt(id)))
|
|
.limit(1);
|
|
|
|
if (existing.length === 0) {
|
|
return NextResponse.json(
|
|
{ error: 'Span annotation not found' },
|
|
{ status: 404 }
|
|
);
|
|
}
|
|
|
|
// Build update object with only provided fields
|
|
const updates: any = {};
|
|
if (label !== undefined) updates.label = label;
|
|
if (confidence !== undefined) updates.confidence = confidence;
|
|
if (outcome !== undefined) updates.outcome = outcome;
|
|
if (notes !== undefined) updates.notes = notes;
|
|
if (sub_spans !== undefined) updates.sub_spans = sub_spans || null;
|
|
|
|
const result = await db
|
|
.update(spanAnnotations)
|
|
.set(updates)
|
|
.where(eq(spanAnnotations.id, parseInt(id)))
|
|
.returning();
|
|
|
|
const s = result[0];
|
|
return NextResponse.json({
|
|
...s,
|
|
start_time: s.start_time instanceof Date ? Math.floor(s.start_time.getTime() / 1000) : s.start_time,
|
|
end_time: s.end_time instanceof Date ? Math.floor(s.end_time.getTime() / 1000) : s.end_time,
|
|
created_at: s.created_at instanceof Date ? Math.floor(s.created_at.getTime() / 1000) : s.created_at,
|
|
});
|
|
} catch (error: any) {
|
|
console.error('Error updating span annotation:', error);
|
|
return NextResponse.json(
|
|
{ error: 'Failed to update span annotation' },
|
|
{ status: 500 }
|
|
);
|
|
}
|
|
}
|
|
|
|
// DELETE - Delete span annotation
|
|
export async function DELETE(
|
|
request: NextRequest,
|
|
{ params }: { params: Promise<{ id: string }> }
|
|
) {
|
|
try {
|
|
const { id } = await params;
|
|
|
|
// Check if the span exists
|
|
const existing = await db
|
|
.select()
|
|
.from(spanAnnotations)
|
|
.where(eq(spanAnnotations.id, parseInt(id)))
|
|
.limit(1);
|
|
|
|
if (existing.length === 0) {
|
|
return NextResponse.json(
|
|
{ error: 'Span annotation not found' },
|
|
{ status: 404 }
|
|
);
|
|
}
|
|
|
|
await db.delete(spanAnnotations).where(eq(spanAnnotations.id, parseInt(id)));
|
|
|
|
return NextResponse.json({ message: 'Span annotation deleted successfully' });
|
|
} catch (error) {
|
|
console.error('Error deleting span annotation:', error);
|
|
return NextResponse.json(
|
|
{ error: 'Failed to delete span annotation' },
|
|
{ status: 500 }
|
|
);
|
|
}
|
|
}
|