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

Debounce token counting in popup plugin

Browse Source
This commit is contained in:
Cohee
2024-03-01 21:42:49 +02:00
parent d57b5ad6f7
commit d1ca855d23
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
public/scripts/extensions/token-counter/index.js
View File

@ -2,7 +2,7 @@ import { callPopup, main_api } from '../../../script.js';
import { getContext } from '../../extensions.js'; import { getContext } from '../../extensions.js';
import { registerSlashCommand } from '../../slash-commands.js'; import { registerSlashCommand } from '../../slash-commands.js';
import { getFriendlyTokenizerName, getTextTokens, getTokenCount, tokenizers } from '../../tokenizers.js'; import { getFriendlyTokenizerName, getTextTokens, getTokenCount, tokenizers } from '../../tokenizers.js';
import { resetScrollHeight } from '../../utils.js'; import { resetScrollHeight, debounce } from '../../utils.js';
function rgb2hex(rgb) { function rgb2hex(rgb) {
rgb = rgb.match(/^rgba?[\s+]?\([\s+]?(\d+)[\s+]?,[\s+]?(\d+)[\s+]?,[\s+]?(\d+)[\s+]?/i); rgb = rgb.match(/^rgba?[\s+]?\([\s+]?(\d+)[\s+]?,[\s+]?(\d+)[\s+]?,[\s+]?(\d+)[\s+]?/i);
@ -38,7 +38,7 @@ async function doTokenCounter() {
</div>`; </div>`;
const dialog = $(html); const dialog = $(html);
dialog.find('#token_counter_textarea').on('input', () => { const countDebounced = debounce(() => {
const text = String($('#token_counter_textarea').val()); const text = String($('#token_counter_textarea').val());
const ids = main_api == 'openai' ? getTextTokens(tokenizers.OPENAI, text) : getTextTokens(tokenizerId, text); const ids = main_api == 'openai' ? getTextTokens(tokenizers.OPENAI, text) : getTextTokens(tokenizerId, text);
@ -59,7 +59,8 @@ async function doTokenCounter() {
resetScrollHeight($('#token_counter_textarea')); resetScrollHeight($('#token_counter_textarea'));
resetScrollHeight($('#token_counter_ids')); resetScrollHeight($('#token_counter_ids'));
}); }, 1000);
dialog.find('#token_counter_textarea').on('input', () => countDebounced());
$('#dialogue_popup').addClass('wide_dialogue_popup'); $('#dialogue_popup').addClass('wide_dialogue_popup');
callPopup(dialog, 'text', '', { wide: true, large: true }); callPopup(dialog, 'text', '', { wide: true, large: true });