@runpod/mcp-server 1.3.0 → 1.4.0

package/dist/stdio.mjs

@@ -6,7 +6,7 @@ import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js"
 // src/server.ts
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 var SERVER_NAME = "Runpod API Server";
-var SERVER_VERSION = true ? "1.3.0" : "dev";
+var SERVER_VERSION = true ? "1.4.0" : "dev";
 function createServer(version = SERVER_VERSION) {
   return new McpServer({
     name: SERVER_NAME,
@@ -19,19 +19,78 @@ function createServer(version = SERVER_VERSION) {
 }
 
 // src/tools.ts
-import { z } from "zod";
+import { z as z2 } from "zod";
 import fetch from "node-fetch";
 import { randomUUID } from "crypto";
+
+// src/pagination.ts
+import { z } from "zod";
+var DEFAULT_LIST_LIMIT = 20;
+var MAX_LIST_LIMIT = 100;
+var listPaginationParams = {
+  limit: z.number().int().positive().max(MAX_LIST_LIMIT).optional().describe(
+    `Maximum number of items to return (default ${DEFAULT_LIST_LIMIT}, max ${MAX_LIST_LIMIT}). Use the returned nextCursor to fetch the next page.`
+  ),
+  cursor: z.string().optional().describe(
+    "Opaque pagination cursor from a previous response (nextCursor). Omit to start from the beginning."
+  )
+};
+function decodeCursorOffset(cursor) {
+  if (!cursor) return 0;
+  try {
+    const decoded = Buffer.from(cursor, "base64").toString("utf8");
+    const offset = Number.parseInt(decoded, 10);
+    return Number.isFinite(offset) && offset >= 0 ? offset : 0;
+  } catch {
+    return 0;
+  }
+}
+function encodeCursorOffset(offset) {
+  return Buffer.from(String(offset), "utf8").toString("base64");
+}
+function capListResult(result, options) {
+  const limit = Math.min(options.limit ?? DEFAULT_LIST_LIMIT, MAX_LIST_LIMIT);
+  const offset = decodeCursorOffset(options.cursor);
+  if (!Array.isArray(result)) {
+    return {
+      content: [
+        { type: "text", text: JSON.stringify(result, null, 2) }
+      ]
+    };
+  }
+  const total = result.length;
+  const page = result.slice(offset, offset + limit);
+  const nextOffset = offset + page.length;
+  const hasMore = nextOffset < total;
+  const envelope = {
+    items: page,
+    pagination: {
+      total,
+      returned: page.length,
+      offset,
+      truncated: hasMore,
+      nextCursor: hasMore ? encodeCursorOffset(nextOffset) : null,
+      note: hasMore ? `Showing ${page.length} of ${total}. Pass cursor=nextCursor to fetch more, or narrow the query with filters.` : void 0
+    }
+  };
+  return {
+    content: [
+      { type: "text", text: JSON.stringify(envelope, null, 2) }
+    ]
+  };
+}
+
+// src/tools.ts
 var API_BASE_URL = process.env.RUNPOD_REST_API_URL ?? "https://rest.runpod.io/v1";
 var SERVERLESS_API_BASE_URL = process.env.RUNPOD_SERVERLESS_API_URL ?? "https://api.runpod.ai/v2";
 var GRAPHQL_URL = process.env.RUNPOD_PUBLIC_GRAPHQL_URL ?? "https://api.runpod.io/graphql";
-var MCP_SERVER_VERSION = true ? "1.3.0" : "dev";
+var MCP_SERVER_VERSION = true ? "1.4.0" : "dev";
 var SESSION_ID = randomUUID();
 function sanitizeUaToken(value) {
   return value.replace(/[()<>@,;:\\"/[\]?={}\s]/g, "_").slice(0, 64);
 }
-function trackingHeaders(server2, ctx) {
-  const info = server2.server.getClientVersion();
+function trackingHeaders(server, ctx) {
+  const info = server.server.getClientVersion();
   const fallbackName = ctx.clientUserAgent || "unknown";
   const name = sanitizeUaToken(info?.name || fallbackName);
   const version = sanitizeUaToken(info?.version || "unknown");
@@ -128,20 +187,20 @@ function createServerlessRequest(apiKey, tracking) {
     }
   };
 }
-function registerTools(server2, ctx) {
-  const tracking = () => trackingHeaders(server2, ctx);
+function registerTools(server, ctx) {
+  const tracking = () => trackingHeaders(server, ctx);
   const runpodRequest = createRunpodRequest(ctx.apiKey, tracking);
   const serverlessRequest = createServerlessRequest(ctx.apiKey, tracking);
-  server2.tool(
+  server.tool(
     "list-gpu-types",
     {
-      minMemoryGb: z.number().optional().describe("Filter to GPUs with at least this much VRAM in GB"),
-      secureCloudOnly: z.boolean().optional().describe("Filter to only GPUs available in secure cloud"),
-      communityCloudOnly: z.boolean().optional().describe("Filter to only GPUs available in community cloud"),
-      searchTerm: z.string().optional().describe(
+      minMemoryGb: z2.number().optional().describe("Filter to GPUs with at least this much VRAM in GB"),
+      secureCloudOnly: z2.boolean().optional().describe("Filter to only GPUs available in secure cloud"),
+      communityCloudOnly: z2.boolean().optional().describe("Filter to only GPUs available in community cloud"),
+      searchTerm: z2.string().optional().describe(
         "Search term to filter GPUs by name (e.g., 'A100', 'RTX 4090')"
       ),
-      includeUnavailable: z.boolean().optional().describe(
+      includeUnavailable: z2.boolean().optional().describe(
         "Include GPUs that are currently out of stock. Default is false"
       )
     },
@@ -218,10 +277,10 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "list-data-centers",
     {
-      region: z.string().optional().describe(
+      region: z2.string().optional().describe(
         "Filter by region/location (e.g., 'United States', 'Europe', 'Canada')"
       )
     },
@@ -255,15 +314,16 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "list-pods",
     {
-      computeType: z.enum(["GPU", "CPU"]).optional().describe("Filter to only GPU or only CPU Pods"),
-      gpuTypeId: z.array(z.string()).optional().describe("Filter to Pods with any of the listed GPU types"),
-      dataCenterId: z.array(z.string()).optional().describe("Filter to Pods in any of the provided data centers"),
-      name: z.string().optional().describe("Filter to Pods with the provided name"),
-      includeMachine: z.boolean().optional().describe("Include information about the machine"),
-      includeNetworkVolume: z.boolean().optional().describe("Include information about attached network volumes")
+      ...listPaginationParams,
+      computeType: z2.enum(["GPU", "CPU"]).optional().describe("Filter to only GPU or only CPU Pods"),
+      gpuTypeId: z2.array(z2.string()).optional().describe("Filter to Pods with any of the listed GPU types"),
+      dataCenterId: z2.array(z2.string()).optional().describe("Filter to Pods in any of the provided data centers"),
+      name: z2.string().optional().describe("Filter to Pods with the provided name"),
+      includeMachine: z2.boolean().optional().describe("Include information about the machine"),
+      includeNetworkVolume: z2.boolean().optional().describe("Include information about attached network volumes")
     },
     async (params) => {
       const queryParams = new URLSearchParams();
@@ -287,22 +347,18 @@ function registerTools(server2, ctx) {
         );
       const queryString = queryParams.toString() ? `?${queryParams.toString()}` : "";
       const result = await runpodRequest(`/pods${queryString}`);
-      return {
-        content: [
-          {
-            type: "text",
-            text: JSON.stringify(result, null, 2)
-          }
-        ]
-      };
+      return capListResult(result, {
+        limit: params.limit,
+        cursor: params.cursor
+      });
     }
   );
-  server2.tool(
+  server.tool(
     "get-pod",
     {
-      podId: z.string().describe("ID of the pod to retrieve"),
-      includeMachine: z.boolean().optional().describe("Include information about the machine"),
-      includeNetworkVolume: z.boolean().optional().describe("Include information about attached network volumes")
+      podId: z2.string().describe("ID of the pod to retrieve"),
+      includeMachine: z2.boolean().optional().describe("Include information about the machine"),
+      includeNetworkVolume: z2.boolean().optional().describe("Include information about attached network volumes")
     },
     async (params) => {
       const queryParams = new URLSearchParams();
@@ -325,21 +381,21 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "create-pod",
     'Create a new GPU/CPU pod on Runpod. If the user does not specify an image, recommend the "Runpod Pytorch 2.8.0" image (runpod/pytorch:1.0.2-cu1281-torch280-ubuntu2404) as the default \u2014 it has the most up-to-date CUDA and PyTorch versions.',
     {
-      name: z.string().optional().describe("Name for the pod"),
-      imageName: z.string().describe("Docker image to use"),
-      cloudType: z.enum(["SECURE", "COMMUNITY"]).optional().describe("SECURE or COMMUNITY cloud"),
-      gpuTypeIds: z.array(z.string()).optional().describe("List of acceptable GPU types"),
-      gpuCount: z.number().optional().describe("Number of GPUs"),
-      containerDiskInGb: z.number().optional().describe("Container disk size in GB"),
-      volumeInGb: z.number().optional().describe("Volume size in GB"),
-      volumeMountPath: z.string().optional().describe("Path to mount the volume"),
-      ports: z.array(z.string()).optional().describe("Ports to expose (e.g., '8888/http', '22/tcp')"),
-      env: z.record(z.string()).optional().describe("Environment variables"),
-      dataCenterIds: z.array(z.string()).optional().describe("List of data centers")
+      name: z2.string().optional().describe("Name for the pod"),
+      imageName: z2.string().describe("Docker image to use"),
+      cloudType: z2.enum(["SECURE", "COMMUNITY"]).optional().describe("SECURE or COMMUNITY cloud"),
+      gpuTypeIds: z2.array(z2.string()).optional().describe("List of acceptable GPU types"),
+      gpuCount: z2.number().optional().describe("Number of GPUs"),
+      containerDiskInGb: z2.number().optional().describe("Container disk size in GB"),
+      volumeInGb: z2.number().optional().describe("Volume size in GB"),
+      volumeMountPath: z2.string().optional().describe("Path to mount the volume"),
+      ports: z2.array(z2.string()).optional().describe("Ports to expose (e.g., '8888/http', '22/tcp')"),
+      env: z2.record(z2.string()).optional().describe("Environment variables"),
+      dataCenterIds: z2.array(z2.string()).optional().describe("List of data centers")
     },
     async (params) => {
       const result = await runpodRequest("/pods", "POST", params);
@@ -353,17 +409,17 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "update-pod",
     {
-      podId: z.string().describe("ID of the pod to update"),
-      name: z.string().optional().describe("New name for the pod"),
-      imageName: z.string().optional().describe("New Docker image"),
-      containerDiskInGb: z.number().optional().describe("New container disk size in GB"),
-      volumeInGb: z.number().optional().describe("New volume size in GB"),
-      volumeMountPath: z.string().optional().describe("New path to mount the volume"),
-      ports: z.array(z.string()).optional().describe("New ports to expose"),
-      env: z.record(z.string()).optional().describe("New environment variables")
+      podId: z2.string().describe("ID of the pod to update"),
+      name: z2.string().optional().describe("New name for the pod"),
+      imageName: z2.string().optional().describe("New Docker image"),
+      containerDiskInGb: z2.number().optional().describe("New container disk size in GB"),
+      volumeInGb: z2.number().optional().describe("New volume size in GB"),
+      volumeMountPath: z2.string().optional().describe("New path to mount the volume"),
+      ports: z2.array(z2.string()).optional().describe("New ports to expose"),
+      env: z2.record(z2.string()).optional().describe("New environment variables")
     },
     async (params) => {
       const { podId, ...updateParams } = params;
@@ -382,10 +438,10 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "start-pod",
     {
-      podId: z.string().describe("ID of the pod to start")
+      podId: z2.string().describe("ID of the pod to start")
     },
     async (params) => {
       const result = await runpodRequest(`/pods/${params.podId}/start`, "POST");
@@ -399,10 +455,10 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "stop-pod",
     {
-      podId: z.string().describe("ID of the pod to stop")
+      podId: z2.string().describe("ID of the pod to stop")
     },
     async (params) => {
       const result = await runpodRequest(`/pods/${params.podId}/stop`, "POST");
@@ -416,10 +472,10 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "delete-pod",
     {
-      podId: z.string().describe("ID of the pod to delete")
+      podId: z2.string().describe("ID of the pod to delete")
     },
     async (params) => {
       const result = await runpodRequest(`/pods/${params.podId}`, "DELETE");
@@ -433,11 +489,12 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "list-endpoints",
     {
-      includeTemplate: z.boolean().optional().describe("Include template information"),
-      includeWorkers: z.boolean().optional().describe("Include information about workers")
+      ...listPaginationParams,
+      includeTemplate: z2.boolean().optional().describe("Include template information"),
+      includeWorkers: z2.boolean().optional().describe("Include information about workers")
     },
     async (params) => {
       const queryParams = new URLSearchParams();
@@ -450,22 +507,18 @@ function registerTools(server2, ctx) {
         queryParams.append("includeWorkers", params.includeWorkers.toString());
       const queryString = queryParams.toString() ? `?${queryParams.toString()}` : "";
       const result = await runpodRequest(`/endpoints${queryString}`);
-      return {
-        content: [
-          {
-            type: "text",
-            text: JSON.stringify(result, null, 2)
-          }
-        ]
-      };
+      return capListResult(result, {
+        limit: params.limit,
+        cursor: params.cursor
+      });
     }
   );
-  server2.tool(
+  server.tool(
     "get-endpoint",
     {
-      endpointId: z.string().describe("ID of the endpoint to retrieve"),
-      includeTemplate: z.boolean().optional().describe("Include template information"),
-      includeWorkers: z.boolean().optional().describe("Include information about workers")
+      endpointId: z2.string().describe("ID of the endpoint to retrieve"),
+      includeTemplate: z2.boolean().optional().describe("Include template information"),
+      includeWorkers: z2.boolean().optional().describe("Include information about workers")
     },
     async (params) => {
       const queryParams = new URLSearchParams();
@@ -490,17 +543,17 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "create-endpoint",
     {
-      name: z.string().optional().describe("Name for the endpoint"),
-      templateId: z.string().describe("Template ID to use"),
-      computeType: z.enum(["GPU", "CPU"]).optional().describe("GPU or CPU endpoint"),
-      gpuTypeIds: z.array(z.string()).optional().describe("List of acceptable GPU types"),
-      gpuCount: z.number().optional().describe("Number of GPUs per worker"),
-      workersMin: z.number().optional().describe("Minimum number of workers"),
-      workersMax: z.number().optional().describe("Maximum number of workers"),
-      dataCenterIds: z.array(z.string()).optional().describe("List of data centers")
+      name: z2.string().optional().describe("Name for the endpoint"),
+      templateId: z2.string().describe("Template ID to use"),
+      computeType: z2.enum(["GPU", "CPU"]).optional().describe("GPU or CPU endpoint"),
+      gpuTypeIds: z2.array(z2.string()).optional().describe("List of acceptable GPU types"),
+      gpuCount: z2.number().optional().describe("Number of GPUs per worker"),
+      workersMin: z2.number().optional().describe("Minimum number of workers"),
+      workersMax: z2.number().optional().describe("Maximum number of workers"),
+      dataCenterIds: z2.array(z2.string()).optional().describe("List of data centers")
     },
     async (params) => {
       const result = await runpodRequest("/endpoints", "POST", params);
@@ -514,16 +567,16 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "update-endpoint",
     {
-      endpointId: z.string().describe("ID of the endpoint to update"),
-      name: z.string().optional().describe("New name for the endpoint"),
-      workersMin: z.number().optional().describe("New minimum number of workers"),
-      workersMax: z.number().optional().describe("New maximum number of workers"),
-      idleTimeout: z.number().optional().describe("New idle timeout in seconds"),
-      scalerType: z.enum(["QUEUE_DELAY", "REQUEST_COUNT"]).optional().describe("Scaler type"),
-      scalerValue: z.number().optional().describe("Scaler value")
+      endpointId: z2.string().describe("ID of the endpoint to update"),
+      name: z2.string().optional().describe("New name for the endpoint"),
+      workersMin: z2.number().optional().describe("New minimum number of workers"),
+      workersMax: z2.number().optional().describe("New maximum number of workers"),
+      idleTimeout: z2.number().optional().describe("New idle timeout in seconds"),
+      scalerType: z2.enum(["QUEUE_DELAY", "REQUEST_COUNT"]).optional().describe("Scaler type"),
+      scalerValue: z2.number().optional().describe("Scaler value")
     },
     async (params) => {
       const { endpointId, ...updateParams } = params;
@@ -542,10 +595,10 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "delete-endpoint",
     {
-      endpointId: z.string().describe("ID of the endpoint to delete")
+      endpointId: z2.string().describe("ID of the endpoint to delete")
     },
     async (params) => {
       const result = await runpodRequest(
@@ -562,26 +615,26 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  const endpointIdSchema = z.string().regex(/^[a-zA-Z0-9_-]+$/, "Invalid endpoint ID format");
-  const jobIdSchema = z.string().regex(/^[a-zA-Z0-9_-]+$/, "Invalid job ID format");
-  const inputSchema = z.record(z.unknown()).describe(
+  const endpointIdSchema = z2.string().regex(/^[a-zA-Z0-9_-]+$/, "Invalid endpoint ID format");
+  const jobIdSchema = z2.string().regex(/^[a-zA-Z0-9_-]+$/, "Invalid job ID format");
+  const inputSchema = z2.record(z2.unknown()).describe(
     "Input payload for the worker handler. The expected fields depend on the deployed model or worker."
   );
-  const webhookSchema = z.string().url().optional().describe(
+  const webhookSchema = z2.string().url().optional().describe(
     "Webhook URL to receive job completion notifications instead of polling"
   );
-  const policySchema = z.object({
-    executionTimeout: z.number().optional().describe("Maximum execution time in milliseconds"),
-    lowPriority: z.boolean().optional().describe("Submit as a low-priority job"),
-    ttl: z.number().optional().describe("Time-to-live for the job result in milliseconds")
+  const policySchema = z2.object({
+    executionTimeout: z2.number().optional().describe("Maximum execution time in milliseconds"),
+    lowPriority: z2.boolean().optional().describe("Submit as a low-priority job"),
+    ttl: z2.number().optional().describe("Time-to-live for the job result in milliseconds")
   }).optional().describe("Execution policy options");
-  const s3ConfigSchema = z.object({
-    accessId: z.string().describe("S3 access key ID"),
-    accessSecret: z.string().describe("S3 secret access key"),
-    bucketName: z.string().describe("S3 bucket name"),
-    endpointUrl: z.string().describe("S3 endpoint URL")
+  const s3ConfigSchema = z2.object({
+    accessId: z2.string().describe("S3 access key ID"),
+    accessSecret: z2.string().describe("S3 secret access key"),
+    bucketName: z2.string().describe("S3 bucket name"),
+    endpointUrl: z2.string().describe("S3 endpoint URL")
   }).optional().describe("S3-compatible storage config for large outputs");
-  server2.tool(
+  server.tool(
     "run-endpoint",
     "Submit an asynchronous job to a Serverless endpoint. Returns a job ID immediately \u2014 use get-job-status to poll for results. Async results are available for 30 minutes after completion.",
     {
@@ -611,7 +664,7 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "runsync-endpoint",
     "Submit a synchronous job to a Serverless endpoint and wait for the result. Best for tasks completing within 90 seconds. If processing exceeds 90 seconds, the response returns a job ID to poll with get-job-status. Max payload: 20 MB. Results expire after 1 minute. Use the wait parameter to extend the server-side wait up to 5 minutes (300000 ms).",
     {
@@ -619,7 +672,7 @@ function registerTools(server2, ctx) {
         "ID of the Serverless endpoint to run synchronously"
       ),
       input: inputSchema,
-      wait: z.number().min(1e3).max(3e5).optional().describe(
+      wait: z2.number().min(1e3).max(3e5).optional().describe(
         "How long in milliseconds the server should wait for a result before returning a job ID to poll (1000\u2013300000). Defaults to 90000 (90 seconds)."
       ),
       webhook: webhookSchema,
@@ -645,7 +698,7 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "get-job-status",
     "Check the status of an asynchronous Serverless job. Returns the current status and output when complete. Job statuses: IN_QUEUE, IN_PROGRESS, COMPLETED, FAILED, CANCELLED, TIMED_OUT.",
     {
@@ -669,7 +722,7 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "stream-job",
     "Retrieve all streaming output from a Serverless job by polling until the job reaches a terminal state. The worker must support streaming output. Polls /stream/{jobId} repeatedly and collects every chunk until status is COMPLETED, FAILED, CANCELLED, or TIMED_OUT.",
     {
@@ -737,7 +790,7 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "cancel-job",
     "Cancel a Serverless job that is queued or in progress.",
     {
@@ -762,7 +815,7 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "retry-job",
     "Retry a failed or timed-out Serverless job. Only works for jobs with FAILED or TIMED_OUT status. The previous output is removed and the job is requeued.",
     {
@@ -787,7 +840,7 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "endpoint-health",
     "Get the health and operational status of a Serverless endpoint, including worker counts and job statistics.",
     {
@@ -807,7 +860,7 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "purge-endpoint-queue",
     "Remove all pending jobs from a Serverless endpoint queue. Only affects queued jobs \u2014 in-progress jobs continue running. Use this for error recovery or clearing outdated requests.",
     {
@@ -831,13 +884,14 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "list-templates",
     `List available templates. By default returns only the user's own templates. Use includeRunpodTemplates to also include official Runpod templates. The recommended default template for new pods is "Runpod Pytorch 2.8.0" (ID: runpod-torch-v280) \u2014 it has the latest CUDA and PyTorch versions.`,
     {
-      includeRunpodTemplates: z.boolean().optional().describe("Include official Runpod templates in the response"),
-      includePublicTemplates: z.boolean().optional().describe("Include community-made public templates in the response"),
-      includeEndpointBoundTemplates: z.boolean().optional().describe(
+      ...listPaginationParams,
+      includeRunpodTemplates: z2.boolean().optional().describe("Include official Runpod templates in the response"),
+      includePublicTemplates: z2.boolean().optional().describe("Include community-made public templates in the response"),
+      includeEndpointBoundTemplates: z2.boolean().optional().describe(
         "Include templates bound to Serverless endpoints in the response"
       )
     },
@@ -853,20 +907,16 @@ function registerTools(server2, ctx) {
       const result = await runpodRequest(
         `/templates${query ? `?${query}` : ""}`
       );
-      return {
-        content: [
-          {
-            type: "text",
-            text: JSON.stringify(result, null, 2)
-          }
-        ]
-      };
+      return capListResult(result, {
+        limit: params.limit,
+        cursor: params.cursor
+      });
     }
   );
-  server2.tool(
+  server.tool(
     "get-template",
     {
-      templateId: z.string().describe("ID of the template to retrieve")
+      templateId: z2.string().describe("ID of the template to retrieve")
     },
     async (params) => {
       const result = await runpodRequest(`/templates/${params.templateId}`);
@@ -880,20 +930,23 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "create-template",
     {
-      name: z.string().describe("Name for the template"),
-      imageName: z.string().describe("Docker image to use"),
-      isServerless: z.boolean().optional().describe("Is this a serverless template"),
-      ports: z.array(z.string()).optional().describe("Ports to expose"),
-      dockerEntrypoint: z.array(z.string()).optional().describe("Docker entrypoint commands"),
-      dockerStartCmd: z.array(z.string()).optional().describe("Docker start commands"),
-      env: z.record(z.string()).optional().describe("Environment variables"),
-      containerDiskInGb: z.number().optional().describe("Container disk size in GB"),
-      volumeInGb: z.number().optional().describe("Volume size in GB"),
-      volumeMountPath: z.string().optional().describe("Path to mount the volume"),
-      readme: z.string().optional().describe("README content in markdown format")
+      name: z2.string().describe("Name for the template"),
+      imageName: z2.string().describe("Docker image to use"),
+      isServerless: z2.boolean().optional().describe("Is this a serverless template"),
+      ports: z2.array(z2.string()).optional().describe("Ports to expose"),
+      dockerEntrypoint: z2.array(z2.string()).optional().describe("Docker entrypoint commands"),
+      dockerStartCmd: z2.array(z2.string()).optional().describe("Docker start commands"),
+      env: z2.record(z2.string()).optional().describe("Environment variables"),
+      containerDiskInGb: z2.number().optional().describe("Container disk size in GB"),
+      volumeInGb: z2.number().optional().describe("Volume size in GB"),
+      volumeMountPath: z2.string().optional().describe("Path to mount the volume"),
+      readme: z2.string().optional().describe("README content in markdown format"),
+      containerRegistryAuthId: z2.string().optional().describe(
+        "ID of a container registry credential (from create-container-registry-auth / list-container-registry-auths) used to pull a private image. Required when imageName points at a private registry; without it serverless workers fail the image pull."
+      )
     },
     async (params) => {
       const result = await runpodRequest("/templates", "POST", params);
@@ -907,15 +960,18 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "update-template",
     {
-      templateId: z.string().describe("ID of the template to update"),
-      name: z.string().optional().describe("New name for the template"),
-      imageName: z.string().optional().describe("New Docker image"),
-      ports: z.array(z.string()).optional().describe("New ports to expose"),
-      env: z.record(z.string()).optional().describe("New environment variables"),
-      readme: z.string().optional().describe("New README content in markdown format")
+      templateId: z2.string().describe("ID of the template to update"),
+      name: z2.string().optional().describe("New name for the template"),
+      imageName: z2.string().optional().describe("New Docker image"),
+      ports: z2.array(z2.string()).optional().describe("New ports to expose"),
+      env: z2.record(z2.string()).optional().describe("New environment variables"),
+      readme: z2.string().optional().describe("New README content in markdown format"),
+      containerRegistryAuthId: z2.string().optional().describe(
+        "ID of a container registry credential (from create-container-registry-auth / list-container-registry-auths) used to pull a private image. Set this to attach a credential to a template referencing a private registry."
+      )
     },
     async (params) => {
       const { templateId, ...updateParams } = params;
@@ -934,10 +990,10 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "delete-template",
     {
-      templateId: z.string().describe("ID of the template to delete")
+      templateId: z2.string().describe("ID of the template to delete")
     },
     async (params) => {
       const result = await runpodRequest(
@@ -954,21 +1010,17 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool("list-network-volumes", {}, async () => {
+  server.tool("list-network-volumes", listPaginationParams, async (params) => {
     const result = await runpodRequest("/networkvolumes");
-    return {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify(result, null, 2)
-        }
-      ]
-    };
+    return capListResult(result, {
+      limit: params.limit,
+      cursor: params.cursor
+    });
   });
-  server2.tool(
+  server.tool(
     "get-network-volume",
     {
-      networkVolumeId: z.string().describe("ID of the network volume to retrieve")
+      networkVolumeId: z2.string().describe("ID of the network volume to retrieve")
     },
     async (params) => {
       const result = await runpodRequest(
@@ -984,12 +1036,12 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "create-network-volume",
     {
-      name: z.string().describe("Name for the network volume"),
-      size: z.number().describe("Size in GB (1-4000)"),
-      dataCenterId: z.string().describe("Data center ID")
+      name: z2.string().describe("Name for the network volume"),
+      size: z2.number().describe("Size in GB (1-4000)"),
+      dataCenterId: z2.string().describe("Data center ID")
     },
     async (params) => {
       const result = await runpodRequest("/networkvolumes", "POST", params);
@@ -1003,12 +1055,12 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "update-network-volume",
     {
-      networkVolumeId: z.string().describe("ID of the network volume to update"),
-      name: z.string().optional().describe("New name for the network volume"),
-      size: z.number().optional().describe("New size in GB (must be larger than current)")
+      networkVolumeId: z2.string().describe("ID of the network volume to update"),
+      name: z2.string().optional().describe("New name for the network volume"),
+      size: z2.number().optional().describe("New size in GB (must be larger than current)")
     },
     async (params) => {
       const { networkVolumeId, ...updateParams } = params;
@@ -1027,10 +1079,10 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "delete-network-volume",
     {
-      networkVolumeId: z.string().describe("ID of the network volume to delete")
+      networkVolumeId: z2.string().describe("ID of the network volume to delete")
     },
     async (params) => {
       const result = await runpodRequest(
@@ -1047,21 +1099,21 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool("list-container-registry-auths", {}, async () => {
-    const result = await runpodRequest("/containerregistryauth");
-    return {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify(result, null, 2)
-        }
-      ]
-    };
-  });
-  server2.tool(
+  server.tool(
+    "list-container-registry-auths",
+    listPaginationParams,
+    async (params) => {
+      const result = await runpodRequest("/containerregistryauth");
+      return capListResult(result, {
+        limit: params.limit,
+        cursor: params.cursor
+      });
+    }
+  );
+  server.tool(
     "get-container-registry-auth",
     {
-      containerRegistryAuthId: z.string().describe("ID of the container registry auth to retrieve")
+      containerRegistryAuthId: z2.string().describe("ID of the container registry auth to retrieve")
     },
     async (params) => {
       const result = await runpodRequest(
@@ -1077,12 +1129,12 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "create-container-registry-auth",
     {
-      name: z.string().describe("Name for the container registry auth"),
-      username: z.string().describe("Registry username"),
-      password: z.string().describe("Registry password")
+      name: z2.string().describe("Name for the container registry auth"),
+      username: z2.string().describe("Registry username"),
+      password: z2.string().describe("Registry password")
     },
     async (params) => {
       const result = await runpodRequest(
@@ -1100,10 +1152,10 @@ function registerTools(server2, ctx) {
       };
     }
   );
-  server2.tool(
+  server.tool(
     "delete-container-registry-auth",
     {
-      containerRegistryAuthId: z.string().describe("ID of the container registry auth to delete")
+      containerRegistryAuthId: z2.string().describe("ID of the container registry auth to delete")
     },
     async (params) => {
       const result = await runpodRequest(
@@ -1123,13 +1175,23 @@ function registerTools(server2, ctx) {
 }
 
 // src/stdio.ts
-var API_KEY = process.env.RUNPOD_API_KEY;
-if (!API_KEY) {
-  console.error("RUNPOD_API_KEY environment variable is required");
-  process.exit(1);
+var WIZARD_SUBCOMMANDS = ["add", "remove"];
+var subcommand = process.argv[2];
+var isWizardMode = WIZARD_SUBCOMMANDS.includes(subcommand);
+if (isWizardMode) {
+  import("./wizard-37ZJSVB5.mjs").then(({ runWizard }) => runWizard(subcommand)).catch((error) => {
+    console.error(error instanceof Error ? error.message : String(error));
+    process.exit(1);
+  });
+} else {
+  const API_KEY = process.env.RUNPOD_API_KEY;
+  if (!API_KEY) {
+    console.error("RUNPOD_API_KEY environment variable is required");
+    process.exit(1);
+  }
+  const server = createServer();
+  registerTools(server, { apiKey: API_KEY, transport: "stdio" });
+  const transport = new StdioServerTransport();
+  server.connect(transport);
 }
-var server = createServer();
-registerTools(server, { apiKey: API_KEY, transport: "stdio" });
-var transport = new StdioServerTransport();
-server.connect(transport);
 //# sourceMappingURL=stdio.mjs.map