agentengine-sdk-python 0.6.9__py3-none-any.whl → 0.7.0__py3-none-any.whl

ksadk/tui/stream_render.py

@@ -0,0 +1,68 @@
+"""TUI 流式渲染共享纯函数。
+
+从 RemoteRunner.stream 的归一化 chunk 提取 delta/usage/终止信号，格式化 usage 文本。
+被 loop.py（交互 TUI）和 cmd_invoke._invoke_once（-m 单次）共用，保证两路径渲染口径一致。
+"""
+from __future__ import annotations
+
+import re
+from typing import Any, Optional
+
+
+def clean_response(text: str) -> str:
+    """清理 LLM 响应中的内部调试伪影。
+
+    只删内部调试残留（Tool Result 标记、python repr 片段、tool_call_id），
+    不动正常的 XML/HTML 标签（用户可能需要 <tag> 输出）。
+    """
+    text = re.sub(r"\[Tool Result:.*?\]", "", text, flags=re.DOTALL)
+    text = re.sub(r"<think>.*?</think>", "", text, flags=re.DOTALL)
+    text = re.sub(r"name='[^']*'\s*tool_call_id='[^']*'", "", text)
+    return text.strip()
+
+
+def extract_stream_delta(chunk: dict) -> tuple[str, dict | None, bool]:
+    """从 RemoteRunner.stream 的 chunk 提取 (delta 文本, usage, 是否终止信号)。
+
+    - text/thinking 等普通 chunk：取 delta，非终止。
+    - final（chat 路径结束，output=完整文本）/ responses_output（responses 路径
+      response.completed，output=list）：只取 usage，不把 output 当 delta——
+      否则末尾会把完整文本或 list 再 append 一遍（重复/乱码）。
+    """
+    chunk_type = str(chunk.get("type") or "text")
+    if chunk_type in {"final", "responses_output"}:
+        usage = chunk.get("usage")
+        usage = dict(usage) if isinstance(usage, dict) else None
+        # terminal chunk 可能带 metadata.last_usage（上下文用量），合并进 usage
+        # 让 TUI 能显示 context 占比，不丢这条 metadata。
+        metadata = chunk.get("metadata")
+        if isinstance(metadata, dict) and isinstance(metadata.get("last_usage"), dict):
+            if usage is None:
+                usage = {}
+            usage.setdefault("last_usage", dict(metadata["last_usage"]))
+        return "", (usage or None), True
+    delta = chunk.get("delta") or ""
+    if not isinstance(delta, str):
+        delta = str(delta or "")
+    return delta, None, False
+
+
+def format_usage(usage: Optional[dict[str, Any]]) -> str:
+    """把 usage dict 格式成紧凑文本：↑输入 ↓输出 ⌀总计。
+
+    兼容 input_tokens/output_tokens（chat 路径）与 prompt_tokens/completion_tokens
+    （responses 路径）两种字段名。
+    """
+    if not isinstance(usage, dict) or not usage:
+        return ""
+    inp = usage.get("input_tokens") or usage.get("prompt_tokens")
+    out = usage.get("output_tokens") or usage.get("completion_tokens")
+    total = usage.get("total_tokens")
+    parts = []
+    if inp:
+        parts.append(f"↑{inp}")
+    if out:
+        parts.append(f"↓{out}")
+    if total:
+        parts.append(f"⌀{total}")
+    return " ".join(parts)

ksadk/version.py

@@ -1,4 +1,4 @@
 """KsADK 版本信息"""
 
-VERSION = "0.6.9"
+VERSION = "0.7.0"
 __version__ = VERSION

ksadk/server/static/assets/architecture-7EHR7CIX-Cr6Xz1_c.js

@@ -1 +0,0 @@
-import{x as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createArchitectureServices};

ksadk/server/static/assets/channel-BSSOwFIb.js

@@ -1 +0,0 @@
-import{Dn as e,En as t}from"./MermaidBlock-DCR8WZ17.js";var n=(n,r)=>e.lang.round(t.parse(n)[r]);export{n as t};

ksadk/server/static/assets/chunk-QZHKN3VN-XXpdiwUL.js

@@ -1 +0,0 @@
-import{kn as e}from"./MermaidBlock-DCR8WZ17.js";var t=class{constructor(e){this.init=e,this.records=this.init()}static{e(this,`ImperativeState`)}reset(){this.records=this.init()}};export{t};

ksadk/server/static/assets/classDiagram-4FO5ZUOK-C5HvGiyK.js

@@ -1 +0,0 @@
-import"./chunk-FMBD7UC4-PrapyfMu.js";import"./chunk-ND2GUHAM-CboKhTZ3.js";import"./chunk-55IACEB6-DOairxzw.js";import"./chunk-2J33WTMH-DLVdkeyQ.js";import{i as e,n as t,r as n,t as r}from"./chunk-727SXJPM-hCO487gs.js";import{kn as i}from"./MermaidBlock-DCR8WZ17.js";var a={parser:t,get db(){return new r},renderer:n,styles:e,init:i(e=>{e.class||={},e.class.arrowMarkerAbsolute=e.arrowMarkerAbsolute},`init`)};export{a as diagram};

ksadk/server/static/assets/classDiagram-v2-Q7XG4LA2-C5HvGiyK.js

@@ -1 +0,0 @@
-import"./chunk-FMBD7UC4-PrapyfMu.js";import"./chunk-ND2GUHAM-CboKhTZ3.js";import"./chunk-55IACEB6-DOairxzw.js";import"./chunk-2J33WTMH-DLVdkeyQ.js";import{i as e,n as t,r as n,t as r}from"./chunk-727SXJPM-hCO487gs.js";import{kn as i}from"./MermaidBlock-DCR8WZ17.js";var a={parser:t,get db(){return new r},renderer:n,styles:e,init:i(e=>{e.class||={},e.class.arrowMarkerAbsolute=e.arrowMarkerAbsolute},`init`)};export{a as diagram};

ksadk/server/static/assets/eventmodeling-FCH6USID-CIgwHJLL.js

@@ -1 +0,0 @@
-import{y as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createEventModelingServices};

ksadk/server/static/assets/gitGraph-WXDBUCRP-BVAJ7uo0.js

@@ -1 +0,0 @@
-import{_ as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createGitGraphServices};

ksadk/server/static/assets/info-J43DQDTF-MT80V4ge.js

@@ -1 +0,0 @@
-import{h as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createInfoServices};

ksadk/server/static/assets/packet-YPE3B663-BLKhbXZB.js

@@ -1 +0,0 @@
-import{p as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createPacketServices};

ksadk/server/static/assets/pie-LRSECV5Y-DFOikjSY.js

@@ -1 +0,0 @@
-import{d as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createPieServices};

ksadk/server/static/assets/radar-GUYGQ44K-095b1Ikw.js

@@ -1 +0,0 @@
-import{l as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createRadarServices};

ksadk/server/static/assets/stateDiagram-v2-BHNVJYJU-J9kCb0nC.js

@@ -1 +0,0 @@
-import"./chunk-55IACEB6-DOairxzw.js";import"./chunk-2J33WTMH-DLVdkeyQ.js";import{i as e,n as t,r as n,t as r}from"./chunk-AQP2D5EJ-XU8vDOuG.js";import{kn as i}from"./MermaidBlock-DCR8WZ17.js";var a={parser:t,get db(){return new r(2)},renderer:n,styles:e,init:i(e=>{e.state||={},e.state.arrowMarkerAbsolute=e.arrowMarkerAbsolute},`init`)};export{a as diagram};

ksadk/server/static/assets/treeView-BLDUP644-yinfJEY6.js

@@ -1 +0,0 @@
-import{s as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createTreeViewServices};

ksadk/server/static/assets/treemap-LRROVOQU-DosUqN1Q.js

@@ -1 +0,0 @@
-import{a as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createTreemapServices};

ksadk/server/static/assets/wardley-L42UT6IY-Bvp7-5uH.js

@@ -1 +0,0 @@
-import{r as e}from"./mermaid-parser.core-D2eMVOgo.js";export{e as createWardleyServices};

ksadk/tui/app.py

@@ -1,400 +0,0 @@
-"""
-KsADK Agent TUI - 简洁交互界面
-
-基本功能：
-- 对话交互
-- 思考过程显示（通过 --show-thinking 参数控制）
-- 工具调用显示
-- 退出：输入 exit/quit 或 Ctrl+C/Ctrl+D
-"""
-
-from __future__ import annotations
-
-import asyncio
-import os
-import uuid
-import re
-from pathlib import Path
-from typing import TYPE_CHECKING, Any, Dict, List, Optional
-
-from textual import on
-from textual.app import App, ComposeResult
-from textual.binding import Binding
-from textual.containers import Container, Vertical, VerticalScroll
-from textual.screen import ModalScreen
-from textual.widgets import Static
-from textual.events import MouseUp
-
-from rich.panel import Panel
-
-from ksadk.tui.clipboard import copy_selection_to_clipboard
-from ksadk.tui.widgets.user import UserMessage
-from ksadk.tui.widgets.assistant import AssistantMessage
-from ksadk.tui.widgets.thinking import ThinkingMessage
-from ksadk.tui.widgets.system import SystemMessage
-from ksadk.tui.widgets.chat_input import ChatInput
-
-if TYPE_CHECKING:
-    from ksadk.runners.base_runner import BaseRunner
-
-
-def _clean_response(text: str) -> str:
-    """清理 LLM 响应中的内部调试信息"""
-    text = re.sub(r'\[Tool Result:.*?\]', '', text, flags=re.DOTALL)
-    text = re.sub(r'<tool_call>.*?</tool_call>', '', text, flags=re.DOTALL)
-    text = re.sub(r"name='[^']*'\s*tool_call_id='[^']*'", '', text)
-    return text.strip()
-
-
-class ApprovalScreen(ModalScreen[bool]):
-    """敏感操作确认弹窗"""
-
-    BINDINGS = [
-        Binding("y", "approve", "确认"),
-        Binding("n", "reject", "取消"),
-        Binding("escape", "reject", "取消"),
-    ]
-
-    def __init__(self, tool_name: str, args: Dict[str, Any], **kwargs):
-        super().__init__(**kwargs)
-        self.tool_name = tool_name
-        self.tool_args = args
-
-    def compose(self) -> ComposeResult:
-        args_str = "\n".join(f"  {k}: {v}" for k, v in self.tool_args.items()) if self.tool_args else "  (无参数)"
-        yield Container(
-            Static(
-                Panel(
-                    f"[bold yellow]⚠️  需要您确认敏感操作[/]\n\n"
-                    f"[bold]操作:[/] {self.tool_name}\n"
-                    f"[bold]参数:[/]\n{args_str}\n\n"
-                    f"[green]y[/] 确认  [red]n[/] 取消",
-                    title="🔒 确认",
-                    border_style="yellow",
-                ),
-            ),
-            id="approval-dialog",
-        )
-
-    def action_approve(self) -> None:
-        self.dismiss(True)
-
-    def action_reject(self) -> None:
-        self.dismiss(False)
-
-    CSS = """
-    #approval-dialog {
-        align: center middle;
-        width: 60;
-        height: auto;
-    }
-    """
-
-
-class AgentTUI(App):
-    """KsADK Agent TUI - 简洁交互界面"""
-
-    CSS = """
-    Screen {
-        background: $background;
-    }
-
-    #main-scroll {
-        height: 100%;
-        padding: 0 1;
-    }
-
-    #title-bar {
-        height: 1;
-        color: $primary;
-        padding: 0;
-    }
-
-    #welcome-area {
-        height: auto;
-        padding: 2 0;
-        content-align: center middle;
-        text-align: center;
-    }
-
-    #chat-log {
-        height: auto;
-        background: transparent;
-        padding: 0;
-    }
-
-    #input-area {
-        height: auto;
-        padding: 1 0;
-    }
-
-    #hint {
-        height: 1;
-        color: $text-muted;
-        padding: 0 0 0 2;
-    }
-
-    .started #welcome-area {
-        display: none;
-    }
-    """
-
-    BINDINGS = [
-        Binding("escape", "interrupt", "中断", show=False, priority=True),
-        Binding("ctrl+c", "quit_or_interrupt", "退出/中断", show=False),
-        Binding("ctrl+d", "quit_app", "退出", show=False, priority=True),
-        Binding("ctrl+q", "quit_app", "退出", show=False),
-    ]
-
-    TITLE = "KsADK"
-
-    def __init__(
-        self,
-        runner: "BaseRunner",
-        show_thinking: bool = False,
-        project_dir: str = ".",
-        **kwargs,
-    ):
-        super().__init__(**kwargs)
-        self.runner = runner
-        self.show_thinking = show_thinking
-        self.project_dir = Path(project_dir).resolve()
-
-        self.session_id = getattr(runner, "session_id", None) or str(uuid.uuid4())[:8]
-        if getattr(self.runner, "session_id", None) is None:
-            self.runner.session_id = self.session_id
-        self.history: List[Dict[str, str]] = []
-        self._is_streaming = False
-        self._started = False
-        self.model_name = os.getenv("MODEL_NAME", "unknown")
-
-        try:
-            from ksadk import __version__
-            self.version = __version__
-        except ImportError:
-            self.version = "0.2.0"
-
-    def compose(self) -> ComposeResult:
-        """构建 UI 布局"""
-        with VerticalScroll(id="main-scroll"):
-            yield Static(f"─ KsADK v{self.version} ─", id="title-bar")
-
-            yield Container(
-                Static(
-                    f"[bold]Welcome![/]\n"
-                    f"🤖\n"
-                    f"[dim]{self.model_name} · Interactive Mode[/]\n"
-                    f"[dim]{self._short_path()}[/]",
-                    id="welcome-content",
-                ),
-                id="welcome-area",
-            )
-
-            with Vertical(id="chat-log"):
-                pass
-
-            yield Container(
-                ChatInput(cwd=self.project_dir, id="chat-input-widget"),
-                Static("Ctrl+C 退出", id="hint"),
-                id="input-area",
-            )
-
-    def _short_path(self) -> str:
-        home = Path.home()
-        try:
-            return "~/" + str(self.project_dir.relative_to(home))
-        except ValueError:
-            return str(self.project_dir)
-
-    def on_mount(self) -> None:
-        self.query_one("#chat-input-widget", ChatInput).focus_input()
-
-    @on(ChatInput.Submitted)
-    async def handle_input(self, event: ChatInput.Submitted) -> None:
-        """处理用户输入"""
-        user_input = event.value.strip()
-
-        if not user_input:
-            return
-
-        # 检查退出
-        if user_input.lower() in ("exit", "quit", "退出"):
-            self.exit()
-            return
-
-        # 隐藏欢迎区域
-        if not self._started:
-            self._started = True
-            self.add_class("started")
-
-        asyncio.create_task(self._stream_response(user_input))
-
-    async def _stream_response(self, user_input: str) -> None:
-        """流式获取 Agent 响应"""
-        chat_log = self.query_one("#chat-log", Vertical)
-        main_scroll = self.query_one("#main-scroll", VerticalScroll)
-
-        self._is_streaming = True
-
-        input_data = {
-            "input": user_input,
-            "session_id": self.session_id,
-            "history": self.history,
-        }
-
-        thinking_msg: Optional[ThinkingMessage] = None
-        assistant_msg: Optional[AssistantMessage] = None
-        full_response_text = ""
-
-        try:
-            await chat_log.mount(UserMessage(user_input))
-            main_scroll.scroll_end(animate=False)
-
-            async for chunk in self.runner.stream(input_data):
-                if not self._is_streaming:
-                    if thinking_msg:
-                        thinking_msg.stop()
-                    if assistant_msg:
-                        await assistant_msg.stop_stream()
-                    await chat_log.mount(SystemMessage("已中断", level="warning"))
-                    break
-
-                chunk_type = chunk.get("type", "text")
-
-                # 思考过程
-                if chunk_type == "thinking":
-                    delta = chunk.get("delta", "")
-                    if delta and self.show_thinking:
-                        if not thinking_msg:
-                            thinking_msg = ThinkingMessage()
-                            await chat_log.mount(thinking_msg)
-                            main_scroll.scroll_end(animate=False)
-                        await thinking_msg.append_content(delta)
-                    continue
-
-                if thinking_msg and thinking_msg.is_active:
-                    thinking_msg.stop()
-
-                # 中断/确认
-                if chunk_type == "interrupt":
-                    await self._handle_interrupt(chunk)
-                    return
-
-                # 工具调用 - 跳过不显示
-                if chunk_type == "tool_call":
-                    continue
-
-                # 文本响应
-                delta = chunk.get("output", "") or chunk.get("delta", "")
-                if delta:
-                    if not assistant_msg:
-                        assistant_msg = AssistantMessage()
-                        await chat_log.mount(assistant_msg)
-                        main_scroll.scroll_end(animate=False)
-                    await assistant_msg.append_content(delta)
-                    full_response_text += delta
-
-            if thinking_msg:
-                thinking_msg.stop()
-            if assistant_msg:
-                await assistant_msg.stop_stream()
-
-            cleaned_response = _clean_response(full_response_text)
-            if cleaned_response:
-                self.history.append({"role": "user", "content": user_input})
-                self.history.append({"role": "model", "content": cleaned_response})
-            elif self._is_streaming and not full_response_text:
-                result = await self.runner.invoke(input_data)
-                response_text = result.get("output", "")
-                cleaned_response = _clean_response(response_text)
-
-                if cleaned_response:
-                    assistant_msg = AssistantMessage()
-                    await chat_log.mount(assistant_msg)
-                    await assistant_msg.append_content(cleaned_response)
-                    await assistant_msg.stop_stream()
-                    self.history.append({"role": "user", "content": user_input})
-                    self.history.append({"role": "model", "content": cleaned_response})
-                else:
-                    await chat_log.mount(SystemMessage("(无响应)", level="warning"))
-
-        except asyncio.CancelledError:
-            await chat_log.mount(SystemMessage("已取消", level="warning"))
-        except Exception as e:
-            await chat_log.mount(SystemMessage(f"错误: {e}", level="error"))
-        finally:
-            self._is_streaming = False
-            main_scroll.scroll_end(animate=False)
-
-    async def _handle_interrupt(self, interrupt_chunk: Dict[str, Any]) -> None:
-        """处理敏感操作确认"""
-        interrupt_info = interrupt_chunk.get("interrupt_info", {})
-
-        if isinstance(interrupt_info, dict):
-            tool_name = interrupt_info.get("tool", "未知操作")
-            args = interrupt_info.get("args", {})
-        elif isinstance(interrupt_info, list) and interrupt_info:
-            first = interrupt_info[0]
-            tool_name = str(getattr(first, "value", first))
-            args = {}
-        else:
-            tool_name = str(interrupt_info)
-            args = {}
-
-        approved = await self.push_screen_wait(ApprovalScreen(tool_name, args))
-        chat_log = self.query_one("#chat-log", Vertical)
-
-        if approved:
-            await chat_log.mount(SystemMessage("✓ 已确认，继续执行...", level="success"))
-
-            resume_data = {
-                "input": "确认",
-                "session_id": interrupt_chunk.get("session_id", self.session_id),
-                "history": self.history,
-                "resume": True,
-            }
-
-            assistant_msg = AssistantMessage()
-            await chat_log.mount(assistant_msg)
-
-            try:
-                async for chunk in self.runner.stream(resume_data):
-                    delta = chunk.get("output", "") or chunk.get("delta", "")
-                    if delta:
-                        await assistant_msg.append_content(delta)
-                await assistant_msg.stop_stream()
-                if assistant_msg.content:
-                    self.history.append({"role": "model", "content": assistant_msg.content})
-            except Exception as e:
-                await chat_log.mount(SystemMessage(f"错误: {e}", level="error"))
-        else:
-            await chat_log.mount(SystemMessage("✗ 已取消操作", level="warning"))
-
-    def action_interrupt(self) -> None:
-        self._is_streaming = False
-
-    def action_quit_or_interrupt(self) -> None:
-        if self._is_streaming:
-            self.action_interrupt()
-        else:
-            self.exit()
-
-    def action_quit_app(self) -> None:
-        self.exit()
-
-    def on_mouse_up(self, event: MouseUp) -> None:
-        copy_selection_to_clipboard(self)
-
-
-def run_tui(
-    runner: "BaseRunner",
-    show_thinking: bool = False,
-    project_dir: str = ".",
-) -> None:
-    """启动 TUI 应用"""
-    app = AgentTUI(
-        runner=runner,
-        show_thinking=show_thinking,
-        project_dir=project_dir,
-    )
-    app.run()

ksadk/tui/clipboard.py

@@ -1,100 +0,0 @@
-"""剪贴板工具 - 支持鼠标选择复制"""
-
-from __future__ import annotations
-
-import base64
-import os
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from textual.app import App
-
-_PREVIEW_MAX_LENGTH = 40
-
-
-def _copy_osc52(text: str) -> None:
-    """使用 OSC 52 转义序列复制（支持 SSH/tmux）"""
-    encoded = base64.b64encode(text.encode("utf-8")).decode("ascii")
-    osc52_seq = f"\033]52;c;{encoded}\a"
-    if os.environ.get("TMUX"):
-        osc52_seq = f"\033Ptmux;\033{osc52_seq}\033\\"
-
-    try:
-        with open("/dev/tty", "w") as tty:
-            tty.write(osc52_seq)
-            tty.flush()
-    except Exception:
-        raise RuntimeError("OSC52 复制失败")
-
-
-def _shorten_preview(texts: list[str]) -> str:
-    """缩短文本预览"""
-    dense_text = "⏎".join(texts).replace("\n", "⏎")
-    if len(dense_text) > _PREVIEW_MAX_LENGTH:
-        return f"{dense_text[: _PREVIEW_MAX_LENGTH - 1]}…"
-    return dense_text
-
-
-def _clipboard_copy_methods(app: App):
-    copy_methods = []
-    if os.name != "nt":
-        copy_methods.append(_copy_osc52)
-
-    try:
-        import pyperclip
-        copy_methods.append(pyperclip.copy)
-    except ImportError:
-        pass
-
-    copy_methods.append(app.copy_to_clipboard)
-    return copy_methods
-
-
-def copy_selection_to_clipboard(app: App) -> None:
-    """复制选中文本到剪贴板
-    
-    遍历所有 widgets 获取选中文本并复制到系统剪贴板
-    """
-    selected_texts = []
-
-    for widget in app.query("*"):
-        if not hasattr(widget, "text_selection") or not widget.text_selection:
-            continue
-
-        selection = widget.text_selection
-
-        try:
-            result = widget.get_selection(selection)
-        except Exception:
-            continue
-
-        if not result:
-            continue
-
-        selected_text, _ = result
-        if selected_text.strip():
-            selected_texts.append(selected_text)
-
-    if not selected_texts:
-        return
-
-    combined_text = "\n".join(selected_texts)
-
-    for copy_fn in _clipboard_copy_methods(app):
-        try:
-            copy_fn(combined_text)
-            app.notify(
-                f'"{_shorten_preview(selected_texts)}" 已复制',
-                severity="information",
-                timeout=2,
-            )
-            return
-        except Exception:
-            continue
-
-    # 所有方式都失败
-    app.notify(
-        "复制失败 - 剪贴板不可用",
-        severity="warning",
-        timeout=3,
-    )

ksadk/tui/widgets/__init__.py

@@ -1,22 +0,0 @@
-"""
-TUI Widgets - 消息组件
-
-提供各类消息组件，用于 TUI 渲染。
-参考 deepagents-cli 的消息组件设计。
-"""
-
-from .base import MessageWidget
-from .user import UserMessage
-from .assistant import AssistantMessage
-from .thinking import ThinkingMessage
-from .tool_call import ToolCallMessage
-from .system import SystemMessage
-
-__all__ = [
-    "MessageWidget",
-    "UserMessage",
-    "AssistantMessage",
-    "ThinkingMessage",
-    "ToolCallMessage",
-    "SystemMessage",
-]