From 098f7ff0273e955dcd2441820198b47c9b03b7ad Mon Sep 17 00:00:00 2001
From: arvinxx <arvinx@foxmail.com>
Date: Sat, 22 Jul 2023 18:06:26 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feat:=20=E8=A1=A5=E5=85=85=20token?=
 =?UTF-8?q?=20=E8=AF=A6=E6=83=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/locales/default/common.ts              |  4 +++
 src/pages/chat/[id]/Conversation/Input.tsx | 39 +++++++++++++---------
 2 files changed, 28 insertions(+), 15 deletions(-)

diff --git a/src/locales/default/common.ts b/src/locales/default/common.ts
index 481a281ce007..c7e192988bff 100644
--- a/src/locales/default/common.ts
+++ b/src/locales/default/common.ts
@@ -12,10 +12,13 @@ export default {
   'agentTag': '标签',
   'agentTagPlaceholder': '请输入标签',
   'archive': '归档',
+  'archiveCurrentMessages': '归档当前会话',
   'autoGenerate': '自动补全',
   'autoGenerateTooltip': '基于提示词自动补全助手描述',
   'cancel': '取消',
+  'clearCurrentMessages': '清空当前会话消息',
   'close': '关闭',
+  'confirmClearCurrentMessages': '即将清空当前会话消息，清空后将无法找回，请确认你的操作',
   'confirmRemoveSessionItemAlert': '即将删除该助手，删除后该将无法找回，请确认你的操作',
   'defaultAgent': '默认助手',
   'edit': '编辑',
@@ -36,6 +39,7 @@ export default {
   'sessionSetting': '会话设置',
   'setting': '设置',
   'share': '分享',
+  'tokenDetail': '系统设定: {{systemRoleToken}} 历史消息: {{chatsToken}}',
   'updateAgent': '更新助理信息',
   'updatePrompt': '更新提示词',
 };
diff --git a/src/pages/chat/[id]/Conversation/Input.tsx b/src/pages/chat/[id]/Conversation/Input.tsx
index 84e32e9df663..4c996c7427da 100644
--- a/src/pages/chat/[id]/Conversation/Input.tsx
+++ b/src/pages/chat/[id]/Conversation/Input.tsx
@@ -1,8 +1,9 @@
-import { ActionIcon, ChatInputArea, DraggablePanel, Icon, TokenTag } from '@lobehub/ui';
-import { Button } from 'antd';
+import { ActionIcon, ChatInputArea, DraggablePanel, Icon, TokenTag, Tooltip } from '@lobehub/ui';
+import { Button, Popconfirm } from 'antd';
 import { encode } from 'gpt-tokenizer';
 import { Archive, Eraser, Languages } from 'lucide-react';
 import { memo, useMemo, useState } from 'react';
+import { useTranslation } from 'react-i18next';
 import { shallow } from 'zustand/shallow';
 
 import { CHAT_TEXTAREA_HEIGHT, HEADER_HEIGHT } from '@/const/layoutTokens';
@@ -11,20 +12,24 @@ import { agentSelectors, chatSelectors, useSessionStore } from '@/store/session'
 import { useSettings } from '@/store/settings';
 
 const ChatInput = () => {
+  const { t } = useTranslation();
   const [expand, setExpand] = useState<boolean>(false);
   const [text, setText] = useState('');
-  const textTokenCount = useMemo(() => encode(text).length, [text]);
+  const inputTokenCount = useMemo(() => encode(text).length, [text]);
 
   const [inputHeight] = useSettings((s) => [s.inputHeight], shallow);
-  const [totalToken, model, sendMessage, clearMessage] = useSessionStore(
-    (s) => [
-      chatSelectors.totalTokenCount(s),
-      agentSelectors.currentAgentModel(s),
-      s.createOrSendMsg,
-      s.clearMessage,
-    ],
-    shallow,
-  );
+  const [totalToken, systemRoleToken, chatsToken, model, sendMessage, clearMessage] =
+    useSessionStore(
+      (s) => [
+        chatSelectors.totalTokenCount(s),
+        chatSelectors.systemRoleTokenCount(s),
+        chatSelectors.chatsTokenCount(s),
+        agentSelectors.currentAgentModel(s),
+        s.createOrSendMsg,
+        s.clearMessage,
+      ],
+      shallow,
+    );
 
   return (
     <DraggablePanel
@@ -46,12 +51,16 @@ const ChatInput = () => {
         actions={
           <>
             <ActionIcon icon={Languages} />
-            <ActionIcon icon={Eraser} onClick={clearMessage} />
-            <TokenTag maxValue={ModelTokens[model]} value={totalToken + textTokenCount} />
+            <Popconfirm onConfirm={() => clearMessage()} title={t('confirmClearCurrentMessages')}>
+              <ActionIcon icon={Eraser} title={t('clearCurrentMessages')} />
+            </Popconfirm>
+            <Tooltip title={t('tokenDetail', { chatsToken, systemRoleToken })}>
+              <TokenTag maxValue={ModelTokens[model]} value={totalToken + inputTokenCount} />
+            </Tooltip>
           </>
         }
         expand={expand}
-        footer={<Button icon={<Icon icon={Archive} />} />}
+        footer={<Button icon={<Icon icon={Archive} title={t('archiveCurrentMessages')} />} />}
         minHeight={CHAT_TEXTAREA_HEIGHT}
         onExpandChange={setExpand}
         onInputChange={setText}