Merge pull request #1996 from aisu-wata0/staging

fix: "Squash system messages" would fail to skip example messages
This commit is contained in:
Cohee 2024-04-02 00:35:57 +03:00 committed by GitHub
commit d498458971
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -2249,8 +2249,12 @@ export class ChatCompletion {
continue;
}
if (!excludeList.includes(message.identifier) && message.role === 'system' && !message.name) {
if (lastMessage && lastMessage.role === 'system') {
const shouldSquash = (message) => {
return !excludeList.includes(message.identifier) && message.role === 'system' && !message.name;
}
if (shouldSquash(message)) {
if (lastMessage && shouldSquash(lastMessage)) {
lastMessage.content += '\n' + message.content;
lastMessage.tokens = tokenHandler.count({ role: lastMessage.role, content: lastMessage.content });
}