voltagegpu-cli 1.0.0__py3-none-any.whl

volt/__init__.py

@@ -0,0 +1,6 @@
+"""VoltageGPU - GPU Cloud Computing CLI and SDK."""
+
+from volt.sdk import VoltageGPUClient, Config
+
+__all__ = ["VoltageGPUClient", "Config"]
+__version__ = "1.0.0"

volt/cli/__init__.py

@@ -0,0 +1,3 @@
+"""VoltageGPU CLI - Command line interface for VoltageGPU."""
+
+__all__ = []

volt/cli/cli.py

@@ -0,0 +1,417 @@
+"""VoltageGPU CLI - Command-line interface for managing GPU cloud resources."""
+
+import click
+import json
+import sys
+from typing import Optional
+
+
+def get_client():
+    """Get an authenticated VoltageGPU client."""
+    try:
+        from volt.sdk import VoltageGPUClient
+        return VoltageGPUClient()
+    except ValueError as e:
+        click.echo(f"Error: {e}", err=True)
+        click.echo("Set your API key: export VOLT_API_KEY=your_key", err=True)
+        click.echo("Or create ~/.volt/config.ini with [api] api_key=your_key", err=True)
+        sys.exit(1)
+    except Exception as e:
+        click.echo(f"Error initializing client: {e}", err=True)
+        sys.exit(1)
+
+
+def format_table(headers, rows):
+    """Format data as a simple aligned table."""
+    if not rows:
+        click.echo("No results found.")
+        return
+    col_widths = [len(h) for h in headers]
+    for row in rows:
+        for i, cell in enumerate(row):
+            col_widths[i] = max(col_widths[i], len(str(cell)))
+    header_line = "  ".join(h.ljust(col_widths[i]) for i, h in enumerate(headers))
+    separator = "  ".join("-" * w for w in col_widths)
+    click.echo(header_line)
+    click.echo(separator)
+    for row in rows:
+        click.echo("  ".join(str(cell).ljust(col_widths[i]) for i, cell in enumerate(row)))
+
+
+@click.group(invoke_without_command=True)
+@click.version_option(version="1.0.0", prog_name="volt")
+@click.pass_context
+def cli(ctx):
+    """VoltageGPU CLI - Manage GPU cloud resources from your terminal.
+
+    Deploy, manage, and scale GPU pods with ease.
+    Get started: https://docs.voltagegpu.com/cli
+    """
+    if ctx.invoked_subcommand is None:
+        click.echo(ctx.get_help())
+
+
+# ==================== PODS ====================
+
+@cli.group()
+def pods():
+    """Manage GPU pods."""
+    pass
+
+
+@pods.command("list")
+@click.option("--json", "as_json", is_flag=True, help="Output as JSON")
+def pods_list(as_json):
+    """List all your pods."""
+    client = get_client()
+    try:
+        pods_data = client.list_pods()
+        if as_json:
+            click.echo(json.dumps([{
+                "id": p.id, "name": p.name, "status": p.status,
+                "gpu_type": p.gpu_type, "gpu_count": p.gpu_count,
+                "hourly_price": p.hourly_price
+            } for p in pods_data], indent=2))
+        else:
+            rows = [(p.id[:12] + "...", p.name, p.status, p.gpu_type,
+                     str(p.gpu_count), f"${p.hourly_price:.2f}/hr") for p in pods_data]
+            format_table(["ID", "NAME", "STATUS", "GPU", "COUNT", "PRICE"], rows)
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@pods.command("get")
+@click.argument("pod_id")
+@click.option("--json", "as_json", is_flag=True, help="Output as JSON")
+def pods_get(pod_id, as_json):
+    """Get details of a specific pod."""
+    client = get_client()
+    try:
+        pod = client.get_pod(pod_id)
+        if as_json:
+            click.echo(json.dumps({
+                "id": pod.id, "name": pod.name, "status": pod.status,
+                "gpu_type": pod.gpu_type, "gpu_count": pod.gpu_count,
+                "hourly_price": pod.hourly_price, "ssh_host": pod.ssh_host,
+                "ssh_port": pod.ssh_port, "template_id": pod.template_id,
+                "created_at": pod.created_at
+            }, indent=2))
+        else:
+            click.echo(f"Pod: {pod.name}")
+            click.echo(f"  ID:       {pod.id}")
+            click.echo(f"  Status:   {pod.status}")
+            click.echo(f"  GPU:      {pod.gpu_count}x {pod.gpu_type}")
+            click.echo(f"  Price:    ${pod.hourly_price:.2f}/hr")
+            if pod.ssh_host:
+                click.echo(f"  SSH:      ssh root@{pod.ssh_host} -p {pod.ssh_port or 22}")
+            if pod.template_id:
+                click.echo(f"  Template: {pod.template_id}")
+            if pod.created_at:
+                click.echo(f"  Created:  {pod.created_at}")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@pods.command("create")
+@click.option("--template", "template_id", required=True, help="Template ID to use")
+@click.option("--name", required=True, help="Name for the pod")
+@click.option("--gpu-count", default=1, type=int, help="Number of GPUs (default: 1)")
+@click.option("--ssh-key", "ssh_key_ids", multiple=True, help="SSH key ID(s) to attach")
+def pods_create(template_id, name, gpu_count, ssh_key_ids):
+    """Deploy a new GPU pod."""
+    client = get_client()
+    try:
+        pod = client.create_pod(
+            template_id=template_id,
+            name=name,
+            gpu_count=gpu_count,
+            ssh_key_ids=list(ssh_key_ids) if ssh_key_ids else None
+        )
+        click.echo(f"Pod created successfully!")
+        click.echo(f"  ID:     {pod.id}")
+        click.echo(f"  Name:   {pod.name}")
+        click.echo(f"  Status: {pod.status}")
+        click.echo(f"  GPU:    {pod.gpu_count}x {pod.gpu_type}")
+        click.echo(f"  Price:  ${pod.hourly_price:.2f}/hr")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@pods.command("start")
+@click.argument("pod_id")
+def pods_start(pod_id):
+    """Start a stopped pod."""
+    client = get_client()
+    try:
+        pod = client.start_pod(pod_id)
+        click.echo(f"Pod {pod.name} started. Status: {pod.status}")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@pods.command("stop")
+@click.argument("pod_id")
+def pods_stop(pod_id):
+    """Stop a running pod."""
+    client = get_client()
+    try:
+        pod = client.stop_pod(pod_id)
+        click.echo(f"Pod {pod.name} stopped. Status: {pod.status}")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@pods.command("delete")
+@click.argument("pod_id")
+@click.option("--yes", is_flag=True, help="Skip confirmation")
+def pods_delete(pod_id, yes):
+    """Terminate and delete a pod permanently."""
+    if not yes:
+        click.confirm(f"Are you sure you want to delete pod {pod_id}?", abort=True)
+    client = get_client()
+    try:
+        client.delete_pod(pod_id)
+        click.echo(f"Pod {pod_id} deleted.")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@pods.command("ssh")
+@click.argument("pod_id")
+def pods_ssh(pod_id):
+    """Show SSH command to connect to a pod."""
+    client = get_client()
+    try:
+        pod = client.get_pod(pod_id)
+        if pod.ssh_host:
+            port = pod.ssh_port or 22
+            cmd = f"ssh root@{pod.ssh_host} -p {port}"
+            click.echo(cmd)
+        else:
+            click.echo("SSH not available for this pod.", err=True)
+            sys.exit(1)
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+# ==================== TEMPLATES ====================
+
+@cli.group()
+def templates():
+    """Browse available pod templates."""
+    pass
+
+
+@templates.command("list")
+@click.option("--category", help="Filter by category (e.g., llm, ml, diffusion)")
+@click.option("--json", "as_json", is_flag=True, help="Output as JSON")
+def templates_list(category, as_json):
+    """List available templates."""
+    client = get_client()
+    try:
+        templates_data = client.list_templates(category=category)
+        if as_json:
+            click.echo(json.dumps([{
+                "id": t.id, "name": t.name, "category": t.category,
+                "docker_image": t.docker_image
+            } for t in templates_data], indent=2))
+        else:
+            rows = [(t.id[:12] + "...", t.name, t.category or "-",
+                     t.docker_image[:40] + "..." if len(t.docker_image) > 40 else t.docker_image)
+                    for t in templates_data]
+            format_table(["ID", "NAME", "CATEGORY", "DOCKER IMAGE"], rows)
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@templates.command("get")
+@click.argument("template_id")
+def templates_get(template_id):
+    """Get details of a specific template."""
+    client = get_client()
+    try:
+        t = client.get_template(template_id)
+        click.echo(f"Template: {t.name}")
+        click.echo(f"  ID:           {t.id}")
+        click.echo(f"  Description:  {t.description}")
+        click.echo(f"  Docker Image: {t.docker_image}")
+        click.echo(f"  Category:     {t.category or '-'}")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+# ==================== SSH KEYS ====================
+
+@cli.group("ssh-keys")
+def ssh_keys():
+    """Manage SSH keys."""
+    pass
+
+
+@ssh_keys.command("list")
+@click.option("--json", "as_json", is_flag=True, help="Output as JSON")
+def ssh_keys_list(as_json):
+    """List your SSH keys."""
+    client = get_client()
+    try:
+        keys = client.list_ssh_keys()
+        if as_json:
+            click.echo(json.dumps([{
+                "id": k.id, "name": k.name, "fingerprint": k.fingerprint,
+                "created_at": k.created_at
+            } for k in keys], indent=2))
+        else:
+            rows = [(k.id[:12] + "...", k.name, k.fingerprint or "-",
+                     k.created_at or "-") for k in keys]
+            format_table(["ID", "NAME", "FINGERPRINT", "CREATED"], rows)
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@ssh_keys.command("add")
+@click.option("--name", required=True, help="Label for this key")
+@click.option("--file", "key_file", type=click.Path(exists=True), help="Path to public key file")
+@click.option("--key", "key_str", help="Public key string directly")
+def ssh_keys_add(name, key_file, key_str):
+    """Add a new SSH key."""
+    if not key_file and not key_str:
+        click.echo("Error: provide --file or --key", err=True)
+        sys.exit(1)
+    if key_file:
+        with open(key_file) as f:
+            public_key = f.read().strip()
+    else:
+        public_key = key_str
+    client = get_client()
+    try:
+        key = client.add_ssh_key(name=name, public_key=public_key)
+        click.echo(f"SSH key added: {key.name} ({key.id})")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@ssh_keys.command("delete")
+@click.argument("key_id")
+def ssh_keys_delete(key_id):
+    """Remove an SSH key."""
+    client = get_client()
+    try:
+        client.delete_ssh_key(key_id)
+        click.echo(f"SSH key {key_id} deleted.")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+# ==================== MACHINES ====================
+
+@cli.group()
+def machines():
+    """View available GPU hardware."""
+    pass
+
+
+@machines.command("list")
+@click.option("--gpu", help="Filter by GPU type (e.g., RTX4090, A100, H100)")
+@click.option("--json", "as_json", is_flag=True, help="Output as JSON")
+def machines_list(gpu, as_json):
+    """List available GPU machines with pricing."""
+    client = get_client()
+    try:
+        machines_data = client.list_machines(gpu_type=gpu)
+        if as_json:
+            click.echo(json.dumps([{
+                "name": m.name, "hourly_price": m.hourly_price,
+                "total_gpu_count": m.total_gpu_count, "available": m.available
+            } for m in machines_data], indent=2))
+        else:
+            rows = [(m.name, str(m.total_gpu_count),
+                     f"${m.hourly_price:.2f}/hr",
+                     "Yes" if m.available else "No") for m in machines_data]
+            format_table(["GPU", "AVAILABLE COUNT", "PRICE", "IN STOCK"], rows)
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+# ==================== ACCOUNT ====================
+
+@cli.group()
+def account():
+    """View account details and balance."""
+    pass
+
+
+@account.command("balance")
+def account_balance():
+    """Check your account credit balance."""
+    client = get_client()
+    try:
+        balance = client.get_balance()
+        click.echo(f"Balance: ${balance:.2f}")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+@account.command("info")
+def account_info():
+    """Show your account details."""
+    client = get_client()
+    try:
+        info = client.get_account_info()
+        for key, value in info.items():
+            click.echo(f"  {key}: {value}")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+    finally:
+        client.close()
+
+
+def main():
+    """Main entry point for the CLI."""
+    cli()
+
+
+if __name__ == "__main__":
+    main()

volt/sdk/__init__.py

@@ -0,0 +1,7 @@
+"""VoltageGPU SDK - Python client for VoltageGPU API."""
+
+from .config import Config
+from .client import VoltageGPUClient
+
+__all__ = ["Config", "VoltageGPUClient"]
+__version__ = "0.1.0"