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 });
|
||
}
|
||
}
|