From 890017d305da8932d2fb32004112340ace2886ca Mon Sep 17 00:00:00 2001 From: zren25 Date: Fri, 14 Mar 2025 18:33:49 +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 +++++++++++++--- .../center/service/impl/DiFyServiceImpl.java | 4 ++-- 2 files changed, 15 insertions(+), 5 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 c60055b..5061287 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 @@ -129,14 +129,14 @@ public class CorpusFailJob { } oldAiAnalysisRequestLogs.setDifyResponse(JSON.toJSONString(execDifyFlow)); - aiAnalysisRequestLogsService.saveAiAnalysisRequestLogs(oldAiAnalysisRequestLogs); + updateDiFyRequest(oldAiAnalysisRequestLogs,oldAiAnalysisRequestLogs.getAiAnalysisRequestId()); aiAnalysisErrors.setRetryCount(aiAnalysisErrors.getRetryCount() + 1); aiAnalysisErrors.setAiAnalysisErrorHandlingStatus("1"); - aiAnalysisErrorsService.saveAiAnalysisErrors(aiAnalysisErrors); + updateAiAnalysisErrors(aiAnalysisErrors,oldAiAnalysisRequestLogs.getAiAnalysisRequestId()); } } catch (Exception e) { aiAnalysisErrors.setAiAnalysisErrorHandlingStatus("0"); - aiAnalysisErrorsService.saveAiAnalysisErrors(aiAnalysisErrors); + updateAiAnalysisErrors(aiAnalysisErrors,aiAnalysisErrors.getAiAnalysisRequestId()); } }); @@ -144,4 +144,14 @@ public class CorpusFailJob { } + + protected void updateDiFyRequest(AiAnalysisRequestLogs oldAiAnalysisRequestLogs, String aiAnalysisRequestId) { + aiAnalysisRequestLogsMapper.update(new AiAnalysisRequestLogs(), + new LambdaQueryWrapper().setEntity(oldAiAnalysisRequestLogs) + .eq(AiAnalysisRequestLogs::getAiAnalysisRequestId, aiAnalysisRequestId)); + } + protected void updateAiAnalysisErrors(AiAnalysisErrors aiAnalysisErrors, String aiAnalysisRequestId) { + aiAnalysisErrorsService.update( new LambdaQueryWrapper().setEntity(aiAnalysisErrors) + .eq(AiAnalysisErrors::getAiAnalysisRequestId, aiAnalysisRequestId)); + } } diff --git a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/DiFyServiceImpl.java b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/DiFyServiceImpl.java index 4cd4819..e6ed00d 100644 --- a/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/DiFyServiceImpl.java +++ b/ai-analytic-center-biz/src/main/java/com/volvo/ai/analytic/center/service/impl/DiFyServiceImpl.java @@ -98,8 +98,8 @@ public class DiFyServiceImpl implements DiFyService{ map.put("inputs",diFyReq.getInputs()); map.put("response_mode","blocking"); map.put("user",diFyReq.getUser()); - JSONObject difyResult = diFyFeign.runWorkflows("Bearer "+diFyReq.getFlowId(),map); - return difyResult; + JSONObject data = difyResult.getJSONObject("data"); + return data; } }