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

Add data attribute during generation

Browse Source 
- Fix reasoning buttons still appearing
- Allow easier customization of HTML elements via CSS that appear during gen
This commit is contained in:
Wolfsblvt
2025-02-11 23:34:21 +01:00
parent 4e94e83540
commit 58b8e7908d
2 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
public/script.js
View File

@ -3160,14 +3160,16 @@ class StreamingProcessor {
}
}
showMessageButtons(messageId) {
markUIGenStarted() {
showStopButton();
$(`#chat .mes[mesid="${messageId}"] .mes_buttons`).css({ 'display': 'none' });
const chatElement = document.getElementById('chat');
chatElement.dataset.generating = 'true';
}
hideMessageButtons(messageId) {
markUIGenStopped() {
hideStopButton();
$(`#chat .mes[mesid="${messageId}"] .mes_buttons`).css({ 'display': 'flex' });
const chatElement = document.getElementById('chat');
delete chatElement.dataset.generating;
}
async onStartStreaming(text) {
@ -3180,7 +3182,7 @@ class StreamingProcessor {
await saveReply(this.type, text, true, '', [], '');
messageId = chat.length - 1;
this.#checkDomElements(messageId);
this.showMessageButtons(messageId);
this.markUIGenStarted();
}
hideSwipeButtons();
scrollChatToBottom();
@ -3268,7 +3270,7 @@ class StreamingProcessor {
}
async onFinishStreaming(messageId, text) {
this.hideMessageButtons(this.messageId);
this.markUIGenStopped();
await this.onProgressStreaming(messageId, text, true);
addCopyToCodeBlocks($(`#chat .mes[mesid="${messageId}"]`));
@ -3313,7 +3315,7 @@ class StreamingProcessor {
this.abortController.abort();
this.isStopped = true;
this.hideMessageButtons(this.messageId);
this.markUIGenStopped();
generatedPromptCache = '';
unblockGeneration();