diff --git a/src/components/MessageBubble.tsx b/src/components/MessageBubble.tsx
index 51e99de..d023387 100644
--- a/src/components/MessageBubble.tsx
+++ b/src/components/MessageBubble.tsx
@@ -51,10 +51,12 @@ function MessageHide({ chat }: HideMessageProps) {
return (
<>
- {getMessageText(chat).split("\n")[0].slice(0, 28)} ...
+ {getMessageText(chat).trim().slice(0, 28)} ...
- Removed from context
+
+ Removed from context
+
>
);
@@ -73,7 +75,7 @@ function MessageDetail({ chat, renderMarkdown }: MessageDetailProps) {
{chat.content.map((mdt) =>
mdt.type === "text" ? (
chat.hide ? (
- mdt.text?.split("\n")[0].slice(0, 16) + " ..."
+ mdt.text?.trim().slice(0, 16) + " ..."
) : renderMarkdown ? (
{mdt.text}
) : (
diff --git a/src/pages/Chatbox.tsx b/src/pages/Chatbox.tsx
index b49e975..50238d7 100644
--- a/src/pages/Chatbox.tsx
+++ b/src/pages/Chatbox.tsx
@@ -54,7 +54,9 @@ const createMessageFromCurrentBuffer = (
tool_calls: tools.length > 0 ? tools : undefined,
// 补全其他必填字段的默认值(根据你的类型定义)
hide: false,
- token: 0, // 需要实际的token计算逻辑
+ token: calculate_token_length(
+ chunkMessages.join("") + reasoningChunks.join("")
+ ), // 需要实际的token计算逻辑
example: false,
audio: null,
logprobs: null,
@@ -284,7 +286,7 @@ export default function ChatBOX() {
token: data.usage?.completion_tokens_details
? data.usage.completion_tokens -
data.usage.completion_tokens_details.reasoning_tokens
- : (data.usage.completion_tokens ?? calculate_token_length(msg.content)),
+ : data.usage.completion_tokens ?? calculate_token_length(msg.content),
example: false,
audio: null,
logprobs: data.choices[0]?.logprobs,
diff --git a/src/translate/zh_CN.ts b/src/translate/zh_CN.ts
index a1ac7d7..ac446e4 100644
--- a/src/translate/zh_CN.ts
+++ b/src/translate/zh_CN.ts
@@ -141,6 +141,7 @@ const LANG_MAP: Record = {
"Configure image generation settings": "配置图片生成设置",
"New Chat": "新对话",
"Delete Chat": "删除对话",
+ "removed from context": "已从上下文中移除",
};
export default LANG_MAP;