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

ksadk/tui/widgets/assistant.py

@@ -1,89 +0,0 @@
-"""
-AssistantMessage - 助手消息组件
-
-支持流式 MarkdownStream 渲染（与 deepagents-cli 一致）
-"""
-
-from textual.containers import Vertical
-from textual.widgets import Markdown
-from textual.widgets._markdown import MarkdownStream
-from typing import Any, Optional
-
-
-class AssistantMessage(Vertical):
-    """助手消息组件
-    
-    使用 MarkdownStream 实现流式 Markdown 渲染
-    """
-    
-    DEFAULT_CSS = """
-    AssistantMessage {
-        height: auto;
-        padding: 0 1;
-        margin: 1 0 0 0;
-        color: #ffffff;
-    }
-
-    AssistantMessage Markdown {
-        padding: 0;
-        margin: 0;
-        color: #ffffff;
-    }
-
-    /* 确保 Markdown 的所有子组件都有正确的颜色 */
-    AssistantMessage Markdown * {
-        color: #ffffff;
-    }
-    """
-
-    def __init__(self, content: str = "", **kwargs: Any) -> None:
-        super().__init__(**kwargs)
-        self._content = content
-        self._markdown: Optional[Markdown] = None
-        self._stream: Optional[MarkdownStream] = None
-
-    def compose(self):
-        """构建助手消息布局"""
-        yield Markdown("", id="assistant-content")
-
-    def on_mount(self) -> None:
-        """缓存 markdown widget 引用"""
-        self._markdown = self.query_one("#assistant-content", Markdown)
-
-    def _get_markdown(self) -> Markdown:
-        """获取 markdown widget"""
-        if self._markdown is None:
-            self._markdown = self.query_one("#assistant-content", Markdown)
-        return self._markdown
-
-    def _ensure_stream(self) -> MarkdownStream:
-        """确保 stream 已初始化"""
-        if self._stream is None:
-            self._stream = Markdown.get_stream(self._get_markdown())
-        return self._stream
-
-    async def append_content(self, text: str) -> None:
-        """追加内容（流式渲染）"""
-        if not text:
-            return
-        self._content += text
-        stream = self._ensure_stream()
-        await stream.write(text)
-
-    async def stop_stream(self) -> None:
-        """停止流式渲染"""
-        if self._stream is not None:
-            await self._stream.stop()
-            self._stream = None
-
-    async def set_content(self, content: str) -> None:
-        """设置完整内容"""
-        await self.stop_stream()
-        self._content = content
-        if self._markdown:
-            await self._markdown.update(content)
-
-    @property
-    def content(self) -> str:
-        return self._content
-

ksadk/tui/widgets/base.py

@@ -1,54 +0,0 @@
-"""
-Base Widget - 消息组件基类
-"""
-
-from textual.widget import Widget
-from textual.reactive import reactive
-from typing import Any, Optional
-
-
-class MessageWidget(Widget):
-    """消息组件基类
-    
-    所有消息组件的抽象基类，提供基础的消息渲染能力。
-    """
-    
-    DEFAULT_CSS = """
-    MessageWidget {
-        width: 100%;
-        padding: 0 1;
-        margin: 0 0 1 0;
-        color: #ffffff;
-    }
-    """
-    
-    content: reactive[str] = reactive("")
-    
-    def __init__(
-        self, 
-        content: str = "", 
-        *,
-        name: Optional[str] = None,
-        id: Optional[str] = None,
-        classes: Optional[str] = None,
-    ) -> None:
-        super().__init__(name=name, id=id, classes=classes)
-        self.content = content
-        self._is_streaming = False
-    
-    async def append_content(self, delta: str) -> None:
-        """追加内容（流式渲染）
-        
-        Args:
-            delta: 增量内容
-        """
-        self._is_streaming = True
-        self.content += delta
-    
-    async def stop_stream(self) -> None:
-        """停止流式渲染"""
-        self._is_streaming = False
-    
-    def watch_content(self, new_content: str) -> None:
-        """监听内容变化，触发重新渲染"""
-        self.refresh()

ksadk/tui/widgets/chat_input.py

@@ -1,246 +0,0 @@
-"""Chat input widget with history support."""
-
-from __future__ import annotations
-
-from pathlib import Path
-from typing import TYPE_CHECKING, Any, ClassVar, Optional, List
-
-from textual import events
-from textual.binding import Binding
-from textual.containers import Horizontal, Vertical
-from textual.message import Message
-from textual.reactive import reactive
-from textual.widgets import Static, TextArea
-
-from ksadk.tui.widgets.history import HistoryManager
-
-if TYPE_CHECKING:
-    from textual.app import ComposeResult
-
-
-class ChatTextArea(TextArea):
-    """TextArea subclass with custom key handling for chat input."""
-
-    BINDINGS: ClassVar[List[Binding]] = [
-        Binding(
-            "shift+enter,ctrl+j,alt+enter,ctrl+enter",
-            "insert_newline",
-            "New Line",
-            show=False,
-            priority=True,
-        ),
-        Binding(
-            "ctrl+a",
-            "select_all_text",
-            "Select All",
-            show=False,
-            priority=True,
-        ),
-    ]
-
-    class Submitted(Message):
-        """Message sent when text is submitted."""
-
-        def __init__(self, value: str) -> None:
-            self.value = value
-            super().__init__()
-
-    class HistoryPrevious(Message):
-        """Request previous history entry."""
-
-        def __init__(self, current_text: str) -> None:
-            self.current_text = current_text
-            super().__init__()
-
-    class HistoryNext(Message):
-        """Request next history entry."""
-
-    def __init__(self, **kwargs: Any) -> None:
-        kwargs.pop("placeholder", None)
-        super().__init__(**kwargs)
-        self._navigating_history = False
-
-    def action_insert_newline(self) -> None:
-        """Insert a newline character."""
-        self.insert("\n")
-
-    def action_select_all_text(self) -> None:
-        """Select all text in the text area."""
-        if not self.text:
-            return
-        lines = self.text.split("\n")
-        end_row = len(lines) - 1
-        end_col = len(lines[end_row])
-        self.selection = ((0, 0), (end_row, end_col))
-
-    async def _on_key(self, event: events.Key) -> None:
-        """Handle key events."""
-        if event.key in ("shift+enter", "ctrl+j", "alt+enter", "ctrl+enter"):
-            event.prevent_default()
-            event.stop()
-            self.insert("\n")
-            return
-
-        if event.key == "enter":
-            event.prevent_default()
-            event.stop()
-            value = self.text.strip()
-            if value:
-                self.post_message(self.Submitted(value))
-            return
-
-        if event.key == "up":
-            row, _ = self.cursor_location
-            if row == 0:
-                event.prevent_default()
-                event.stop()
-                self._navigating_history = True
-                self.post_message(self.HistoryPrevious(self.text))
-                return
-
-        if event.key == "down":
-            row, _ = self.cursor_location
-            total_lines = self.text.count("\n") + 1
-            if row == total_lines - 1:
-                event.prevent_default()
-                event.stop()
-                self._navigating_history = True
-                self.post_message(self.HistoryNext())
-                return
-
-        await super()._on_key(event)
-
-    def set_text_from_history(self, text: str) -> None:
-        """Set text from history navigation."""
-        self._navigating_history = True
-        self.text = text
-        lines = text.split("\n")
-        last_row = len(lines) - 1
-        last_col = len(lines[last_row])
-        self.move_cursor((last_row, last_col))
-        self._navigating_history = False
-
-    def clear_text(self) -> None:
-        """Clear the text area."""
-        self.text = ""
-        self.move_cursor((0, 0))
-
-
-class ChatInput(Vertical):
-    """Chat input widget with prompt indicator, multi-line text, and history."""
-
-    DEFAULT_CSS = """
-    ChatInput {
-        height: auto;
-        min-height: 3;
-        max-height: 12;
-        padding: 0;
-        background: $surface;
-        border: solid $primary;
-    }
-
-    ChatInput .input-row {
-        height: auto;
-        width: 100%;
-    }
-
-    ChatInput .input-prompt {
-        width: 3;
-        height: 1;
-        padding: 0 1;
-        color: $primary;
-        text-style: bold;
-    }
-
-    ChatInput ChatTextArea {
-        width: 1fr;
-        height: auto;
-        min-height: 1;
-        max-height: 8;
-        border: none;
-        background: transparent;
-        padding: 0;
-    }
-
-    ChatInput ChatTextArea:focus {
-        border: none;
-    }
-    """
-
-    class Submitted(Message):
-        """Message sent when input is submitted."""
-
-        def __init__(self, value: str, mode: str = "normal") -> None:
-            super().__init__()
-            self.value = value
-            self.mode = mode
-
-    mode: reactive[str] = reactive("normal")
-
-    def __init__(
-        self,
-        cwd: Optional[str | Path] = None,
-        history_file: Optional[Path] = None,
-        **kwargs: Any,
-    ) -> None:
-        super().__init__(**kwargs)
-        self._cwd = Path(cwd) if cwd else Path.cwd()
-        self._text_area: Optional[ChatTextArea] = None
-
-        if history_file is None:
-            history_file = Path.home() / ".ksadk" / "history.jsonl"
-        self._history = HistoryManager(history_file)
-
-    def compose(self) -> ComposeResult:
-        """Compose the chat input layout."""
-        with Horizontal(classes="input-row"):
-            yield Static(">", classes="input-prompt", id="prompt")
-            yield ChatTextArea(id="chat-input")
-
-    def on_mount(self) -> None:
-        """Initialize components after mount."""
-        self._text_area = self.query_one("#chat-input", ChatTextArea)
-        self._text_area.focus()
-
-    def on_chat_text_area_submitted(self, event: ChatTextArea.Submitted) -> None:
-        """Handle text submission."""
-        value = event.value
-        if value:
-            self._history.add(value)
-            self.post_message(self.Submitted(value, self.mode))
-            if self._text_area:
-                self._text_area.clear_text()
-            self.mode = "normal"
-
-    def on_chat_text_area_history_previous(self, event: ChatTextArea.HistoryPrevious) -> None:
-        """Handle history previous request."""
-        entry = self._history.get_previous(event.current_text)
-        if entry is not None and self._text_area:
-            self._text_area.set_text_from_history(entry)
-
-    def on_chat_text_area_history_next(self, event: ChatTextArea.HistoryNext) -> None:
-        """Handle history next request."""
-        entry = self._history.get_next()
-        if entry is not None and self._text_area:
-            self._text_area.set_text_from_history(entry)
-
-    def focus_input(self) -> None:
-        if self._text_area:
-            self._text_area.focus()
-
-    @property
-    def value(self) -> str:
-        if self._text_area:
-            return self._text_area.text
-        return ""
-
-    @value.setter
-    def value(self, val: str) -> None:
-        if self._text_area:
-            self._text_area.text = val
-
-    def set_disabled(self, *, disabled: bool) -> None:
-        if self._text_area:
-            self._text_area.disabled = disabled
-            if disabled:
-                self._text_area.blur()

ksadk/tui/widgets/history.py

@@ -1,156 +0,0 @@
-"""Command history manager for input persistence."""
-
-from __future__ import annotations
-
-import json
-from pathlib import Path  # noqa: TC003 - used at runtime in type hints
-from typing import List, Optional
-
-
-class HistoryManager:
-    """Manages command history with file persistence.
-
-    Uses append-only writes for concurrent safety. Multiple agents can
-    safely write to the same history file without corruption.
-    """
-
-    def __init__(self, history_file: Path, max_entries: int = 100) -> None:
-        """Initialize the history manager.
-
-        Args:
-            history_file: Path to the JSON-lines history file
-            max_entries: Maximum number of entries to keep
-        """
-        self.history_file = history_file
-        self.max_entries = max_entries
-        self._entries: List[str] = []
-        self._current_index: int = -1
-        self._temp_input: str = ""
-        self._load_history()
-
-    def _load_history(self) -> None:
-        """Load history from file."""
-        if not self.history_file.exists():
-            return
-
-        try:
-            with self.history_file.open("r", encoding="utf-8") as f:
-                entries = []
-                for raw_line in f:
-                    line = raw_line.rstrip("\n\r")
-                    if not line:
-                        continue
-                    try:
-                        entry = json.loads(line)
-                    except json.JSONDecodeError:
-                        entry = line
-                    entries.append(entry if isinstance(entry, str) else str(entry))
-                self._entries = entries[-self.max_entries :]
-        except (OSError, UnicodeDecodeError):
-            self._entries = []
-
-    def _append_to_file(self, text: str) -> None:
-        """Append a single entry to history file (concurrent-safe)."""
-        try:
-            self.history_file.parent.mkdir(parents=True, exist_ok=True)
-            with self.history_file.open("a", encoding="utf-8") as f:
-                f.write(json.dumps(text) + "\n")
-        except OSError:
-            pass
-
-    def _compact_history(self) -> None:
-        """Rewrite history file to remove old entries.
-
-        Only called when entries exceed 2x max_entries to minimize rewrites.
-        """
-        try:
-            self.history_file.parent.mkdir(parents=True, exist_ok=True)
-            with self.history_file.open("w", encoding="utf-8") as f:
-                for entry in self._entries:
-                    f.write(json.dumps(entry) + "\n")
-        except OSError:
-            pass
-
-    def add(self, text: str) -> None:
-        """Add a command to history.
-
-        Args:
-            text: The command text to add
-        """
-        text = text.strip()
-        # Skip empty or slash commands ? No, allow slash commands in history usually.
-        # DeepAgents code skips empty or slash commands.
-        # But user might want to recall /clear or /thinking.
-        # Let's keep slash commands in history for ksadk.
-        if not text:
-            return
-
-        # Skip duplicates of the last entry
-        if self._entries and self._entries[-1] == text:
-            return
-
-        self._entries.append(text)
-
-        # Append to file (fast, concurrent-safe)
-        self._append_to_file(text)
-
-        # Compact only when we have 2x max entries (rare operation)
-        if len(self._entries) > self.max_entries * 2:
-            self._entries = self._entries[-self.max_entries :]
-            self._compact_history()
-
-        self.reset_navigation()
-
-    def get_previous(self, current_input: str, prefix: str = "") -> Optional[str]:
-        """Get the previous history entry.
-
-        Args:
-            current_input: Current input text (saved on first navigation)
-            prefix: Optional prefix to filter entries
-
-        Returns:
-            Previous matching entry or None
-        """
-        if not self._entries:
-            return None
-
-        # Save current input on first navigation
-        if self._current_index == -1:
-            self._temp_input = current_input
-            self._current_index = len(self._entries)
-
-        # Search backwards for matching entry
-        for i in range(self._current_index - 1, -1, -1):
-            if self._entries[i].startswith(prefix):
-                self._current_index = i
-                return self._entries[i]
-
-        return None
-
-    def get_next(self, prefix: str = "") -> Optional[str]:
-        """Get the next history entry.
-
-        Args:
-            prefix: Optional prefix to filter entries
-
-        Returns:
-            Next matching entry, original input at end, or None
-        """
-        if self._current_index == -1:
-            return None
-
-        # Search forwards for matching entry
-        for i in range(self._current_index + 1, len(self._entries)):
-            if self._entries[i].startswith(prefix):
-                self._current_index = i
-                return self._entries[i]
-
-        # Return to original input at the end
-        result = self._temp_input
-        self.reset_navigation()
-        return result
-
-    def reset_navigation(self) -> None:
-        """Reset navigation state."""
-        self._current_index = -1
-        self._temp_input = ""

ksadk/tui/widgets/system.py

@@ -1,87 +0,0 @@
-"""
-SystemMessage - 系统消息组件
-
-用于显示系统通知、错误等。
-"""
-
-from textual.widgets import Static
-from typing import Optional
-
-from .base import MessageWidget
-
-
-class SystemMessage(MessageWidget):
-    """系统消息组件
-    
-    用于显示系统级通知：
-    - 信息提示
-    - 警告
-    - 错误
-    """
-    
-    DEFAULT_CSS = """
-    SystemMessage {
-        width: 100%;
-        padding: 0 1;
-        margin: 0 0 1 0;
-        color: #bbbbbb;
-        background: $surface;
-    }
-    
-    SystemMessage.info {
-        color: #66ccff;
-    }
-
-    SystemMessage.info * {
-        color: #66ccff;
-    }
-    
-    SystemMessage.warning {
-        color: #ffcc00;
-    }
-
-    SystemMessage.warning * {
-        color: #ffcc00;
-    }
-    
-    SystemMessage.error {
-        color: #ff6666;
-    }
-
-    SystemMessage.error * {
-        color: #ff6666;
-    }
-    
-    SystemMessage.success {
-        color: #66ff66;
-    }
-
-    SystemMessage.success * {
-        color: #66ff66;
-    }
-    """
-    
-    def __init__(
-        self,
-        content: str,
-        level: str = "info",
-        **kwargs
-    ) -> None:
-        super().__init__(content=content, **kwargs)
-        self._level = level
-        self.add_class(level)
-    
-    def compose(self):
-        """构建组件"""
-        icon = self._get_icon()
-        yield Static(f"{icon} {self.content}")
-    
-    def _get_icon(self) -> str:
-        """获取图标"""
-        icons = {
-            "info": "ℹ️",
-            "warning": "⚠️",
-            "error": "❌",
-            "success": "✅",
-        }
-        return icons.get(self._level, "ℹ️")