langchain-agentx-cli 0.5.2__tar.gz → 0.5.3__tar.gz

{langchain-agentx-cli-0.5.2 → langchain-agentx-cli-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-agentx-cli
-Version: 0.5.2
+Version: 0.5.3
 Summary: Terminal CLI/TUI for AgentX: migrate Claude Code Ink UI, backed by langchain-agentx-python SDK.
 Author: GoodMood2008
 License: Apache-2.0

{langchain-agentx-cli-0.5.2 → langchain-agentx-cli-0.5.3}/langchain_agentx_cli/__init__.py

@@ -1,3 +1,3 @@
 """langchain_agentx_cli：依托 langchain_agentx_python SDK 的终端 CLI/TUI（CC Ink 迁移）。"""
 
-__version__ = "0.5.2"
+__version__ = "0.5.3"

{langchain-agentx-cli-0.5.2 → langchain-agentx-cli-0.5.3}/langchain_agentx_cli/widgets/message_list.py

@@ -35,6 +35,7 @@ from langchain_agentx_cli.widgets.messages import (
     HookMessageWidget,
     HookProgressed,
     HookStarted,
+    StreamingAssistantMessage,
     TaskProgress,
     TaskStarted,
     ThinkingContentDelta,
@@ -221,7 +222,7 @@ class MessageListWidget(ScrollableContainer):
 
     def __init__(self, *args, **kwargs) -> None:
         super().__init__(*args, **kwargs)
-        self._active_assistant: AssistantMessageWidget | None = None
+        self._active_assistant: StreamingAssistantMessage | AssistantMessageWidget | None = None
         self._welcome_shown = False
         self._pending_permission: InlinePermissionWidget | None = None
         self._turn_spinner: SpinnerWidget | None = None
@@ -695,7 +696,7 @@ class MessageListWidget(ScrollableContainer):
 
         if not chunk.delta and self._active_assistant is None:
             info("[ML] creating new assistant widget (empty delta)")
-            line = AssistantMessageWidget(streaming=True)
+            line = StreamingAssistantMessage(streaming=True)
             self._mount_keeping_spinner_below(line)
             self._active_assistant = line
             self._scroll_to_end(force=True)
@@ -704,7 +705,7 @@ class MessageListWidget(ScrollableContainer):
         if chunk.delta:
             if self._active_assistant is None:
                 info("[ML] creating new assistant widget (first delta)")
-                line = AssistantMessageWidget(streaming=True)
+                line = StreamingAssistantMessage(streaming=True)
                 self._mount_keeping_spinner_below(line)
                 self._active_assistant = line
                 self._scroll_to_end(force=True)

{langchain-agentx-cli-0.5.2 → langchain-agentx-cli-0.5.3}/langchain_agentx_cli/widgets/messages/__init__.py

@@ -38,6 +38,9 @@ from langchain_agentx_cli.widgets.message_events import (
 )
 from langchain_agentx_cli.widgets.messages.assistant_message import AssistantMessageWidget
 from langchain_agentx_cli.widgets.messages.error_message import ErrorMessageWidget
+from langchain_agentx_cli.widgets.messages.streaming_assistant_message import (
+    StreamingAssistantMessage,
+)
 from langchain_agentx_cli.widgets.messages.hook_message import HookMessageWidget
 from langchain_agentx_cli.widgets.messages.stop_hook_summary import StopHookSummaryWidget
 from langchain_agentx_cli.widgets.messages.thinking_message import ThinkingMessageWidget
@@ -63,6 +66,7 @@ __all__ = [
     "PermissionPendingShown",
     "SpinnerCompletion",
     "StopHookSummaryWidget",
+    "StreamingAssistantMessage",
     "TaskProgress",
     "TaskStarted",
     "ThinkingContentDelta",

langchain-agentx-cli-0.5.3/langchain_agentx_cli/widgets/messages/streaming_assistant_message.py

@@ -0,0 +1,323 @@
+"""
+widgets/messages/streaming_assistant_message.py — 双 Widget 流式渲染组件。
+
+职责：
+    对齐 CC StreamingMarkdown：stable 用 Static（不解析），unstable 用 Markdown（流式）。
+    边界单调前进，stable 只增不减，避免重新解析已完成内容。
+
+链路位置：
+    MessageListWidget.handle_assistant_chunk → StreamingAssistantMessage。
+
+架构对比：
+    旧版（AssistantMessageWidget）：单一 Markdown Widget，每次都重新解析整个内容
+    新版（本组件）：双 Widget，stable 部分 Static 不触发 Markdown 解析
+
+性能优化：
+    stable 长文本不再触发 Markdown.lexer()，O(1) vs O(n)
+    unstable 部分仍是 Markdown 解析，但长度受限（当前段落）
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+from typing import TYPE_CHECKING
+
+import time
+
+from textual.containers import Vertical
+from textual.widgets import Markdown, Static
+
+from langchain_agentx_cli.observation import info
+from langchain_agentx_cli.widgets.messages.rendering import (
+    MAX_COLLAPSE_LINES,
+    STREAM_RENDER_INTERVAL_S,
+    STREAM_RENDER_MIN_CHARS,
+    escape_markdown_unsafe,
+    preprocess_assistant_markdown,
+    truncate_lines,
+)
+
+if TYPE_CHECKING:
+    from langchain_agentx_cli.app import ReplApp
+
+_PARAGRAPH_SEPARATOR = "\n\n"
+_METADATA_SEPARATOR = " · "
+_BLACK_CIRCLE = "●"
+_PLAIN_TEXT_SCAN_CHARS = 500
+_MARKDOWN_PATTERN = __import__("re").compile(
+    r"(?m)^#{1,6}\s|"  # 标题
+    r"`|"  # 行内代码
+    r"\*\*|"  # 粗体
+    r"\*|"  # 斜体
+    r"\[.*?\]\(|"  # 链接
+    r"^\s*[-*+]\s|"  # 无序列表
+    r"^\s*\d+\.\s|"  # 有序列表
+    r"^\s*>\s|"  # 引用
+    r"\|"  # 表格
+)
+
+
+def _looks_like_plain_text(text: str) -> bool:
+    """快速检测文本是否为纯文本。"""
+    if not text:
+        return True
+    scan = text[:_PLAIN_TEXT_SCAN_CHARS]
+    return not bool(_MARKDOWN_PATTERN.search(scan))
+
+
+def _format_timestamp() -> str:
+    """格式化时间戳（HH:MM 格式）。"""
+    return datetime.now().strftime("%H:%M")
+
+
+def _format_metadata(model: str | None, tokens: int | None = None) -> str:
+    """格式化元数据行。"""
+    parts: list[str] = []
+    if model:
+        parts.append(model)
+    if tokens is not None and tokens > 0:
+        parts.append(f"{tokens} tokens")
+    parts.append(_format_timestamp())
+    return _METADATA_SEPARATOR.join(parts)
+
+
+def _render_text_fast(text: str) -> str:
+    """快速渲染文本：纯文本跳过 Markdown 解析。"""
+    if not text:
+        return ""
+    if _looks_like_plain_text(text):
+        return escape_markdown_unsafe(text)
+    return preprocess_assistant_markdown(text)
+
+
+def _with_dot_prefix(content: str) -> str:
+    """添加 BLACK_CIRCLE 前缀。"""
+    if not content:
+        return f"{_BLACK_CIRCLE} "
+    return f"{_BLACK_CIRCLE} {content}"
+
+
+class StreamingAssistantMessage(Vertical):
+    """双 Widget 流式助手消息（对齐 CC StreamingMarkdown）。
+
+    架构：
+        - _stable_widget: Static，已完成的段落，不再触发 Markdown 解析
+        - _unstable_widget: Markdown，当前增长中的段落，流式更新
+
+    边界前进：
+        - stable 只增不减（单调）
+        - 段落边界（\n\n）触发内容从 unstable 移动到 stable
+    """
+
+    DEFAULT_CSS = """
+    StreamingAssistantMessage {
+        width: 100%;
+        height: auto;
+        margin: 1 0;
+        padding: 0 1;
+        background: $surface;
+    }
+    StreamingAssistantMessage.-streaming {
+        border-left: solid $accent;
+    }
+    StreamingAssistantMessage .stable-content {
+        width: 100%;
+        height: auto;
+    }
+    StreamingAssistantMessage .unstable-content {
+        width: 100%;
+        height: auto;
+    }
+    StreamingAssistantMessage .message-metadata {
+        text-align: right;
+        color: $text-muted;
+        text-style: dim;
+        margin: 1 0 0 0;
+        padding: 0 1;
+    }
+    """
+
+    def __init__(self, content: str = "", *, streaming: bool = True) -> None:
+        classes = "message-assistant"
+        if streaming:
+            classes += " -streaming"
+        super().__init__(classes=classes)
+
+        # 内部状态
+        self._streaming = streaming
+        self._stable_text: str = ""  # 已完成的段落（原始文本）
+        self._stable_rendered_blocks: list[str] = []  # 已渲染的段落块（增量追加）
+        self._last_stable_rendered: str = ""  # 上次渲染的 stable（避免无谓更新）
+        self._unstable_buffer: str = ""  # 当前增长中的段落
+        self._chars_since_render: int = 0
+        self._last_render_at: float = 0.0
+
+        # 子 Widget（延迟挂载，在首次 _ensure_widgets 时挂载）
+        self._stable_widget: Static | None = None
+        self._unstable_widget: Markdown | None = None
+        self._widgets_mounted: bool = False
+
+        # 初始内容
+        if content:
+            self._unstable_buffer = content
+
+        # 初始渲染（延迟到首次 update_content 或 finalize）
+        self._initial_content = content
+
+    @property
+    def text(self) -> str:
+        """获取完整文本（stable + unstable）。"""
+        return self._stable_text + self._unstable_buffer
+
+    def _ensure_widgets(self) -> None:
+        """确保子 Widget 已挂载（延迟挂载以支持测试环境）。"""
+        if self._widgets_mounted:
+            return
+
+        # 创建子 Widget
+        self._stable_widget = Static("", classes="stable-content")
+        self._unstable_widget = Markdown("", classes="unstable-content")
+
+        # 尝试挂载（如果没有 app，跳过挂载但保持内部状态）
+        try:
+            self.mount(self._stable_widget)
+            self.mount(self._unstable_widget)
+            self._widgets_mounted = True
+
+            # 初始渲染
+            if self._initial_content:
+                self._update_unstable_widget()
+            else:
+                self._unstable_widget.update(_with_dot_prefix(""))
+        except Exception:
+            # 无 app 环境（测试），跳过挂载
+            pass
+
+    def update_content(self, delta: str) -> None:
+        """接收流式 delta，更新 unstable，检测段落边界。"""
+        # 确保子 Widget 已挂载
+        self._ensure_widgets()
+
+        if delta:
+            self._unstable_buffer += delta
+            self._chars_since_render += len(delta)
+            # 检查是否有段落边界（\n\n），有则移入 stable
+            self._try_flush_unstable_to_stable()
+
+        if self._should_render_now():
+            self._flush_render()
+
+    def _try_flush_unstable_to_stable(self) -> None:
+        """检测段落边界，将已完成段落移入 stable（对齐 CC 边界单调前进）。"""
+        # 查找最后一个完整的段落边界（保留最后一个可能不完整的段落）
+        last_separator_idx = self._unstable_buffer.rfind(_PARAGRAPH_SEPARATOR)
+        if last_separator_idx == -1:
+            return
+
+        # 将边界之前的内容移入 stable
+        to_stable = self._unstable_buffer[: last_separator_idx + 2]  # +2 保留 \n\n
+        self._unstable_buffer = self._unstable_buffer[last_separator_idx + 2 :]
+
+        # 更新 stable 文本
+        self._stable_text += to_stable
+
+        # 增量渲染：只渲染新增的 to_stable，避免重新处理整个 stable
+        newly_rendered = _render_text_fast(to_stable)
+        self._stable_rendered_blocks.append(newly_rendered)
+
+        # 拼接所有已渲染的块（用于 widget 更新）
+        stable_rendered = "\n\n".join(self._stable_rendered_blocks)
+
+        # 只在内容变化时更新（避免无谓重绘）
+        if stable_rendered != self._last_stable_rendered:
+            if self._stable_widget is not None:
+                self._stable_widget.update(_with_dot_prefix(stable_rendered))
+            self._last_stable_rendered = stable_rendered
+
+        info(
+            "[SAM] boundary advanced stable_len=%d unstable_len=%d blocks=%d",
+            len(self._stable_text),
+            len(self._unstable_buffer),
+            len(self._stable_rendered_blocks),
+        )
+
+    def _should_render_now(self) -> bool:
+        """判断是否应该现在渲染（节流）。"""
+        if not self._streaming:
+            return True
+        if self._chars_since_render >= STREAM_RENDER_MIN_CHARS:
+            return True
+        if self._last_render_at == 0.0:
+            return True
+        return (time.monotonic() - self._last_render_at) >= STREAM_RENDER_INTERVAL_S
+
+    def _flush_render(self) -> None:
+        """更新 unstable widget 的显示。"""
+        self._update_unstable_widget()
+        self._chars_since_render = 0
+        self._last_render_at = time.monotonic()
+
+    def _update_unstable_widget(self) -> str:
+        """更新 unstable widget（Markdown 解析）。"""
+        # unstable 始终需要 Markdown 解析（因为可能在增长中）
+        unstable_rendered = _render_text_fast(self._unstable_buffer)
+        content = _with_dot_prefix(unstable_rendered) if unstable_rendered else _with_dot_prefix("")
+        if self._unstable_widget is not None:
+            self._unstable_widget.update(content)
+        return content
+
+    def finalize(
+        self,
+        *,
+        model: str | None = None,
+        tokens: int | None = None,
+    ) -> None:
+        """完成流式渲染，添加元数据行。"""
+        info(
+            "[SAM] finalize model=%s stable_len=%d unstable_len=%d",
+            model,
+            len(self._stable_text),
+            len(self._unstable_buffer),
+        )
+
+        self._streaming = False
+        self.remove_class("-streaming")
+
+        # 将剩余 unstable 全部移入 stable
+        if self._unstable_buffer:
+            self._stable_text += self._unstable_buffer
+            # 增量渲染：把剩余部分也加入 blocks
+            newly_rendered = _render_text_fast(self._unstable_buffer)
+            if newly_rendered:
+                self._stable_rendered_blocks.append(newly_rendered)
+            self._unstable_buffer = ""
+
+        # 截断超长内容
+        body = self._stable_text
+        truncated, was_truncated = truncate_lines(body, MAX_COLLAPSE_LINES)
+        if was_truncated:
+            truncated += f"\n\n… 输出超过 {MAX_COLLAPSE_LINES} 行，已截断"
+            self._stable_text = truncated
+            # 截断后需要重新构建 blocks（简单方式：重新渲染整个截断后的内容）
+            self._stable_rendered_blocks = [_render_text_fast(truncated)]
+
+        # 最终渲染（使用已渲染的块）
+        final_rendered = "\n\n".join(self._stable_rendered_blocks)
+        final_content = _with_dot_prefix(final_rendered) if final_rendered else _with_dot_prefix("")
+
+        # 追加元数据行
+        metadata = _format_metadata(model, tokens)
+        final_content += f"\n\n<div class='message-metadata'>{metadata}</div>"
+
+        # 移除 unstable widget（全部移入 stable）
+        if self._stable_widget is not None:
+            self._stable_widget.update(final_content)
+        if self._unstable_widget is not None and self._unstable_widget.is_attached:
+            self._unstable_widget.remove()
+
+        info("[SAM] finalized content_len=%d", len(final_content))
+
+    def set_verbose(self, value: bool) -> None:
+        """设置 verbose 模式（占位，未来可能用于控制显示细节）。"""
+        # 对齐现有 API，当前暂无实现
+        pass

{langchain-agentx-cli-0.5.2 → langchain-agentx-cli-0.5.3}/langchain_agentx_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-agentx-cli
-Version: 0.5.2
+Version: 0.5.3
 Summary: Terminal CLI/TUI for AgentX: migrate Claude Code Ink UI, backed by langchain-agentx-python SDK.
 Author: GoodMood2008
 License: Apache-2.0

{langchain-agentx-cli-0.5.2 → langchain-agentx-cli-0.5.3}/langchain_agentx_cli.egg-info/SOURCES.txt

@@ -166,6 +166,7 @@ langchain_agentx_cli/widgets/messages/error_message.py
 langchain_agentx_cli/widgets/messages/hook_message.py
 langchain_agentx_cli/widgets/messages/rendering.py
 langchain_agentx_cli/widgets/messages/stop_hook_summary.py
+langchain_agentx_cli/widgets/messages/streaming_assistant_message.py
 langchain_agentx_cli/widgets/messages/thinking_message.py
 langchain_agentx_cli/widgets/messages/tool_result.py
 langchain_agentx_cli/widgets/presentation/__init__.py

{langchain-agentx-cli-0.5.2 → langchain-agentx-cli-0.5.3}/langchain_agentx_cli.egg-info/requires.txt

@@ -1,4 +1,4 @@
-langchain-agentx-python<0.9.0,>=0.8.3
+langchain-agentx-python<0.9.0,>=0.8.4
 click>=8.1
 textual>=0.79.0
 

{langchain-agentx-cli-0.5.2 → langchain-agentx-cli-0.5.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "langchain-agentx-cli"
-version = "0.5.2"
+version = "0.5.3"
 description = "Terminal CLI/TUI for AgentX: migrate Claude Code Ink UI, backed by langchain-agentx-python SDK."
 readme = "README.md"
 requires-python = ">=3.11"
@@ -26,7 +26,7 @@ classifiers = [
   "Topic :: Software Development",
 ]
 dependencies = [
-  "langchain-agentx-python>=0.8.3,<0.9.0",
+  "langchain-agentx-python>=0.8.4,<0.9.0",
   "click>=8.1",
   "textual>=0.79.0",
 ]