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

lobehub · Jul 22, 2023 · 17815c6 · 17815c6
1 parent 6cb4828
commit 17815c6
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/src/store/session/slices/chat/selectors.ts b/src/store/session/slices/chat/selectors.ts
@@ -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,
 };