Command Execution Examples

This directory demonstrates how to execute shell commands in the AGB environment.

Overview

The Command Execution API allows you to run shell commands (bash) directly in the cloud environment. This is useful for:

• Installing system dependencies (apt-get, pip)
• Managing files and directories
• Running background processes
• System diagnostics

Examples

Basic Command Execution (main.py)

Shows how to run simple commands, handle long-running processes, and inspect exit codes/output.

"""
AGB Command Execution Example

This example demonstrates how to execute shell commands in the cloud environment using the AGB SDK.
It covers:
1. Basic System Information
2. File Operations via Shell
3. Network Diagnostics
4. Advanced Data Processing Pipeline
5. System Monitoring Script
"""

import os
import sys
import time

from agb import AGB
from agb.session_params import CreateSessionParams

def main():
    # 1. Initialize AGB client
    api_key = os.getenv("AGB_API_KEY")
    if not api_key:
        print("Error: AGB_API_KEY environment variable is not set")
        return

    agb = AGB(api_key=api_key)

    # 2. Create a session
    print("Creating session...")
    params = CreateSessionParams(image_id="agb-code-space-1")
    result = agb.create(params)

    if not result.success:
        print(f"Failed to create session: {result.error_message}")
        return

    session = result.session
    print(f"Session created: {session.session_id}")

    try:
        # 3. System Information
        print("\n=== 1. System Information ===")
        commands = ["uname -a", "whoami", "date", "free -h"]

        for cmd in commands:
            print(f"> {cmd}")
            cmd_result = session.command.execute_command(cmd)
            if cmd_result.success:
                print(cmd_result.output.strip())
            else:
                print(f"Error: {cmd_result.error_message}")

        # 4. File Operations via Shell
        print("\n=== 2. File Operations (Shell) ===")

        # Create a file
        print("> echo 'Hello via Shell' > /tmp/shell_test.txt")
        session.command.execute_command("echo 'Hello via Shell' > /tmp/shell_test.txt")

        # Read it back
        print("> cat /tmp/shell_test.txt")
        read_result = session.command.execute_command("cat /tmp/shell_test.txt")
        print(read_result.output.strip())

        # List details
        print("> ls -l /tmp/shell_test.txt")
        ls_result = session.command.execute_command("ls -l /tmp/shell_test.txt")
        print(ls_result.output.strip())

        # 5. Network Diagnostics
        print("\n=== 3. Network Diagnostics ===")
        print("> curl -I https://www.google.com (Head Request)")
        curl_result = session.command.execute_command("curl -I -s --connect-timeout 5 https://www.google.com")
        if curl_result.success:
            # Print just the first few lines of headers
            print("\n".join(curl_result.output.splitlines()[:3]))
        else:
            print(f"Error: {curl_result.error_message}")

        # 6. Advanced Data Processing Pipeline (from Guide)
        print("\n=== 4. Data Processing Pipeline (Advanced) ===")

        # Step A: Create sample CSV data
        csv_data = """name,age,city,salary
Alice,25,New York,50000
Bob,30,London,60000
Charlie,35,Paris,70000
Diana,28,Tokyo,55000
Eve,32,Berlin,65000
Frank,27,Sydney,52000"""

        print("Creating CSV file...")
        session.command.execute_command(f"cat > /tmp/employees.csv << 'EOF'\n{csv_data}\nEOF")

        # Step B: Analyze with awk
        print("Analyzing data with awk...")
        awk_cmd = "awk -F',' 'NR>1 {sum+=$4; count++} END {print \"Average salary: $\" sum/count}' /tmp/employees.csv"
        print(f"> {awk_cmd}")
        awk_result = session.command.execute_command(awk_cmd)
        print(awk_result.output.strip())

        # Step C: Filter high earners
        filter_cmd = "awk -F',' 'NR>1 && $4>55000 {print $1 \": $\" $4}' /tmp/employees.csv"
        print(f"> {filter_cmd}")
        filter_result = session.command.execute_command(filter_cmd)
        print("High earners:")
        print(filter_result.output.strip())

        # 7. System Monitoring Script (from Guide)
        print("\n=== 5. System Monitoring Script ===")
        health_check_script = """
#!/bin/bash
echo "System Health Check - $(date)"
echo "================================"
# Check disk space
DISK_USAGE=$(df / | awk 'NR==2 {print $5}' | sed 's/%//')
echo "Disk usage: ${DISK_USAGE}%"
# Check memory
MEM_USAGE=$(free | awk 'NR==2{printf "%.0f", $3*100/$2}')
echo "Memory usage: ${MEM_USAGE}%"
echo "Health check completed"
"""
        # Write script
        print("Deploying health check script...")
        session.command.execute_command(f"cat > /tmp/health_check.sh << 'EOF'\n{health_check_script}\nEOF")
        session.command.execute_command("chmod +x /tmp/health_check.sh")

        # Run script
        print("Running health check...")
        health_result = session.command.execute_command("/tmp/health_check.sh")
        if health_result.success:
            print(health_result.output.strip())
        else:
            print(f"Error: {health_result.error_message}")

    finally:
        # 8. Cleanup
        print("\nCleaning up...")
        agb.delete(session)
        print("Session deleted")

if __name__ == "__main__":
    main()

Notes

• Root Access: You generally have sudo/root access in the container.
• Persistence: Changes to the filesystem (e.g., mkdir, apt-get install) persist for the duration of the session.
• State: Environment variables set in one command (e.g., export VAR=1) do not persist to the next command unless you chain them (e.g., export VAR=1 && echo $VAR) or write them to .bashrc (depending on the shell initialization).