diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/yiyan/YiYanChatClient.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/yiyan/YiYanChatClient.java
index a613af6db2..c7b0bf372b 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/yiyan/YiYanChatClient.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/yiyan/YiYanChatClient.java
@@ -85,22 +85,20 @@ public class YiYanChatClient implements ChatClient, StreamingChatClient {
     @Override
     public Flux<ChatResponse> stream(Prompt prompt) {
         YiYanChatCompletionRequest request = this.createRequest(prompt, true);
-        // TODO @fan：return this.retryTemplate.execute(ctx -> {
-        // 调用 callWithFunctionSupport 发送请求
-        Flux<YiYanChatCompletionResponse> response = this.yiYanApi.chatCompletionStream(request);
-        // TODO @fan：下面的 doOnComplete 是不是可以删除哈？
-        response.doOnComplete(new Runnable() {
-            @Override
-            public void run() {
-                String a = ";";
-            }
-        });
-        return response.map(chunk -> {
-            // TODO @fan：ChatResponseMetadata chatResponseMetadata
-            return new ChatResponse(List.of(new Generation(chunk.getResult())));
+        // TODO done @fan：return this.retryTemplate.execute(ctx -> {
+        return retryTemplate.execute(ctx -> {
+            // 调用 callWithFunctionSupport 发送请求
+            Flux<YiYanChatCompletionResponse> response = this.yiYanApi.chatCompletionStream(request);
+            return response.map(chunk -> {
+//                System.err.println("---".concat(chunk.getResult()));
+                // TODO @fan：ChatResponseMetadata chatResponseMetadata
+                return new ChatResponse(List.of(new Generation(chunk.getResult())));
+            });
         });
     }
 
+
+
     private YiYanChatCompletionRequest createRequest(Prompt prompt, boolean stream) {
         // 参考 https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t 文档，system 是独立字段
         // 1.1 获取 user 和 assistant