comate-cli 0.7.0a1__tar.gz → 0.7.0a3__tar.gz

{comate_cli-0.7.0a1 → comate_cli-0.7.0a3}/CHANGELOG.md

@@ -17,6 +17,6 @@
 - Resume picker 重做：搜索、跨 cwd 切换、预览模式、视觉抛光
 - Auto compact 期间显式提示 `Compacting context`
 - Team / subagent 事件透出与渲染统一
-- 与 SDK 0.8.0a1 配套的 tool envelope / token 账本展示
+- 与 SDK 0.8.0a2 配套的 tool envelope / token 账本展示
 
 详细变更请参考 git log。

{comate_cli-0.7.0a1 → comate_cli-0.7.0a3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comate-cli
-Version: 0.7.0a1
+Version: 0.7.0a3
 Summary: Comate terminal CLI built on comate-agent-sdk
 Project-URL: Homepage, https://github.com/AndyLee1024/agent-sdk
 Project-URL: Repository, https://github.com/AndyLee1024/agent-sdk

{comate_cli-0.7.0a1 → comate_cli-0.7.0a3}/comate_cli/terminal_agent/error_display.py

@@ -30,7 +30,7 @@ def format_error(exc: Exception) -> tuple[str, str, str]:
                     "Context limit exceeded",
                     "error",
                 )
-            return f"API error: {_truncate(exc_msg, 80)}", "API error", "error"
+            return f"API error: {exc_msg}", "API error", "error"
 
         if code == 401:
             return "Invalid or expired API key", "Auth failed", "error"
@@ -40,7 +40,7 @@ def format_error(exc: Exception) -> tuple[str, str, str]:
             return "Model not found or invalid API path", "Model not found", "error"
         if code and code >= 500:
             return f"Server error ({code})", "Server error", "warning"
-        return f"API error: {_truncate(exc_msg, 80)}", "API error", "error"
+        return f"API error: {exc_msg}", "API error", "error"
 
     # Session errors
     if exc_type == "ChatSessionClosedError":
@@ -54,7 +54,7 @@ def format_error(exc: Exception) -> tuple[str, str, str]:
         return "Connection failed", "Connection failed", "error"
 
     # Generic fallback
-    return f"Error: {_truncate(exc_msg, 80)}", "Error occurred", "error"
+    return f"Error: {exc_msg}", "Error occurred", "error"
 
 
 def _parse_context_size_error(msg: str) -> tuple[str, str] | None:
@@ -77,7 +77,3 @@ def _parse_context_size_error(msg: str) -> tuple[str, str] | None:
         return None  # Type matches but can't parse numbers — fall through to generic 400
 
     return None
-
-
-def _truncate(s: str, max_len: int) -> str:
-    return s[:max_len] + "..." if len(s) > max_len else s

{comate_cli-0.7.0a1 → comate_cli-0.7.0a3}/comate_cli/terminal_agent/logging_adapter.py

@@ -113,11 +113,6 @@ class TUILoggingHandler(logging.Handler):
             parts = msg.split("timeout_ms=")
             msg = parts[0].rstrip(": ,")
 
-        # 限制长度
-        max_len = 100
-        if len(msg) > max_len:
-            msg = msg[:max_len] + "..."
-
         return msg
 
     def _get_message_key(self, record: logging.LogRecord) -> str:

{comate_cli-0.7.0a1 → comate_cli-0.7.0a3}/comate_cli/terminal_agent/mention_completer.py

@@ -3,8 +3,9 @@ from __future__ import annotations
 import os
 import re
 import subprocess
+import threading
 import time
-from collections.abc import Iterable
+from collections.abc import Callable, Iterable
 from dataclasses import dataclass
 from pathlib import Path, PurePosixPath
 
@@ -112,6 +113,9 @@ class LocalFileMentionCompleter(Completer):
 
         self._deep_cache_time: float = 0.0
         self._deep_cached_paths: list[str] = []
+        self._deep_cache_lock = threading.RLock()
+        self._deep_warmup_lock = threading.Lock()
+        self._deep_warmup_thread: threading.Thread | None = None
         self._directory_cached_paths: dict[str, tuple[float, list[str]]] = {}
 
     @classmethod
@@ -300,16 +304,68 @@ class LocalFileMentionCompleter(Completer):
         return self._top_cached_paths
 
     def _get_deep_paths(self) -> list[str]:
-        now = time.monotonic()
-        if now - self._deep_cache_time <= self._refresh_interval:
-            return self._deep_cached_paths
+        with self._deep_cache_lock:
+            now = time.monotonic()
+            if now - self._deep_cache_time <= self._refresh_interval:
+                return self._deep_cached_paths
 
-        file_paths = self._load_indexed_file_paths()
-        paths = self._build_deep_paths(file_paths)
+            file_paths = self._load_indexed_file_paths()
+            paths = self._build_deep_paths(file_paths)
 
-        self._deep_cached_paths = paths
-        self._deep_cache_time = now
-        return self._deep_cached_paths
+            self._deep_cached_paths = paths
+            self._deep_cache_time = time.monotonic()
+            return self._deep_cached_paths
+
+    def start_deep_cache_warmup(
+        self,
+        *,
+        on_complete: Callable[[], None] | None = None,
+    ) -> bool:
+        """Build the deep path cache in a daemon thread if it is stale."""
+        with self._deep_warmup_lock:
+            if (
+                self._deep_warmup_thread is not None
+                and self._deep_warmup_thread.is_alive()
+            ):
+                return False
+
+        with self._deep_cache_lock:
+            now = time.monotonic()
+            if now - self._deep_cache_time <= self._refresh_interval:
+                return False
+
+        def _warm() -> None:
+            try:
+                self._get_deep_paths()
+            except Exception:
+                logger.debug("deep mention cache warmup failed", exc_info=True)
+            finally:
+                with self._deep_warmup_lock:
+                    if self._deep_warmup_thread is threading.current_thread():
+                        self._deep_warmup_thread = None
+                if on_complete is not None:
+                    try:
+                        on_complete()
+                    except Exception:
+                        logger.debug(
+                            "deep mention cache warmup callback failed",
+                            exc_info=True,
+                        )
+
+        thread = threading.Thread(
+            target=_warm,
+            name="comate-mention-index-warmup",
+            daemon=True,
+        )
+        with self._deep_warmup_lock:
+            if (
+                self._deep_warmup_thread is not None
+                and self._deep_warmup_thread.is_alive()
+            ):
+                return False
+            self._deep_warmup_thread = thread
+        thread.start()
+        return True
 
     def _load_indexed_file_paths(self) -> list[str]:
         git_paths = self._get_paths_from_git()
@@ -368,12 +424,17 @@ class LocalFileMentionCompleter(Completer):
                 "%s mention index command exited with %s: %s",
                 source_name,
                 completed.returncode,
-                completed.stderr.strip(),
+                (completed.stderr or "").strip(),
             )
             return None
 
+        stdout = completed.stdout
+        if stdout is None:
+            logger.debug("%s mention index command returned no stdout", source_name)
+            return None
+
         indexed_paths: set[str] = set()
-        for line in completed.stdout.splitlines():
+        for line in stdout.splitlines():
             normalized = self._normalize_indexed_path(line)
             if normalized is None or not self._should_include_relative_path(normalized):
                 continue

{comate_cli-0.7.0a1 → comate_cli-0.7.0a3}/comate_cli/terminal_agent/resume_selector.py

@@ -1,4 +1,4 @@
-"""Resume picker 入口：把 SDK SessionInfo 喂给 ResumePickerApp。"""
+"""Resume picker entry point: feeds SDK SessionInfo into ResumePickerApp."""
 
 from __future__ import annotations
 
@@ -52,7 +52,7 @@ def list_resume_sessions_via_sdk(*, cwd: Path | str | None) -> list[ResumeSessio
 async def select_resume_session_id(console: Console, *, cwd: Path | None = None) -> str | None:
     items = list_resume_sessions_via_sdk(cwd=cwd)
     if not items:
-        console.print("[dim]暂无可恢复会话。[/]")
+        console.print("[dim]No resumable sessions.[/]")
         return None
 
     cwd_by_session: dict[str, str | None] = {item.session_id: item.raw.cwd for item in items}
@@ -82,6 +82,6 @@ async def select_resume_session_id(console: Console, *, cwd: Path | None = None)
     with patch_stdout(raw=True):
         result = await app.run_async()
     if not isinstance(result, str) or not result.strip():
-        console.print("[dim]已取消恢复。[/]")
+        console.print("[dim]Resume cancelled.[/]")
         return None
     return result

{comate_cli-0.7.0a1 → comate_cli-0.7.0a3}/comate_cli/terminal_agent/tui.py

@@ -329,6 +329,11 @@ class TerminalAgentTUI(
             if self._busy:
                 return
             doc = self._input_area.buffer.document
+            mention_context = self._mention_completer.extract_context(
+                doc.text_before_cursor
+            )
+            if mention_context is not None:
+                self._start_mention_cache_warmup()
             if self._completion_context_active(
                 doc.text_before_cursor,
                 doc.text_after_cursor,
@@ -1325,6 +1330,43 @@ class TerminalAgentTUI(
 
         task.add_done_callback(_done)
 
+    def _start_mention_cache_warmup(self) -> None:
+        mention_completer = getattr(self, "_mention_completer", None)
+        start_warmup = getattr(mention_completer, "start_deep_cache_warmup", None)
+        if not callable(start_warmup):
+            return
+        start_warmup(on_complete=self._on_mention_cache_warmed)
+
+    def _on_mention_cache_warmed(self) -> None:
+        app = self._app
+        if app is None or self._closing:
+            return
+        loop = getattr(app, "loop", None)
+        if loop is None or loop.is_closed():
+            return
+        loop.call_soon_threadsafe(self._restart_active_mention_completion)
+        app.invalidate()
+
+    def _restart_active_mention_completion(self) -> None:
+        if self._closing or self._busy or self._ui_mode != UIMode.NORMAL:
+            return
+        input_area = getattr(self, "_input_area", None)
+        if input_area is None:
+            return
+        buffer = input_area.buffer
+        doc = buffer.document
+        if not self._completion_context_active(
+            doc.text_before_cursor,
+            doc.text_after_cursor,
+        ):
+            return
+        complete_state = buffer.complete_state
+        if complete_state is not None:
+            self._invalidate()
+            return
+        buffer.start_completion(select_first=False)
+        self._invalidate()
+
     def _refresh_layers(self) -> None:
         self._sync_focus_for_mode()
         self._render_dirty = True
@@ -1546,6 +1588,7 @@ class TerminalAgentTUI(
         if self._app is None:
             return
 
+        self._start_mention_cache_warmup()
         self._refresh_layers()
 
         try:

{comate_cli-0.7.0a1 → comate_cli-0.7.0a3}/comate_cli/terminal_agent/tui_parts/commands.py

@@ -43,12 +43,12 @@ if TYPE_CHECKING:
 
 class CommandsMixin:
     def _begin_llm_long_task(self) -> None:
-        """标记 LLM 远端长任务开始：开启 busy 并记录运行起点。"""
+        """Mark the start of a long-running LLM task: enable busy state and record the start time."""
         self._set_busy(True)
         self._run_start_time = time.monotonic()
 
     def _end_llm_long_task(self) -> None:
-        """标记 LLM 远端长任务结束：清理运行计时并退出 busy。"""
+        """Mark the end of a long-running LLM task: clear the run timer and exit busy state."""
         self._run_start_time = None
         self._set_busy(False)
 
@@ -78,7 +78,7 @@ class CommandsMixin:
     def _cycle_agent_mode(self) -> None:
         if self._busy:
             self._renderer.append_system_message(
-                "当前任务运行中，请在本轮结束后再切换模式。",
+                "A task is currently running. Please switch modes after this turn completes.",
             )
             self._refresh_layers()
             return
@@ -126,7 +126,7 @@ class CommandsMixin:
             is_busy = self._busy
             if is_busy and not entry.allow_when_busy:
                 self._append_slash_command_result(
-                    f"当前任务运行中，暂不可执行 /{entry.spec.name}。",
+                    f"A task is currently running. /{entry.spec.name} is not available right now.",
                     severity="error",
                 )
                 self._refresh_layers()
@@ -265,7 +265,7 @@ class CommandsMixin:
         self._open_mcp_server_list_menu(rows)
 
     def _slash_btw(self, args: str) -> None:
-        """/btw <question> — 进入 BTW 面板，进行不污染主上下文的旁路问答。"""
+        """/btw <question> — Enter the BTW panel for side-channel Q&A without polluting the main context."""
         normalized = args.strip()
         if not normalized:
             self._append_slash_command_result(
@@ -274,7 +274,7 @@ class CommandsMixin:
             )
             return
         if self._ui_mode == UIMode.BTW:
-            # 已经在 BTW 模式（理论上不可达：输入框被隐藏），防御性兜底
+            # Already in BTW mode (theoretically unreachable: input is hidden); defensive fallback.
             self._append_slash_command_result(
                 "Already in /btw mode. Press Esc to exit first.",
                 severity="error",
@@ -282,7 +282,7 @@ class CommandsMixin:
             return
 
         def _on_exit() -> None:
-            # handle_escape 内部调用，UI mode 切换由 key binding 兜底完成。
+            # Called from inside handle_escape; UI mode switch is finalized by the key binding.
             self._ui_mode = UIMode.NORMAL
             self._sync_focus_for_mode()
             self._invalidate()
@@ -606,7 +606,7 @@ class CommandsMixin:
         *,
         enabled: bool,
     ) -> None:
-        """启动 MCP server 启用/禁用流程。"""
+        """Start the MCP server enable/disable flow."""
         agent_runtime = self._session._agent
         row = rows_by_alias.get(alias, {})
         config_scope = str(row.get("config_scope", "")).strip().lower()
@@ -692,7 +692,7 @@ class CommandsMixin:
         self._invalidate()
 
     def _start_mcp_reconnect(self, alias: str, rows_by_alias: dict[str, dict[str, Any]]) -> None:
-        """启动 MCP server 重连流程。"""
+        """Start the MCP server reconnect flow."""
         agent_runtime = self._session._agent
         row = rows_by_alias.get(alias, {})
         server_type = str(row.get("server_type", "")).strip().upper()
@@ -711,17 +711,17 @@ class CommandsMixin:
             else:
                 message = f"Failed to reconnect to {alias}: {error_message}"
 
-            # 将 /mcp 命令和结果刷入 scrollback
+            # Flush the /mcp command and result into scrollback.
             self._renderer.seed_user_message("/mcp")
             self._renderer.append_subtitle(message)
 
             if success:
-                # 成功：回到 NORMAL 模式，不返回菜单
+                # Success: return to NORMAL mode without going back to the menu.
                 self._ui_mode = UIMode.NORMAL
                 self._sync_focus_for_mode()
                 self._refresh_layers()
             else:
-                # 失败：回到 detail 菜单，方便用户再次重试
+                # Failure: return to the detail menu so the user can retry.
                 refreshed_rows = self._collect_mcp_cached_rows()
                 if refreshed_rows:
                     refreshed_by_alias = {str(r["alias"]): r for r in refreshed_rows}
@@ -1050,14 +1050,14 @@ class CommandsMixin:
             )
         except CustomSlashExpandError as exc:
             self._append_slash_command_result(
-                f"/{command_name} 执行失败: {exc}",
+                f"/{command_name} failed: {exc}",
                 severity="error",
             )
             return
         except Exception as exc:
             logger.exception("custom slash command rendering failed")
             self._append_slash_command_result(
-                f"/{command_name} 执行失败: {exc}",
+                f"/{command_name} failed: {exc}",
                 severity="error",
             )
             return
@@ -1090,10 +1090,11 @@ class CommandsMixin:
         self._is_compacting = True
         self._compact_cancel_requested = False
         self._compact_task = asyncio.current_task()
-        # 不在此处提前 seed `> /compact` anchor：compact 必须 await 长任务，
-        # 若提前写入，user entry 的尾部空行会落到 scrollback 而无法被后续 drain 的
-        # `⎿` subtitle pop 掉（见 history_printer.py:223 的批内 pop 约束）。
-        # anchor 由结果阶段的 _append_slash_command_result 与 subtitle 同批写入。
+        # Do not pre-seed the `> /compact` anchor here: compact must await a long task,
+        # and seeding it early would leave the user entry's trailing blank line in scrollback
+        # where the subsequent drain's `⎿` subtitle cannot pop it
+        # (see the in-batch pop constraint in history_printer.py:223).
+        # The anchor is written together with the subtitle by _append_slash_command_result in the result phase.
         self._refresh_layers()
 
         try:
@@ -1132,19 +1133,19 @@ class CommandsMixin:
 
         if result.compacted:
             self._append_slash_command_result(
-                f"/compact completed: {result.tokens_before:,} {INJECTED_ARROW} {result.tokens_after:,} tokens",
+                f"compact completed: {result.tokens_before:,} {INJECTED_ARROW} {result.tokens_after:,} tokens",
             )
             return
 
         if not result.attempted:
             self._append_slash_command_result(
-                f"/compact skipped: {result.reason}",
+                f"compact skipped: {result.reason}",
                 severity="warning",
             )
             return
 
         self._append_slash_command_result(
-            f"/compact made no changes: {result.reason}",
+            f"compact made no changes: {result.reason}",
             severity="warning",
         )
 
@@ -1157,7 +1158,7 @@ class CommandsMixin:
             return
         if self._busy:
             self._append_slash_command_result(
-                "当前已有任务在运行，请稍后再执行 /rewind。",
+                "A task is currently running. Please try /rewind again later.",
                 severity="error",
             )
             return
@@ -1165,8 +1166,9 @@ class CommandsMixin:
         targets = self._session.list_rewind_targets()
         if not targets:
             self._append_slash_command_result(
-                "暂时没有可以回到的文字消息。/rewind 目前支持回到本次会话中仍可恢复的普通文字输入；"
-                "图片、团队消息、队列自动插入的消息，以及已压缩进摘要的较早内容暂不支持。"
+                "No rewind-able text messages are available. /rewind currently supports plain text "
+                "messages from this session that can still be restored; images, team messages, "
+                "queue-injected messages, and content already compacted into the summary are not supported."
             )
             return
         self._show_rewind_target_menu(targets)
@@ -1175,7 +1177,7 @@ class CommandsMixin:
         self._request_exit()
 
     def _slash_plugin(self, args: str) -> None:
-        """Handle /plugin command — 进入嵌入式 plugin picker 模式。"""
+        """Handle /plugin command — enter the embedded plugin picker mode."""
         from comate_agent_sdk.plugins import (
             MarketplaceManager,
             PluginLoader,
@@ -1188,7 +1190,7 @@ class CommandsMixin:
         registry = create_plugin_registry()
         mkt_mgr = MarketplaceManager(registry)
 
-        # 快捷路径：/plugin marketplace add <repo> → 直接进入轻量安装面板
+        # Shortcut: /plugin marketplace add <repo> → go straight to the lightweight install panel.
         if context.action == "add" and context.target_marketplace:
             self._install_view.enter(
                 repo=context.target_marketplace,
@@ -1238,7 +1240,7 @@ class CommandsMixin:
         self._append_slash_command_result(msg)
 
     def _handle_plugin_action(self, action: PluginPickerAction | None) -> None:
-        """处理 plugin picker 的 handler 返回值，切换回 NORMAL 模式。"""
+        """Handle the plugin picker handler return value and switch back to NORMAL mode."""
         from comate_cli.terminal_agent.plugins.plugin_picker import PluginPickerAction
         if action is None or not isinstance(action, PluginPickerAction):
             return
@@ -1254,13 +1256,13 @@ class CommandsMixin:
         self._refresh_layers()
 
     def _exit_install_view(self) -> None:
-        """关闭 marketplace install 面板，切回 NORMAL 模式。"""
+        """Close the marketplace install panel and switch back to NORMAL mode."""
         repo = self._install_view._repo
         result = self._install_view.take_result()
         self._ui_mode = UIMode.NORMAL
         self._sync_focus_for_mode()
-        # seed_user_message + append_subtitle 必须在同一次 drain 批次中，
-        # 否则 user entry 的尾部空行会在 scrollback 中残留。
+        # seed_user_message + append_subtitle must occur in the same drain batch;
+        # otherwise the user entry's trailing blank line will remain in scrollback.
         self._renderer.seed_user_message(f"/plugin marketplace add {repo}")
         if result.installed:
             self._renderer.append_subtitle(
@@ -1272,7 +1274,7 @@ class CommandsMixin:
         self._refresh_layers()
 
     def _on_marketplace_install_done(self) -> None:
-        """install_view 安装成功后的回调 — 自动关闭面板。"""
+        """Callback invoked after install_view succeeds — automatically closes the panel."""
         self._exit_install_view()
 
     def _slash_model(self, args: str) -> None:
@@ -1308,7 +1310,7 @@ class CommandsMixin:
             )
             logger.info(f"Model level switched: {event}")
 
-            # Update status bar model name - 使用 event 中的新模型名
+            # Update status bar model name using the new model name from event.
             self._status_bar.set_model_name(new_model)
             self._invalidate()
         except Exception as e:
@@ -1524,7 +1526,7 @@ class CommandsMixin:
             return
         if self._busy:
             self._append_slash_command_result(
-                "当前已有任务在运行，请稍后再执行 /rewind。",
+                "A task is currently running. Please try /rewind again later.",
                 severity="error",
             )
             self._refresh_layers()
@@ -1623,7 +1625,7 @@ class CommandsMixin:
             self._exit_selection_mode()
             return
 
-        # 回调可能已经切换了 ui_mode（如 MCP_CONNECTING），尊重回调的意图
+        # The callback may have switched ui_mode (e.g., MCP_CONNECTING); respect its intent.
         if self._ui_mode == UIMode.SELECTION:
             self._exit_selection_mode()
 

{comate_cli-0.7.0a1 → comate_cli-0.7.0a3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "comate-cli"
-version = "0.7.0a1"
+version = "0.7.0a3"
 description = "Comate terminal CLI built on comate-agent-sdk"
 readme = "README.md"
 requires-python = ">=3.11"

comate_cli-0.7.0a3/tests/test_completion_context_activation.py

@@ -0,0 +1,119 @@
+from __future__ import annotations
+
+import tempfile
+import unittest
+from pathlib import Path
+from types import SimpleNamespace
+
+from prompt_toolkit.document import Document
+
+from comate_cli.terminal_agent.mention_completer import LocalFileMentionCompleter
+from comate_cli.terminal_agent.tui import TerminalAgentTUI
+from comate_cli.terminal_agent.tui_parts import UIMode
+
+
+class TestCompletionContextActivation(unittest.TestCase):
+    def setUp(self) -> None:
+        self._tmpdir = tempfile.TemporaryDirectory()
+        self.tui = TerminalAgentTUI.__new__(TerminalAgentTUI)
+        self.tui._mention_completer = LocalFileMentionCompleter(
+            Path(self._tmpdir.name),
+            refresh_interval=0.0,
+            limit=200,
+        )
+
+    def tearDown(self) -> None:
+        self._tmpdir.cleanup()
+
+    def test_slash_completion_requires_cursor_at_end(self) -> None:
+        self.assertTrue(self.tui._completion_context_active("/model", ""))
+        self.assertFalse(self.tui._completion_context_active("/model", " trailing"))
+
+    def test_mention_completion_triggers_at_line_end(self) -> None:
+        self.assertTrue(self.tui._completion_context_active("message @file", ""))
+
+    def test_mention_completion_triggers_at_line_start(self) -> None:
+        self.assertTrue(self.tui._completion_context_active("@file", " message"))
+
+    def test_mention_completion_triggers_in_middle_with_space_boundaries(self) -> None:
+        self.assertTrue(
+            self.tui._completion_context_active("message @file", " message")
+        )
+
+    def test_mention_completion_rejects_middle_without_right_space(self) -> None:
+        self.assertFalse(
+            self.tui._completion_context_active("message @file", "message")
+        )
+
+    def test_mention_completion_rejects_middle_without_left_space(self) -> None:
+        self.assertFalse(
+            self.tui._completion_context_active("message@file", " message")
+        )
+
+    def test_mention_completion_rejects_inside_token_editing(self) -> None:
+        self.assertFalse(
+            self.tui._completion_context_active("message @fi", "le message")
+        )
+
+    def test_mention_cache_warmup_registers_completion_callback(self) -> None:
+        callbacks = []
+
+        class _FakeMentionCompleter:
+            def start_deep_cache_warmup(self, *, on_complete):
+                callbacks.append(on_complete)
+                return True
+
+        self.tui._mention_completer = _FakeMentionCompleter()
+
+        self.tui._start_mention_cache_warmup()
+
+        self.assertEqual(len(callbacks), 1)
+        self.assertTrue(callable(callbacks[0]))
+
+    def test_warmup_callback_does_not_cancel_in_flight_completion(self) -> None:
+        class _FakeBuffer:
+            def __init__(self) -> None:
+                self.document = Document("@穆", cursor_position=2)
+                self.complete_state = SimpleNamespace(completions=[])
+                self.start_calls = 0
+
+            def start_completion(self, *, select_first: bool) -> None:
+                del select_first
+                self.start_calls += 1
+
+        buffer = _FakeBuffer()
+        self.tui._closing = False
+        self.tui._busy = False
+        self.tui._ui_mode = UIMode.NORMAL
+        self.tui._app = SimpleNamespace(invalidate=lambda: None)
+        self.tui._input_area = SimpleNamespace(buffer=buffer)
+
+        self.tui._restart_active_mention_completion()
+
+        self.assertEqual(buffer.start_calls, 0)
+
+    def test_warmup_callback_restarts_when_no_completion_is_active(self) -> None:
+        class _FakeBuffer:
+            def __init__(self) -> None:
+                self.document = Document("@穆", cursor_position=2)
+                self.complete_state = None
+                self.start_calls = 0
+
+            def start_completion(self, *, select_first: bool) -> None:
+                del select_first
+                self.start_calls += 1
+
+        buffer = _FakeBuffer()
+        self.tui._closing = False
+        self.tui._busy = False
+        self.tui._ui_mode = UIMode.NORMAL
+        self.tui._app = SimpleNamespace(invalidate=lambda: None)
+        self.tui._input_area = SimpleNamespace(buffer=buffer)
+
+        self.tui._restart_active_mention_completion()
+
+        self.assertEqual(buffer.start_calls, 1)
+
+
+if __name__ == "__main__":
+    unittest.main(verbosity=2)