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<AiAnalysisRequestLogs>().setEntity(oldAiAnalysisRequestLogs)
+                        .eq(AiAnalysisRequestLogs::getAiAnalysisRequestId, aiAnalysisRequestId));
+    }
+    protected void updateAiAnalysisErrors(AiAnalysisErrors aiAnalysisErrors, String aiAnalysisRequestId) {
+        aiAnalysisErrorsService.update(  new LambdaQueryWrapper<AiAnalysisErrors>().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;
     }
 }