diff --git a/src/modules/ai/ai.module.ts b/src/modules/ai/ai.module.ts
index 0e07349..8de815d 100644
--- a/src/modules/ai/ai.module.ts
+++ b/src/modules/ai/ai.module.ts
@@ -10,7 +10,7 @@ import { FeynmanEvaluationWorkflow } from './workflows/feynman-evaluation.workfl
 import { KnowledgeImportWorkflow } from './workflows/knowledge-import.workflow';
 import { ReviewCardGenerationWorkflow } from './workflows/review-card-generation.workflow';
 import { LearningTrendWorkflow } from './workflows/learning-trend.workflow';
-import { AiController } from './ai.controller';
+import { AdminAiGatewayController } from './ai.controller';
 import { MockAiProvider } from './providers/mock-ai.provider';
 import { DeepSeekProvider } from './providers/deepseek.provider';
 import { MiniMaxProvider } from './providers/minimax.provider';
@@ -18,7 +18,7 @@ import type { AiProvider } from './providers/ai-provider.interface';
 
 @Module({
   imports: [ConfigModule],
-  controllers: [AiController],
+  controllers: [AdminAiGatewayController],
   providers: [
     ModelRouter,
     PromptTemplateService,