74 lines
2.4 KiB
JavaScript
74 lines
2.4 KiB
JavaScript
|
import { NextResponse } from 'next/server';
|
|||
|
|
import { getQuestionsForChunk } from '@/lib/db/questions';
|
|||
|
|
import logger from '@/lib/util/logger';
|
|||
|
|
import questionService from '@/lib/services/questions';
|
|||
|
|
|
|||
|
|
// 为指定文本块生成问题
|
|||
|
|
export async function POST(request, { params }) {
|
|||
|
|
try {
|
|||
|
|
const { projectId, chunkId } = params;
|
|||
|
|
|
|||
|
|
// 验证项目ID和文本块ID
|
|||
|
|
if (!projectId || !chunkId) {
|
|||
|
|
return NextResponse.json({ error: 'Project ID or text block ID cannot be empty' }, { status: 400 });
|
|||
|
|
} // 获取请求体
|
|||
|
|
const { model, language = '中文', number, enableGaExpansion = false } = await request.json();
|
|||
|
|
|
|||
|
|
if (!model) {
|
|||
|
|
return NextResponse.json({ error: 'Model cannot be empty' }, { status: 400 });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 后续会根据是否有GA对来选择是否启用GA扩展选择服务函数
|
|||
|
|
const serviceFunc = questionService.generateQuestionsForChunkWithGA;
|
|||
|
|
|
|||
|
|
// 使用问题生成服务
|
|||
|
|
const result = await serviceFunc(projectId, chunkId, {
|
|||
|
|
model,
|
|||
|
|
language,
|
|||
|
|
number,
|
|||
|
|
enableGaExpansion
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 统一返回格式，确保包含GA扩展信息
|
|||
|
|
const response = {
|
|||
|
|
chunkId,
|
|||
|
|
questions: result.questions || result.labelQuestions || [],
|
|||
|
|
total: result.total || (result.questions || result.labelQuestions || []).length,
|
|||
|
|
gaExpansionUsed: result.gaExpansionUsed || false,
|
|||
|
|
gaPairsCount: result.gaPairsCount || 0,
|
|||
|
|
expectedTotal: result.expectedTotal || result.total
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 返回生成的问题
|
|||
|
|
return NextResponse.json(response);
|
|||
|
|
} catch (error) {
|
|||
|
|
logger.error('Error generating questions:', error);
|
|||
|
|
return NextResponse.json({ error: error.message || 'Error generating questions' }, { status: 500 });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取指定文本块的问题
|
|||
|
|
export async function GET(request, { params }) {
|
|||
|
|
try {
|
|||
|
|
const { projectId, chunkId } = params;
|
|||
|
|
|
|||
|
|
// 验证项目ID和文本块ID
|
|||
|
|
if (!projectId || !chunkId) {
|
|||
|
|
return NextResponse.json({ error: 'The item ID or text block ID cannot be empty' }, { status: 400 });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取文本块的问题
|
|||
|
|
const questions = await getQuestionsForChunk(projectId, chunkId);
|
|||
|
|
|
|||
|
|
// 返回问题列表
|
|||
|
|
return NextResponse.json({
|
|||
|
|
chunkId,
|
|||
|
|
questions,
|
|||
|
|
total: questions.length
|
|||
|
|
});
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('Error getting questions:', String(error));
|
|||
|
|
return NextResponse.json({ error: error.message || 'Error getting questions' }, { status: 500 });
|
|||
|
|
}
|
|||
|
|
}
|