kova/youchat_proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adjust heartbeat

Browse Source
This commit is contained in:
被遗忘的记忆
2024-12-21 12:45:12 +08:00
committed by GitHub
parent 2f45b5016a
commit 24b0bd4bb2
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
you_providers/youProvider.mjs
+12 -1
View File
@@ -1354,6 +1354,17 @@ class YouProvider {
await setupEventSource(page, url, traceId, customEndMarker); 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; return true;
} catch (error) { } catch (error) {
console.error("重新发送请求时发生错误:", error); console.error("重新发送请求时发生错误:", error);
@@ -1409,7 +1420,7 @@ class YouProvider {
if (stream) { if (stream) {
heartbeatInterval = setInterval(() => { heartbeatInterval = setInterval(() => {
if (!isEnding && !clientState.isClosed()) { if (!isEnding && !clientState.isClosed()) {
emitter.emit("completion", traceId, " "); emitter.emit("completion", traceId, `\r`);
} else { } else {
clearInterval(heartbeatInterval); clearInterval(heartbeatInterval);
heartbeatInterval = null; heartbeatInterval = null;