voltagegpu-cli 1.0.0__py3-none-any.whl

volt/sdk/utils.py

@@ -0,0 +1,94 @@
+"""Utility helpers for the Lium SDK."""
+
+import hashlib
+import random
+import re
+import time
+from functools import wraps
+from typing import Callable, TypeVar
+
+import requests
+
+from .exceptions import LiumRateLimitError, LiumServerError
+
+F = TypeVar("F", bound=Callable[..., object])
+
+# Human-friendly ID parts
+ADJECTIVES = ["swift", "brave", "calm", "eager", "gentle", "cosmic", "golden", "lunar", "zesty", "noble"]
+NOUNS = ["hawk", "lion", "eagle", "fox", "wolf", "shark", "raven", "matrix", "comet", "orbit"]
+
+
+def generate_huid(id_str: str) -> str:
+    """Generate human-readable ID from UUID."""
+    if not id_str:
+        return "invalid"
+
+    digest = hashlib.md5(id_str.encode()).hexdigest()
+    adj = ADJECTIVES[int(digest[:4], 16) % len(ADJECTIVES)]
+    noun = NOUNS[int(digest[4:8], 16) % len(NOUNS)]
+    return f"{adj}-{noun}-{digest[-2:]}"
+
+
+def extract_gpu_type(machine_name: str) -> str:
+    """Extract GPU type from machine name."""
+    patterns = [
+        (r"RTX\s*(\d{4})", lambda m: f"RTX{m.group(1)}"),
+        (r"([HBL])(\d{2,3}S?)", lambda m: f"{m.group(1)}{m.group(2)}"),
+        (r"A(\d{2,4})", lambda m: f"A{m.group(1)}"),
+    ]
+    for pattern, fmt in patterns:
+        if match := re.search(pattern, machine_name, re.I):
+            return fmt(match)
+    return machine_name.split()[-1] if machine_name else "Unknown"
+
+
+def expand_gpu_shorthand(gpu_short: str) -> str:
+    """Expand GPU shorthand to a pattern that matches full machine names.
+
+    Examples:
+        A100 -> "A100" (matches "NVIDIA A100-SXM4-80GB", "NVIDIA A100-PCIE-40GB", etc.)
+        H200 -> "H200" (matches "NVIDIA H200", etc.)
+        RTX4090 -> "RTX 4090" (matches "NVIDIA GeForce RTX 4090", etc.)
+
+    Args:
+        gpu_short: Short GPU name like "A100", "H200", "RTX4090"
+
+    Returns:
+        Pattern string that can be used to filter machine names.
+    """
+    # Already a full name or pattern, return as-is
+    if len(gpu_short) > 10 or " " in gpu_short:
+        return gpu_short
+
+    gpu_upper = gpu_short.upper()
+
+    # Handle RTX cards - need to add space between RTX and number
+    if gpu_upper.startswith("RTX"):
+        # RTX4090 -> RTX 4090
+        match = re.match(r"RTX(\d+)", gpu_upper)
+        if match:
+            return f"RTX {match.group(1)}"
+
+    # For A-series (A100, A6000, etc.) and H-series (H100, H200, etc.)
+    # Just return as-is since the API does substring matching
+    # "A100" will match "NVIDIA A100-SXM4-80GB"
+    return gpu_short
+
+
+def with_retry(max_attempts: int = 3, delay: float = 1.0):
+    """Retry decorator for API calls."""
+    def decorator(func: F) -> F:
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            for attempt in range(max_attempts):
+                try:
+                    return func(*args, **kwargs)
+                except (LiumRateLimitError, LiumServerError, requests.RequestException):
+                    if attempt == max_attempts - 1:
+                        raise
+                    time.sleep(delay * (2 ** attempt) + random.uniform(0, 0.5))
+        return wrapper  # type: ignore[misc]
+    return decorator
+
+
+__all__ = ["generate_huid", "extract_gpu_type", "expand_gpu_shorthand", "with_retry"]

voltagegpu_cli-1.0.0.dist-info/METADATA

@@ -0,0 +1,288 @@
+Metadata-Version: 2.4
+Name: voltagegpu-cli
+Version: 1.0.0
+Summary: VoltageGPU CLI - Command-line interface for managing GPU cloud resources on voltagegpu.com. Deploy, manage, and scale your GPU workloads with ease.
+Project-URL: Homepage, https://voltagegpu.com
+Project-URL: Documentation, https://docs.voltagegpu.com
+Project-URL: Repository, https://github.com/voltagegpu/voltagegpu-cli
+Project-URL: Issues, https://github.com/voltagegpu/voltagegpu-cli/issues
+Author-email: VoltageGPU <support@voltagegpu.com>
+Keywords: ai,cli,cloud,deep-learning,gpu,machine-learning,voltagegpu
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: System :: Distributed Computing
+Requires-Python: >=3.9
+Requires-Dist: click>=8.0
+Requires-Dist: httpx>=0.24.0
+Provides-Extra: dev
+Requires-Dist: black; extra == 'dev'
+Requires-Dist: pytest; extra == 'dev'
+Requires-Dist: wheel; extra == 'dev'
+Provides-Extra: docs
+Requires-Dist: sphinx-rtd-theme>=1.3; extra == 'docs'
+Requires-Dist: sphinx>=7.3; extra == 'docs'
+Description-Content-Type: text/markdown
+
+# VoltageGPU CLI
+
+Command-line interface and Python SDK for VoltageGPU - affordable GPU cloud computing for AI/ML workloads.
+
+## Installation
+
+```bash
+pip install voltagegpu-cli
+```
+
+Or install from source:
+
+```bash
+git clone https://github.com/voltagegpu/voltagegpu-cli.git
+cd voltagegpu-cli
+pip install -e .
+```
+
+## Quick Start
+
+### 1. Configure your API key
+
+Set the environment variable:
+
+```bash
+export VOLT_API_KEY="your_api_key_here"
+```
+
+Or create a config file at `~/.volt/config.ini`:
+
+```ini
+[api]
+api_key = your_api_key_here
+```
+
+### 2. List available templates
+
+```bash
+volt templates list
+```
+
+### 3. Create a pod
+
+```bash
+volt pods create --template <template_id> --name my-gpu-pod
+```
+
+### 4. SSH into your pod
+
+```bash
+volt pods ssh <pod_id>
+```
+
+## CLI Commands
+
+### Pods
+
+```bash
+# List all pods
+volt pods list
+volt pods list --json
+
+# Get pod details
+volt pods get <pod_id>
+
+# Create a new pod
+volt pods create --template <template_id> --name <name> [--gpu-count 1] [--ssh-key <key_id>]
+
+# Start/Stop/Delete pods
+volt pods start <pod_id>
+volt pods stop <pod_id>
+volt pods delete <pod_id> [--yes]
+
+# Get SSH command
+volt pods ssh <pod_id>
+```
+
+### Templates
+
+```bash
+# List available templates
+volt templates list
+volt templates list --category llm
+volt templates list --json
+
+# Get template details
+volt templates get <template_id>
+```
+
+### SSH Keys
+
+```bash
+# List SSH keys
+volt ssh-keys list
+
+# Add a new SSH key
+volt ssh-keys add --name "my-key" --file ~/.ssh/id_ed25519.pub
+volt ssh-keys add --name "my-key" --key "ssh-ed25519 AAAA..."
+
+# Delete an SSH key
+volt ssh-keys delete <key_id>
+```
+
+### Machines
+
+```bash
+# List available machines
+volt machines list
+volt machines list --gpu RTX4090
+volt machines list --json
+```
+
+### Account
+
+```bash
+# Check balance
+volt account balance
+
+# Get account info
+volt account info
+```
+
+### Configuration
+
+```bash
+# Show current configuration
+volt config
+```
+
+## Python SDK
+
+You can also use VoltageGPU as a Python library:
+
+```python
+from volt import VoltageGPUClient
+
+# Initialize client (uses VOLT_API_KEY env var or ~/.volt/config.ini)
+client = VoltageGPUClient()
+
+# List pods
+pods = client.list_pods()
+for pod in pods:
+    print(f"{pod.name}: {pod.status} ({pod.gpu_type})")
+
+# Create a pod
+pod = client.create_pod(
+    template_id="template-123",
+    name="my-training-pod",
+    gpu_count=2
+)
+print(f"Created pod: {pod.id}")
+
+# Get SSH connection info
+pod = client.get_pod(pod.id)
+print(f"SSH: ssh -p {pod.ssh_port} root@{pod.ssh_host}")
+
+# Stop and delete
+client.stop_pod(pod.id)
+client.delete_pod(pod.id)
+```
+
+### Context Manager
+
+```python
+from volt import VoltageGPUClient
+
+with VoltageGPUClient() as client:
+    templates = client.list_templates()
+    for t in templates:
+        print(f"{t.name}: ${t.hourly_price}/hr")
+```
+
+### Custom Configuration
+
+```python
+from volt.sdk import Config, VoltageGPUClient
+
+config = Config(
+    api_key="your_api_key",
+    base_url="https://voltagegpu.com/api"
+)
+client = VoltageGPUClient(config=config)
+```
+
+## Environment Variables
+
+| Variable | Description | Default |
+|----------|-------------|---------|
+| `VOLT_API_KEY` | Your VoltageGPU API key | - |
+| `VOLT_BASE_URL` | API base URL | `https://voltagegpu.com/api` |
+| `LIUM_API_KEY` | Legacy API key (fallback) | - |
+
+## Configuration File
+
+Create `~/.volt/config.ini`:
+
+```ini
+[api]
+api_key = your_api_key_here
+```
+
+## Output Formats
+
+Most list commands support `--json` flag for machine-readable output:
+
+```bash
+volt pods list --json | jq '.[] | select(.status == "running")'
+```
+
+## Examples
+
+### Launch a training job
+
+```bash
+# Find a suitable template
+volt templates list --category ml
+
+# Create pod with SSH key
+volt pods create \
+  --template pytorch-cuda12 \
+  --name training-job \
+  --gpu-count 4 \
+  --ssh-key my-key-id
+
+# Get SSH command
+volt pods ssh <pod_id>
+```
+
+### Monitor costs
+
+```bash
+# Check balance
+volt account balance
+
+# List running pods with costs
+volt pods list | grep running
+```
+
+### Batch operations with jq
+
+```bash
+# Stop all running pods
+volt pods list --json | jq -r '.[] | select(.status == "running") | .id' | xargs -I {} volt pods stop {}
+```
+
+## Support
+
+- Website: https://voltagegpu.com
+- Documentation: https://docs.voltagegpu.com
+- Email: support@voltagegpu.com
+
+## License
+
+MIT License - see LICENSE file for details.

voltagegpu_cli-1.0.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+volt/__init__.py,sha256=5mj6qn88CHB7HTYEfxpyf6KVNGAPLOAwkLnWRxqPQ8U,169
+volt/cli/__init__.py,sha256=C30c4Qb9If5qp3-DL5izT4aqghi09o2DCjjMEoh1LqY,79
+volt/cli/cli.py,sha256=dVUygtLU-sq8B2kpeQ8F5v8vnMyCMzm4iDx6QefYrd8,13060
+volt/sdk/__init__.py,sha256=4Cv6YSKLInZBrO2WIaF3BTwxtj7CxUk_QxtoEK3Q-Xo,193
+volt/sdk/client.py,sha256=Tl02DK-elpGmb0HLYgrt5chBEmrECnYkqag6SeFyysE,9351
+volt/sdk/config.py,sha256=j01zsvw6GOnZ98CaSVdE3iKCZLf3T5OfTQNmTEWISh8,2286
+volt/sdk/decorators.py,sha256=wtWhqEA4VvS9xf19yzWz7tNd0c4BjXHKpY5buhOecvA,7849
+volt/sdk/exceptions.py,sha256=BK_RC1ef5LV50f6A-Z5BZaeN9KKKqjN5MRcQzAZmtKA,539
+volt/sdk/models.py,sha256=PC3zpXadXF4VE6vHagTBBC_GFYtM0q7fa7QbgpheB_E,3125
+volt/sdk/utils.py,sha256=NP58Hi5_Z5bDQYF5qaA8c-TmWfOoMIqEf0c8ZplTH9o,3276
+voltagegpu_cli-1.0.0.dist-info/METADATA,sha256=1qK4A2iXZDJdV09gKglUvnHkDAtjlJDBbHuhsbyFY_0,5887
+voltagegpu_cli-1.0.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+voltagegpu_cli-1.0.0.dist-info/entry_points.txt,sha256=oncTVsfHJOq_-Fl4QdWbbXfSVXbCRlYH3bbzxTcgu8s,43
+voltagegpu_cli-1.0.0.dist-info/RECORD,,

voltagegpu_cli-1.0.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any

voltagegpu_cli-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+volt = volt.cli.cli:main