feat(M-AI-07-05): quiz generation execution router + controller wiring
Some checks failed
Deploy API Server / build-and-unit (push) Successful in 36s
Deploy API Server / current-integration (push) Successful in 31s
Deploy API Server / backward-compat (push) Successful in 0s
Deploy API Server / deploy (push) Failing after 50s

- QuizExecutionRouter: QUIZ_GENERATION_ENGINE_MODE flag routing
- legacy: UserAiService.createAnalysisJob()
- unified: SnapshotBuilder + AiJobCreationService.createJob()
- Controller: quiz_generation routes through Router
- Response compat: legacy fields + optional engineMode/lifecycleStatus

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
wangdl 2026-06-22 20:46:54 +08:00
parent a59446266a
commit 40e772cc3a
4 changed files with 131 additions and 7 deletions

View File

@ -1,6 +1,7 @@
import { Module } from '@nestjs/common';
import { ConfigModule } from '@nestjs/config';
import { PrismaModule } from '../../infrastructure/database/prisma.module';
import { AiJobModule } from '../ai-job/ai-job.module';
import { UserAiController } from './user-ai.controller';
import { UserAiService } from './user-ai.service';
import { CredentialEncryptionService } from './credential-encryption.service';
@ -12,11 +13,12 @@ import { SnapshotBuilderService } from './snapshot-builder.service';
import { PriorityRulesService } from './priority-rules.service';
import { SnapshotCleanupService } from './snapshot-cleanup.service';
import { JobReaperService } from './job-reaper.service';
import { QuizExecutionRouter } from './quiz-execution-router';
@Module({
imports: [ConfigModule, PrismaModule],
imports: [ConfigModule, PrismaModule, AiJobModule],
controllers: [UserAiController, RuntimeInternalController],
providers: [UserAiService, CredentialEncryptionService, RuntimeInternalService, UserAiQuotaService, PlatformBudgetService, SnapshotBuilderService, PriorityRulesService, SnapshotCleanupService, JobReaperService],
providers: [UserAiService, CredentialEncryptionService, RuntimeInternalService, UserAiQuotaService, PlatformBudgetService, SnapshotBuilderService, PriorityRulesService, SnapshotCleanupService, JobReaperService, QuizExecutionRouter],
exports: [UserAiService, CredentialEncryptionService, RuntimeInternalService, UserAiQuotaService, PlatformBudgetService, SnapshotBuilderService, PriorityRulesService, SnapshotCleanupService, JobReaperService],
})
export class AiRuntimeModule {}

View File

@ -0,0 +1,101 @@
import { Injectable, Logger } from '@nestjs/common';
import { FeatureFlagService } from '../config/feature-flag.service';
import { QuizGenerationSnapshotBuilder } from '../ai-job/quiz-generation-snapshot-builder';
import type { QuizGenerationSnapshotInput } from '../ai-job/quiz-generation-snapshot-builder';
import { AiJobCreationService } from '../ai-job/ai-job-creation.service';
import { UserAiService } from './user-ai.service';
import type { CreateAnalysisJobDto } from './user-ai.dto';
/**
* M-AI-07-05: Quiz Execution Router
*
* QUIZ_GENERATION_ENGINE_MODE Feature Flag Quiz
* - 'legacy' UserAiService.createAnalysisJob()
* - 'unified' SnapshotBuilder AiJobCreationService
*/
const FLAG_NAME = 'QUIZ_GENERATION_ENGINE_MODE';
@Injectable()
export class QuizExecutionRouter {
private readonly logger = new Logger(QuizExecutionRouter.name);
constructor(
private readonly featureFlag: FeatureFlagService,
private readonly snapshotBuilder: QuizGenerationSnapshotBuilder,
private readonly creationService: AiJobCreationService,
private readonly legacyService: UserAiService,
) {}
async generateQuiz(userId: string, dto: CreateAnalysisJobDto) {
// 1. 判断模式
const useUnified = await this.shouldUseUnified(userId);
if (!useUnified) {
this.logger.log(`QUIZ_GENERATION_ENGINE_MODE=legacy for userId=${userId}`);
return this.legacyService.createAnalysisJob(userId, dto);
}
// ═════════════════════════════════════════════════════════
// Unified 路径
// ═════════════════════════════════════════════════════════
this.logger.log(`QUIZ_GENERATION_ENGINE_MODE=unified for userId=${userId}`);
// 2. 确定 submissionId幂等键来源
const submissionId = dto.idempotencyKey ??
`${dto.targetType}:${dto.targetId}:${Date.now().toString(36)}`;
// 3. 构建 Snapshot
const snapshotInput: QuizGenerationSnapshotInput = {
userId,
targetType: dto.targetType,
targetId: dto.targetId,
submissionId,
questionCount: dto.questionCount,
questionTypes: dto.questionTypes,
difficultyLevel: dto.difficultyLevel,
knowledgePointIds: dto.knowledgePointIds,
};
const snapshot = await this.snapshotBuilder.build(snapshotInput);
// 4. 幂等键
const idempotencyKey = dto.idempotencyKey ?? `quiz-generation:${submissionId}`;
// 5. 通过 AiJobCreationService 创建 Job原子Job + Snapshot + Outbox
const job = await this.creationService.createJob({
userId,
jobType: 'quiz_generation',
triggerType: 'user_api',
targetType: dto.targetType,
targetId: dto.targetId,
idempotencyKey,
retrySnapshotContent: snapshot as unknown as Record<string, unknown>,
});
this.logger.log(
`Quiz Unified: jobId=${job.id} userId=${userId} ` +
`targetType=${dto.targetType} targetId=${dto.targetId} ` +
`questionCount=${snapshot.snapshot.questionCount}`,
);
// 6. 返回兼容响应
return {
jobId: job.id,
status: 'queued',
engineMode: 'unified' as const,
lifecycleStatus: 'queued',
planId: undefined, // Unified 不创建 QuestionGenerationPlan
};
}
private async shouldUseUnified(userId: string): Promise<boolean> {
try {
return await this.featureFlag.isEnabled(FLAG_NAME, userId);
} catch (err: any) {
this.logger.warn(`FeatureFlag query failed, falling back to legacy: ${err.message}`);
return false;
}
}
}

View File

@ -3,6 +3,7 @@ import { UserAiController } from './user-ai.controller';
describe('UserAiController', () => {
let controller: UserAiController;
let mockService: any;
let mockQuizRouter: any;
beforeEach(() => {
mockService = {
@ -35,7 +36,10 @@ describe('UserAiController', () => {
getFlashcard: jest.fn(),
listFlashcards: jest.fn(),
};
controller = new UserAiController(mockService);
mockQuizRouter = {
generateQuiz: jest.fn(),
};
controller = new UserAiController(mockService, mockQuizRouter);
});
const req = (id = 'u1') => ({ user: { id } }) as any;
@ -136,12 +140,22 @@ describe('UserAiController', () => {
// ═══════════════════════════════════════════════════════════════════
describe('Analysis Jobs', () => {
it('POST jobs creates analysis job', async () => {
it('quiz_generation routes through QuizExecutionRouter', async () => {
const dto = { jobType: 'quiz_generation', targetType: 'knowledge_base', targetId: 'kb1' };
mockService.createAnalysisJob.mockResolvedValue({ jobId: 'j1' });
mockQuizRouter.generateQuiz.mockResolvedValue({ jobId: 'j1', status: 'queued' });
const result = await controller.createAnalysisJob(req(), dto as any);
expect(mockQuizRouter.generateQuiz).toHaveBeenCalledWith('u1', dto);
expect(mockService.createAnalysisJob).not.toHaveBeenCalled();
expect(result).toEqual({ jobId: 'j1', status: 'queued' });
});
it('non-quiz jobType calls legacy service', async () => {
const dto = { jobType: 'learning_state_analysis', targetType: 'knowledge_base', targetId: 'kb1' };
mockService.createAnalysisJob.mockResolvedValue({ jobId: 'j2' });
const result = await controller.createAnalysisJob(req(), dto as any);
expect(mockService.createAnalysisJob).toHaveBeenCalledWith('u1', dto);
expect(result).toEqual({ jobId: 'j1' });
expect(mockQuizRouter.generateQuiz).not.toHaveBeenCalled();
expect(result).toEqual({ jobId: 'j2' });
});
it('POST jobs/:jobId/cancel cancels job', async () => {

View File

@ -1,10 +1,14 @@
import { Controller, Get, Put, Post, Delete, Param, Body, Req, Query } from '@nestjs/common';
import { UserAiService } from './user-ai.service';
import { QuizExecutionRouter } from './quiz-execution-router';
import { SaveLearningProfileDto, UpdateAiSettingsDto, CreateCredentialDto, UpdateCredentialDto, CreateAnalysisJobDto } from './user-ai.dto';
@Controller('ai')
export class UserAiController {
constructor(private readonly service: UserAiService) {}
constructor(
private readonly service: UserAiService,
private readonly quizRouter: QuizExecutionRouter,
) {}
// ── Profile ──
@ -62,6 +66,9 @@ export class UserAiController {
@Post('jobs')
async createAnalysisJob(@Req() req: any, @Body() dto: CreateAnalysisJobDto) {
if (dto.jobType === 'quiz_generation') {
return this.quizRouter.generateQuiz(req.user.id, dto);
}
return this.service.createAnalysisJob(req.user.id, dto);
}