centersl/lobehub
1
0
Fork 0
mirror of https://github.com/lobehub/lobehub.git synced 2026-03-27 13:29:15 +07:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix: 修正total token 计算不正确的问题

Browse Source
This commit is contained in:
arvinxx
2023-07-22 17:46:00 +08:00
parent 6cb4828b96
commit 17815c6989
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/store/session/slices/chat/selectors.ts
View File

@@ -59,25 +59,24 @@ const systemRoleSel = (s: SessionStore): string => {
return config.systemRole;
};
const totalTokens = (s: SessionStore): number[] => {
const chats = currentChats(s);
return encode(chats.map((m) => m.content).join(''));
};
const systemRoleTokens = (s: SessionStore): number[] => {
const systemRole = systemRoleSel(s);
return encode(systemRole || '');
};
const totalTokenCount = (s: SessionStore) => totalTokens(s).length;
const chatsTokens = (s: SessionStore): number[] => {
const chats = currentChats(s);
return encode(chats.map((m) => m.content).join(''));
};
const systemRoleTokenCount = (s: SessionStore) => systemRoleTokens(s).length;
const totalTokenCount = (s: SessionStore) => chatsTokens(s).length + systemRoleTokenCount(s);
export const chatSelectors = {
currentChats,
systemRoleTokenCount,
systemRoleTokens,
totalTokenCount,
totalTokens,
};