Skill: VAST.ai GPU Rental

Skill: VAST.ai GPU Rental

Overview

This skill allows you to provision on-demand GPU infrastructure. You must have a user-provided API Key before performing write actions.

Capabilities

1. Search: Find machines by GPU model (e.g., "RTX 4090") and max hourly price.
2. Rent: Instantiate a container (default: PyTorch) on a specific Offer ID.
3. Connect: Retrieve the SSH connection string for an active instance.
4. Balance: Check available credit and current hourly burn rate across all active machines.

Usage Protocol

• Step 1: Ask the user for their VAST API Key if not already in context.
• Pre-flight Check: Before renting, call balance to ensure the user has sufficient funds.
• Step 2: Search for offers and present the top 3 cheapest options to the user.
• Step 3: Upon confirmation, call rent.
• Reporting: If credit is below $5.00, warn the user after every successful rental.
• Step 4: Wait 30-60 seconds, then call connect to provide the SSH string.

Tool Definitions

• search(gpu: string, price: number)
• rent(id: number, image: string)
• connect(id: number)
• balance()

Execution

Run the CLI wrapper for these tools. Command: node /Users/sschepis/Development/vast-ai/dist/cli.js <action> [params] Env: VAST_API_KEY must be set.

Examples

• Search: node dist/cli.js search --gpu "RTX 4090" --price 0.5
• Rent: node dist/cli.js rent --id 12345 --image "pytorch/pytorch"
• Connect: node dist/cli.js connect --id 12345
• Balance: node dist/cli.js balance