diff --git a/src/main/java/com/zsc/edu/dify/modules/dify/controller/V1WorkflowController.java b/src/main/java/com/zsc/edu/dify/modules/dify/controller/V1WorkflowController.java
index c26c6c6..688dba3 100644
--- a/src/main/java/com/zsc/edu/dify/modules/dify/controller/V1WorkflowController.java
+++ b/src/main/java/com/zsc/edu/dify/modules/dify/controller/V1WorkflowController.java
@@ -37,7 +37,7 @@ public class V1WorkflowController {
      */
     @PostMapping("/run/{appId}")
     @PreAuthorize("hasAuthority('dify:workflow:run')")
-    public WorkflowData runWorkflow(@RequestBody WorkflowRunRequest request, @PathVariable String appId) {
+    public WorkflowRunResponse runWorkflow(@RequestBody WorkflowRunRequest request, @PathVariable String appId) {
         return difyWorkflowService.run(request, appId);
     }
 
@@ -60,7 +60,7 @@ public class V1WorkflowController {
      * @return
      */
     @PatchMapping("/stop/{appId}")
-    public WorkflowStopResponse stopWorkflowStream( String taskId, String userId, @PathVariable String appId) {
+    public WorkflowStopResponse stopWorkflowStream(String taskId, String userId, @PathVariable String appId) {
         String apiKey = appEntityRepository.selectApiKey(appId);
         return difyWorkflow.stopWorkflowStream(apiKey, taskId, userId);
     }
diff --git a/src/main/java/com/zsc/edu/dify/modules/dify/service/DifyWorkflowService.java b/src/main/java/com/zsc/edu/dify/modules/dify/service/DifyWorkflowService.java
index 118e99c..af33588 100644
--- a/src/main/java/com/zsc/edu/dify/modules/dify/service/DifyWorkflowService.java
+++ b/src/main/java/com/zsc/edu/dify/modules/dify/service/DifyWorkflowService.java
@@ -3,7 +3,9 @@ package com.zsc.edu.dify.modules.dify.service;
 import com.baomidou.mybatisplus.extension.service.IService;
 import com.zsc.edu.dify.modules.dify.entity.WorkflowData;
 import io.github.guoshiqiufeng.dify.workflow.dto.request.WorkflowRunRequest;
+import io.github.guoshiqiufeng.dify.workflow.dto.response.WorkflowRunResponse;
+
 
 public interface DifyWorkflowService extends IService<WorkflowData> {
-    WorkflowData run(WorkflowRunRequest request, String appId);
+    WorkflowRunResponse run(WorkflowRunRequest request, String appId);
 }
diff --git a/src/main/java/com/zsc/edu/dify/modules/dify/service/Impl/DifyWorkflowServiceImpl.java b/src/main/java/com/zsc/edu/dify/modules/dify/service/Impl/DifyWorkflowServiceImpl.java
index d070834..e39ba73 100644
--- a/src/main/java/com/zsc/edu/dify/modules/dify/service/Impl/DifyWorkflowServiceImpl.java
+++ b/src/main/java/com/zsc/edu/dify/modules/dify/service/Impl/DifyWorkflowServiceImpl.java
@@ -8,6 +8,7 @@ import com.zsc.edu.dify.modules.dify.repo.WorkflowRepository;
 import com.zsc.edu.dify.modules.dify.service.DifyWorkflowService;
 import io.github.guoshiqiufeng.dify.workflow.DifyWorkflow;
 import io.github.guoshiqiufeng.dify.workflow.dto.request.WorkflowRunRequest;
+import io.github.guoshiqiufeng.dify.workflow.dto.response.WorkflowRunResponse;
 import jakarta.annotation.Resource;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
@@ -23,12 +24,13 @@ public class DifyWorkflowServiceImpl extends ServiceImpl<WorkflowRepository, Wor
     WorkflowMapper mapper;
 
     @Override
-    public WorkflowData run(WorkflowRunRequest request, String appId){
+    public WorkflowRunResponse run(WorkflowRunRequest request, String appId){
         String apiKey = appEntityRepository.selectApiKey(appId);
         request.setApiKey(apiKey);
-        WorkflowData  workflowData =mapper.toEntity(difyWorkflow.runWorkflow(request));
+        WorkflowRunResponse response = difyWorkflow.runWorkflow(request);
+        WorkflowData  workflowData =mapper.toEntity(response);
         workflowData.setAppId(appId);
         baseMapper.insert(workflowData);
-        return workflowData;
+        return response;
     }
 }