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

Switch word counting to Segmenter

Browse Source
This commit is contained in:
Wolfsblvt
2024-05-02 02:02:12 +02:00
parent 0799090a1a
commit 04a798b229
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/endpoints/stats.js
View File

@@ -16,8 +16,9 @@ const MIN_TIMESTAMP = 0;
const MAX_TIMESTAMP = new Date('9999-12-31T23:59:59.999Z').getTime(); const MAX_TIMESTAMP = new Date('9999-12-31T23:59:59.999Z').getTime();
const MIN_DATE = new Date(MIN_TIMESTAMP); const MIN_DATE = new Date(MIN_TIMESTAMP);
const MAX_DATE = new Date(MAX_TIMESTAMP); const MAX_DATE = new Date(MAX_TIMESTAMP);
const STATS_LANGUAGE = 'en';
const STATS_FILE = 'stats.json'; const STATS_FILE = 'stats.json';
const CURRENT_STATS_VERSION = '1.1'; const CURRENT_STATS_VERSION = '1.2';
/** @type {Map<string, UserStatsCollection>} The stats collections for each user, accessable via their key - gets set/built on init */ /** @type {Map<string, UserStatsCollection>} The stats collections for each user, accessable via their key - gets set/built on init */
const STATS = new Map(); const STATS = new Map();
@@ -676,7 +677,9 @@ function removeModelUsage(obj, model, tokens, count = 1) {
* @returns {number} - The number of words in the string. * @returns {number} - The number of words in the string.
*/ */
function countWordsInString(str) { function countWordsInString(str) {
return str.match(/\b\w+\b/g)?.length ?? 0; const words = Array.from(new Intl.Segmenter(STATS_LANGUAGE ?? 'en', { granularity: 'word' }).segment(str))
.filter(it => it.isWordLike);
return words.length;
} }
/** /**