[fix] tokenizer

This commit is contained in:
kwaroran
2023-06-10 02:27:54 +09:00
parent e88f95102b
commit 6fc60805e6

View File

@@ -49,8 +49,10 @@ export class ChatTokenizer {
this.useName = useName this.useName = useName
} }
async tokenizeChat(data:OpenAIChat) { async tokenizeChat(data:OpenAIChat) {
const encoded = (await encode(data.content)).length let encoded = (await encode(data.content)).length + this.chatAdditonalTokens
+ this.useName === 'name' ? (await encode(data.name)).length : 0 + this.chatAdditonalTokens if(data.name && this.useName ==='name'){
encoded += (await encode(data.name)).length
}
return encoded return encoded
} }