回调更新report

ai-analytic-center-api/src/main/java/com/volvo/ai/analytic/center/dto/req/SummaryQuestionRequestDTO.java

@@ -0,0 +1,12 @@
+package com.volvo.ai.analytic.center.dto.req;
+
+import lombok.Data;
+
+@Data
+public class SummaryQuestionRequestDTO {
+
+    private String summaryReportId;
+
+    private String preliminarySummaryResults;
+
+}

ai-analytic-center-api/src/main/java/com/volvo/ai/analytic/center/enums/BusinessTypeEnum.java

@@ -29,6 +29,7 @@ public enum BusinessTypeEnum {
     SMART_ASSISTANT_NAMEPLATE("SMART_ASSISTANT_NAMEPLATE", "智能助手-铭牌"),
 
     IM_CORPUS_ANALYSIS("IM_CORPUS_ANALYSIS", "IM语料库"),
+    SUMMARY_REPORT("SUMMARY_REPORT", "问题总结"),
 
 
     //特邀发言官

ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/controller/AiAnalysisDifyController.java

@@ -9,9 +9,11 @@ import com.fasterxml.jackson.databind.ObjectMapper;
 import com.volvo.ai.analytic.center.dto.req.AnalysisQueryReq;
 import com.volvo.ai.analytic.center.dto.req.AnalysisReq;
 import com.volvo.ai.analytic.center.dto.req.ConsultingRequestDTO;
+import com.volvo.ai.analytic.center.dto.req.SummaryQuestionRequestDTO;
 import com.volvo.ai.analytic.center.dto.resp.AnalysisResp;
 import com.volvo.ai.analytic.center.service.AiAnalysisDifyService;
 import com.volvo.ai.analytic.center.service.CorpusQuestionService;
+import com.volvo.ai.analytic.center.service.UserQuestionSummaryService;
 import com.volvo.common.core.util.ResultMsg;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -30,14 +32,11 @@ import org.springframework.web.bind.annotation.*;
 @RequestMapping("analysis")
 public class AiAnalysisDifyController {
 
-    @Autowired
-    private RocketMQTemplate rocketMQTemplate;
-
     @Autowired
     private AiAnalysisDifyService aiAnalysisDifyService;
 
     @Autowired
-    private CorpusQuestionService corpusQuestionService;
+    private UserQuestionSummaryService userQuestionSummaryService;
 
     @PostMapping("/updateByAiId")
     @ApiOperation(value = "更新dify结果")
@@ -79,13 +78,12 @@ public class AiAnalysisDifyController {
 
 
 
-    @PostMapping("/corpusQuestion")
+    @PostMapping("/corpusSummaryQuestion")
     @ApiOperation(value = "单条语料分析结论入库")
-    public AnalysisResp<Object> corpusQuestion(@RequestHeader(value = "appKey") String appKey,
+    public AnalysisResp<Object> corpusSummaryQuestion(@RequestBody SummaryQuestionRequestDTO requestDTO) {
-                                        @RequestBody ConsultingRequestDTO requestDTO) {
+        log.info("aiAnalyze corpusSummaryQuestion requestDTO: {}", JSON.toJSONString(requestDTO));
-        log.info("aiAnalyze corpusQuestion requestDTO: {} ,appKey:{}", JSON.toJSONString(requestDTO),appKey);
         // 打印接收到的参数
-        corpusQuestionService.runCorpusQuestionDify(requestDTO);
+        userQuestionSummaryService.updateSummery(requestDTO);
         return AnalysisResp .success("ok");
     }
 

ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/mapper/TmCorpusQuestionSummaryReportMapper.java

@@ -1,8 +1,12 @@
 package com.volvo.ai.analytic.center.mapper;
 
+import org.apache.ibatis.annotations.Param;
+
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.volvo.ai.analytic.center.entity.TmCorpusQuestionSummaryReport;
 
 public interface TmCorpusQuestionSummaryReportMapper extends BaseMapper<TmCorpusQuestionSummaryReport> {
 
+    int updateBySummaryReportId(@Param("summaryReportId") String summaryReportId, @Param("preliminarySummaryResults") String preliminarySummaryResults);
+
 }

ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/UserQuestionSummaryService.java

@@ -1,6 +1,7 @@
 package com.volvo.ai.analytic.center.service;
 
 import com.baomidou.mybatisplus.extension.service.IService;
+import com.volvo.ai.analytic.center.dto.req.SummaryQuestionRequestDTO;
 import com.volvo.ai.analytic.center.entity.TmCorpusQuestionSummaryReport;
 /**
  * @description 用户问题总结
@@ -11,5 +12,7 @@ public interface UserQuestionSummaryService extends IService<TmCorpusQuestionSum
 
 	void runUserQuestionSummary(String paramJson);
 
+	void updateSummery(SummaryQuestionRequestDTO summaryQuestionRequestDTO);
+
 
 }

ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/UserQuestionSummaryServiceImpl.java

@@ -4,11 +4,14 @@ import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.volvo.ai.analytic.center.dto.corpus.ImCurMessageDTO;
 import com.volvo.ai.analytic.center.dto.req.DiFyReq;
+import com.volvo.ai.analytic.center.dto.req.SummaryQuestionRequestDTO;
 import com.volvo.ai.analytic.center.entity.*;
+import com.volvo.ai.analytic.center.enums.BusinessTypeEnum;
 import com.volvo.ai.analytic.center.enums.QuestionMainCategoryEnum;
 import com.volvo.ai.analytic.center.mapper.TmCorpusQuestionReportMapper;
 import com.volvo.ai.analytic.center.mapper.TmCorpusQuestionSummaryReportMapper;
 import com.volvo.ai.analytic.center.service.*;
+import com.volvo.ai.analytic.center.utils.AiAnalysisUtils;
 import com.volvo.ai.analytic.center.utils.ConstantStr;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
@@ -83,6 +86,15 @@ public class UserQuestionSummaryServiceImpl extends ServiceImpl<TmCorpusQuestion
         log.info("处理问题结束 耗时：{}",System.currentTimeMillis()-startTime);
 	}
 
+
+	@Override
+	public void updateSummery(SummaryQuestionRequestDTO summaryQuestionRequestDTO) {
+		if (summaryQuestionRequestDTO == null || summaryQuestionRequestDTO.getSummaryReportId() == null || StringUtils.isBlank(summaryQuestionRequestDTO.getPreliminarySummaryResults())) {
+			return;
+		}
+		tmCorpusQuestionSummaryReportMapper.updateBySummaryReportId(summaryQuestionRequestDTO.getSummaryReportId(), summaryQuestionRequestDTO.getPreliminarySummaryResults());
+	}
+
 	private void processItem(Date statTime, Date endTime, String mainCategoryString) {
 		log.info("用户问题总结处理：mainCategory：{}", mainCategoryString);
 
@@ -92,6 +104,13 @@ public class UserQuestionSummaryServiceImpl extends ServiceImpl<TmCorpusQuestion
 		if (resultList == null || resultList.isEmpty()) {
 			return;
 		}
+
+		TmCorpusQuestionSummaryReport summaryReport = new TmCorpusQuestionSummaryReport();
+		summaryReport.setMainCategory(mainCategoryString);
+		summaryReport.setSourceTime(statTime);
+		String aiAnalysisRequestId = AiAnalysisUtils.getAiAnalysisRequestId(BusinessTypeEnum.SUMMARY_REPORT.getCode());
+		summaryReport.setSummaryReportId(aiAnalysisRequestId);
+		tmCorpusQuestionSummaryReportMapper.insert(summaryReport);
         
 		// 组装语料信息
 		StringBuffer demandList = new StringBuffer();
@@ -102,6 +121,7 @@ public class UserQuestionSummaryServiceImpl extends ServiceImpl<TmCorpusQuestion
 		});
 		Map<String, Object> inputMap = new HashMap<>();
 		inputMap.put("chat", demandList.toString());
+		inputMap.put("summaryReportId", aiAnalysisRequestId);
 
 		DiFyReq diFyImageReq = new DiFyReq();
 		diFyImageReq.setUser(ConstantStr.corpus_user);
@@ -111,15 +131,6 @@ public class UserQuestionSummaryServiceImpl extends ServiceImpl<TmCorpusQuestion
 		JSONObject resultJsonObject = (JSONObject) diFyService.getDiFyObject(diFyImageReq);
 		log.info("runDify userQuestionSummary difyflow {}", resultJsonObject);
 
-		// 保存结果
-        if (resultJsonObject != null) {
-			TmCorpusQuestionSummaryReport summaryReport = new TmCorpusQuestionSummaryReport();
-			summaryReport.setPreliminarySummaryResults(resultJsonObject.getString("preliminarySummaryResults"));
-			summaryReport.setMainCategory(mainCategoryString);
-			summaryReport.setSourceTime(statTime);
-			tmCorpusQuestionSummaryReportMapper.insert(summaryReport);
-		}
-
 	}
 
 	/**

ai-analytic-center-biz/src/main/resources/mapper/TmCorpusQuestionSummaryReportMapper.xml

@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.volvo.ai.analytic.center.mapper.TmCorpusQuestionSummaryReportMapper">
+
+    <update id="updateBySummaryReportId">
+        update tm_corpus_question_summary_report
+        set preliminary_summary_results = #{preliminarySummaryResults}
+        where summary_report_id = #{summaryReportId}
+    </update>
+
+</mapper>

ai-analytic-center-biz/target/classes/mapper/TmCorpusQuestionSummaryReportMapper.xml

@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.volvo.ai.analytic.center.mapper.TmCorpusQuestionSummaryReportMapper">
+
+    <update id="updateBySummaryReportId">
+        update tm_corpus_question_summary_report
+        set preliminary_summary_results = #{preliminarySummaryResults}
+        where summary_report_id = #{summaryReportId}
+    </update>
+
+</mapper>