2fc9a34626
Allows users to configure the max_tokens sent to the AI endpoint for each activity (tagging, description, extraction, translation) individually, with per-library overrides following the same pattern as model overrides. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
49 lines
2.2 KiB
TypeScript
49 lines
2.2 KiB
TypeScript
import { NextRequest, NextResponse } from 'next/server'
|
|
import { requireAdmin } from '@/lib/auth'
|
|
import { getLibraryAiOverrides, setLibraryAiOverrides } from '@/lib/app-settings'
|
|
|
|
export async function GET(
|
|
request: NextRequest,
|
|
{ params }: { params: Promise<{ id: string }> }
|
|
) {
|
|
const auth = await requireAdmin(request)
|
|
if (auth instanceof NextResponse) return auth
|
|
|
|
const { id } = await params
|
|
return NextResponse.json(getLibraryAiOverrides(id))
|
|
}
|
|
|
|
export async function PUT(
|
|
request: NextRequest,
|
|
{ params }: { params: Promise<{ id: string }> }
|
|
) {
|
|
const auth = await requireAdmin(request)
|
|
if (auth instanceof NextResponse) return auth
|
|
|
|
const { id } = await params
|
|
|
|
let body: Record<string, unknown>
|
|
try {
|
|
body = await request.json()
|
|
} catch {
|
|
return NextResponse.json({ error: 'Invalid JSON body' }, { status: 400 })
|
|
}
|
|
|
|
setLibraryAiOverrides(id, {
|
|
modelTagging: typeof body.modelTagging === 'string' ? body.modelTagging : undefined,
|
|
modelDescribe: typeof body.modelDescribe === 'string' ? body.modelDescribe : undefined,
|
|
modelExtract: typeof body.modelExtract === 'string' ? body.modelExtract : undefined,
|
|
modelTranslate: typeof body.modelTranslate === 'string' ? body.modelTranslate : undefined,
|
|
promptDescribe: typeof body.promptDescribe === 'string' ? body.promptDescribe : undefined,
|
|
promptTagger: typeof body.promptTagger === 'string' ? body.promptTagger : undefined,
|
|
promptExtract: typeof body.promptExtract === 'string' ? body.promptExtract : undefined,
|
|
promptTranslate: typeof body.promptTranslate === 'string' ? body.promptTranslate : undefined,
|
|
maxTokensTag: typeof body.maxTokensTag === 'number' ? body.maxTokensTag : (body.maxTokensTag === null ? null : undefined),
|
|
maxTokensDescribe: typeof body.maxTokensDescribe === 'number' ? body.maxTokensDescribe : (body.maxTokensDescribe === null ? null : undefined),
|
|
maxTokensExtract: typeof body.maxTokensExtract === 'number' ? body.maxTokensExtract : (body.maxTokensExtract === null ? null : undefined),
|
|
maxTokensTranslate: typeof body.maxTokensTranslate === 'number' ? body.maxTokensTranslate : (body.maxTokensTranslate === null ? null : undefined),
|
|
})
|
|
|
|
return NextResponse.json(getLibraryAiOverrides(id))
|
|
}
|