comate-cli 0.10.7__tar.gz → 0.10.9__tar.gz

{comate_cli-0.10.7 → comate_cli-0.10.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comate-cli
-Version: 0.10.7
+Version: 0.10.9
 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.10.7 → comate_cli-0.10.9}/comate_cli/terminal_agent/tui.py

@@ -10,7 +10,7 @@ import logging
 import random
 import time
 from collections.abc import Callable
-from contextlib import suppress
+from contextlib import nullcontext, suppress
 from dataclasses import dataclass
 from pathlib import Path
 from typing import Any
@@ -48,6 +48,7 @@ from comate_agent_sdk.agent.events import (
 from comate_agent_sdk.agent.recap import generate_recap
 from comate_agent_sdk.agent.goal_state import GoalStatus
 from comate_agent_sdk.agent.queue_types import MessageOrigin
+from comate_agent_sdk.utils.first_turn_profile import FirstTurnProfiler
 
 from comate_cli.terminal_agent.animations import (
     DEFAULT_STATUS_PHRASES,
@@ -232,6 +233,7 @@ class TerminalAgentTUI(
         self._session = session
         self._approval_bridge = approval_bridge
         self._yolo = yolo
+        self._first_turn_profiler = FirstTurnProfiler.from_env(logger=logger).child("tui")
         self._status_bar = status_bar
         try:
             self._status_bar.set_mode(self._current_permission_mode())
@@ -1351,17 +1353,20 @@ class TerminalAgentTUI(
         submit_text: str,
         display_text: str,
     ) -> None:
+        self._first_turn_mark("enter_staging.start")
         if self._staging_payload is not None:
             if bool(getattr(self, "_staging_submitted", False)):
                 self._flush_staging_to_scrollback()
             else:
                 self._commit_staging()
 
+        with self._first_turn_span("enter_staging.compute_file_ref_hints"):
+            file_ref_hints = tuple(self._renderer.compute_file_ref_hints(display_text))
         self._staging_payload = _StagedMessagePayload(
             raw_text=raw_text,
             submit_text=submit_text,
             display_text=display_text,
-            file_ref_hints=tuple(self._renderer.compute_file_ref_hints(display_text)),
+            file_ref_hints=file_ref_hints,
         )
         self._staging_deadline = None
         self._staging_recallable = True
@@ -1378,6 +1383,7 @@ class TerminalAgentTUI(
         self._schedule_background(self._animation_controller.start())
         self._render_dirty = True
         self._invalidate()
+        self._first_turn_mark("enter_staging.done")
 
     def _recall_latest_turn_after_interrupt(self) -> bool:
         if not bool(getattr(self, "_pending_latest_recall_on_interrupt", False)):
@@ -1456,10 +1462,13 @@ class TerminalAgentTUI(
         return False
 
     def _commit_staging(self) -> bool:
+        self._first_turn_mark("commit_staging.start")
         payload = self._staging_payload
         if payload is None:
+            self._first_turn_mark("commit_staging.no_payload")
             return False
         if bool(getattr(self, "_staging_submitted", False)):
+            self._first_turn_mark("commit_staging.already_submitted")
             return False
 
         # 不在此处清除 staging — 延迟到 _drain_history_async 的 run_in_terminal
@@ -1473,19 +1482,38 @@ class TerminalAgentTUI(
             self._render_dirty = True
             self._invalidate()
             self._schedule_background(self._queue_staged_busy_message(payload))
+            self._first_turn_mark("commit_staging.scheduled_busy_queue")
         else:
             self._staging_deadline = None
             self._staging_recallable = False
             self._staging_submitted = True
             self._schedule_background(
-                self._submit_user_message(
+                self._submit_staged_user_message_after_first_repaint(
                     payload.submit_text,
                     display_text=payload.display_text,
                     defer_user_seed=True,
                 )
             )
+            self._first_turn_mark("commit_staging.scheduled_submit")
+        self._first_turn_mark("commit_staging.done")
         return True
 
+    async def _submit_staged_user_message_after_first_repaint(
+        self,
+        text: str,
+        *,
+        display_text: str | None,
+        defer_user_seed: bool,
+    ) -> None:
+        self._first_turn_mark("submit_staged.yield_for_first_repaint.start")
+        await asyncio.sleep(0)
+        self._first_turn_mark("submit_staged.yield_for_first_repaint.done")
+        await self._submit_user_message(
+            text,
+            display_text=display_text,
+            defer_user_seed=defer_user_seed,
+        )
+
     async def _queue_staged_busy_message(self, payload: _StagedMessagePayload) -> None:
         try:
             message_id = await self._session.send(payload.submit_text)
@@ -1588,10 +1616,12 @@ class TerminalAgentTUI(
         defer_user_seed: bool = False,
         allow_latest_recall_on_interrupt: bool = True,
     ) -> None:
+        self._first_turn_mark("submit_user_message.start")
         if self._busy:
             self._renderer.append_system_message(
                 "A task is currently running. Please wait.", severity="error"
             )
+            self._first_turn_mark("submit_user_message.rejected_busy")
             return
 
         if self._ui_mode == UIMode.QUESTION:
@@ -1601,6 +1631,7 @@ class TerminalAgentTUI(
         self._interrupt_requested_at = None
 
         self._set_busy(True)
+        self._first_turn_mark("submit_user_message.busy_set")
         self._allow_latest_recall_on_interrupt = allow_latest_recall_on_interrupt
         self._run_start_time = time.monotonic()
         self._waiting_for_input = False
@@ -1628,29 +1659,33 @@ class TerminalAgentTUI(
             )
         self._last_turn_user_preview = preview_text
         self._refresh_layers()
+        self._first_turn_mark("submit_user_message.layers_refreshed")
 
         # 启动提交动画
         await self._animation_controller.start()
+        self._first_turn_mark("submit_user_message.animation_started")
 
         # @path context hint injection
-        try:
-            path_hint_source = (
-                text.provider_content
-                if isinstance(text, RenderedAuthoredTurnInput)
-                else text
-            )
-            hint = (
-                build_path_context_hint(path_hint_source, Path.cwd())
-                if isinstance(path_hint_source, str)
-                else None
-            )
-            if hint is not None:
-                self._session._agent._context.add_system_reminder_message(hint)
-        except Exception:
-            logger.debug("path context hint injection failed", exc_info=True)
+        with self._first_turn_span("submit_user_message.path_context_hint"):
+            try:
+                path_hint_source = (
+                    text.provider_content
+                    if isinstance(text, RenderedAuthoredTurnInput)
+                    else text
+                )
+                hint = (
+                    build_path_context_hint(path_hint_source, Path.cwd())
+                    if isinstance(path_hint_source, str)
+                    else None
+                )
+                if hint is not None:
+                    self._session._agent._context.add_system_reminder_message(hint)
+            except Exception:
+                logger.debug("path context hint injection failed", exc_info=True)
 
         try:
-            await self._session.send(text)
+            with self._first_turn_span("submit_user_message.session_send"):
+                await self._session.send(text)
         except Exception as exc:
             logger.exception("send failed")
             await self._handle_error(exc)
@@ -1919,6 +1954,19 @@ class TerminalAgentTUI(
 
         task.add_done_callback(_done)
 
+    def _first_turn_mark(self, phase: str) -> None:
+        profiler = getattr(self, "_first_turn_profiler", None)
+        mark = getattr(profiler, "mark", None)
+        if callable(mark):
+            mark(phase)
+
+    def _first_turn_span(self, phase: str) -> Any:
+        profiler = getattr(self, "_first_turn_profiler", None)
+        span = getattr(profiler, "span", None)
+        if callable(span):
+            return span(phase)
+        return nullcontext()
+
     def _start_mention_cache_warmup(self) -> None:
         mention_completer = getattr(self, "_mention_completer", None)
         start_warmup = getattr(mention_completer, "start_deep_cache_warmup", None)

{comate_cli-0.10.7 → comate_cli-0.10.9}/comate_cli/terminal_agent/tui_parts/input_behavior.py

@@ -20,6 +20,9 @@ from comate_cli.terminal_agent.tui_parts.ui_mode import UIMode
 
 
 class InputBehaviorMixin:
+    def _first_turn_mark(self, phase: str) -> None:
+        del phase
+
     def _sync_local_interactive_ui_barrier(self) -> None:
         return
 
@@ -454,15 +457,19 @@ class InputBehaviorMixin:
         return True
 
     def _submit_from_input(self) -> None:
+        self._first_turn_mark("input.submit_from_input.start")
         if self._ui_mode != UIMode.NORMAL:
+            self._first_turn_mark("input.submit_from_input.non_normal_mode")
             return
 
         raw_text = self._input_area.text
         display_text, submit_text = self._resolve_submit_texts(raw_text)
         if not submit_text.strip():
             if bool(getattr(self, "_staging_submitted", False)):
+                self._first_turn_mark("input.submit_from_input.empty_submitted_staging")
                 return
             self._commit_staging()
+            self._first_turn_mark("input.submit_from_input.done")
             return
 
         recall_text = raw_text
@@ -514,6 +521,7 @@ class InputBehaviorMixin:
 
         if raw_text.startswith("/"):
             self._schedule_background(self._execute_command(display_text.strip()))
+            self._first_turn_mark("input.submit_from_input.slash_scheduled")
             return
 
         previous_staging_payload = getattr(self, "_staging_payload", None)
@@ -528,3 +536,4 @@ class InputBehaviorMixin:
             and current_staging_payload is not previous_staging_payload
         ):
             self._commit_staging()
+        self._first_turn_mark("input.submit_from_input.done")

{comate_cli-0.10.7 → comate_cli-0.10.9}/docs/startup-performance.md

@@ -33,6 +33,25 @@
      `tests/test_startup_import_budget.py` 是否仍然全绿。
    - 是否新增了 logo 前同步 import、无 timeout 子进程、或状态栏后台轮询。
 
+## 排查首条消息 Enter 卡顿
+
+启动已进入 TUI 但第一次按 Enter 后 3-4 秒才出现提交/loading 时，运行
+`COMATE_FIRST_ENTER_PROFILE=1 comate`，退出后查看 `~/.comate/logs/agent.log`
+中的 `first_turn_profile phase=... elapsed_ms=... delta_ms=...` 行。
+
+优先看这些阶段：
+
+- `tui.enter_staging.compute_file_ref_hints.*`：普通输入是否卡在本地 `@path` hint。
+- `tui.submit_user_message.path_context_hint.*`：提交前是否卡在 `@path` context hint。
+- `chat_session.header_snapshot.write_once.*` 与 `transcript.append.header-snapshot.*`：
+  新 session 首条消息前的 header snapshot 落盘是否被 Windows 文件系统、OneDrive
+  或 Defender 放大。
+- `budget.token_counter.init_encoder.*`、`budget.tiktoken_cache.prepare.*`、
+  `budget.token_counter.get_encoding.*`：首轮 token 估算是否在前台付了 tiktoken
+  初始化或 cache seed 成本。
+- `chat_session.run_turn_stream.first_event.*`：从进入 SDK turn 到第一个 runtime event
+  的等待时间。
+
 ## 当前基准口径
 
 `packages/comate_cli/tools/startup_bench.py` 使用 `COMATE_STARTUP_PROFILE=1`