@prismnetwork/mcp 0.1.0

package/README.md

@@ -0,0 +1,43 @@
+# @prismnetwork/mcp
+
+An MCP server that lets Claude (or any MCP client) lease and run on real GPUs through [Prism Network](https://prismnetwork.tech). Give it a wallet; it handles auth, on-chain payment, provisioning, and SSH.
+
+## Tools
+
+- `prism_wallet`: the agent's address and USDG/ETH balances.
+- `prism_list_gpus`: GPUs available to lease, with price per second and per hour.
+- `prism_lease_and_run`: lease a GPU, run a command, return the output (one shot).
+- `prism_lease`: lease a GPU and keep it; returns a `lease_id` and SSH access.
+- `prism_run`: run a command on an existing lease.
+- `prism_end_lease`: release a lease.
+
+## Install
+
+Not yet published to npm. Until it is, install from the repo:
+
+```
+cd prism-public/mcp && npm install
+```
+
+Then point your MCP client at the local entrypoint (Claude Desktop / Code):
+
+```json
+{
+  "mcpServers": {
+    "prism": {
+      "command": "node",
+      "args": ["/path/to/prism-public/mcp/server.mjs"],
+      "env": {
+        "PRISM_AGENT_KEY": "0x<agent wallet private key>",
+        "PRISM_ESCROW": "0x71Df0eF3bc81022cB3bec0b1a05f52f12bAfcDeD"
+      }
+    }
+  }
+}
+```
+
+The wallet needs USDG (`0x5fc5360D0400a0Fd4f2af552ADD042D716F1d168`) and Robinhood-Chain ETH for gas. See the SDK's Funding section for how to fund a fresh wallet.
+
+## Timing
+
+`prism_lease` and `prism_lease_and_run` block while a GPU provisions (usually one to four minutes, occasionally longer on a slow host). Configure your MCP client to allow long tool calls.

package/package.json

@@ -0,0 +1,24 @@
+{
+  "name": "@prismnetwork/mcp",
+  "version": "0.1.0",
+  "description": "MCP server for leasing and running on Prism Network GPUs.",
+  "type": "module",
+  "bin": { "prism-mcp": "server.mjs" },
+  "files": ["server.mjs", "README.md"],
+  "engines": { "node": ">=20" },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "@prismnetwork/agent-sdk": "^0.1.0",
+    "viem": "^2"
+  },
+  "keywords": ["prism", "mcp", "gpu", "agent", "claude", "compute"],
+  "homepage": "https://prismnetwork.tech",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/prismnetwork-tech/prism.git",
+    "directory": "mcp"
+  },
+  "bugs": { "url": "https://github.com/prismnetwork-tech/prism/issues" },
+  "license": "Apache-2.0",
+  "publishConfig": { "access": "public" }
+}

package/server.mjs

@@ -0,0 +1,189 @@
+#!/usr/bin/env node
+// Prism Network MCP server: lets an MCP client (Claude, agents) lease and run on
+// real GPUs. Configure with a wallet: PRISM_AGENT_KEY, PRISM_ESCROW.
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { DEFAULT_IMAGE, PrismAgent } from "@prismnetwork/agent-sdk";
+
+const IMAGE = process.env.PRISM_DEFAULT_IMAGE ?? DEFAULT_IMAGE;
+
+function requireEnv(name) {
+  const value = process.env[name];
+  if (!value) throw new Error(`${name} is required`);
+  return value;
+}
+
+let agent;
+try {
+  agent = new PrismAgent({
+    privateKey: requireEnv("PRISM_AGENT_KEY"),
+    escrow: requireEnv("PRISM_ESCROW"),
+    apiBase: process.env.PRISM_API_BASE ?? "https://prismnetwork.tech",
+    rpcUrl: process.env.PRISM_RPC_URL,
+  });
+} catch (err) {
+  console.error(`prism mcp config error: ${err.message}. Set PRISM_AGENT_KEY and PRISM_ESCROW in the server env.`);
+  process.exit(1);
+}
+
+const leases = new Map();
+const usdg = (micros) => `${(Number(micros) / 1e6).toFixed(6)} USDG`;
+
+function sweepExpiredLeases() {
+  const now = Date.now();
+  for (const [id, lease] of leases) {
+    const expiry = Date.parse(lease.access?.expires_at ?? "");
+    if (Number.isFinite(expiry) && expiry < now) {
+      agent.endLease(lease);
+      leases.delete(id);
+    }
+  }
+}
+
+function leaseId(value) {
+  const id = Number(value);
+  if (!Number.isInteger(id) || id <= 0) throw new Error("lease_id must be a positive integer");
+  return id;
+}
+
+const TOOLS = [
+  {
+    name: "prism_wallet",
+    description: "Show the agent's wallet address and on-chain balances (USDG and ETH for gas) on Robinhood Chain. Check this before leasing to confirm the wallet can pay.",
+    inputSchema: { type: "object", properties: {} },
+  },
+  {
+    name: "prism_list_gpus",
+    description: "List GPUs currently available to lease on Prism Network, with model, VRAM, and price per second in USDG.",
+    inputSchema: { type: "object", properties: {} },
+  },
+  {
+    name: "prism_lease_and_run",
+    description: "Lease a GPU, run one shell command on it, and return the output. The lease stays alive (use prism_run for more commands, prism_end_lease to release). Prefer this for a single command; use prism_lease when you'll run several.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        command: { type: "string", description: "Shell command to run on the GPU (e.g. 'nvidia-smi')." },
+        duration_seconds: { type: "integer", description: "Lease length in seconds (default 900, max 21600)." },
+        min_vram_mib: { type: "integer", description: "Minimum GPU memory in MiB (default 16000)." },
+      },
+      required: ["command"],
+    },
+  },
+  {
+    name: "prism_lease",
+    description: "Lease a GPU and keep it running. Returns a lease_id and SSH access. Use prism_run to execute commands and prism_end_lease when done.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        duration_seconds: { type: "integer", description: "Lease length in seconds (default 900, max 21600)." },
+        min_vram_mib: { type: "integer", description: "Minimum GPU memory in MiB (default 16000)." },
+      },
+    },
+  },
+  {
+    name: "prism_run",
+    description: "Run a shell command on a GPU you already leased with prism_lease.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        lease_id: { type: "integer", description: "The lease_id returned by prism_lease." },
+        command: { type: "string", description: "Shell command to run." },
+        timeout_seconds: { type: "integer", description: "Max seconds to wait (default 120)." },
+      },
+      required: ["lease_id", "command"],
+    },
+  },
+  {
+    name: "prism_end_lease",
+    description: "Release a lease's local access. The on-chain lease settles at the end of its paid duration.",
+    inputSchema: {
+      type: "object",
+      properties: { lease_id: { type: "integer" } },
+      required: ["lease_id"],
+    },
+  },
+];
+
+async function handle(name, args) {
+  if (name === "prism_wallet") {
+    const b = await agent.balances();
+    return { address: b.address, usdg: usdg(b.usdg), eth_wei: b.eth };
+  }
+  if (name === "prism_list_gpus") {
+    await ensureAuth();
+    const offers = await agent.offers();
+    return {
+      available: offers.length,
+      gpus: offers.map((o) => ({
+        model: o.gpu.model,
+        vram_mib: o.gpu.vram_mib,
+        price_per_second: usdg(o.rate_per_second),
+        price_per_hour: usdg(o.rate_per_second * 3600),
+      })),
+    };
+  }
+  if (name === "prism_lease_and_run" || name === "prism_lease") {
+    if (name === "prism_lease_and_run" && !args.command) throw new Error("command is required");
+    await ensureAuth();
+    sweepExpiredLeases();
+    const lease = await agent.lease({
+      image: IMAGE,
+      durationSeconds: args.duration_seconds ?? 900,
+      minVramMib: args.min_vram_mib ?? 16000,
+    });
+    leases.set(lease.leaseId, lease);
+    const summary = {
+      lease_id: lease.leaseId,
+      ssh: { host: lease.access.ssh_host, port: lease.access.ssh_port, user: lease.access.ssh_user },
+      expires_at: lease.access.expires_at,
+    };
+    if (name === "prism_lease") return summary;
+    const out = await agent.run(lease, args.command);
+    return { ...summary, command: args.command, exit_code: out.code, stdout: out.stdout, stderr: out.stderr };
+  }
+  if (name === "prism_run") {
+    if (!args.command) throw new Error("command is required");
+    const id = leaseId(args.lease_id);
+    const lease = leases.get(id);
+    if (!lease) throw new Error(`no active lease ${id} in this session`);
+    const out = await agent.run(lease, args.command, {
+      timeoutMs: (args.timeout_seconds ?? 120) * 1000,
+    });
+    return { lease_id: id, exit_code: out.code, stdout: out.stdout, stderr: out.stderr };
+  }
+  if (name === "prism_end_lease") {
+    const id = leaseId(args.lease_id);
+    const lease = leases.get(id);
+    if (lease) {
+      agent.endLease(lease);
+      leases.delete(id);
+    }
+    return { lease_id: id, released: Boolean(lease) };
+  }
+  throw new Error(`unknown tool ${name}`);
+}
+
+let authPromise = null;
+function ensureAuth() {
+  authPromise ??= agent.authenticate().catch((err) => {
+    authPromise = null;
+    throw err;
+  });
+  return authPromise;
+}
+
+const server = new Server({ name: "prism", version: "0.1.0" }, { capabilities: { tools: {} } });
+server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: TOOLS }));
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  try {
+    const result = await handle(request.params.name, request.params.arguments ?? {});
+    return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
+  } catch (err) {
+    return { isError: true, content: [{ type: "text", text: `error: ${err.message ?? err}` }] };
+  }
+});
+
+await server.connect(new StdioServerTransport());
+console.error("prism mcp server ready");