From 87c04408deebc1dfec68eff3e970a7ba3ad86363 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Mon, 22 Jul 2024 14:50:40 +0800
Subject: [PATCH] feat: add contextCount to inputbar

---
 src/renderer/src/i18n/index.ts                      | 2 +-
 src/renderer/src/pages/home/components/Inputbar.tsx | 3 +++
 src/renderer/src/pages/home/components/Message.tsx  | 2 +-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/i18n/index.ts b/src/renderer/src/i18n/index.ts
index 7e12a30b..e837e9c4 100644
--- a/src/renderer/src/i18n/index.ts
+++ b/src/renderer/src/i18n/index.ts
@@ -67,7 +67,7 @@ const resources = {
         'input.send': 'Send',
         'input.pause': 'Pause',
         'input.settings': 'Settings',
-        'input.estimated_tokens': 'Estimated tokens: ',
+        'input.estimated_tokens': 'Estimated Tokens: ',
         'settings.temperature': 'Temperature',
         'settings.temperature.tip':
           'Lower values make the model more creative and unpredictable, while higher values make it more deterministic and precise.',
diff --git a/src/renderer/src/pages/home/components/Inputbar.tsx b/src/renderer/src/pages/home/components/Inputbar.tsx
index a5268cbf..709b596a 100644
--- a/src/renderer/src/pages/home/components/Inputbar.tsx
+++ b/src/renderer/src/pages/home/components/Inputbar.tsx
@@ -26,6 +26,7 @@ import { useTranslation } from 'react-i18next'
 import styled from 'styled-components'
 import AssistantSettings from './AssistantSettings'
 import SendMessageSetting from './SendMessageSetting'
+import { DEFAULT_CONEXTCOUNT } from '@renderer/config/constant'
 
 interface Props {
   assistant: Assistant
@@ -193,6 +194,7 @@ const Inputbar: FC<Props> = ({ assistant, setActiveTopic }) => {
         styles={{ textarea: { paddingLeft: 0 } }}
       />
       <TextCount>
+        <HistoryOutlined /> {assistant?.settings?.contextCount ?? DEFAULT_CONEXTCOUNT} |{' '}
         {t('assistant.input.estimated_tokens')}: {`${inputTokenCount}/${estimateTokenCount}`}
       </TextCount>
     </Container>
@@ -257,6 +259,7 @@ const TextCount = styled.div`
   bottom: 8px;
   font-size: 11px;
   color: var(--color-text-3);
+  z-index: 10;
 `
 
 export default Inputbar
diff --git a/src/renderer/src/pages/home/components/Message.tsx b/src/renderer/src/pages/home/components/Message.tsx
index 39d4cd1b..e9fbca15 100644
--- a/src/renderer/src/pages/home/components/Message.tsx
+++ b/src/renderer/src/pages/home/components/Message.tsx
@@ -105,7 +105,7 @@ const MessageItem: FC<Props> = ({ message, index, showMenu, onDeleteMessage }) =
             {message.usage && (
               <>
                 <MessageMetadata>
-                  Tokens: {message.usage.total_tokens} (IN:{message.usage.prompt_tokens}/OUT:
+                  tokens: {message.usage.total_tokens} (in:{message.usage.prompt_tokens}/out:
                   {message.usage.completion_tokens})
                 </MessageMetadata>
               </>