处理语音

keyBoard/Class/AiTalk/V/Chat/KBPersonaChatCell.m

@@ -503,6 +503,10 @@ static NSString * const KBChatSessionDidResetNotification = @"KBChatSessionDidRe
     }
     
     [self ensureOpeningMessageAtTop];
+    
+    // 查找并移除 loading 消息
+    [self removeLoadingAssistantMessage];
+    
     KBAiChatMessage *message = [KBAiChatMessage assistantMessageWithText:text
                                                                  audioId:audioId];
     message.needsTypewriterEffect = YES;
@@ -510,6 +514,33 @@ static NSString * const KBChatSessionDidResetNotification = @"KBChatSessionDidRe
     [self.chatView addMessage:message autoScroll:YES];
 }
 
+/// 添加 loading AI 消息
+- (void)appendLoadingAssistantMessage {
+    if (!self.messages) {
+        self.messages = [NSMutableArray array];
+    }
+    
+    [self ensureOpeningMessageAtTop];
+    KBAiChatMessage *message = [KBAiChatMessage loadingAssistantMessage];
+    [self.messages addObject:message];
+    [self.chatView addMessage:message autoScroll:YES];
+}
+
+/// 移除 loading AI 消息
+- (void)removeLoadingAssistantMessage {
+    // 从数据源中移除
+    for (NSInteger i = self.messages.count - 1; i >= 0; i--) {
+        KBAiChatMessage *message = self.messages[i];
+        if (message.type == KBAiChatMessageTypeAssistant && message.isLoading) {
+            [self.messages removeObjectAtIndex:i];
+            break;
+        }
+    }
+    
+    // 从 chatView 中移除
+    [self.chatView removeLoadingAssistantMessage];
+}
+
 - (void)updateChatViewBottomInset:(CGFloat)bottomInset {
     [self.chatView updateContentBottomInset:bottomInset];
 }