Use proper tokenizer for Poe

This commit is contained in:
SillyLossy
2023-04-13 16:26:08 +03:00
parent 661b41341e
commit 14cc5ba937
10 changed files with 539 additions and 28 deletions

View File

@ -208,14 +208,14 @@ async function summarizeChat(context) {
memoryBuffer.push(entry);
// check if token limit was reached
if (context.encode(getMemoryString()).length >= extension_settings.memory.shortMemoryLength) {
if (context.getTokenCount(getMemoryString()) >= extension_settings.memory.shortMemoryLength) {
break;
}
}
const resultingString = getMemoryString();
if (context.encode(resultingString).length < extension_settings.memory.shortMemoryLength) {
if (context.getTokenCount(resultingString) < extension_settings.memory.shortMemoryLength) {
return;
}