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

ksadk/tui/widgets/thinking.py

@@ -1,110 +0,0 @@
-"""
-ThinkingMessage - 思考过程组件
-
-带 spinner 动画的思考过程展示
-"""
-
-from textual.widgets import Static
-from textual.reactive import reactive
-from textual.timer import Timer
-from typing import ClassVar, Any
-from time import time
-
-from .base import MessageWidget
-
-
-class ThinkingMessage(MessageWidget):
-    """思考过程组件
-    
-    显示 AI 思考过程，支持：
-    - spinner 动画
-    - 实时计时
-    - 流式内容追加
-    - 可折叠
-    """
-    
-    DEFAULT_CSS = """
-    ThinkingMessage {
-        height: auto;
-        padding: 0 1;
-        margin: 0 0 1 0;
-        border-left: wide #7777aa;
-        color: #ffffff;
-        background: $surface;
-    }
-    
-    ThinkingMessage .thinking-header {
-        color: #cccccc;
-        width: 100%;
-    }
-    
-    ThinkingMessage .thinking-content {
-        color: #ffffff;
-        margin-left: 2;
-        width: 100%;
-    }
-
-    /* 确保所有子组件都有正确的颜色 */
-    ThinkingMessage * {
-        color: #ffffff;
-    }
-    """
-    
-    # Spinner 动画帧
-    _SPINNER_FRAMES: ClassVar[tuple[str, ...]] = ("⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏")
-    
-    is_active: reactive[bool] = reactive(True)
-    
-    def __init__(self, content: str = "", **kwargs: Any) -> None:
-        super().__init__(content=content, **kwargs)
-        self._thinking_content = content
-        self._spinner_position = 0
-        self._start_time: float = time()
-        self._animation_timer: Timer | None = None
-        self._header_widget: Static | None = None
-        self._content_widget: Static | None = None
-    
-    def compose(self):
-        """构建思考消息布局"""
-        self._header_widget = Static("🧠 Thinking...", classes="thinking-header")
-        yield self._header_widget
-        
-        self._content_widget = Static(self._thinking_content, classes="thinking-content")
-        yield self._content_widget
-    
-    def on_mount(self) -> None:
-        """启动动画"""
-        if self.is_active:
-            self._animation_timer = self.set_interval(0.1, self._update_animation)
-    
-    def _update_animation(self) -> None:
-        """更新 spinner 动画"""
-        if not self.is_active or self._header_widget is None:
-            return
-        
-        frame = self._SPINNER_FRAMES[self._spinner_position]
-        self._spinner_position = (self._spinner_position + 1) % len(self._SPINNER_FRAMES)
-        
-        elapsed = int(time() - self._start_time)
-        self._header_widget.update(f"[bold #ffffff]{frame}[/] [italic #dddddd]Thinking... ({elapsed}s)[/]")
-    
-    async def append_content(self, delta: str) -> None:
-        """追加思考内容"""
-        self._thinking_content += delta
-        if self._content_widget:
-            self._content_widget.update(self._thinking_content)
-    
-    def stop(self) -> None:
-        """停止思考动画"""
-        self.is_active = False
-        if self._animation_timer:
-            self._animation_timer.stop()
-            self._animation_timer = None
-        
-        if self._header_widget:
-            elapsed = int(time() - self._start_time)
-            self._header_widget.update(f"[bold #ffffff]💭[/] [italic #aaaaaa]Thought for {elapsed}s[/]")
-    
-    @property
-    def thinking_content(self) -> str:
-        return self._thinking_content

ksadk/tui/widgets/tool_call.py

@@ -1,161 +0,0 @@
-"""
-ToolCallMessage - 工具调用消息组件
-
-显示 Tool Call 状态和参数。
-"""
-
-from textual.widgets import Static
-from textual.reactive import reactive
-from textual.containers import Vertical
-from typing import Any, Dict, Optional
-import json
-
-from .base import MessageWidget
-
-
-class ToolStatus:
-    """Tool Call 状态"""
-    PENDING = "pending"
-    RUNNING = "running"
-    SUCCESS = "success"
-    ERROR = "error"
-    REJECTED = "rejected"
-
-
-class ToolCallMessage(MessageWidget):
-    """工具调用消息组件
-    
-    用于显示 Tool Call 的状态和参数：
-    - pending: 等待执行（黄色）
-    - running: 执行中（蓝色动画）
-    - success: 成功（绿色）
-    - error: 失败（红色）
-    - rejected: 被拒绝（灰色）
-    """
-    
-    DEFAULT_CSS = """
-    ToolCallMessage {
-        width: 100%;
-        padding: 1;
-        margin: 0 0 1 0;
-        border: round $secondary;
-        background: $surface;
-        color: #ffffff;
-    }
-    
-    ToolCallMessage.pending {
-        border: round $warning;
-    }
-    
-    ToolCallMessage.running {
-        border: round $primary;
-    }
-    
-    ToolCallMessage.success {
-        border: round $success;
-    }
-    
-    ToolCallMessage.error {
-        border: round $error;
-    }
-    
-    ToolCallMessage.rejected {
-        border: round $surface-darken-2;
-        opacity: 0.7;
-    }
-    
-    ToolCallMessage .tool-header {
-        text-style: bold;
-        color: #ffffff;
-    }
-    
-    ToolCallMessage .tool-args {
-        color: #aaaaaa;
-        margin-left: 2;
-    }
-    
-    ToolCallMessage .tool-output {
-        margin-top: 1;
-        padding: 1;
-        background: $surface-darken-1;
-        color: #ffffff;
-    }
-    """
-    
-    status: reactive[str] = reactive(ToolStatus.PENDING)
-    output: reactive[str] = reactive("")
-    
-    def __init__(
-        self,
-        tool_name: str,
-        args: Dict[str, Any],
-        tool_id: Optional[str] = None,
-        **kwargs
-    ) -> None:
-        super().__init__(**kwargs)
-        self._tool_name = tool_name
-        self._args = args
-        self._tool_id = tool_id
-        self.add_class("pending")
-    
-    def compose(self):
-        """构建组件"""
-        status_icon = self._get_status_icon()
-        yield Static(
-            f"{status_icon} {self._tool_name}",
-            classes="tool-header"
-        )
-        
-        # 显示参数（截断过长的值）
-        args_display = self._format_args(self._args)
-        yield Static(args_display, classes="tool-args")
-        
-        # 输出区域
-        if self.output:
-            yield Static(self.output, classes="tool-output")
-    
-    def _get_status_icon(self) -> str:
-        """获取状态图标"""
-        icons = {
-            ToolStatus.PENDING: "⏳",
-            ToolStatus.RUNNING: "🔄",
-            ToolStatus.SUCCESS: "✅",
-            ToolStatus.ERROR: "❌",
-            ToolStatus.REJECTED: "🚫",
-        }
-        return icons.get(self.status, "⏳")
-    
-    def _format_args(self, args: Dict[str, Any], max_len: int = 100) -> str:
-        """格式化参数显示"""
-        try:
-            args_str = json.dumps(args, ensure_ascii=False, indent=2)
-            if len(args_str) > max_len:
-                args_str = args_str[:max_len] + "..."
-            return args_str
-        except Exception:
-            return str(args)[:max_len]
-    
-    def _update_status(self, new_status: str) -> None:
-        """更新状态"""
-        self.remove_class(self.status)
-        self.status = new_status
-        self.add_class(new_status)
-        self.refresh()
-    
-    def set_running(self) -> None:
-        """设置为运行中"""
-        self._update_status(ToolStatus.RUNNING)
-    
-    def set_success(self, output: str = "") -> None:
-        """设置为成功"""
-        self.output = output
-        self._update_status(ToolStatus.SUCCESS)
-    
-    def set_error(self, error: str = "") -> None:
-        """设置为失败"""
-        self.output = f"Error: {error}"
-        self._update_status(ToolStatus.ERROR)
-    
-    def set_rejected(self) -> None:
-        """设置为被拒绝"""
-        self._update_status(ToolStatus.REJECTED)

ksadk/tui/widgets/user.py

@@ -1,37 +0,0 @@
-"""
-UserMessage - 用户消息组件
-"""
-
-from textual.widgets import Static
-from rich.text import Text
-from typing import Any
-
-from .base import MessageWidget
-
-
-class UserMessage(MessageWidget):
-    """用户消息组件
-    
-    显示用户输入，左边框绿色
-    """
-    
-    DEFAULT_CSS = """
-    UserMessage {
-        height: auto;
-        padding: 0 1;
-        margin: 1 0 0 0;
-        background: transparent;
-        border-left: wide #10b981;
-    }
-    """
-
-    def __init__(self, content: str, **kwargs: Any) -> None:
-        super().__init__(content=content, **kwargs)
-        self._content = content
-
-    def compose(self):
-        """构建用户消息布局"""
-        text = Text()
-        text.append("> ", style="bold #10b981")
-        text.append(self._content, style="#ffffff")
-        yield Static(text)