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 token counting slash command.

Browse Source
This commit is contained in:
city-unit
2023-10-21 01:04:16 -04:00
parent eaadb1c5c2
commit 63ecca1fe2
1 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
public/scripts/power-user.js
View File

@ -28,7 +28,7 @@ import {
} from "./instruct-mode.js"; } from "./instruct-mode.js";
import { registerSlashCommand } from "./slash-commands.js"; import { registerSlashCommand } from "./slash-commands.js";
import { tokenizers } from "./tokenizers.js"; import { tokenizers, getTokenCount } from "./tokenizers.js";
import { countOccurrences, debounce, delay, isOdd, resetScrollHeight, sortMoments, timestampToMoment } from "./utils.js"; import { countOccurrences, debounce, delay, isOdd, resetScrollHeight, sortMoments, timestampToMoment } from "./utils.js";
@ -1554,6 +1554,21 @@ function doResetPanels() {
$("#movingUIreset").trigger('click'); $("#movingUIreset").trigger('click');
} }
function doCount() {
// get all of the messages in the chat
const messages = $('#chat .mes');
//concat all the messages into a single string
const allMessages = messages.toArray().map(x => $(x).find('.mes_text').text()).join(' ');
console.log(allMessages);
//toastr success with the token count of the chat
toastr.success(`Token count: ${getTokenCount(allMessages)}`);
}
function setAvgBG() { function setAvgBG() {
const bgimg = new Image(); const bgimg = new Image();
bgimg.src = $('#bg1') bgimg.src = $('#bg1')
@ -2396,4 +2411,5 @@ $(document).ready(() => {
registerSlashCommand('cut', doMesCut, [], '<span class="monospace">(number)</span> – cuts the specified message from the chat', true, true); registerSlashCommand('cut', doMesCut, [], '<span class="monospace">(number)</span> – cuts the specified message from the chat', true, true);
registerSlashCommand('resetpanels', doResetPanels, ['resetui'], '– resets UI panels to original state.', true, true); registerSlashCommand('resetpanels', doResetPanels, ['resetui'], '– resets UI panels to original state.', true, true);
registerSlashCommand('bgcol', setAvgBG, [], '– WIP test of auto-bg avg coloring', true, true); registerSlashCommand('bgcol', setAvgBG, [], '– WIP test of auto-bg avg coloring', true, true);
registerSlashCommand('count', doCount, [], '– counts the number of tokens in the current chat', true, false);
}); });