rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix stream error parsing when using Smooth Streaming

This commit is contained in:
Cohee 2024-04-04 21:20:10 +03:00
parent ecc638a76d
commit 813476d72a
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
public/scripts/sse-stream.js
View File

@ -227,7 +227,7 @@ async function* parseStreamData(json) {
}
}
return null;
yield null;
}
/**
@ -243,6 +243,12 @@ export class SmoothEventSourceStream extends EventSourceStream {
const data = event.data;
try {
const hasFocus = document.hasFocus();
if (data === '[DONE]') {
lastStr = '';
return controller.enqueue(event);
}
const json = JSON.parse(data);
if (!json) {
@ -261,7 +267,8 @@ export class SmoothEventSourceStream extends EventSourceStream {
lastStr = parsed.chunk;
hasFocus && await eventSource.emit(event_types.SMOOTH_STREAM_TOKEN_RECEIVED, parsed.chunk);
}
} catch {
} catch (error) {
console.error('Smooth Streaming parsing error', error);
controller.enqueue(event);
}
},