From 54a8727c3cde462d8a50c8d9cbe6c9231b659f5e Mon Sep 17 00:00:00 2001
From: zren25 <zhen.ren.2@volvocars.com>
Date: Mon, 17 Mar 2025 11:08:09 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=A4=B1=E8=B4=A5=E8=A1=A5?=
 =?UTF-8?q?=E5=81=BF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ai/analytic/center/job/CorpusFailJob.java    | 16 ++++++++++------
 .../impl/TmTelephoneCorpusServiceImpl.java       |  1 +
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/job/CorpusFailJob.java b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/job/CorpusFailJob.java
index 5061287..0557f61 100644
--- a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/job/CorpusFailJob.java
+++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/job/CorpusFailJob.java
@@ -22,6 +22,8 @@ import org.apache.rocketmq.spring.core.RocketMQTemplate;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Component;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RestController;
 
 import javax.annotation.Resource;
 import java.util.Date;
@@ -32,6 +34,7 @@ import java.util.Map;
 
 @Slf4j
 @Component
+@RestController
 public class CorpusFailJob {
 
     @Autowired
@@ -61,6 +64,7 @@ public class CorpusFailJob {
      * 企微语料处理
      */
     @XxlJob("corpusFailTask")
+    @GetMapping("corpusFailTask")
     public void corpusFailTask() {
 
         log.info("语料解析失败重试处理");
@@ -93,7 +97,7 @@ public class CorpusFailJob {
                             ltoMap.put("analysisRecordId", oldAiAnalysisRequestLogs.getAiAnalysisRequestId());
                             ltoMap.put("analysisScene", corpusReportDTO.getAnalysisScene() + "");
                             String tag = "";
-                            if (corpusReportDTO.getAnalysisScene() == 1) { //企微
+                            if (null != corpusReportDTO && corpusReportDTO.getAnalysisScene() == 1) { //企微
                                 ltoMap.put("unionId", corpusReportDTO.getUnionId());
                                 ltoMap.put("consultantId", corpusReportDTO.getUserId());
                                 tag = CategoryEnum.ENTERPRISE_WECHAT.getCode();
@@ -136,6 +140,7 @@ public class CorpusFailJob {
                     }
                 } catch (Exception e) {
                     aiAnalysisErrors.setAiAnalysisErrorHandlingStatus("0");
+                    aiAnalysisErrors.setRetryCount(aiAnalysisErrors.getRetryCount()+ 1);
                     updateAiAnalysisErrors(aiAnalysisErrors,aiAnalysisErrors.getAiAnalysisRequestId());
                 }
             });
@@ -146,12 +151,11 @@ public class CorpusFailJob {
 
 
     protected void updateDiFyRequest(AiAnalysisRequestLogs oldAiAnalysisRequestLogs, String aiAnalysisRequestId) {
-        aiAnalysisRequestLogsMapper.update(new AiAnalysisRequestLogs(),
-                new LambdaQueryWrapper<AiAnalysisRequestLogs>().setEntity(oldAiAnalysisRequestLogs)
-                        .eq(AiAnalysisRequestLogs::getAiAnalysisRequestId, aiAnalysisRequestId));
+        aiAnalysisRequestLogsMapper.update(oldAiAnalysisRequestLogs,
+                new LambdaQueryWrapper<AiAnalysisRequestLogs>().eq(AiAnalysisRequestLogs::getAiAnalysisRequestId, aiAnalysisRequestId));
     }
     protected void updateAiAnalysisErrors(AiAnalysisErrors aiAnalysisErrors, String aiAnalysisRequestId) {
-        aiAnalysisErrorsService.update(  new LambdaQueryWrapper<AiAnalysisErrors>().setEntity(aiAnalysisErrors)
-                        .eq(AiAnalysisErrors::getAiAnalysisRequestId, aiAnalysisRequestId));
+        aiAnalysisErrorsService.update(aiAnalysisErrors,
+                new LambdaQueryWrapper<AiAnalysisErrors>().eq(AiAnalysisErrors::getAiAnalysisRequestId, aiAnalysisRequestId));
     }
 }
diff --git a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/TmTelephoneCorpusServiceImpl.java b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/TmTelephoneCorpusServiceImpl.java
index b655ec3..78011af 100644
--- a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/TmTelephoneCorpusServiceImpl.java
+++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/TmTelephoneCorpusServiceImpl.java
@@ -140,6 +140,7 @@ public class TmTelephoneCorpusServiceImpl extends ServiceImpl<TmTelephoneCorpusM
 			CorpusReportDTO corpusReportDTO = new CorpusReportDTO();
 			corpusReportDTO.setCorpusTime(formattedDateStartTime);
 			corpusReportDTO.setRecordId(aicorpusTelephone.getSourceId());
+			corpusReportDTO.setAnalysisScene(2l);
 			// 获取配置
 			JSONObject execDifyFlow = diFyService.executeDifyFlow(diFyImageReq, BusinessTypeEnum.SMART_ASSISTANT.getCode(), JSONObject.toJSONString(corpusReportDTO));
 			log.info("runDify execDifyFlow {}",execDifyFlow);