// AI

galaxy/galaxy/Classes/Modules/AI/AI-Dialogue(AI对话)/C/YHAIMainChatViewController.swift

@@ -54,16 +54,7 @@ class YHAIMainChatViewController: YHBaseViewController {
         v.stopSendBlock = {
             [weak self] in
             guard let self = self else { return }
-            self.manager.stopChat(chatId: self.manager.chatId, conversationId: self.conversationId) { success, error in
-                self.bottomInputView.status = self.bottomInputView.textView.text.isEmpty ? .disableSend : .enableSend
-                if success {
-                    self.manager.request?.cancel()
-                    self.chatConfig.disableHandleMessage = true
-                    self.chatConfig.removeThinkingMessageFromChatList(&self.messages)
-                    self.bottomInputView.status = self.bottomInputView.textView.text.isEmpty ? .disableSend : .enableSend
-                    self.tableView.reloadData()
-                }
-            }
+            self.stopAutoResponse()
         }
         return v
     }()
@@ -190,6 +181,20 @@ class YHAIMainChatViewController: YHBaseViewController {
         }
     }
     
+    func stopAutoResponse(completion:((Bool)->())? = nil) {
+        self.manager.stopChat(chatId: self.manager.chatId, conversationId: self.conversationId) { success, error in
+            self.bottomInputView.status = self.bottomInputView.textView.text.isEmpty ? .disableSend : .enableSend
+            if success {
+                self.manager.request?.cancel()
+                self.chatConfig.disableHandleMessage = true
+                self.chatConfig.removeThinkingMessageFromChatList(&self.messages)
+                self.bottomInputView.status = self.bottomInputView.textView.text.isEmpty ? .disableSend : .enableSend
+                self.tableView.reloadData()
+                completion?(success)
+            }
+        }
+    }
+    
     func uploadEvaluationMessage(_ msg: YHAIChatMessage, callback:((Bool)->())? = nil) {
         viewModel.createMessage(conversationId: conversationId, role: "assistant", msg: msg) {
             [weak self] success, error in
@@ -238,8 +243,11 @@ extension YHAIMainChatViewController: UITableViewDelegate, UITableViewDataSource
                 cell.messageClick = {
                     [weak self] text in
                     guard let self = self else { return }
-                    self.sendMessage(text)
-                }
+                    self.stopAutoResponse { success in
+                        if success {
+                            self.sendMessage(text)
+                        }
+                    }                }
                 return cell
                 
             } else if msgType == .thinking {

galaxy/galaxy/Classes/Modules/AI/AI-Dialogue(AI对话)/C/YHAIRobotChatViewController.swift

@@ -60,15 +60,7 @@ class YHAIRobotChatViewController: YHBaseViewController {
         v.stopSendBlock = {
             [weak self] in
             guard let self = self else { return }
-            self.manager.stopChat(chatId: self.manager.chatId, conversationId: self.conversationId) { success, error in
-                self.bottomInputView.status = self.bottomInputView.textView.text.isEmpty ? .disableSend : .enableSend
-                if success {
-                    self.chatConfig.disableHandleMessage = true
-                    self.chatConfig.removeThinkingMessageFromChatList(&self.messages)
-                    self.bottomInputView.status = self.bottomInputView.textView.text.isEmpty ? .disableSend : .enableSend
-                    self.tableView.reloadData()
-                }
-            }
+            self.stopAutoResponse()
         }
         return v
     }()
@@ -220,6 +212,20 @@ class YHAIRobotChatViewController: YHBaseViewController {
         }
     }
     
+    func stopAutoResponse(completion:((Bool)->())? = nil) {
+        self.manager.stopChat(chatId: self.manager.chatId, conversationId: self.conversationId) { success, error in
+            self.bottomInputView.status = self.bottomInputView.textView.text.isEmpty ? .disableSend : .enableSend
+            if success {
+                self.manager.request?.cancel()
+                self.chatConfig.disableHandleMessage = true
+                self.chatConfig.removeThinkingMessageFromChatList(&self.messages)
+                self.bottomInputView.status = self.bottomInputView.textView.text.isEmpty ? .disableSend : .enableSend
+                self.tableView.reloadData()
+                completion?(success)
+            }
+        }
+    }
+    
     @objc func didCleanButtonClicked() {
         
         YHCommonAlertView.show("删除历史记录", "删除后记录无法恢复", "取消", "删除", fullGuestureEnable: false) {
@@ -295,7 +301,11 @@ extension YHAIRobotChatViewController: UITableViewDelegate, UITableViewDataSourc
                 cell.messageClick = {
                     [weak self] text in
                     guard let self = self else { return }
-                    self.sendMessage(text)
+                    self.stopAutoResponse { success in
+                        if success {
+                            self.sendMessage(text)
+                        }
+                    }
                 }
                 return cell
                 
@@ -395,7 +405,11 @@ extension YHAIRobotChatViewController: UITableViewDelegate, UITableViewDataSourc
         view.selectFlowMsgBlock = {
             [weak self] text in
             guard let self = self else { return }
-            self.sendMessage(text)
+            self.stopAutoResponse { success in
+                if success {
+                    self.sendMessage(text)
+                }
+            }
         }
         view.selectBannerItemBlock = {
             [weak self] model in
@@ -408,7 +422,11 @@ extension YHAIRobotChatViewController: UITableViewDelegate, UITableViewDataSourc
                 text = model.desc
             }
 
-            self.sendMessage(text)
+            self.stopAutoResponse { success in
+                if success {
+                    self.sendMessage(text)
+                }
+            }
         }
         return view
     }