rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Append continue message timer instead of rewriting

Browse Source
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
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'); throw new Error('Generation function for streaming is not hooked up');
} }
constructor(type, force_name2) { constructor(type, force_name2, timeStarted) {
this.result = ""; this.result = "";
this.messageId = -1; this.messageId = -1;
this.type = type; this.type = type;
@@ -2270,7 +2270,7 @@ class StreamingProcessor {
this.generator = this.nullStreamingGeneration; this.generator = this.nullStreamingGeneration;
this.abortController = new AbortController(); this.abortController = new AbortController();
this.firstMessageText = '...'; this.firstMessageText = '...';
this.timeStarted = new Date(); this.timeStarted = timeStarted;
} }
async generate() { async generate() {
@@ -2420,6 +2420,18 @@ async function Generate(type, { automatic_trigger, force_name2, resolve, reject,
const isContinue = type == 'continue'; 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) { if (!dryRun) {
deactivateSendButtons(); deactivateSendButtons();
} }
@@ -2665,7 +2677,7 @@ async function Generate(type, { automatic_trigger, force_name2, resolve, reject,
console.debug('calling runGenerate'); console.debug('calling runGenerate');
if (!dryRun) { if (!dryRun) {
streamingProcessor = isStreamingEnabled() ? new StreamingProcessor(type, force_name2) : false; streamingProcessor = isStreamingEnabled() ? new StreamingProcessor(type, force_name2, generation_started) : false;
} }
if (isContinue) { if (isContinue) {