From cbf7a1c587caee27cb98676f0ccffdcbf7b1dc49 Mon Sep 17 00:00:00 2001
From: cherishsince <cherishsince@aliyun.com>
Date: Thu, 27 Jun 2024 17:13:49 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BB=A3=E7=A0=81=E4=BC=98=E5=8C=96?=
 =?UTF-8?q?=E3=80=91XingHuoApi=20stream=20=E5=8F=98=E9=87=8F=E6=B2=A1?=
 =?UTF-8?q?=E6=9C=89=E5=85=B6=E4=BB=96=E5=9C=B0=E6=96=B9=E4=BD=BF=E7=94=A8?=
 =?UTF-8?q?=EF=BC=8C=E7=9B=B4=E6=8E=A5return?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../yudao/framework/ai/core/model/xinghuo/api/XingHuoApi.java  | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/xinghuo/api/XingHuoApi.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/xinghuo/api/XingHuoApi.java
index b267de84c..eeceb525a 100644
--- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/xinghuo/api/XingHuoApi.java
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/xinghuo/api/XingHuoApi.java
@@ -134,7 +134,7 @@ public class XingHuoApi {
         URI uri = URI.create(authUrl);
         // 发起 wss 请求并处理响应
         // 创建一个 Flux 来处理接收到的消息
-        Flux<XingHuoChatCompletion> messageFlux = Flux.create(sink -> {
+        return Flux.create(sink -> {
             socketClient.execute(uri, session ->
                             session.send(Mono.just(session.textMessage(JSONUtil.toJsonStr(request))))
                                     .thenMany(session.receive()
@@ -145,6 +145,5 @@ public class XingHuoApi {
                                     .then())
                     .subscribe(); // 订阅以开始会话
         });
-        return messageFlux;
     }
 }