From 24b0bd4bb21fc083ca7449b2af9207f245d5d6aa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=A2=AB=E9=81=97=E5=BF=98=E7=9A=84=E8=AE=B0=E5=BF=86?=
 <32097237+YIWANG-sketch@users.noreply.github.com>
Date: Sat, 21 Dec 2024 12:45:12 +0800
Subject: [PATCH] adjust heartbeat

---
 you_providers/youProvider.mjs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/you_providers/youProvider.mjs b/you_providers/youProvider.mjs
index fa7f312..6d125fc 100644
--- a/you_providers/youProvider.mjs
+++ b/you_providers/youProvider.mjs
@@ -1354,6 +1354,17 @@ class YouProvider {
 
                 await setupEventSource(page, url, traceId, customEndMarker);
 
+                if (stream) {
+                    heartbeatInterval = setInterval(() => {
+                        if (!isEnding && !clientState.isClosed()) {
+                            emitter.emit("completion", traceId, `\r`);
+                        } else {
+                            clearInterval(heartbeatInterval);
+                            heartbeatInterval = null;
+                        }
+                    }, 5000);
+                }
+
                 return true;
             } catch (error) {
                 console.error("重新发送请求时发生错误:", error);
@@ -1409,7 +1420,7 @@ class YouProvider {
             if (stream) {
                 heartbeatInterval = setInterval(() => {
                     if (!isEnding && !clientState.isClosed()) {
-                        emitter.emit("completion", traceId, " ");
+                        emitter.emit("completion", traceId, `\r`);
                     } else {
                         clearInterval(heartbeatInterval);
                         heartbeatInterval = null;