supyagent 0.1.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

supyagent/default_tools/__init__.py

@@ -0,0 +1,74 @@
+"""
+Default supypowers tools bundled with supyagent.
+
+These tools are copied to the user's project when running `supyagent init`.
+"""
+
+import shutil
+from pathlib import Path
+
+# Path to the bundled default tools
+TOOLS_DIR = Path(__file__).parent
+
+
+def get_bundled_tools() -> list[Path]:
+    """Get list of bundled tool files."""
+    return [f for f in TOOLS_DIR.glob("*.py") if f.name != "__init__.py"]
+
+
+def install_default_tools(target_dir: Path | str = "supypowers") -> int:
+    """
+    Install default tools to a target directory.
+
+    Args:
+        target_dir: Directory to install tools to (default: supypowers/)
+
+    Returns:
+        Number of files installed
+    """
+    target = Path(target_dir)
+    target.mkdir(parents=True, exist_ok=True)
+
+    installed = 0
+    for tool_file in get_bundled_tools():
+        dest = target / tool_file.name
+        if not dest.exists():
+            shutil.copy(tool_file, dest)
+            installed += 1
+
+    # Create __init__.py if not exists
+    init_file = target / "__init__.py"
+    if not init_file.exists():
+        init_file.write_text('"""Supypowers tools for this project."""\n')
+        installed += 1
+
+    return installed
+
+
+def list_default_tools() -> list[dict]:
+    """
+    List available default tools.
+
+    Returns:
+        List of tool info dicts
+    """
+    tools = []
+    for tool_file in get_bundled_tools():
+        # Read first docstring
+        content = tool_file.read_text()
+        description = ""
+        if '"""' in content:
+            start = content.find('"""') + 3
+            end = content.find('"""', start)
+            if end > start:
+                description = content[start:end].strip().split("\n")[0]
+
+        tools.append(
+            {
+                "name": tool_file.stem,
+                "file": tool_file.name,
+                "description": description,
+            }
+        )
+
+    return tools

supyagent/default_tools/files.py

@@ -0,0 +1,439 @@
+# /// script
+# dependencies = ["pydantic"]
+# ///
+"""
+File system operation tools.
+
+Allows agents to read, write, and manage files and directories.
+"""
+
+import os
+import shutil
+from pathlib import Path
+from typing import Optional, List
+
+from pydantic import BaseModel, Field
+
+
+# =============================================================================
+# Read File
+# =============================================================================
+
+
+class ReadFileInput(BaseModel):
+    """Input for read_file function."""
+
+    path: str = Field(description="Path to the file to read")
+    encoding: str = Field(default="utf-8", description="File encoding")
+
+
+class ReadFileOutput(BaseModel):
+    """Output for read_file function."""
+
+    ok: bool
+    content: Optional[str] = None
+    size: Optional[int] = None
+    path: Optional[str] = None
+    error: Optional[str] = None
+
+
+def read_file(input: ReadFileInput) -> ReadFileOutput:
+    """
+    Read the contents of a file.
+
+    Examples:
+        >>> read_file({"path": "README.md"})
+        >>> read_file({"path": "data.txt", "encoding": "latin-1"})
+    """
+    try:
+        path = os.path.expanduser(input.path)
+        p = Path(path)
+
+        if not p.exists():
+            return ReadFileOutput(ok=False, error=f"File not found: {path}")
+
+        if not p.is_file():
+            return ReadFileOutput(ok=False, error=f"Not a file: {path}")
+
+        size = p.stat().st_size
+        if size > 10 * 1024 * 1024:
+            return ReadFileOutput(
+                ok=False,
+                error=f"File too large ({size} bytes). Maximum is 10MB.",
+            )
+
+        content = p.read_text(encoding=input.encoding)
+        return ReadFileOutput(
+            ok=True,
+            content=content,
+            size=size,
+            path=str(p.absolute()),
+        )
+
+    except UnicodeDecodeError:
+        return ReadFileOutput(
+            ok=False,
+            error=f"Cannot decode file as {input.encoding}",
+        )
+    except PermissionError:
+        return ReadFileOutput(ok=False, error=f"Permission denied: {input.path}")
+    except Exception as e:
+        return ReadFileOutput(ok=False, error=str(e))
+
+
+# =============================================================================
+# Write File
+# =============================================================================
+
+
+class WriteFileInput(BaseModel):
+    """Input for write_file function."""
+
+    path: str = Field(description="Path to the file to write")
+    content: str = Field(description="Content to write")
+    encoding: str = Field(default="utf-8", description="File encoding")
+    create_dirs: bool = Field(
+        default=True, description="Create parent directories if needed"
+    )
+
+
+class WriteFileOutput(BaseModel):
+    """Output for write_file function."""
+
+    ok: bool
+    path: Optional[str] = None
+    size: Optional[int] = None
+    error: Optional[str] = None
+
+
+def write_file(input: WriteFileInput) -> WriteFileOutput:
+    """
+    Write content to a file.
+
+    Examples:
+        >>> write_file({"path": "output.txt", "content": "Hello, world!"})
+        >>> write_file({"path": "data/results.json", "content": '{"status": "ok"}'})
+    """
+    try:
+        path = os.path.expanduser(input.path)
+        p = Path(path)
+
+        if input.create_dirs:
+            p.parent.mkdir(parents=True, exist_ok=True)
+
+        p.write_text(input.content, encoding=input.encoding)
+
+        return WriteFileOutput(
+            ok=True,
+            path=str(p.absolute()),
+            size=len(input.content.encode(input.encoding)),
+        )
+
+    except PermissionError:
+        return WriteFileOutput(ok=False, error=f"Permission denied: {input.path}")
+    except Exception as e:
+        return WriteFileOutput(ok=False, error=str(e))
+
+
+# =============================================================================
+# List Directory
+# =============================================================================
+
+
+class ListDirectoryInput(BaseModel):
+    """Input for list_directory function."""
+
+    path: str = Field(default=".", description="Directory path")
+    pattern: Optional[str] = Field(
+        default=None, description="Optional glob pattern (e.g., '*.py')"
+    )
+    recursive: bool = Field(default=False, description="List recursively")
+
+
+class FileInfo(BaseModel):
+    """Information about a single file/directory."""
+
+    name: str
+    path: str
+    type: str  # "file" or "directory"
+    size: Optional[int] = None
+
+
+class ListDirectoryOutput(BaseModel):
+    """Output for list_directory function."""
+
+    ok: bool
+    items: List[FileInfo] = []
+    count: int = 0
+    error: Optional[str] = None
+
+
+def list_directory(input: ListDirectoryInput) -> ListDirectoryOutput:
+    """
+    List files and directories.
+
+    Examples:
+        >>> list_directory({"path": "."})
+        >>> list_directory({"path": "src", "pattern": "*.py"})
+        >>> list_directory({"path": ".", "pattern": "*.md", "recursive": True})
+    """
+    try:
+        path = os.path.expanduser(input.path)
+        p = Path(path)
+
+        if not p.exists():
+            return ListDirectoryOutput(
+                ok=False, error=f"Directory not found: {path}"
+            )
+
+        if not p.is_dir():
+            return ListDirectoryOutput(ok=False, error=f"Not a directory: {path}")
+
+        items = []
+
+        if input.pattern:
+            matches = p.rglob(input.pattern) if input.recursive else p.glob(input.pattern)
+        else:
+            matches = p.iterdir()
+
+        for item in sorted(matches):
+            try:
+                stat = item.stat()
+                items.append(
+                    FileInfo(
+                        name=item.name,
+                        path=str(item),
+                        type="directory" if item.is_dir() else "file",
+                        size=stat.st_size if item.is_file() else None,
+                    )
+                )
+            except (PermissionError, OSError):
+                pass  # Skip inaccessible files
+
+        return ListDirectoryOutput(ok=True, items=items, count=len(items))
+
+    except Exception as e:
+        return ListDirectoryOutput(ok=False, error=str(e))
+
+
+# =============================================================================
+# File Info
+# =============================================================================
+
+
+class FileInfoInput(BaseModel):
+    """Input for file_info function."""
+
+    path: str = Field(description="Path to get info for")
+
+
+class FileInfoOutput(BaseModel):
+    """Output for file_info function."""
+
+    ok: bool
+    path: Optional[str] = None
+    name: Optional[str] = None
+    type: Optional[str] = None
+    size: Optional[int] = None
+    extension: Optional[str] = None
+    error: Optional[str] = None
+
+
+def file_info(input: FileInfoInput) -> FileInfoOutput:
+    """
+    Get detailed information about a file or directory.
+
+    Examples:
+        >>> file_info({"path": "README.md"})
+    """
+    try:
+        path = os.path.expanduser(input.path)
+        p = Path(path)
+
+        if not p.exists():
+            return FileInfoOutput(ok=False, error=f"Path not found: {path}")
+
+        stat = p.stat()
+
+        return FileInfoOutput(
+            ok=True,
+            path=str(p.absolute()),
+            name=p.name,
+            type="directory" if p.is_dir() else "file",
+            size=stat.st_size,
+            extension=p.suffix if p.is_file() else None,
+        )
+
+    except Exception as e:
+        return FileInfoOutput(ok=False, error=str(e))
+
+
+# =============================================================================
+# Delete File
+# =============================================================================
+
+
+class DeleteFileInput(BaseModel):
+    """Input for delete_file function."""
+
+    path: str = Field(description="Path to the file to delete")
+
+
+class DeleteFileOutput(BaseModel):
+    """Output for delete_file function."""
+
+    ok: bool
+    deleted: Optional[str] = None
+    error: Optional[str] = None
+
+
+def delete_file(input: DeleteFileInput) -> DeleteFileOutput:
+    """
+    Delete a file.
+
+    Examples:
+        >>> delete_file({"path": "temp.txt"})
+    """
+    try:
+        path = os.path.expanduser(input.path)
+        p = Path(path)
+
+        if not p.exists():
+            return DeleteFileOutput(ok=False, error=f"File not found: {path}")
+
+        if p.is_dir():
+            return DeleteFileOutput(
+                ok=False, error=f"Use delete_directory for directories: {path}"
+            )
+
+        p.unlink()
+        return DeleteFileOutput(ok=True, deleted=str(p))
+
+    except Exception as e:
+        return DeleteFileOutput(ok=False, error=str(e))
+
+
+# =============================================================================
+# Create Directory
+# =============================================================================
+
+
+class CreateDirectoryInput(BaseModel):
+    """Input for create_directory function."""
+
+    path: str = Field(description="Directory path to create")
+    parents: bool = Field(
+        default=True, description="Create parent directories if needed"
+    )
+
+
+class CreateDirectoryOutput(BaseModel):
+    """Output for create_directory function."""
+
+    ok: bool
+    path: Optional[str] = None
+    error: Optional[str] = None
+
+
+def create_directory(input: CreateDirectoryInput) -> CreateDirectoryOutput:
+    """
+    Create a directory.
+
+    Examples:
+        >>> create_directory({"path": "new_folder"})
+        >>> create_directory({"path": "a/b/c/deep"})
+    """
+    try:
+        path = os.path.expanduser(input.path)
+        p = Path(path)
+
+        p.mkdir(parents=input.parents, exist_ok=True)
+
+        return CreateDirectoryOutput(ok=True, path=str(p.absolute()))
+
+    except Exception as e:
+        return CreateDirectoryOutput(ok=False, error=str(e))
+
+
+# =============================================================================
+# Copy File
+# =============================================================================
+
+
+class CopyFileInput(BaseModel):
+    """Input for copy_file function."""
+
+    source: str = Field(description="Source file path")
+    destination: str = Field(description="Destination path")
+
+
+class CopyFileOutput(BaseModel):
+    """Output for copy_file function."""
+
+    ok: bool
+    source: Optional[str] = None
+    destination: Optional[str] = None
+    error: Optional[str] = None
+
+
+def copy_file(input: CopyFileInput) -> CopyFileOutput:
+    """
+    Copy a file.
+
+    Examples:
+        >>> copy_file({"source": "file.txt", "destination": "backup/file.txt"})
+    """
+    try:
+        source = os.path.expanduser(input.source)
+        destination = os.path.expanduser(input.destination)
+
+        # Create destination directory if needed
+        dest_path = Path(destination)
+        dest_path.parent.mkdir(parents=True, exist_ok=True)
+
+        shutil.copy2(source, destination)
+
+        return CopyFileOutput(ok=True, source=source, destination=destination)
+
+    except Exception as e:
+        return CopyFileOutput(ok=False, error=str(e))
+
+
+# =============================================================================
+# Move File
+# =============================================================================
+
+
+class MoveFileInput(BaseModel):
+    """Input for move_file function."""
+
+    source: str = Field(description="Source path")
+    destination: str = Field(description="Destination path")
+
+
+class MoveFileOutput(BaseModel):
+    """Output for move_file function."""
+
+    ok: bool
+    source: Optional[str] = None
+    destination: Optional[str] = None
+    error: Optional[str] = None
+
+
+def move_file(input: MoveFileInput) -> MoveFileOutput:
+    """
+    Move or rename a file or directory.
+
+    Examples:
+        >>> move_file({"source": "old.txt", "destination": "new.txt"})
+    """
+    try:
+        source = os.path.expanduser(input.source)
+        destination = os.path.expanduser(input.destination)
+
+        shutil.move(source, destination)
+
+        return MoveFileOutput(ok=True, source=source, destination=destination)
+
+    except Exception as e:
+        return MoveFileOutput(ok=False, error=str(e))