@nookplot/cli 0.6.78 → 0.6.80

package/dist/commands/gpu.d.ts

@@ -1,12 +1,17 @@
 /**
  * GPU marketplace CLI commands.
  *
- * nookplot gpu benchmark  — Run standardized benchmark (detects GPU specs)
- * nookplot gpu register   — Submit attestation on-chain
- * nookplot gpu update     — Re-run benchmark + update attestation
- * nookplot gpu status     — Show attestation, availability, active jobs
- * nookplot gpu list       — Browse available GPUs
- * nookplot gpu challenge  — Challenge a provider's benchmark
+ * nookplot gpu benchmark     — Run standardized benchmark (detects GPU specs)
+ * nookplot gpu register      — Submit attestation on-chain
+ * nookplot gpu update        — Re-run benchmark + update attestation
+ * nookplot gpu status        — Show attestation, availability, active jobs
+ * nookplot gpu list          — Browse available GPUs
+ * nookplot gpu challenge     — Challenge a provider's benchmark
+ * nookplot gpu rent          — Rent a GPU by creating a service agreement
+ * nookplot gpu rent-swarm    — Rent multiple GPUs for a research swarm
+ * nookplot gpu provision     — Interactive setup on current machine
+ * nookplot gpu setup-script  — Generate a bash provisioning script
+ * nookplot gpu serve         — Go online, send heartbeats, listen for jobs
  *
  * @module commands/gpu
  */

package/dist/commands/gpu.js

@@ -1,18 +1,25 @@
 /**
  * GPU marketplace CLI commands.
  *
- * nookplot gpu benchmark  — Run standardized benchmark (detects GPU specs)
- * nookplot gpu register   — Submit attestation on-chain
- * nookplot gpu update     — Re-run benchmark + update attestation
- * nookplot gpu status     — Show attestation, availability, active jobs
- * nookplot gpu list       — Browse available GPUs
- * nookplot gpu challenge  — Challenge a provider's benchmark
+ * nookplot gpu benchmark     — Run standardized benchmark (detects GPU specs)
+ * nookplot gpu register      — Submit attestation on-chain
+ * nookplot gpu update        — Re-run benchmark + update attestation
+ * nookplot gpu status        — Show attestation, availability, active jobs
+ * nookplot gpu list          — Browse available GPUs
+ * nookplot gpu challenge     — Challenge a provider's benchmark
+ * nookplot gpu rent          — Rent a GPU by creating a service agreement
+ * nookplot gpu rent-swarm    — Rent multiple GPUs for a research swarm
+ * nookplot gpu provision     — Interactive setup on current machine
+ * nookplot gpu setup-script  — Generate a bash provisioning script
+ * nookplot gpu serve         — Go online, send heartbeats, listen for jobs
  *
  * @module commands/gpu
  */
 import chalk from "chalk";
 import ora from "ora";
+import { writeFileSync } from "node:fs";
 import { ethers } from "ethers";
+import inquirer from "inquirer";
 import { loadConfig, validateConfig } from "../config.js";
 import { gatewayRequest, isGatewayError } from "../utils/http.js";
 export function registerGpuCommand(program) {
@@ -135,6 +142,85 @@ export function registerGpuCommand(program) {
         }
     });
     // ============================================================
+    //  nookplot gpu rent
+    // ============================================================
+    gpu
+        .command("rent <listingId>")
+        .description("Rent a GPU by creating a marketplace service agreement")
+        .option("--terms <string>", "Agreement terms")
+        .option("--duration <hours>", "Rental duration in hours", parseInt)
+        .option("--amount <usdc>", "Token amount to pay")
+        .option("--token <usdc|nook>", "Payment token (usdc or nook)")
+        .option("--json", "Output raw JSON")
+        .option("-y, --yes", "Skip confirmation prompt")
+        .action(async (listingId, opts) => {
+        try {
+            await runRent(program.opts(), listingId, opts);
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            console.error(chalk.red(`\nRental failed: ${msg}`));
+            process.exit(1);
+        }
+    });
+    // ============================================================
+    //  nookplot gpu setup-script
+    // ============================================================
+    gpu
+        .command("setup-script")
+        .description("Generate a bash setup script for GPU provisioning")
+        .option("--agreement-id <id>", "Service agreement ID")
+        .option("--output <file>", "Write script to file instead of stdout")
+        .action(async (opts) => {
+        try {
+            await runSetupScript(program.opts(), opts);
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            console.error(chalk.red(`\nSetup script generation failed: ${msg}`));
+            process.exit(1);
+        }
+    });
+    // ============================================================
+    //  nookplot gpu rent-swarm
+    // ============================================================
+    gpu
+        .command("rent-swarm")
+        .description("Rent multiple GPUs for a research swarm")
+        .option("--count <n>", "Number of GPUs to rent", parseInt)
+        .option("--min-vram <gb>", "Minimum VRAM in GB per GPU", parseInt)
+        .option("--gpu-model <model>", "Preferred GPU model (e.g. A100, H100)")
+        .option("--duration <hours>", "Rental duration in hours", parseInt)
+        .option("--json", "Output raw JSON")
+        .action(async (opts) => {
+        try {
+            await runRentSwarm(program.opts(), opts);
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            console.error(chalk.red(`\nSwarm rental failed: ${msg}`));
+            process.exit(1);
+        }
+    });
+    // ============================================================
+    //  nookplot gpu provision
+    // ============================================================
+    gpu
+        .command("provision")
+        .description("Interactive setup: detect GPU, install deps, configure autoresearch")
+        .option("--repo <path>", "Path to autoresearch git repo")
+        .option("--listing-id <id>", "GPU listing ID to validate against", parseInt)
+        .action(async (opts) => {
+        try {
+            await runProvision(program.opts(), opts);
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            console.error(chalk.red(`\nProvisioning failed: ${msg}`));
+            process.exit(1);
+        }
+    });
+    // ============================================================
     //  nookplot gpu serve
     // ============================================================
     gpu
@@ -546,4 +632,391 @@ async function runServe(globalOpts, cmdOpts) {
     // Keep process alive
     await new Promise(() => { });
 }
+async function runRent(globalOpts, listingId, cmdOpts) {
+    const config = loadConfig({
+        configPath: globalOpts.config,
+        gatewayOverride: globalOpts.gateway,
+        apiKeyOverride: globalOpts.apiKey,
+    });
+    const errors = validateConfig(config);
+    if (errors.length > 0) {
+        for (const e of errors)
+            console.error(chalk.red(`  \u2717 ${e}`));
+        process.exit(1);
+    }
+    if (!config.privateKey) {
+        console.error(chalk.red("  \u2717 Private key required. Set NOOKPLOT_AGENT_PRIVATE_KEY."));
+        process.exit(1);
+    }
+    const lid = parseInt(listingId, 10);
+    if (isNaN(lid) || lid <= 0) {
+        console.error(chalk.red("  \u2717 Invalid listing ID. Must be a positive integer."));
+        process.exit(1);
+    }
+    const durationHours = cmdOpts.duration ?? 24;
+    const tokenAmount = cmdOpts.amount ?? "0";
+    // Resolve token address
+    let tokenAddress;
+    if (cmdOpts.token === "nook") {
+        tokenAddress = "0xb233BDFFD437E60fA451F62c6c09D3804d285Ba3";
+    }
+    // Fetch listing details first
+    const spinner = ora("Fetching listing details...").start();
+    const listingRes = await gatewayRequest(config.gateway, "GET", `/v1/gpu/availability/${lid}`, { apiKey: config.apiKey });
+    if (isGatewayError(listingRes)) {
+        spinner.fail("Failed to fetch listing details");
+        console.error(chalk.red(`  ${listingRes.error}`));
+        process.exit(1);
+    }
+    const listing = listingRes.data.availability;
+    spinner.stop();
+    // Show confirmation
+    console.log(chalk.bold("\n  GPU Rental Summary\n"));
+    console.log(`  Listing:    #${lid}`);
+    console.log(`  GPU:        ${chalk.cyan(listing?.gpu_model || "Unknown")}`);
+    console.log(`  VRAM:       ${chalk.cyan((listing?.vram_gb ?? 0) + " GB")}`);
+    console.log(`  Status:     ${listing?.status === "online" ? chalk.green(listing.status) : chalk.yellow(listing?.status ?? "unknown")}`);
+    console.log(`  Provider:   ${chalk.dim(listing?.provider_address?.slice(0, 10) + "\u2026" || "unknown")}`);
+    console.log(`  Duration:   ${durationHours} hours`);
+    console.log(`  Amount:     ${tokenAmount} ${cmdOpts.token?.toUpperCase() ?? "USDC"}`);
+    console.log("");
+    if (!cmdOpts.yes) {
+        const { confirm } = await inquirer.prompt([{
+                type: "confirm",
+                name: "confirm",
+                message: "Proceed with GPU rental?",
+                default: false,
+            }]);
+        if (!confirm) {
+            console.log(chalk.dim("\n  Rental cancelled.\n"));
+            return;
+        }
+    }
+    const wallet = new ethers.Wallet(config.privateKey);
+    const rentSpinner = ora("Creating GPU rental agreement on-chain...").start();
+    const deadline = Math.floor(Date.now() / 1000) + durationHours * 3600;
+    // Step 1: Prepare
+    const prepResult = await gatewayRequest(config.gateway, "POST", "/v1/prepare/service/agree", {
+        apiKey: config.apiKey,
+        body: {
+            listingId: lid,
+            terms: cmdOpts.terms ?? "GPU rental via Nookplot marketplace",
+            deadline: String(deadline),
+            tokenAmount,
+            tokenAddress,
+        },
+    });
+    if (isGatewayError(prepResult)) {
+        rentSpinner.fail("Failed to prepare rental agreement");
+        console.error(chalk.red(`  ${prepResult.error}`));
+        process.exit(1);
+    }
+    // Step 2: Sign
+    const { forwardRequest, domain, types } = prepResult.data;
+    const sig = await wallet.signTypedData(domain, types, forwardRequest);
+    // Step 3: Relay
+    const relayResult = await gatewayRequest(config.gateway, "POST", "/v1/relay", { apiKey: config.apiKey, body: { ...forwardRequest, signature: sig } });
+    if (isGatewayError(relayResult)) {
+        rentSpinner.fail("Failed to relay rental agreement");
+        console.error(chalk.red(`  ${relayResult.error}`));
+        process.exit(1);
+    }
+    if (cmdOpts.json) {
+        rentSpinner.stop();
+        console.log(JSON.stringify({
+            txHash: relayResult.data.txHash,
+            agreementId: relayResult.data.agreementId,
+            listingId: lid,
+            gpuModel: listing?.gpu_model,
+            durationHours,
+        }, null, 2));
+        return;
+    }
+    rentSpinner.succeed(chalk.green("GPU rental agreement created"));
+    console.log(`    TX:           ${relayResult.data.txHash}`);
+    if (relayResult.data.agreementId) {
+        console.log(`    Agreement:    #${relayResult.data.agreementId}`);
+    }
+    console.log(`    GPU:          ${listing?.gpu_model || "Unknown"}`);
+    console.log(`    Duration:     ${durationHours} hours`);
+    console.log("");
+    console.log(chalk.dim("  Use `nookplot gpu setup-script` to generate a provisioning script."));
+    console.log("");
+}
+async function runSetupScript(_globalOpts, cmdOpts) {
+    const script = `#!/usr/bin/env bash
+# Nookplot GPU Provisioning Script
+# Generated by: nookplot gpu setup-script
+# ─────────────────────────────────────────────────────────────
+set -euo pipefail
+
+echo "=== Nookplot GPU Setup ==="
+echo ""
+
+# 1. Check Python version
+PYTHON=""
+for cmd in python3.12 python3.11 python3.10 python3; do
+  if command -v "$cmd" &> /dev/null; then
+    PYTHON="$cmd"
+    break
+  fi
+done
+
+if [ -z "$PYTHON" ]; then
+  echo "ERROR: Python 3.10+ is required. Install it first."
+  exit 1
+fi
+
+PY_VERSION=$($PYTHON -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')
+echo "Using $PYTHON (version $PY_VERSION)"
+
+# 2. Check nvidia-smi
+if ! command -v nvidia-smi &> /dev/null; then
+  echo "WARNING: nvidia-smi not found. GPU experiments may not work."
+else
+  echo "GPU: $(nvidia-smi --query-gpu=name --format=csv,noheader 2>/dev/null || echo 'unknown')"
+  echo "VRAM: $(nvidia-smi --query-gpu=memory.total --format=csv,noheader 2>/dev/null || echo 'unknown')"
+fi
+
+# 3. Create virtualenv
+VENV_DIR="\${HOME}/.nookplot/gpu-venv"
+if [ ! -d "$VENV_DIR" ]; then
+  echo ""
+  echo "Creating virtualenv at $VENV_DIR..."
+  $PYTHON -m venv "$VENV_DIR"
+fi
+source "$VENV_DIR/bin/activate"
+
+# 4. Install nookplot-autoresearch
+echo ""
+echo "Installing nookplot-autoresearch..."
+pip install --upgrade pip > /dev/null 2>&1
+pip install nookplot-autoresearch > /dev/null 2>&1 || {
+  echo "WARNING: nookplot-autoresearch not on PyPI yet."
+  echo "Install from source: pip install -e /path/to/integrations/autoresearch"
+}
+
+# 5. Copy credentials
+CRED_DIR="\${HOME}/.nookplot"
+mkdir -p "$CRED_DIR"
+
+if [ -f "nookplot.yaml" ]; then
+  cp nookplot.yaml "$CRED_DIR/nookplot.yaml"
+  echo "Copied nookplot.yaml to $CRED_DIR"
+fi
+
+if [ -f ".env" ] && grep -q "NOOKPLOT_" ".env"; then
+  # Extract NOOKPLOT_ vars only
+  grep "^NOOKPLOT_" .env >> "$CRED_DIR/.env" 2>/dev/null || true
+  echo "Copied NOOKPLOT_ env vars to $CRED_DIR/.env"
+fi
+
+${cmdOpts.agreementId ? `# 6. Record agreement ID
+echo "${cmdOpts.agreementId}" > "$CRED_DIR/gpu-agreement-id"
+echo "Agreement ID: ${cmdOpts.agreementId}"
+` : `# 6. Agreement ID (none specified)
+# Pass --agreement-id <id> to nookplot gpu setup-script to auto-configure.
+`}
+# 7. Verify setup
+echo ""
+echo "=== Setup Complete ==="
+echo ""
+echo "Next steps:"
+echo "  1. Activate the venv:  source $VENV_DIR/bin/activate"
+echo "  2. Run experiments:    nookplot-autoresearch run --config research.yaml"
+echo "  3. Monitor GPU:        watch nvidia-smi"
+echo ""
+`;
+    if (cmdOpts.output) {
+        writeFileSync(cmdOpts.output, script, { mode: 0o755 });
+        console.log(chalk.green(`\n  Setup script written to ${cmdOpts.output}`));
+        console.log(chalk.dim(`  Run it with: bash ${cmdOpts.output}\n`));
+    }
+    else {
+        process.stdout.write(script);
+    }
+}
+async function runRentSwarm(globalOpts, cmdOpts) {
+    const config = loadConfig({
+        configPath: globalOpts.config,
+        gatewayOverride: globalOpts.gateway,
+        apiKeyOverride: globalOpts.apiKey,
+    });
+    const errors = validateConfig(config);
+    if (errors.length > 0) {
+        for (const e of errors)
+            console.error(chalk.red(`  \u2717 ${e}`));
+        process.exit(1);
+    }
+    if (!config.privateKey) {
+        console.error(chalk.red("  \u2717 Private key required. Set NOOKPLOT_AGENT_PRIVATE_KEY."));
+        process.exit(1);
+    }
+    const count = cmdOpts.count ?? 3;
+    const durationHours = cmdOpts.duration ?? 24;
+    const wallet = new ethers.Wallet(config.privateKey);
+    // Search for available GPUs
+    const spinner = ora(`Searching for ${count} available GPUs...`).start();
+    const params = new URLSearchParams({ limit: String(count + 2), status: "online" });
+    if (cmdOpts.minVram)
+        params.set("minVram", String(cmdOpts.minVram));
+    if (cmdOpts.gpuModel)
+        params.set("gpuModel", cmdOpts.gpuModel);
+    const searchRes = await gatewayRequest(config.gateway, "GET", `/v1/gpu/availability?${params}`, { apiKey: config.apiKey });
+    if (isGatewayError(searchRes)) {
+        spinner.fail("Failed to search GPUs");
+        console.error(chalk.red(`  ${searchRes.error}`));
+        process.exit(1);
+    }
+    const available = searchRes.data.gpus || [];
+    if (available.length < count) {
+        spinner.fail(`Only ${available.length} GPUs found (need ${count})`);
+        process.exit(1);
+    }
+    spinner.succeed(`Found ${available.length} available GPUs`);
+    // Rent each GPU
+    const agreements = [];
+    const deadline = Math.floor(Date.now() / 1000) + durationHours * 3600;
+    for (let i = 0; i < count; i++) {
+        const gpu = available[i];
+        const rentSpinner = ora(`  Renting GPU #${gpu.listing_id} (${gpu.gpu_model}, ${gpu.vram_gb}GB)...`).start();
+        try {
+            const prepResult = await gatewayRequest(config.gateway, "POST", "/v1/prepare/service/agree", {
+                apiKey: config.apiKey,
+                body: {
+                    listingId: gpu.listing_id,
+                    terms: "GPU rental for autoresearch swarm via Nookplot",
+                    deadline: String(deadline),
+                    tokenAmount: "0",
+                },
+            });
+            if (isGatewayError(prepResult)) {
+                rentSpinner.fail(`  GPU #${gpu.listing_id}: ${prepResult.error}`);
+                agreements.push({ listingId: gpu.listing_id, error: prepResult.error });
+                continue;
+            }
+            const { forwardRequest, domain, types } = prepResult.data;
+            const sig = await wallet.signTypedData(domain, types, forwardRequest);
+            const relayResult = await gatewayRequest(config.gateway, "POST", "/v1/relay", { apiKey: config.apiKey, body: { ...forwardRequest, signature: sig } });
+            if (isGatewayError(relayResult)) {
+                rentSpinner.fail(`  GPU #${gpu.listing_id}: relay failed`);
+                agreements.push({ listingId: gpu.listing_id, error: relayResult.error });
+                continue;
+            }
+            rentSpinner.succeed(`  GPU #${gpu.listing_id}: rented (tx: ${relayResult.data.txHash?.slice(0, 10)}...)`);
+            agreements.push({
+                listingId: gpu.listing_id,
+                txHash: relayResult.data.txHash,
+                agreementId: relayResult.data.agreementId,
+            });
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            rentSpinner.fail(`  GPU #${gpu.listing_id}: ${msg}`);
+            agreements.push({ listingId: gpu.listing_id, error: msg });
+        }
+    }
+    const succeeded = agreements.filter((a) => a.txHash);
+    const failed = agreements.filter((a) => a.error);
+    if (cmdOpts.json) {
+        console.log(JSON.stringify({ agreements, succeeded: succeeded.length, failed: failed.length }, null, 2));
+        return;
+    }
+    console.log(chalk.bold(`\n  Swarm GPU Rental Summary\n`));
+    console.log(`  Rented:   ${chalk.green(String(succeeded.length))} / ${count}`);
+    if (failed.length > 0) {
+        console.log(`  Failed:   ${chalk.red(String(failed.length))}`);
+    }
+    console.log(`  Duration: ${durationHours} hours`);
+    console.log("");
+    for (const a of succeeded) {
+        console.log(`    #${a.listingId}: agreement ${a.agreementId ?? "pending"} (tx: ${a.txHash?.slice(0, 10)}...)`);
+    }
+    console.log("");
+}
+async function runProvision(globalOpts, cmdOpts) {
+    const config = loadConfig({
+        configPath: globalOpts.config,
+        gatewayOverride: globalOpts.gateway,
+        apiKeyOverride: globalOpts.apiKey,
+    });
+    console.log(chalk.bold("\n  GPU Provision — Interactive Setup\n"));
+    // Step 1: Check Python
+    const { execSync } = await import("child_process");
+    let pythonCmd = "";
+    for (const cmd of ["python3.12", "python3.11", "python3.10", "python3"]) {
+        try {
+            execSync(`${cmd} --version`, { stdio: "pipe" });
+            pythonCmd = cmd;
+            break;
+        }
+        catch {
+            // try next
+        }
+    }
+    if (!pythonCmd) {
+        console.error(chalk.red("  \u2717 Python 3.10+ not found. Install it first."));
+        process.exit(1);
+    }
+    console.log(chalk.green(`  \u2713 Python: ${pythonCmd}`));
+    // Step 2: Check nvidia-smi
+    try {
+        const gpuName = execSync("nvidia-smi --query-gpu=name --format=csv,noheader", { stdio: "pipe" }).toString().trim();
+        const gpuVram = execSync("nvidia-smi --query-gpu=memory.total --format=csv,noheader", { stdio: "pipe" }).toString().trim();
+        console.log(chalk.green(`  \u2713 GPU: ${gpuName} (${gpuVram})`));
+    }
+    catch {
+        console.log(chalk.yellow("  \u26a0 nvidia-smi not found. GPU experiments may not work."));
+    }
+    // Step 3: Validate against listing if specified
+    if (cmdOpts.listingId) {
+        const valErrors = validateConfig(config);
+        if (valErrors.length === 0) {
+            const spinner = ora("  Validating GPU against listing...").start();
+            const listingRes = await gatewayRequest(config.gateway, "GET", `/v1/gpu/availability/${cmdOpts.listingId}`, { apiKey: config.apiKey });
+            if (isGatewayError(listingRes)) {
+                spinner.warn(`  Could not validate listing: ${listingRes.error}`);
+            }
+            else {
+                const listing = listingRes.data.availability;
+                spinner.succeed(`  Listing #${cmdOpts.listingId}: ${listing?.gpu_model || "Unknown"} (${listing?.vram_gb ?? 0}GB, ${listing?.status || "unknown"})`);
+            }
+        }
+    }
+    // Step 4: Check autoresearch repo
+    if (cmdOpts.repo) {
+        const { existsSync } = await import("fs");
+        if (!existsSync(cmdOpts.repo)) {
+            console.error(chalk.red(`  \u2717 Repo path not found: ${cmdOpts.repo}`));
+            process.exit(1);
+        }
+        if (!existsSync(`${cmdOpts.repo}/train.py`)) {
+            console.log(chalk.yellow(`  \u26a0 No train.py found in ${cmdOpts.repo} — may not be an autoresearch repo`));
+        }
+        else {
+            console.log(chalk.green(`  \u2713 Autoresearch repo: ${cmdOpts.repo}`));
+        }
+    }
+    // Step 5: Check pip install
+    try {
+        execSync(`${pythonCmd} -c "import nookplot_autoresearch"`, { stdio: "pipe" });
+        console.log(chalk.green("  \u2713 nookplot-autoresearch: installed"));
+    }
+    catch {
+        console.log(chalk.yellow("  \u26a0 nookplot-autoresearch not installed"));
+        console.log(chalk.dim("    Install: pip install nookplot-autoresearch"));
+        console.log(chalk.dim("    Or from source: pip install -e /path/to/integrations/autoresearch"));
+    }
+    // Summary
+    console.log(chalk.bold("\n  Next Steps:\n"));
+    console.log("  1. Install nookplot-autoresearch if not already installed");
+    console.log("  2. Set environment variables: NOOKPLOT_API_KEY, AGENT_PRIVATE_KEY");
+    if (cmdOpts.repo) {
+        console.log(`  3. Run: nookplot-autoresearch watch ${cmdOpts.repo}`);
+    }
+    else {
+        console.log("  3. Run: nookplot-autoresearch watch /path/to/autoresearch-repo");
+    }
+    console.log("");
+}
 //# sourceMappingURL=gpu.js.map