feat: 修复 AI Messages 的空白问题

Author: KonghaYao

Cargo.lock

@@ -0,0 +1,11 @@
+[target.x86_64-unknown-linux-gnu]
+pre-build = [
+    "apt-get update",
+    "apt-get install -y libxcb1-dev libxkbcommon-dev libxkbcommon-x11-dev libwayland-dev libssl-dev pkg-config",
+]
+
+[target.aarch64-unknown-linux-gnu]
+pre-build = [
+    "apt-get update",
+    "apt-get install -y libxcb1-dev libxkbcommon-dev libxkbcommon-x11-dev libwayland-dev libssl-dev pkg-config",
+]

Cross.toml

@@ -4,6 +4,10 @@ version = "0.1.0"
 edition = "2021"
 description = "TUI interface for Rust Agent - interactive terminal playground"
 
+[lib]
+name = "rust_agent_tui"
+path = "src/lib.rs"
+
 [[bin]]
 name = "agent-tui"
 path = "src/main.rs"

rust-agent-tui/Cargo.toml

@@ -256,15 +256,26 @@ impl App {
                         let _ = self.render_tx.send(RenderEvent::UpdateLastMessage(vm));
                     }
                 } else {
-                    match self.view_messages.last_mut() {
-                        Some(m) if m.is_assistant() => m.append_chunk(&chunk),
-                        _ => {
-                            let vm = MessageViewModel::assistant();
-                            self.view_messages.push(vm.clone());
-                            let _ = self.render_tx.send(RenderEvent::AddMessage(vm));
+                    // 如果 chunk 为空且没有现有的 assistant bubble，跳过创建空的 bubble
+                    // 避免 AI 只发起工具调用时显示空白消息
+                    if chunk.is_empty() {
+                        match self.view_messages.last_mut() {
+                            Some(m) if m.is_assistant() => m.append_chunk(&chunk),
+                            _ => {
+                                // 没有现有的 assistant bubble，chunk 为空，不创建新的空 bubble
+                            }
+                        }
+                    } else {
+                        match self.view_messages.last_mut() {
+                            Some(m) if m.is_assistant() => m.append_chunk(&chunk),
+                            _ => {
+                                let vm = MessageViewModel::assistant();
+                                self.view_messages.push(vm.clone());
+                                let _ = self.render_tx.send(RenderEvent::AddMessage(vm));
+                            }
                         }
+                        let _ = self.render_tx.send(RenderEvent::AppendChunk(chunk));
                     }
-                    let _ = self.render_tx.send(RenderEvent::AppendChunk(chunk));
                 }
                 (true, false, false)
             }

rust-agent-tui/src/app/agent_ops.rs

@@ -0,0 +1,42 @@
+//! TUI interface for Rust Agent - interactive terminal playground
+
+pub mod app;
+pub mod command;
+pub mod config;
+pub mod event;
+pub mod langfuse;
+pub mod prompt;
+pub mod relay_adapter;
+pub mod thread;
+pub mod ui;
+
+/// CLI 参数解析结果：--remote-control [url] [--relay-token <token>] [--relay-name <name>]
+/// url 为空字符串表示 `--remote-control` 无参数模式（从配置读取）
+pub struct RelayCli {
+    pub url: String,
+    pub token: Option<String>,
+    pub name: Option<String>,
+}
+
+pub fn parse_relay_args(args: &[String]) -> Option<RelayCli> {
+    // 查找 --remote-control 参数位置
+    let remote_idx = args.iter().position(|a| a == "--remote-control")?;
+
+    // 检查是否有值（即 --remote-control <url> 格式）
+    // 有值条件：下一个参数存在且不以 -- 开头
+    let url = if remote_idx + 1 < args.len() && !args[remote_idx + 1].starts_with("--") {
+        args[remote_idx + 1].clone()
+    } else {
+        // --remote-control 无参数，返回空字符串标记"从配置读取"
+        String::new()
+    };
+
+    let token = args.windows(2)
+        .find(|w| w[0] == "--relay-token")
+        .map(|w| w[1].clone());
+    let name = args.windows(2)
+        .find(|w| w[0] == "--relay-name")
+        .map(|w| w[1].clone());
+
+    Some(RelayCli { url, token, name })
+}

rust-agent-tui/src/lib.rs

@@ -1,13 +1,3 @@
-mod app;
-mod command;
-mod config;
-mod event;
-mod langfuse;
-mod prompt;
-mod relay_adapter;
-mod thread;
-mod ui;
-
 use anyhow::Result;
 use ratatui::{
     crossterm::{
@@ -19,6 +9,11 @@ use ratatui::{
 };
 use std::io;
 
+use rust_agent_tui::app::App;
+use rust_agent_tui::event;
+use rust_agent_tui::ui;
+use rust_agent_tui::{parse_relay_args, RelayCli};
+
 fn main() -> Result<()> {
     // 加载 .env 文件（仅开发环境，文件不存在时静默忽略）
     let _ = dotenvy::dotenv();
@@ -71,39 +66,8 @@ fn main() -> Result<()> {
     Ok(())
 }
 
-/// CLI 参数解析结果：--remote-control [url] [--relay-token <token>] [--relay-name <name>]
-/// url 为空字符串表示 `--remote-control` 无参数模式（从配置读取）
-pub struct RelayCli {
-    pub url: String,
-    pub token: Option<String>,
-    pub name: Option<String>,
-}
-
-fn parse_relay_args(args: &[String]) -> Option<RelayCli> {
-    // 查找 --remote-control 参数位置
-    let remote_idx = args.iter().position(|a| a == "--remote-control")?;
-
-    // 检查是否有值（即 --remote-control <url> 格式）
-    // 有值条件：下一个参数存在且不以 -- 开头
-    let url = if remote_idx + 1 < args.len() && !args[remote_idx + 1].starts_with("--") {
-        args[remote_idx + 1].clone()
-    } else {
-        // --remote-control 无参数，返回空字符串标记"从配置读取"
-        String::new()
-    };
-
-    let token = args.windows(2)
-        .find(|w| w[0] == "--relay-token")
-        .map(|w| w[1].clone());
-    let name = args.windows(2)
-        .find(|w| w[0] == "--relay-name")
-        .map(|w| w[1].clone());
-
-    Some(RelayCli { url, token, name })
-}
-
 async fn run_app(terminal: &mut Terminal<CrosstermBackend<io::Stdout>>, relay_cli: Option<RelayCli>) -> Result<()> {
-    let mut app = app::App::new();
+    let mut app = App::new();
 
     // 尝试连接 Relay Server（CLI 参数优先，其次读 settings.json）
     app.try_connect_relay(relay_cli.as_ref()).await;

rust-agent-tui/src/main.rs

@@ -493,4 +493,85 @@ mod tests {
         let has_tool_call_text = snap.iter().any(|l| l.contains("bash") || l.contains("Bash"));
         assert!(has_tool_call_text, "ToolCall 创建的 ToolBlock 应在快照中可见，但实际内容为:\n{}", snap.join("\n"));
     }
+
+    #[tokio::test]
+    async fn test_empty_assistant_chunk_no_bubble() {
+        // 空 AssistantChunk 不应创建空白的 AssistantBubble
+        let (mut app, _handle) = App::new_headless(120, 30);
+
+        // 发送空 chunk，不应创建 AssistantBubble
+        app.push_agent_event(AgentEvent::AssistantChunk("".into()));
+        app.process_pending_events();
+
+        // view_messages 应为空（没有创建空白气泡）
+        assert!(
+            app.view_messages.is_empty(),
+            "空 AssistantChunk 不应创建 AssistantBubble，实际: {:?}",
+            app.view_messages.len()
+        );
+
+        // 发送多个空 chunk，仍不应创建气泡
+        app.push_agent_event(AgentEvent::AssistantChunk("".into()));
+        app.push_agent_event(AgentEvent::AssistantChunk("".into()));
+        app.process_pending_events();
+
+        assert!(
+            app.view_messages.is_empty(),
+            "多个空 AssistantChunk 仍不应创建 AssistantBubble"
+        );
+    }
+
+    #[tokio::test]
+    async fn test_empty_then_nonempty_assistant_chunk() {
+        // 空_chunk → 非空_chunk：非空 chunk 应正常创建气泡
+        let (mut app, mut handle) = App::new_headless(120, 30);
+
+        // 先发送空 chunk
+        app.push_agent_event(AgentEvent::AssistantChunk("".into()));
+        app.process_pending_events();
+
+        // 再发送非空 chunk
+        let notify = Arc::clone(&handle.render_notify);
+        let n1 = notify.notified();
+        let n2 = notify.notified();
+        app.push_agent_event(AgentEvent::AssistantChunk("Hello".into()));
+        app.push_agent_event(AgentEvent::Done);
+        app.process_pending_events();
+        tokio::join!(n1, n2);
+
+        handle.terminal.draw(|f| main_ui::render(f, &mut app)).unwrap();
+
+        // 应该只有 1 个 AssistantBubble，内容为 "Hello"
+        assert_eq!(app.view_messages.len(), 1, "应只有 1 条消息");
+        assert!(app.view_messages[0].is_assistant(), "应为 AssistantBubble");
+        assert!(handle.contains("Hello"), "应显示 Hello 内容");
+    }
+
+    #[tokio::test]
+    async fn test_tool_call_without_assistant_chunk_no_bubble() {
+        // 模拟 AI 只调用工具不输出文本的场景
+        let (mut app, mut handle) = App::new_headless(120, 30);
+
+        // 直接发送 ToolCall 事件（无 AssistantChunk）
+        let notified = handle.render_notify.notified();
+        app.push_agent_event(AgentEvent::ToolCall {
+            tool_call_id: "tc1".into(),
+            name: "bash".into(),
+            display: "Bash".into(),
+            args: Some("ls".into()),
+            is_error: false,
+        });
+        app.process_pending_events();
+        notified.await;
+
+        handle.terminal.draw(|f| main_ui::render(f, &mut app)).unwrap();
+
+        // 应该有 1 个 ToolBlock，不应有空白 AssistantBubble
+        assert_eq!(app.view_messages.len(), 1, "应有 1 条消息（ToolBlock）");
+        // 确保不是 AssistantBubble（空白气泡）
+        assert!(
+            !app.view_messages[0].is_assistant(),
+            "不应创建 AssistantBubble，应为 ToolBlock"
+        );
+    }
 }