From 9a466694dedbe03d4456fe619c14e8217ff6d2c2 Mon Sep 17 00:00:00 2001
From: ShineKOT <1917095344@qq.com>
Date: Thu, 3 Apr 2025 11:54:21 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8DAI=E5=B7=A5=E5=85=B7?=
 =?UTF-8?q?=E6=A0=8F=E6=98=BE=E7=A4=BA=E5=92=8C=E7=82=B9=E5=87=BB=E5=BC=82?=
 =?UTF-8?q?=E5=B8=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/chat-container/chat-container.tsx |  4 +++-
 src/components/chat-toolbar/chat-toolbar.tsx     | 14 +++++++++++---
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/components/chat-container/chat-container.tsx b/src/components/chat-container/chat-container.tsx
index b70b58e..91b8829 100644
--- a/src/components/chat-container/chat-container.tsx
+++ b/src/components/chat-container/chat-container.tsx
@@ -580,8 +580,9 @@ export class ChatContainer extends Component<
                     />
                   </div>
                   <ChatToolbar
-                    data={this.props.aiTopic.activedTopic.value}
                     type='footer'
+                    mode={this.props.mode}
+                    data={this.props.aiTopic.activedTopic.value}
                     className={`${this.ns.e('toolbar')} ${this.ns.is(
                       'has-materials',
                       this.props.aiChat.materials.value.length > 0,
@@ -607,6 +608,7 @@ export class ChatContainer extends Component<
                 </div>
                 <ChatToolbar
                   type='footer'
+                  mode={this.props.mode}
                   data={this.props.aiTopic.activedTopic.value}
                   className={`${this.ns.e('toolbar')} ${this.ns.is(
                     'has-materials',
diff --git a/src/components/chat-toolbar/chat-toolbar.tsx b/src/components/chat-toolbar/chat-toolbar.tsx
index 7b85d50..92773ac 100644
--- a/src/components/chat-toolbar/chat-toolbar.tsx
+++ b/src/components/chat-toolbar/chat-toolbar.tsx
@@ -57,12 +57,20 @@ export interface ChatToolbarProps {
    * @memberof ChatToolbarProps
    */
   className?: string;
+
+  /**
+   * AI模式
+   *
+   * @type {('DEFAULT' | 'TOPIC')}
+   * @memberof ChatToolbarProps
+   */
+  mode: 'DEFAULT' | 'TOPIC';
 }
 
 const ns = new Namespace('chat-toolbar');
 
 export const ChatToolbar = (props: ChatToolbarProps) => {
-  const { controller, items = [], data, type, className } = props;
+  const { controller, items = [], data, type, className, mode } = props;
   const containerContext = useContext(ContainerContext);
 
   let toolbarItems: IChatToolbarItem[] = [];
@@ -95,7 +103,7 @@ export const ChatToolbar = (props: ChatToolbarProps) => {
         {
           label: '新建对话',
           title: '新建对话',
-          hidden: props.controller.opts.mode !== 'TOPIC',
+          hidden: mode !== 'TOPIC',
           icon: () => {
             return <NewDialogueSvg />;
           },
@@ -209,9 +217,9 @@ export const ChatToolbar = (props: ChatToolbarProps) => {
       // 计算真实文本
       tempData.msg.realcontent = data.realcontent;
     } else {
+      if (!tempData.data) tempData.data = {};
       Object.assign(tempData.data, { messages: controller.messages.value });
     }
-
     if (item.onClick && typeof item.onClick === 'function') {
       item.onClick(e, item, controller.context, controller.params, tempData);
     } else {
-- 
Gitee