Spaces:
Sleeping
Sleeping
File size: 1,458 Bytes
a01cf9a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import { NextResponse } from 'next/server'
import { runInference } from '@/lib/huggingface'
import { createServerSupabaseClient } from '@/lib/supabase-server'
import type { InferenceRequest } from '@/types'
export async function POST(request: Request) {
try {
const body: InferenceRequest = await request.json()
const { model, inputs, parameters } = body
if (!model || !inputs) {
return NextResponse.json(
{ error: 'Model and inputs are required' },
{ status: 400 }
)
}
// Run inference
const result = await runInference(model, inputs, parameters)
// Store in Supabase
const supabase = await createServerSupabaseClient()
const { data, error } = await supabase
.from('model_interactions')
.insert({
user_input: typeof inputs === 'string' ? inputs : JSON.stringify(inputs),
model_output: JSON.stringify(result),
model_name: model,
})
.select()
.single()
if (error) {
console.error('Supabase error:', error)
// Still return result even if DB insert fails
return NextResponse.json({ result }, { status: 200 })
}
return NextResponse.json({ result, data }, { status: 200 })
} catch (error: any) {
console.error('API Error:', error)
return NextResponse.json(
{ error: error.message || 'Internal server error' },
{ status: 500 }
)
}
}
|