From ef45bedf83e04aec59ad4373c060dfcc403bea31 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B6=85=E7=BA=A7=E8=80=81=E7=99=BD=E5=85=94?=
 <fsmecx@gmail.com>
Date: Wed, 22 Oct 2025 17:05:35 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9EAI=E5=AF=B9=E8=AF=9D=E5=8A=A0?=
 =?UTF-8?q?=E8=BD=BD=E7=8A=B6=E6=80=81=E7=AE=A1=E7=90=86=EF=BC=8C=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96=E6=B6=88=E6=81=AF=E8=BE=93=E5=85=A5=E7=BB=84=E4=BB=B6?=
 =?UTF-8?q?=E4=BB=A5=E6=94=AF=E6=8C=81=E5=8A=A0=E8=BD=BD=E6=8C=87=E7=A4=BA?=
 =?UTF-8?q?=EF=BC=8C=E6=8F=90=E5=8D=87=E7=94=A8=E6=88=B7=E4=BD=93=E9=AA=8C?=
 =?UTF-8?q?=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../components/MessageEnter/index.tsx         |  60 ++++++----
 .../src/store/module/weChat/weChat.data.ts    |   2 +
 Touchkebao/src/store/module/weChat/weChat.ts  | 108 ++++++++++--------
 3 files changed, 102 insertions(+), 68 deletions(-)

diff --git a/Touchkebao/src/pages/pc/ckbox/weChat/components/ChatWindow/components/MessageEnter/index.tsx b/Touchkebao/src/pages/pc/ckbox/weChat/components/ChatWindow/components/MessageEnter/index.tsx
index 16f6c179..f49ceccb 100644
--- a/Touchkebao/src/pages/pc/ckbox/weChat/components/ChatWindow/components/MessageEnter/index.tsx
+++ b/Touchkebao/src/pages/pc/ckbox/weChat/components/ChatWindow/components/MessageEnter/index.tsx
@@ -1,5 +1,5 @@
 import React, { useEffect, useState } from "react";
-import { Layout, Input, Button, Modal } from "antd";
+import { Layout, Input, Button, Modal, Spin } from "antd";
 import {
   SendOutlined,
   FolderOutlined,
@@ -7,6 +7,7 @@ import {
   ExportOutlined,
   CloseOutlined,
   MessageOutlined,
+  LoadingOutlined,
 } from "@ant-design/icons";
 import { ContractData, weChatGroup } from "@/pages/pc/ckbox/data";
 import { useWebSocketStore } from "@/store/module/websocket/websocket";
@@ -45,7 +46,7 @@ const MessageEnter: React.FC<MessageEnterProps> = ({ contract }) => {
   const quoteMessageContent = useWeChatStore(
     state => state.quoteMessageContent,
   );
-
+  const isLoadingAiChat = useWeChatStore(state => state.isLoadingAiChat);
   useEffect(() => {
     if (quoteMessageContent) {
       setInputValue(quoteMessageContent);
@@ -224,29 +225,42 @@ const MessageEnter: React.FC<MessageEnterProps> = ({ contract }) => {
                 </div>
               </div>
             </div>
-            <div className={styles.inputArea}>
-              <div className={styles.inputWrapper}>
-                <TextArea
-                  value={inputValue}
-                  onChange={e => setInputValue(e.target.value)}
-                  onKeyDown={handleKeyPress}
-                  placeholder="输入消息..."
-                  className={styles.messageInput}
-                  autoSize={{ minRows: 2, maxRows: 6 }}
-                />
-                <div className={styles.sendButtonArea}>
-                  <Button
-                    type="primary"
-                    icon={<SendOutlined />}
-                    onClick={handleSend}
-                    disabled={!inputValue.trim()}
-                    className={styles.sendButton}
-                  >
-                    发送
-                  </Button>
+            {isLoadingAiChat ? (
+              <Spin
+                indicator={<LoadingOutlined spin />}
+                spinning={isLoadingAiChat}
+              >
+                <div style={{ height: "100px" }}>
+                  <div>Ai思考中...请稍后</div>
+                </div>
+              </Spin>
+            ) : (
+              <div className={styles.inputArea}>
+                <div className={styles.inputWrapper}>
+                  <TextArea
+                    value={inputValue}
+                    onChange={e => setInputValue(e.target.value)}
+                    onKeyDown={handleKeyPress}
+                    placeholder="输入消息..."
+                    className={styles.messageInput}
+                    autoSize={{ minRows: 2, maxRows: 6 }}
+                  />
+
+                  <div className={styles.sendButtonArea}>
+                    <Button
+                      type="primary"
+                      icon={<SendOutlined />}
+                      onClick={handleSend}
+                      disabled={!inputValue.trim()}
+                      className={styles.sendButton}
+                    >
+                      发送
+                    </Button>
+                  </div>
                 </div>
               </div>
-            </div>
+            )}
+
             <div className={styles.inputHint}>
               按下Ctrl+Enter换行，Enter发送
             </div>
diff --git a/Touchkebao/src/store/module/weChat/weChat.data.ts b/Touchkebao/src/store/module/weChat/weChat.data.ts
index 29a1f2d6..b3f1bf2c 100644
--- a/Touchkebao/src/store/module/weChat/weChat.data.ts
+++ b/Touchkebao/src/store/module/weChat/weChat.data.ts
@@ -14,6 +14,8 @@ export interface WeChatState {
   updateShowChatRecordModel: (show: boolean) => void;
   aiQuoteMessageContent: number;
   updateAiQuoteMessageContent: (message: number) => void;
+  isLoadingAiChat: boolean;
+  updateIsLoadingAiChat: (loading: boolean) => void;
   quoteMessageContent: string;
   updateQuoteMessageContent: (value: string) => void;
   // ==================== Transmit Module =========Start===========
diff --git a/Touchkebao/src/store/module/weChat/weChat.ts b/Touchkebao/src/store/module/weChat/weChat.ts
index 5068f3c0..c1746dfc 100644
--- a/Touchkebao/src/store/module/weChat/weChat.ts
+++ b/Touchkebao/src/store/module/weChat/weChat.ts
@@ -46,6 +46,11 @@ export const useWeChatStore = create<WeChatState>()(
       updateQuoteMessageContent: (message: string) => {
         set({ quoteMessageContent: message });
       },
+      //正在加载ai对话
+      isLoadingAiChat: false,
+      updateIsLoadingAiChat: (loading: boolean) => {
+        set({ isLoadingAiChat: loading });
+      },
       // ==================== Transmit Module =========Start===========
       /** 选中的聊天记录列表 */
       selectedChatRecords: [],
@@ -307,60 +312,73 @@ export const useWeChatStore = create<WeChatState>()(
           message?.wechatChatroomId || message.wechatFriendId;
         const isWechatGroup = message?.wechatChatroomId;
 
-        // 如果是当前选中的聊天，直接添加到消息列表
-        if (currentContract && currentContract.id == getMessageId) {
-          set(state => ({
-            currentMessages: [...state.currentMessages, message],
-          }));
-          //把数据传到存客宝
-          const params: any = {
-            type: "CmdNewMessage",
-            wechatAccountId: currentContract.wechatAccountId,
-          };
-          if (isWechatGroup) {
-            params.chatroomMessage = [message];
-          } else {
-            params.friendMessage = [message];
-          }
-          const dataProcessingResult = await dataProcessing(params);
-          //如果成功，就请求ai对话接口
-          if (dataProcessingResult) {
-            const messageContent = await aiChat({
-              friendId: getMessageId,
+        try {
+          // 如果是当前选中的聊天，直接添加到消息列表
+          if (currentContract && currentContract.id == getMessageId) {
+            set(state => ({
+              currentMessages: [...state.currentMessages, message],
+            }));
+            //把数据传到存客宝
+            const params: any = {
+              type: "CmdNewMessage",
               wechatAccountId: currentContract.wechatAccountId,
-              message: message,
-            });
-            console.log("ai对话接口返回", messageContent);
-          }
-        } else {
-          // 更新其他聊天的未读消息数
-          const chatSessions = useCkChatStore.getState().chatSessions;
-          const session = chatSessions.find(item => item.id == getMessageId);
-          if (session) {
-            session.config.unreadCount = Number(session.config.unreadCount) + 1;
-            updateChatSession(session);
-            // 将接收到新消息的会话置顶到列表顶部
-            pinChatSessionToTop(getMessageId);
-          } else {
-            // 如果会话不存在，创建新会话
+            };
             if (isWechatGroup) {
-              const [group] = await weChatGroupService.findByIds(getMessageId);
-              if (group) {
+              params.chatroomMessage = [message];
+            } else {
+              params.friendMessage = [message];
+            }
+            const dataProcessingResult = await dataProcessing(params);
+            //如果成功，就请求ai对话接口
+            set(() => ({
+              isLoadingAiChat: true,
+            }));
+            if (!dataProcessingResult) {
+              const messageContent = await aiChat({
+                friendId: getMessageId,
+                wechatAccountId: currentContract.wechatAccountId,
+                message: message,
+              });
+
+              set(() => ({
+                quoteMessageContent: messageContent,
+                isLoadingAiChat: false,
+              }));
+            }
+          } else {
+            // 更新其他聊天的未读消息数
+            const chatSessions = useCkChatStore.getState().chatSessions;
+            const session = chatSessions.find(item => item.id == getMessageId);
+            if (session) {
+              session.config.unreadCount =
+                Number(session.config.unreadCount) + 1;
+              updateChatSession(session);
+              // 将接收到新消息的会话置顶到列表顶部
+              pinChatSessionToTop(getMessageId);
+            } else {
+              // 如果会话不存在，创建新会话
+              if (isWechatGroup) {
+                const [group] =
+                  await weChatGroupService.findByIds(getMessageId);
+                if (group) {
+                  addChatSession({
+                    ...group,
+                    config: { unreadCount: 1 },
+                  });
+                  // 新创建的会话会自动添加到列表顶部，无需额外置顶
+                }
+              } else {
+                const [user] = await contractService.findByIds(getMessageId);
                 addChatSession({
-                  ...group,
+                  ...user,
                   config: { unreadCount: 1 },
                 });
                 // 新创建的会话会自动添加到列表顶部，无需额外置顶
               }
-            } else {
-              const [user] = await contractService.findByIds(getMessageId);
-              addChatSession({
-                ...user,
-                config: { unreadCount: 1 },
-              });
-              // 新创建的会话会自动添加到列表顶部，无需额外置顶
             }
           }
+        } catch (error) {
+          console.error("接收新消息失败:", error);
         }
       },