From a0975d3f7bfbe66b6a9dc6967079db6b6ff5a3e3 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 02 Feb 2026 14:27:29 +0800
Subject: [PATCH] feat(ai): 添加AI产品题目生成任务功能

---
 src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java |   38 ++++++++++++++++++++++++++++++++++++--
 1 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java b/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java
index ca3cb00..8fc94e9 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java
@@ -1,7 +1,8 @@
 package cc.mrbird.febs.ai.controller.productQuestion;
 
-import cc.mrbird.febs.ai.entity.AiProductPoint;
 import cc.mrbird.febs.ai.entity.AiProductQuestion;
+import cc.mrbird.febs.ai.entity.AiProductQuestionJob;
+import cc.mrbird.febs.ai.req.AiProductQuestionAiDto;
 import cc.mrbird.febs.ai.service.AiProductQuestionService;
 import cc.mrbird.febs.common.annotation.ControllerEndpoint;
 import cc.mrbird.febs.common.controller.BaseController;
@@ -30,8 +31,19 @@
 
     @GetMapping("list")
     public FebsResponse list(AiProductQuestion dto, QueryRequest request) {
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
 
         Map<String, Object> data = getDataTable(aiProductQuestionService.listInPage(dto, request));
+        return new FebsResponse().success().data(data);
+    }
+
+    @GetMapping("jobList")
+    public FebsResponse jobList(AiProductQuestionJob dto, QueryRequest request) {
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
+
+        Map<String, Object> data = getDataTable(aiProductQuestionService.listJobInPage(dto, request));
         return new FebsResponse().success().data(data);
     }
 
@@ -49,7 +61,21 @@
     @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
     public FebsResponse add(@RequestBody @Valid AiProductQuestion dto) {
 
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         return aiProductQuestionService.add(dto);
+    }
+
+    @PostMapping("aiAdd")
+    @ControllerEndpoint(operation = "AI生成题目", exceptionMessage = "操作失败")
+    public FebsResponse aiAdd(@RequestBody @Valid AiProductQuestionAiDto dto) {
+
+//        String companyId = getCurrentUserCompanyId();
+//        dto.setCompanyId(companyId);
+//        return aiProductQuestionService.aiAdd(dto);
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
+        return aiProductQuestionService.aiAddV2(dto);
     }
 
     @PostMapping("update")
@@ -57,6 +83,12 @@
     public FebsResponse update(@RequestBody @Valid AiProductQuestion dto) {
 
         return aiProductQuestionService.update(dto);
+    }
+
+    @PostMapping(value = "stateUpdate/{type}/{ids}")
+    @ControllerEndpoint(operation = "更新", exceptionMessage = "更新失败")
+    public FebsResponse stateUpdate(@PathVariable("ids") String ids, @PathVariable("type") Integer type) {
+        return aiProductQuestionService.stateUpdate(ids, type);
     }
 
     @GetMapping("delete/{id}")
@@ -71,6 +103,8 @@
     @GetMapping(value = "/questionTree")
     public FebsResponse questionTree() {
 
-        return new FebsResponse().success().data(aiProductQuestionService.questionTree());
+
+        String companyId = getCurrentUserCompanyId();
+        return new FebsResponse().success().data(aiProductQuestionService.questionTree(companyId));
     }
 }

--
Gitblit v1.9.1