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

Append continue message timer instead of rewriting

This commit is contained in:
Cohee 2023-09-18 02:12:06 +03:00
parent dc1c477d62
commit 3a8383ab79
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
public/script.js
View File

@ -2260,7 +2260,7 @@ class StreamingProcessor {
throw new Error('Generation function for streaming is not hooked up');
}
constructor(type, force_name2) {
constructor(type, force_name2, timeStarted) {
this.result = "";
this.messageId = -1;
this.type = type;
@ -2270,7 +2270,7 @@ class StreamingProcessor {
this.generator = this.nullStreamingGeneration;
this.abortController = new AbortController();
this.firstMessageText = '...';
this.timeStarted = new Date();
this.timeStarted = timeStarted;
}
async generate() {
@ -2420,6 +2420,18 @@ async function Generate(type, { automatic_trigger, force_name2, resolve, reject,
const isContinue = type == 'continue';
// Rewrite the generation timer to account for the time passed for all the continuations.
if (isContinue && chat.length) {
const prevFinished = chat[chat.length - 1]['gen_finished'];
const prevStarted = chat[chat.length - 1]['gen_started'];
if (prevFinished && prevStarted) {
const timePassed = prevFinished - prevStarted;
generation_started = new Date(Date.now() - timePassed);
chat[chat.length - 1]['gen_started'] = generation_started;
}
}
if (!dryRun) {
deactivateSendButtons();
}
@ -2665,7 +2677,7 @@ async function Generate(type, { automatic_trigger, force_name2, resolve, reject,
console.debug('calling runGenerate');
if (!dryRun) {
streamingProcessor = isStreamingEnabled() ? new StreamingProcessor(type, force_name2) : false;
streamingProcessor = isStreamingEnabled() ? new StreamingProcessor(type, force_name2, generation_started) : false;
}
if (isContinue) {