xg-cli 1.0__py3-none-any.whl

xg/tui/renderables.py

@@ -0,0 +1,252 @@
+"""Rich renderables for transcript items."""
+
+from __future__ import annotations
+
+import json
+
+from rich.console import Console, ConsoleOptions, Group, RenderResult
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.text import Text
+
+from xg.tui.diagrams import FlowchartModel, FlowchartParseError, parse_flowchart, render_flowchart, split_mermaid_blocks
+from xg.tui.plan_renderables import PlanReviewCard
+from xg.tui.state import AgentGroupState, TranscriptItem
+
+
+def _truncate(value: str, limit: int = 20_000) -> str:
+    return value if len(value) <= limit else value[:limit] + "\n… (输出已截断)"
+
+
+def _mermaid_source(source: str) -> str:
+    return "```mermaid\n" + _truncate(source) + "\n```"
+
+
+def _trace_status(item: TranscriptItem) -> str:
+    return {
+        "streaming": "进行中",
+        "running": "执行中",
+        "success": "成功",
+        "failed": "失败",
+        "blocked": "已阻塞",
+        "needs_input": "等待输入",
+        "skipped": "已跳过",
+        "cancelled": "已取消",
+        "done": "已完成",
+    }.get(item.status, item.status)
+
+
+def _trace_summary(item: TranscriptItem) -> str:
+    if item.kind == "thinking":
+        label = "思考"
+    elif item.kind == "tool_call":
+        label = f"工具调用 · {item.tool_name}"
+    elif item.kind == "tool_result":
+        label = f"工具结果 · {item.tool_name}"
+    else:
+        label = item.kind
+    source = item.text or item.tool_args or "无详细内容"
+    first_line = next((line.strip() for line in source.splitlines() if line.strip()), "无详细内容")
+    return f"{label} · {_trace_status(item)} · {first_line[:100]}"
+
+
+def trace_renderable(item: TranscriptItem):
+    """Render a collapsible trace item; the widget owns click behavior."""
+    marker = "▶" if item.collapsed else "▼"
+    if item.kind == "thinking":
+        label = "思考"
+        detail = _truncate(item.text, 12_000)
+    elif item.kind == "tool_call":
+        label = f"工具调用 · {item.tool_name}"
+        try:
+            detail = json.dumps(json.loads(item.tool_args), ensure_ascii=False, indent=2)
+        except (TypeError, ValueError):
+            detail = item.tool_args
+        detail = _truncate(detail, 4_000)
+    elif item.kind == "tool_result":
+        label = f"工具结果 · {item.tool_name}"
+        detail = _truncate(item.text, 12_000)
+    else:
+        label = f"审批 · {item.tool_name}"
+        detail = _truncate(item.text or "工具审批结果", 2_000)
+    body = detail if not item.collapsed else _trace_summary(item)
+    style = {
+        "streaming": "yellow",
+        "running": "yellow",
+        "success": "green",
+        "failed": "red",
+        "blocked": "yellow",
+        "skipped": "cyan",
+        "cancelled": "yellow",
+    }.get(item.status, "cyan")
+    title = f"{marker} {label} · {_trace_status(item)}"
+    return Panel(Text(body), title=title, border_style=style)
+
+
+def _agent_group_status(status: str) -> str:
+    return {
+        "pending": "待执行",
+        "running": "执行中",
+        "reviewing": "审查中",
+        "repairing": "修复中",
+        "done": "已完成",
+        "failed": "失败",
+        "blocked": "已阻塞",
+        "needs_input": "等待输入",
+        "skipped": "已跳过",
+        "cancelled": "已取消",
+    }.get(status, status or "未知")
+
+
+def agent_group_renderable(group: AgentGroupState):
+    """Render one AgentRun as a collapsible group in the shared transcript."""
+    marker = "▶" if group.collapsed else "▼"
+    identity = f"{group.role}/{group.task_id}" if group.task_id else group.role
+    stats = f"工具 {group.tool_count} · Artifact {group.artifact_count}"
+    header = f"{marker} [{identity}] {group.task_title} · {_agent_group_status(group.status)} · {group.resource_scope_mode} · {stats}"
+    if group.effective_steps:
+        header += f" · 预算 {group.effective_steps} 步"
+    if group.attempt > 1:
+        header += f" · 第 {group.attempt} 次"
+    if group.failure_category:
+        header += f" · {group.failure_category}"
+    if group.repair_attempt:
+        header += f" · 修复 {group.repair_attempt}"
+    latest = group.latest_error or group.latest_summary
+    if group.collapsed and latest:
+        header += f"\n  {latest.replace(chr(10), ' ')[:180]}"
+    parts: list[object] = [Text(header)]
+    if not group.collapsed:
+        if group.entries:
+            parts.append(Group(*(render_item(item) for item in group.entries)))
+        else:
+            parts.append(Text("（尚无事件）", style="dim"))
+    style = {
+        "running": "yellow",
+        "reviewing": "yellow",
+        "repairing": "magenta",
+        "done": "green",
+        "failed": "red",
+        "blocked": "yellow",
+        "skipped": "cyan",
+        "cancelled": "yellow",
+    }.get(group.status, "cyan")
+    return Panel(Group(*parts), title="Agent", border_style=style)
+
+
+class DiagramCard:
+    """Rich renderable for one Mermaid block inside the transcript."""
+
+    def __init__(
+        self,
+        source: str,
+        model: FlowchartModel | None,
+        *,
+        error: str = "",
+        source_visible: bool = False,
+    ) -> None:
+        self.source = source
+        self.model = model
+        self.error = error
+        self.source_visible = source_visible
+
+    def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:
+        available = max(1, options.max_width - 4)
+        if self.model is None:
+            body = f"Mermaid flowchart 暂无法渲染：{self.error}\n\n{_mermaid_source(self.source)}"
+            title = "Flowchart · parse failed"
+            yield Panel(Text(_truncate(body)), title=title, border_style="bright_blue")
+            return
+        else:
+            result = render_flowchart(self.model, width=available)
+            title = f"Flowchart · {self.model.direction} · {len(self.model.nodes)} nodes · {len(self.model.edges)} edges"
+            if result.mode != "unicode":
+                title += f" · {result.mode}"
+            parts: list[object] = [Text(result.text, no_wrap=True, overflow="crop")]
+            if result.warnings:
+                parts.append(Text("\n\n⚠ " + "；".join(result.warnings)))
+            if self.source_visible:
+                parts.append(Text("\n\n" + _mermaid_source(self.source)))
+            parts.append(Text("\n\n[d] 查看/隐藏 Mermaid 源码 · [r] 重新布局"))
+            yield Panel(Group(*parts), title=title, border_style="bright_blue")
+            return
+
+
+def _assistant_renderable(item: TranscriptItem):
+    parts: list[object] = []
+    for text, block in split_mermaid_blocks(_truncate(item.text)):
+        if block is None:
+            if text:
+                try:
+                    parts.append(Markdown(text))
+                except Exception:
+                    parts.append(Text(text))
+            continue
+        try:
+            model = parse_flowchart(block.source)
+            parts.append(DiagramCard(block.source, model, source_visible=item.diagram_source_visible))
+        except FlowchartParseError as exc:
+            # An unsupported block remains visible as source; a malformed
+            # diagram must never prevent the rest of the assistant message.
+            parts.append(DiagramCard(block.source, None, error=str(exc), source_visible=True))
+    if not parts:
+        parts.append(Text("…"))
+    return Panel(Group(*parts), title="XG", border_style="green" if not item.streaming else "yellow")
+
+
+def render_item(item: TranscriptItem):
+    if item.kind == "user":
+        return Panel(Text(item.text), title="你", border_style="cyan")
+    if item.kind == "progress":
+        return Panel(
+            Text(f"… {item.text}"),
+            title="XG · Working",
+            border_style="yellow",
+        )
+    if item.kind == "assistant":
+        return _assistant_renderable(item)
+    if item.kind in ("thinking", "tool_call", "tool_result", "approval") and item.collapsible:
+        return trace_renderable(item)
+    if item.kind == "tool_call":
+        try:
+            args = json.dumps(json.loads(item.tool_args), ensure_ascii=False, indent=2)
+        except (TypeError, ValueError):
+            args = item.tool_args
+        return Panel(Text(_truncate(args, 4_000)), title=f"工具调用 · {item.tool_name}", border_style="cyan")
+    if item.kind == "tool_result":
+        style = "green" if item.tool_ok else "red"
+        return Panel(Text(_truncate(item.text)), title=f"工具结果 · {item.tool_name}", border_style=style)
+    if item.kind == "approval":
+        return Panel(Text(item.text or "工具审批结果"), title=f"审批 · {item.tool_name}", border_style="yellow")
+    if item.kind == "context":
+        return Panel(Text(item.text), title="上下文", border_style="blue")
+    if item.kind == "help":
+        return Panel(Text(_truncate(item.text)), title="命令帮助", border_style="cyan")
+    if item.kind == "plan":
+        plan = item.plan
+        if plan is None:
+            return Panel(Text(item.text), title="计划", border_style="magenta")
+        if item.plan_review:
+            return PlanReviewCard(item)
+        lines = [f"目标：{plan.goal}", f"共 {len(plan.batches)} 轮"]
+        for batch_no, batch in enumerate(plan.batches, 1):
+            lines.append(f"第 {batch_no} 轮：{', '.join(batch)}")
+            for task_id in batch:
+                task = plan.task_by_id(task_id)
+                if task is None:
+                    continue
+                lines.append(f"  [{task.status}] {task.id}  {task.title}")
+                if not item.collapsed:
+                    deps = f"（依赖：{', '.join(task.deps)}）" if task.deps else ""
+                    lines.append(f"      {task.description}{deps}")
+                    mode = getattr(task, "resource_scope_mode", "")
+                    if mode:
+                        lines.append(f"      资源模式：{mode}")
+                    for claim in getattr(task, "resource_claims", ()):
+                        lines.append(f"      资源声明：{claim.access} {claim.pattern}")
+        lines.append("")
+        lines.append("Enter 执行 · d 展开/折叠详情 · r 重规划 · Esc 取消")
+        return Panel(Text("\n".join(lines)), title="计划审阅", border_style="magenta")
+    if item.kind == "error":
+        return Panel(Text(item.text), title="错误", border_style="red")
+    return Text(item.text, style="dim")

xg/tui/state.py

@@ -0,0 +1,240 @@
+"""Pure state types consumed by the Textual widgets."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any, Literal
+
+from xg.tui.i18n import UiLanguage
+
+
+TranscriptKind = Literal[
+    "user", "assistant", "thinking", "tool_call", "tool_result", "approval",
+    "context", "plan", "help", "error", "system", "progress", "agent_group",
+]
+TranscriptStatus = Literal[
+    "streaming", "running", "success", "failed", "cancelled", "done",
+]
+ProgressKind = Literal["response", "plan", "context"]
+QueueItemKind = Literal["task", "plan", "command"]
+InspectorView = Literal["session", "plan", "memory", "safety"]
+TuiPhase = Literal[
+    "idle", "running", "awaiting_approval", "awaiting_plan_review",
+    "awaiting_team_input", "error",
+]
+
+
+@dataclass
+class TranscriptItem:
+    id: str
+    kind: TranscriptKind
+    progress_kind: ProgressKind = "response"
+    text: str = ""
+    tool_name: str = ""
+    tool_args: str = ""
+    tool_call_id: str = ""
+    plan: Any | None = None
+    plan_review: bool = False
+    tool_ok: bool | None = None
+    collapsed: bool = True
+    collapsible: bool = False
+    user_collapsed: bool | None = None
+    diagram_source_visible: bool = False
+    streaming: bool = False
+    turn_id: str = ""
+    trace_id: str = ""
+    parent_call_id: str = ""
+    status: TranscriptStatus = "done"
+    elapsed_ms: int | None = None
+    agent_group_id: str = ""
+
+
+@dataclass
+class AgentGroupState:
+    """One logical AgentRun rendered inside the shared Team transcript."""
+
+    group_id: str
+    team_id: str
+    agent_id: str
+    role: str
+    task_id: str
+    task_title: str
+    resource_scope_mode: str = "targeted"
+    status: str = "pending"
+    collapsed: bool = True
+    user_toggled: bool = False
+    event_count: int = 0
+    tool_count: int = 0
+    artifact_count: int = 0
+    repair_attempt: int = 0
+    attempt: int = 0
+    effective_steps: int = 0
+    failure_category: str = ""
+    latest_summary: str = ""
+    latest_error: str = ""
+    entries: list[TranscriptItem] = field(default_factory=list)
+
+
+@dataclass(frozen=True)
+class QueueItem:
+    id: str
+    text: str
+    kind: QueueItemKind
+    status: Literal["queued"] = "queued"
+
+
+@dataclass(frozen=True)
+class UsageSnapshot:
+    """Current context estimate plus provider request accounting."""
+
+    estimated_prompt_tokens: int = 0
+    context_window: int = 0
+    request_token_limit: int = 0
+    window_ratio: float = 0.0
+    budget_usage_ratio: float = 0.0
+    last_prompt_tokens: int = 0
+    last_completion_tokens: int = 0
+    last_total_tokens: int = 0
+    session_prompt_tokens: int = 0
+    session_completion_tokens: int = 0
+    session_total_tokens: int = 0
+    usage_source: Literal["provider", "estimated", "unavailable"] = "unavailable"
+    compaction_count: int = 0
+    last_compaction_before: int = 0
+    last_compaction_after: int = 0
+
+
+@dataclass(frozen=True)
+class SessionInspectorSnapshot:
+    provider: str = ""
+    model: str = ""
+    status: str = "idle"
+
+
+@dataclass(frozen=True)
+class PlanTaskSnapshot:
+    id: str
+    title: str
+    status: str = "pending"
+
+
+@dataclass(frozen=True)
+class PlanInspectorSnapshot:
+    goal: str = ""
+    status: str = "idle"
+    current_round: int = 0
+    total_rounds: int = 0
+    completed_tasks: int = 0
+    total_tasks: int = 0
+    failure_count: int = 0
+    tasks: tuple[PlanTaskSnapshot, ...] = ()
+
+
+@dataclass(frozen=True)
+class MemoryInspectorSnapshot:
+    project_root: str = ""
+    xg_loaded: bool = False
+    xg_local_loaded: bool = False
+    warning_count: int = 0
+    memory_count: int = 0
+    store_available: bool = True
+    last_operation: str = ""
+
+
+@dataclass(frozen=True)
+class SafetyInspectorSnapshot:
+    hitl_enabled: bool = True
+    session_allow_all: bool = False
+    approval_status: str = "idle"
+    current_tool: str = ""
+    current_level: str = ""
+    last_decision: str = ""
+    last_reason: str = ""
+    path_guard_active: bool = True
+    command_guard_active: bool = True
+    last_rejection: str = ""
+
+
+@dataclass(frozen=True)
+class SmartRouterTierSnapshot:
+    """One tier entry rendered on the Header routing row."""
+
+    tier: str
+    provider: str
+    model: str
+    is_active: bool = False
+    configured: bool = False
+
+
+@dataclass(frozen=True)
+class SmartRouterSnapshot:
+    """SmartRouter routing state consumed by HeaderBar (phase-02).
+
+    Off 态 Header 渲染不读取该快照，因此默认值不影响既有渲染路径。
+    """
+
+    enabled: bool = False
+    tiers: tuple[SmartRouterTierSnapshot, ...] = ()
+    active_tier: str = ""
+    # 第 3 期校准后的置信度展示位，本期恒为 None。
+    confidence: float | None = None
+
+
+@dataclass
+class InspectorState:
+    active_view: InspectorView = "session"
+    session: SessionInspectorSnapshot = field(default_factory=SessionInspectorSnapshot)
+    plan: PlanInspectorSnapshot = field(default_factory=PlanInspectorSnapshot)
+    memory: MemoryInspectorSnapshot = field(default_factory=MemoryInspectorSnapshot)
+    safety: SafetyInspectorSnapshot = field(default_factory=SafetyInspectorSnapshot)
+    smart_router: SmartRouterSnapshot = field(default_factory=SmartRouterSnapshot)
+    provider: str = ""
+    model: str = ""
+    usage: UsageSnapshot = field(default_factory=UsageSnapshot)
+    # Compatibility fields retained for adapters written before UsageSnapshot.
+    context_tokens: int = 0
+    context_window: int = 0
+    memory_count: int = 0
+    hitl_enabled: bool = True
+    plan_status: str = "idle"
+    batch: str = ""
+
+
+@dataclass
+class TuiState:
+    phase: TuiPhase = "idle"
+    transcript: list[TranscriptItem] = field(default_factory=list)
+    inspector: InspectorState = field(default_factory=InspectorState)
+    active_turn_id: str = ""
+    active_input: str = ""
+    pending_approval: Any | None = None
+    pending_plan: Any | None = None
+    pending_confirmation: ConfirmationRequest | None = None
+    team_input_task_id: str = ""
+    team_input_category: str = ""
+    team_input_message: str = ""
+    team_input_scope: list[Any] = field(default_factory=list)
+    notification: str = ""
+    notification_level: Literal["info", "warning", "error"] = "info"
+    plan_tasks: dict[str, str] = field(default_factory=dict)
+    agent_groups: dict[str, AgentGroupState] = field(default_factory=dict)
+    agent_group_order: list[str] = field(default_factory=list)
+    queue: list[QueueItem] = field(default_factory=list)
+    # UI preference only; never serialized into Agent messages or memory.
+    ui_language: UiLanguage = "en"
+
+
+@dataclass(frozen=True)
+class ApprovalRequest:
+    tool_name: str
+    level: str
+    args: dict
+    turn_id: str = ""
+
+
+@dataclass(frozen=True)
+class ConfirmationRequest:
+    kind: str
+    title: str
+    body: str
+    payload: object | None = None

xg/tui/theme.tcss

@@ -0,0 +1,208 @@
+Screen {
+    background: $surface;
+}
+
+#shell {
+    width: 1fr;
+    height: 1fr;
+}
+
+#main-column {
+    width: 1fr;
+    height: 1fr;
+}
+
+#header {
+    height: 2fr;
+    min-height: 5;
+    max-height: 20;
+    color: $text;
+    background: $panel;
+    padding: 0 2;
+    content-align: center middle;
+    text-align: center;
+    text-style: none;
+}
+
+#header.usage-warning {
+    color: $warning;
+}
+
+#header.usage-error {
+    color: $error;
+}
+
+#transcript {
+    width: 1fr;
+    height: 8fr;
+    min-height: 8;
+    padding: 1 2;
+    background: $surface;
+    scrollbar-size: 1 1;
+}
+
+.transcript-empty-state {
+    width: 1fr;
+    height: auto;
+    margin-top: 6;
+    padding: 1 2;
+    color: $text-muted;
+    text-align: center;
+}
+
+.inline-action-card {
+    width: 1fr;
+    height: auto;
+    margin: 1 0 1 0;
+    padding: 1 2;
+    background: $panel;
+    border: round $warning;
+}
+
+.inline-approval-card {
+    border: round $error;
+}
+
+.confirmation-memory_clear {
+    border: round $accent;
+}
+
+.inline-action-card > Static {
+    width: 1fr;
+    height: auto;
+    margin-bottom: 1;
+}
+
+#inspector {
+    width: 36;
+    min-width: 32;
+    height: 1fr;
+    border-left: solid $primary;
+    padding: 1;
+    color: $text-muted;
+}
+
+#inspector-tabs {
+    height: 2;
+    width: 1fr;
+    color: $text-muted;
+}
+
+InspectorTab {
+    width: 1fr;
+    height: 2;
+    content-align: center middle;
+    padding: 0;
+    text-wrap: nowrap;
+    overflow: hidden;
+}
+
+InspectorTab:hover {
+    color: $text;
+    background: $panel;
+}
+
+InspectorTab.active {
+    color: $accent;
+    text-style: bold;
+    border-bottom: heavy $accent;
+}
+
+#inspector-content {
+    height: 1fr;
+    overflow-y: auto;
+}
+
+#inspector-content > Static {
+    width: 1fr;
+    height: auto;
+    color: $text;
+}
+
+#notification {
+    height: auto;
+    max-height: 2;
+    padding: 0 1;
+    color: $warning;
+}
+
+#composer-area {
+    width: 1fr;
+    height: auto;
+    margin: 0;
+}
+
+#queue-status {
+    height: auto;
+    max-height: 5;
+    padding: 0 1;
+    color: $text-muted;
+    background: $panel;
+}
+
+#command-suggestions {
+    width: 1fr;
+    height: auto;
+    max-height: 8;
+    padding: 0 1;
+    color: $text-muted;
+    background: $panel;
+    border: round $primary;
+    scrollbar-size: 1 1;
+}
+
+#command-suggestions > .option-list--option-highlighted {
+    color: $text;
+    background: $accent;
+}
+
+#composer-label {
+    height: 1;
+    padding: 0 1;
+    color: $text;
+    background: $panel;
+    text-style: bold;
+}
+
+#composer {
+    height: 3;
+    margin: 0;
+    border: round $primary;
+    background: $surface;
+}
+
+#composer:focus {
+    border: double $accent;
+}
+
+#footer {
+    width: 1fr;
+    height: 1;
+    color: $text-muted;
+    background: $panel;
+    padding: 0 1;
+}
+
+ApprovalModal, PlanModal {
+    align: center middle;
+}
+
+#approval-dialog {
+    width: 70;
+    max-width: 90%;
+    height: auto;
+    max-height: 80%;
+    padding: 1 2;
+    background: $panel;
+    border: thick $warning;
+}
+
+#confirm-dialog {
+    width: 70;
+    max-width: 90%;
+    height: auto;
+    max-height: 80%;
+    padding: 1 2;
+    background: $panel;
+    border: thick $warning;
+}

xg/tui/widgets/__init__.py

@@ -0,0 +1 @@
+"""Reusable fullscreen UI widgets."""