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

supyagent/default_tools/shell.py

@@ -0,0 +1,217 @@
+# /// script
+# dependencies = ["pydantic"]
+# ///
+"""
+Shell command execution tools.
+
+Allows agents to run bash/shell commands on the system.
+Use with caution - consider restricting via agent tool permissions.
+"""
+
+import os
+import subprocess
+import shutil
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class RunCommandInput(BaseModel):
+    """Input for run_command function."""
+
+    command: str = Field(description="The shell command to execute")
+    working_dir: Optional[str] = Field(
+        default=None, description="Working directory for the command"
+    )
+    timeout: int = Field(
+        default=60, description="Maximum seconds to wait for command"
+    )
+
+
+class RunCommandOutput(BaseModel):
+    """Output for run_command function."""
+
+    ok: bool
+    stdout: str
+    stderr: str
+    exit_code: int
+    command: str
+
+
+def run_command(input: RunCommandInput) -> RunCommandOutput:
+    """
+    Execute a shell command and return the result.
+
+    Examples:
+        >>> run_command({"command": "ls -la"})
+        >>> run_command({"command": "echo 'hello' | grep hello"})
+        >>> run_command({"command": "pwd", "working_dir": "/tmp"})
+    """
+    try:
+        working_dir = input.working_dir
+        if working_dir:
+            working_dir = os.path.expanduser(working_dir)
+            working_dir = os.path.expandvars(working_dir)
+
+        result = subprocess.run(
+            input.command,
+            shell=True,
+            capture_output=True,
+            text=True,
+            timeout=input.timeout,
+            cwd=working_dir,
+        )
+
+        return RunCommandOutput(
+            ok=result.returncode == 0,
+            stdout=result.stdout,
+            stderr=result.stderr,
+            exit_code=result.returncode,
+            command=input.command,
+        )
+
+    except subprocess.TimeoutExpired:
+        return RunCommandOutput(
+            ok=False,
+            stdout="",
+            stderr=f"Command timed out after {input.timeout} seconds",
+            exit_code=-1,
+            command=input.command,
+        )
+    except Exception as e:
+        return RunCommandOutput(
+            ok=False,
+            stdout="",
+            stderr=str(e),
+            exit_code=-1,
+            command=input.command,
+        )
+
+
+class RunScriptInput(BaseModel):
+    """Input for run_script function."""
+
+    script: str = Field(description="Multi-line script content")
+    interpreter: str = Field(
+        default="/bin/bash", description="Script interpreter"
+    )
+    working_dir: Optional[str] = Field(
+        default=None, description="Working directory"
+    )
+    timeout: int = Field(default=120, description="Max seconds to wait")
+
+
+class RunScriptOutput(BaseModel):
+    """Output for run_script function."""
+
+    ok: bool
+    stdout: str
+    stderr: str
+    exit_code: int
+
+
+def run_script(input: RunScriptInput) -> RunScriptOutput:
+    """
+    Execute a multi-line shell script.
+
+    Examples:
+        >>> run_script({"script": "cd /tmp\\necho $(pwd)\\nls"})
+    """
+    try:
+        working_dir = input.working_dir
+        if working_dir:
+            working_dir = os.path.expanduser(working_dir)
+
+        result = subprocess.run(
+            [input.interpreter],
+            input=input.script,
+            capture_output=True,
+            text=True,
+            timeout=input.timeout,
+            cwd=working_dir,
+        )
+
+        return RunScriptOutput(
+            ok=result.returncode == 0,
+            stdout=result.stdout,
+            stderr=result.stderr,
+            exit_code=result.returncode,
+        )
+
+    except subprocess.TimeoutExpired:
+        return RunScriptOutput(
+            ok=False,
+            stdout="",
+            stderr=f"Script timed out after {input.timeout} seconds",
+            exit_code=-1,
+        )
+    except Exception as e:
+        return RunScriptOutput(
+            ok=False,
+            stdout="",
+            stderr=str(e),
+            exit_code=-1,
+        )
+
+
+class WhichInput(BaseModel):
+    """Input for which function."""
+
+    command: str = Field(description="Command name to look up")
+
+
+class WhichOutput(BaseModel):
+    """Output for which function."""
+
+    ok: bool
+    path: Optional[str] = None
+    error: Optional[str] = None
+
+
+def which(input: WhichInput) -> WhichOutput:
+    """
+    Find the path to an executable.
+
+    Examples:
+        >>> which({"command": "python"})
+        >>> which({"command": "git"})
+    """
+    path = shutil.which(input.command)
+    if path:
+        return WhichOutput(ok=True, path=path)
+    else:
+        return WhichOutput(ok=False, error=f"Command not found: {input.command}")
+
+
+class GetEnvInput(BaseModel):
+    """Input for get_env function."""
+
+    name: str = Field(description="Environment variable name")
+    default: Optional[str] = Field(
+        default=None, description="Default if not set"
+    )
+
+
+class GetEnvOutput(BaseModel):
+    """Output for get_env function."""
+
+    ok: bool
+    value: Optional[str] = None
+    error: Optional[str] = None
+
+
+def get_env(input: GetEnvInput) -> GetEnvOutput:
+    """
+    Get an environment variable value.
+
+    Examples:
+        >>> get_env({"name": "HOME"})
+        >>> get_env({"name": "MY_VAR", "default": "fallback"})
+    """
+    value = os.environ.get(input.name, input.default)
+    if value is not None:
+        return GetEnvOutput(ok=True, value=value)
+    else:
+        return GetEnvOutput(
+            ok=False, error=f"Environment variable not set: {input.name}"
+        )

{supyagent-0.1.0.dist-info → supyagent-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: supyagent
-Version: 0.1.0
+Version: 0.2.1
 Summary: LLM agents powered by supypowers - build AI agents with tool use, multi-agent orchestration, and secure credential management
 Project-URL: Homepage, https://github.com/ergodic-ai/supyagent
 Project-URL: Documentation, https://github.com/ergodic-ai/supyagent#readme
@@ -68,6 +68,12 @@ uv pip install supyagent
 ## Quick Start
 
 ```bash
+# Initialize supyagent (sets up default tools)
+supyagent init
+
+# Set up your API key (stored securely)
+supyagent config set ANTHROPIC_API_KEY
+
 # Create your first agent
 supyagent new myagent
 
@@ -133,39 +139,40 @@ supyagent plan "Build a Python library for data validation"
 # The planner will delegate to specialist agents (coder, writer, researcher)
 ```
 
-## Agent Configuration
+## Configuration
 
-Agents are defined in YAML files in the `agents/` directory:
+### Setting Up API Keys
 
-```yaml
-name: researcher
-description: An AI research assistant
-type: interactive  # or "execution"
+Supyagent securely stores your LLM API keys so you don't need to export them every time:
 
-model:
-  provider: anthropic/claude-3-5-sonnet-20241022
-  temperature: 0.7
-  max_tokens: 4096
+```bash
+# Interactive setup (recommended)
+supyagent config set
+# Shows a menu of common providers to choose from
 
-system_prompt: |
-  You are a helpful research assistant...
+# Set a specific key
+supyagent config set ANTHROPIC_API_KEY
+supyagent config set OPENAI_API_KEY
 
-tools:
-  allow:
-    - "*"           # Allow all tools
-    # or be specific:
-    # - "web:*"     # All functions in web.py
-    # - "math:calc" # Specific function
-  deny:
-    - "dangerous:*" # Block specific tools
+# Import from a .env file
+supyagent config import .env
 
-# For multi-agent support
-delegates:
-  - coder
-  - writer
+# Import only specific keys
+supyagent config import .env --filter OPENAI
+
+# List configured keys
+supyagent config list
+
+# Export keys to backup
+supyagent config export backup.env
+
+# Delete a key
+supyagent config delete OPENAI_API_KEY
 ```
 
-### LLM Providers
+Keys are encrypted and stored in `~/.supyagent/config/`. They're automatically loaded when running any agent command.
+
+### Supported Providers
 
 Supyagent uses LiteLLM, supporting 100+ providers:
 
@@ -187,11 +194,36 @@ model:
   provider: gemini/gemini-pro
 ```
 
-Set API keys as environment variables:
+## Agent Configuration
 
-```bash
-export OPENAI_API_KEY=sk-...
-export ANTHROPIC_API_KEY=sk-ant-...
+Agents are defined in YAML files in the `agents/` directory:
+
+```yaml
+name: researcher
+description: An AI research assistant
+type: interactive  # or "execution"
+
+model:
+  provider: anthropic/claude-3-5-sonnet-20241022
+  temperature: 0.7
+  max_tokens: 4096
+
+system_prompt: |
+  You are a helpful research assistant...
+
+tools:
+  allow:
+    - "*"           # Allow all tools
+    # or be specific:
+    # - "web:*"     # All functions in web.py
+    # - "math:calc" # Specific function
+  deny:
+    - "dangerous:*" # Block specific tools
+
+# For multi-agent support
+delegates:
+  - coder
+  - writer
 ```
 
 ## CLI Reference
@@ -200,6 +232,7 @@ export ANTHROPIC_API_KEY=sk-ant-...
 
 | Command | Description |
 |---------|-------------|
+| `supyagent init` | Initialize project with default tools |
 | `supyagent new <name>` | Create a new agent |
 | `supyagent list` | List all agents |
 | `supyagent show <name>` | Show agent details |
@@ -223,6 +256,16 @@ export ANTHROPIC_API_KEY=sk-ant-...
 | `supyagent agents` | List active agent instances |
 | `supyagent cleanup` | Remove completed instances |
 
+### Config Commands
+
+| Command | Description |
+|---------|-------------|
+| `supyagent config set [KEY]` | Set an API key (interactive menu if no key specified) |
+| `supyagent config list` | List all configured keys |
+| `supyagent config import <file>` | Import keys from .env file |
+| `supyagent config export <file>` | Export keys to .env file |
+| `supyagent config delete <key>` | Delete a stored key |
+
 ### In-Chat Commands
 
 While chatting, use these commands:
@@ -242,6 +285,25 @@ While chatting, use these commands:
 | `/clear` | Clear conversation history |
 | `/quit` | Exit the chat |
 
+## Bundled Tools
+
+Running `supyagent init` installs these default tools:
+
+### Shell (`shell.py`)
+- `run_command` - Execute shell commands
+- `run_script` - Run multi-line bash scripts
+- `which` - Find executable paths
+- `get_env` - Get environment variables
+
+### Files (`files.py`)
+- `read_file` / `write_file` - File I/O
+- `list_directory` - List files with glob patterns
+- `copy_file` / `move_file` / `delete_file` - File operations
+- `create_directory` - Create directories
+- `file_info` - Get file metadata
+
+You can add your own tools by creating Python files in `supypowers/`.
+
 ## Project Structure
 
 ```
@@ -251,8 +313,9 @@ your-project/
 │   ├── planner.yaml
 │   └── researcher.yaml
 ├── supypowers/          # Tool definitions (Python)
-│   ├── hello.py
-│   └── web_search.py
+│   ├── shell.py         # Shell commands (bundled)
+│   ├── files.py         # File operations (bundled)
+│   └── my_tools.py      # Your custom tools
 └── .supyagent/          # Runtime data (gitignore this)
     ├── sessions/        # Conversation history
     ├── credentials/     # Encrypted secrets

{supyagent-0.1.0.dist-info → supyagent-0.2.1.dist-info}/RECORD

@@ -1,9 +1,10 @@
 supyagent/__init__.py,sha256=JrrQbk1bmDhUb2qOqxqDGBq1V-oArwVS4f5f9Qu-VcM,77
 supyagent/__main__.py,sha256=szUvsccEIA428c2_5J_X1qzq4L44qLvDvHfuRPYoy7c,121
 supyagent/cli/__init__.py,sha256=7jb5kxHtzce4rNKd9zrAiATp3Zxj7UQvtpQsyA97BTc,32
-supyagent/cli/main.py,sha256=5oM_IZ85C3gyW_ypPmFi3x-khkgd7tZrbowW_kJ_JhY,31028
-supyagent/core/__init__.py,sha256=5GOA6Ca5UBWG_d6lhfZxYdKdDjnKE4UdVooe0lUDpe0,626
+supyagent/cli/main.py,sha256=6jXcEZjLRgE_9PnPvyg-6BWGg5_yB71xZOD8_cejSno,38515
+supyagent/core/__init__.py,sha256=7MI6K2-OR_Fg0wcMziC4u3gNa0YxWmzJXO0RtOCevUg,727
 supyagent/core/agent.py,sha256=mkiV6CnK6Vk0Wg-DiAMm29wAViMa52-1Yb33Y5-FvQY,12655
+supyagent/core/config.py,sha256=kuTpuxy-qr-fhou7rJ-JLF82cdzdHQTu46kCZccSw1Y,10041
 supyagent/core/context.py,sha256=ywen6IjWxcudxtipeqA_PHpZKdjK52eqOb9xy7QKnTM,4565
 supyagent/core/credentials.py,sha256=35JBrzMOEDzHnn3XuyGmRjXVdSICL3tups3BLO_NCXQ,7600
 supyagent/core/delegation.py,sha256=B6OYlLjkZaRTPMQsK9nIGeXVO9JeC7RDpxXsOk652xs,9391
@@ -12,13 +13,16 @@ supyagent/core/llm.py,sha256=z16cjqjPfwRGtEpUhZe1XlzOXJdlgKHjckNdeJnEWYw,1898
 supyagent/core/registry.py,sha256=1CGCPsOcrIc_63_oJaFmHY8WwMgJEvbwsv8mnsOBWtY,7326
 supyagent/core/session_manager.py,sha256=ngqrbofR8V0wZSuOfGHTtSYE0C9U6noDrzfDsO8wi24,7389
 supyagent/core/tools.py,sha256=exIynhwulsMJTzzW5NmDqBeXepULAt4FlVhGbtrvdXE,6627
+supyagent/default_tools/__init__.py,sha256=VwVjwNSPxjflmzJ-Mv_7L79Yjn9LdxyeXNUkWzu3wFo,1890
+supyagent/default_tools/files.py,sha256=tUpjkP7TKbcK75lQ6jzEDpVDBSeupzTSqx_g2-rkON0,12059
+supyagent/default_tools/shell.py,sha256=fYXdy693Nb0hsNLMzTFgj3zaHq0wFSQ3SxpqHbVayjM,5478
 supyagent/models/__init__.py,sha256=y7fIdzh-yD0LKAK0zfY_MZdtjJhxofwRWBqrHDU93h0,259
 supyagent/models/agent_config.py,sha256=sQ7S7zExoFs0g-IMgACsGH2SMiDRjJTnCVZwhN6DnkY,2570
 supyagent/models/session.py,sha256=_Dwuija_Uvt_KzcNp82ja8qP5ltNHHz43Llug1vq1ic,1136
 supyagent/utils/__init__.py,sha256=l4HdE3n8axphMInBex1J1dB_O1MjzbZz6ot1l8Syf0g,39
 supyagent/utils/paths.py,sha256=Xum9kX_UjRvmshifSq423QsAKynpnyAVfpWuFjlsdDQ,692
-supyagent-0.1.0.dist-info/METADATA,sha256=ooG14oWAy-YelwL-QC3DDZL8aFhDzM6Gs9mdjFW3xw4,8430
-supyagent-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-supyagent-0.1.0.dist-info/entry_points.txt,sha256=7dFXWtRewhLfA5XUckV7Yu1E5_c6c9lq8Gz-7BFs6TM,53
-supyagent-0.1.0.dist-info/licenses/LICENSE,sha256=35fw1cvTM-IhiR3xaUohbiFBc2OBNRjI5z1b1cF7vZI,1067
-supyagent-0.1.0.dist-info/RECORD,,
+supyagent-0.2.1.dist-info/METADATA,sha256=5sn1RwCPZ0Ja2ZGLatdfHlgg7KZnCr3v0u7zywAgpSs,10387
+supyagent-0.2.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+supyagent-0.2.1.dist-info/entry_points.txt,sha256=7dFXWtRewhLfA5XUckV7Yu1E5_c6c9lq8Gz-7BFs6TM,53
+supyagent-0.2.1.dist-info/licenses/LICENSE,sha256=35fw1cvTM-IhiR3xaUohbiFBc2OBNRjI5z1b1cF7vZI,1067
+supyagent-0.2.1.dist-info/RECORD,,