python-agent-harness 1.5.0__py3-none-any.whl

python_agent_harness/tools/skill.py

@@ -0,0 +1,32 @@
+"""Skill tool: load a skill into the current conversation."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from .base import Tool, ToolContext
+
+
+class Skill(Tool):
+    name = "Skill"
+    description = (
+        "Load a skill to get detailed instructions for a specific task. "
+        "Use this when a task matches an available skill's description. "
+        "Invoke with the skill name and optional args."
+    )
+    parameters = {
+        "type": "object",
+        "properties": {
+            "skill": {"type": "string", "description": "Name of the skill to load"},
+            "args": {"type": "string", "description": "Optional arguments for the skill"},
+        },
+        "required": ["skill"],
+    }
+
+    def run(self, args: dict, ctx: ToolContext) -> str:
+        name = args.get("skill", "")
+        skill_path = ctx.find_skill(name)
+        if not skill_path:
+            return f"Error: skill {name!r} not found"
+        text = Path(skill_path).read_text(encoding="utf-8", errors="replace")
+        return f"[Skill: {name}]\n\n{text}"

python_agent_harness/tools/todo.py

@@ -0,0 +1,60 @@
+"""TodoWrite tool: maintain a structured task list during execution."""
+
+from __future__ import annotations
+
+import json
+
+from .base import Tool, ToolContext
+
+DESCRIPTION = (
+    "Create and manage a structured task list for your session. Helps track "
+    "progress on complex, multi-step tasks.\n\n"
+    "Use it when:\n"
+    "- The task has 3+ distinct steps or phases\n"
+    "- The task is non-trivial and benefits from planning\n"
+    "- You start a task (mark it in_progress) or finish a task (mark it completed)\n\n"
+    "Task states: pending (not started), in_progress (currently working on), "
+    "completed (finished). Only one task can be in_progress at a time. Send "
+    "the entire todo list with each call (not just changed items)."
+)
+
+PARAMETERS = {
+    "type": "object",
+    "properties": {
+        "todos": {
+            "type": "array",
+            "items": {
+                "type": "object",
+                "properties": {
+                    "content": {
+                        "type": "string",
+                        "minLength": 1,
+                        "description": "Task description (imperative form)",
+                    },
+                    "status": {
+                        "type": "string",
+                        "enum": ["pending", "in_progress", "completed"],
+                    },
+                    "activeForm": {
+                        "type": "string",
+                        "minLength": 1,
+                        "description": "Present continuous form",
+                    },
+                },
+                "required": ["content", "status"],
+            },
+        }
+    },
+    "required": ["todos"],
+}
+
+
+class TodoWrite(Tool):
+    name = "TodoWrite"
+    description = DESCRIPTION
+    parameters = PARAMETERS
+
+    def run(self, args: dict, ctx: ToolContext) -> str:
+        todos = args.get("todos") or []
+        ctx.update_todos(todos)
+        return json.dumps({"todos": todos, "count": len(todos)}, ensure_ascii=False)

python_agent_harness/tools/write.py

@@ -0,0 +1,56 @@
+"""Write tool: create/overwrite a file, recording a diff for the UI."""
+
+from __future__ import annotations
+
+import os
+
+from ..diffrender import unified_diff
+from .base import Tool, ToolContext
+
+
+class Write(Tool):
+    name = "Write"
+    description = (
+        "Create a new file with the given content. Overwrites an existing file — use with care!"
+    )
+    parameters = {
+        "type": "object",
+        "properties": {
+            "path": {"type": "string", "description": "Directory for the file"},
+            "filename": {"type": "string", "description": "File name"},
+            "content": {"type": "string", "description": "Full file content"},
+        },
+        "required": ["path", "filename", "content"],
+    }
+
+    def run(self, args: dict, ctx: ToolContext) -> str:
+        dir_path = args.get("path") or "."
+        filename = args.get("filename") or ""
+        content = args.get("content") or ""
+        # LLM may put the full file path in "filename" or in "path"
+        if filename:
+            path = os.path.realpath(os.path.abspath(os.path.join(dir_path, filename)))
+        else:
+            path = os.path.realpath(os.path.abspath(dir_path))
+        if not filename:
+            filename = os.path.basename(path) or os.path.basename(dir_path)
+        existed = os.path.exists(path)
+        old_content = ""
+        if existed:
+            try:
+                with open(path, encoding="utf-8") as f:
+                    old_content = f.read()
+            except OSError:
+                old_content = ""
+        try:
+            parent = os.path.dirname(path)
+            if parent:
+                os.makedirs(parent, exist_ok=True)
+            with open(path, "w", encoding="utf-8") as f:
+                f.write(content)
+        except OSError as e:
+            return f"Error: {e}"
+        diff_text = unified_diff(old_content, content, path)
+        if diff_text:
+            ctx.record_diff(diff_text)
+        return f"Created file {filename} in {dir_path}"

python_agent_harness/tui/__init__.py

@@ -0,0 +1,68 @@
+"""TUI package for the agent harness.
+
+Re-exports the public API so that ``from python_agent_harness.tui import Tui``
+and other imports continue to work unchanged after the split into
+submodules.
+"""
+
+from __future__ import annotations
+
+from .commands import CommandMixin
+from .core import Tui
+from .input import (
+    SLASH_COMMANDS,
+    SlashCompleter,
+    UiQuestion,
+    _custom_slash_commands,
+    _history_path,
+    _make_key_bindings,
+    _make_prompt_session,
+    _resolve_keyed_choice,
+    _resolve_numbered_choice,
+)
+from .render import (
+    _FC_DANGLING_RE,
+    _FC_HEADER,
+    _FC_LABEL,
+    _FINAL_CHECK_RE,
+    ASSISTANT_STYLE,
+    SPINNER_FRAMES,
+    USER_STYLE,
+    _head_chars,
+    _head_lines,
+    _is_injected_user_text,
+    _strip_final_check,
+    _strip_reasoning,
+    _tail_chars,
+    _tail_lines,
+    _tool_result_preview,
+)
+
+__all__ = [
+    "CommandMixin",
+    "Tui",
+    "UiQuestion",
+    "SlashCompleter",
+    "SLASH_COMMANDS",
+    "SPINNER_FRAMES",
+    "USER_STYLE",
+    "ASSISTANT_STYLE",
+    "_resolve_keyed_choice",
+    "_resolve_numbered_choice",
+    "_make_key_bindings",
+    "_make_prompt_session",
+    "_history_path",
+    "_custom_slash_commands",
+    "_strip_final_check",
+    "_strip_reasoning",
+    "_is_injected_user_text",
+    "_tail_lines",
+    "_tail_chars",
+    "_head_lines",
+    "_head_chars",
+    "_tool_result_preview",
+    "_FINAL_CHECK_RE",
+    "_FC_DANGLING_RE",
+    "_FC_HEADER",
+    "_FC_LABEL",
+]