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

ksadk/tui/stream_render.py

@@ -0,0 +1,69 @@
+"""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/ui_config.py

@@ -6,7 +6,6 @@ from dataclasses import dataclass
 from typing import Any, Dict, Optional, Tuple
 from urllib.parse import urlsplit
 
-
 UI_PROFILE_AUTO = "auto"
 UI_PROFILE_ADK = "adk"
 UI_PROFILE_LANGCHAIN = "langchain"
@@ -89,11 +88,14 @@ def default_ui_path(profile: str) -> str:
     return _DEFAULT_PATH_BY_PROFILE.get(profile, "/")
 
 
-def extract_ui_state(state: Optional[Dict[str, Any]]) -> Tuple[Optional[str], Optional[str], Optional[str]]:
+def extract_ui_state(
+    state: Optional[Dict[str, Any]],
+) -> Tuple[Optional[str], Optional[str], Optional[str]]:
     if not isinstance(state, dict):
         return None, None, None
 
-    nested = state.get("ui") if isinstance(state.get("ui"), dict) else {}
+    nested_value = state.get("ui")
+    nested: Dict[str, Any] = nested_value if isinstance(nested_value, dict) else {}
 
     profile = state.get("ui_profile") or nested.get("profile")
     path = state.get("ui_path") or nested.get("path")

ksadk/version.py

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

ksadk_runtime_common/schemas/runtime_event_v1.json

@@ -0,0 +1,95 @@
+{
+  "$schema": "https://json-schema.org/draft/2020-12/schema",
+  "$id": "https://kingsoftcloud.github.io/ksadk-python/schemas/runtime_event_v1.json",
+  "title": "RuntimeEvent v1",
+  "description": "RuntimeEvent v1 信封 (goal-02 / G0.2 冻结)。additive 演进:只增字段/事件类型,不改既有字段语义。",
+  "type": "object",
+  "required": [
+    "schema_version",
+    "event_id",
+    "event_type",
+    "timestamp",
+    "agent_id",
+    "user_id",
+    "session_id",
+    "invocation_id",
+    "seq_id",
+    "payload"
+  ],
+  "properties": {
+    "schema_version": {"const": 1},
+    "event_id": {"type": "string", "minLength": 1},
+    "event_type": {"type": "string", "minLength": 1},
+    "timestamp": {"type": "number"},
+    "agent_id": {"type": "string"},
+    "user_id": {"type": "string"},
+    "session_id": {"type": "string"},
+    "invocation_id": {"type": "string"},
+    "seq_id": {"type": "integer"},
+    "phase": {"enum": ["commentary", "final_answer"]},
+    "payload": {"type": "object"}
+  },
+  "additionalProperties": false,
+  "allOf": [
+    {
+      "if": {"properties": {"event_type": {"enum": ["text.delta", "text.completed"]}}},
+      "then": {"properties": {"payload": {"required": ["text"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"enum": ["reasoning.delta", "reasoning.completed"]}}},
+      "then": {"properties": {"payload": {"required": ["text"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"enum": ["tool.call.begin", "tool.call.end"]}}},
+      "then": {"properties": {"payload": {"required": ["call_id", "name"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"enum": ["artifact.created", "artifact.updated"]}}},
+      "then": {"properties": {"payload": {"required": ["name", "version"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"const": "approval.requested"}}},
+      "then": {"properties": {"payload": {"required": ["approval_id", "call_id", "kind"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"const": "approval.resolved"}}},
+      "then": {"properties": {"payload": {"required": ["approval_id", "call_id", "decision"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"enum": ["run.started", "run.progress", "run.interrupted", "run.completed", "run.canceled"]}}},
+      "then": {"properties": {"payload": {"required": ["status"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"const": "run.failed"}}},
+      "then": {"properties": {"payload": {"required": ["status", "error"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"const": "checkpoint.created"}}},
+      "then": {"properties": {"payload": {"required": ["checkpoint_id", "granularity"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"const": "checkpoint.resumed"}}},
+      "then": {"properties": {"payload": {"required": ["checkpoint_id"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"const": "usage.reported"}}},
+      "then": {"properties": {"payload": {"required": ["input_tokens", "output_tokens", "total_tokens"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"enum": ["a2ui.surface.begin", "a2ui.surface.update", "a2ui.surface.end", "a2ui.interaction", "a2ui.action"]}}},
+      "then": {"properties": {"payload": {"required": ["surface_id"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"const": "a2a.task.created"}}},
+      "then": {"properties": {"payload": {"required": ["task_id", "origin"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"const": "a2a.task.status"}}},
+      "then": {"properties": {"payload": {"required": ["task_id", "origin", "status"]}}}
+    },
+    {
+      "if": {"properties": {"event_type": {"const": "a2a.task.artifact"}}},
+      "then": {"properties": {"payload": {"required": ["task_id", "origin"]}}}
+    }
+  ]
+}