@runpod/mcp-server 1.2.0 → 1.4.0

package/dist/http.cjs

@@ -0,0 +1,1276 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/http.ts
+var http_exports = {};
+__export(http_exports, {
+  handleMcpRequest: () => handleMcpRequest,
+  registerTools: () => registerTools
+});
+module.exports = __toCommonJS(http_exports);
+var import_streamableHttp = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
+
+// src/server.ts
+var import_mcp = require("@modelcontextprotocol/sdk/server/mcp.js");
+var SERVER_NAME = "Runpod API Server";
+var SERVER_VERSION = true ? "1.4.0" : "dev";
+function createServer(version = SERVER_VERSION) {
+  return new import_mcp.McpServer({
+    name: SERVER_NAME,
+    version,
+    capabilities: {
+      resources: {},
+      tools: {}
+    }
+  });
+}
+
+// src/tools.ts
+var import_zod2 = require("zod");
+var import_node_fetch = __toESM(require("node-fetch"), 1);
+var import_node_crypto = require("crypto");
+
+// src/pagination.ts
+var import_zod = require("zod");
+var DEFAULT_LIST_LIMIT = 20;
+var MAX_LIST_LIMIT = 100;
+var listPaginationParams = {
+  limit: import_zod.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: import_zod.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.4.0" : "dev";
+var SESSION_ID = (0, import_node_crypto.randomUUID)();
+function sanitizeUaToken(value) {
+  return value.replace(/[()<>@,;:\\"/[\]?={}\s]/g, "_").slice(0, 64);
+}
+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");
+  const serverVersion = ctx.serverVersion || MCP_SERVER_VERSION;
+  const userAgent = `runpod-mcp-server/${serverVersion} (caller=mcp; client=${name}; client_version=${version}; transport=${ctx.transport})`;
+  return {
+    "User-Agent": userAgent,
+    "X-Runpod-Session-Id": SESSION_ID
+  };
+}
+async function graphqlRequest(query, tracking) {
+  const response = await (0, import_node_fetch.default)(GRAPHQL_URL, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      ...tracking()
+    },
+    body: JSON.stringify({ query })
+  });
+  const result = await response.json();
+  if (result.errors && result.errors.length > 0) {
+    throw new Error(
+      `GraphQL Error: ${result.errors.map((e) => e.message).join(", ")}`
+    );
+  }
+  if (!result.data) {
+    throw new Error("No data returned from GraphQL query");
+  }
+  return result.data;
+}
+function createRunpodRequest(apiKey, tracking) {
+  return async function runpodRequest(endpoint, method = "GET", body) {
+    const url = `${API_BASE_URL}${endpoint}`;
+    const headers = {
+      Authorization: `Bearer ${apiKey}`,
+      "Content-Type": "application/json",
+      ...tracking()
+    };
+    const options = {
+      method,
+      headers
+    };
+    if (body && (method === "POST" || method === "PATCH")) {
+      options.body = JSON.stringify(body);
+    }
+    try {
+      const response = await (0, import_node_fetch.default)(url, options);
+      if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`Runpod API Error: ${response.status} - ${errorText}`);
+      }
+      const contentType = response.headers.get("content-type");
+      if (contentType && contentType.includes("application/json")) {
+        return await response.json();
+      }
+      return { success: true, status: response.status };
+    } catch (error) {
+      console.error("Error calling Runpod API:", error);
+      throw error;
+    }
+  };
+}
+function createServerlessRequest(apiKey, tracking) {
+  return async function serverlessRequest(endpointId, path, method = "GET", body) {
+    const url = `${SERVERLESS_API_BASE_URL}/${endpointId}${path}`;
+    const headers = {
+      Authorization: `Bearer ${apiKey}`,
+      "Content-Type": "application/json",
+      ...tracking()
+    };
+    const options = {
+      method,
+      headers
+    };
+    if (body && (method === "POST" || method === "PATCH")) {
+      options.body = JSON.stringify(body);
+    }
+    try {
+      const response = await (0, import_node_fetch.default)(url, options);
+      if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(
+          `Runpod Serverless API Error: ${response.status} - ${errorText}`
+        );
+      }
+      const contentType = response.headers.get("content-type");
+      if (contentType && contentType.includes("application/json")) {
+        return await response.json();
+      }
+      return { success: true, status: response.status };
+    } catch (error) {
+      console.error("Error calling Runpod Serverless API:", error);
+      throw error;
+    }
+  };
+}
+function registerTools(server, ctx) {
+  const tracking = () => trackingHeaders(server, ctx);
+  const runpodRequest = createRunpodRequest(ctx.apiKey, tracking);
+  const serverlessRequest = createServerlessRequest(ctx.apiKey, tracking);
+  server.tool(
+    "list-gpu-types",
+    {
+      minMemoryGb: import_zod2.z.number().optional().describe("Filter to GPUs with at least this much VRAM in GB"),
+      secureCloudOnly: import_zod2.z.boolean().optional().describe("Filter to only GPUs available in secure cloud"),
+      communityCloudOnly: import_zod2.z.boolean().optional().describe("Filter to only GPUs available in community cloud"),
+      searchTerm: import_zod2.z.string().optional().describe(
+        "Search term to filter GPUs by name (e.g., 'A100', 'RTX 4090')"
+      ),
+      includeUnavailable: import_zod2.z.boolean().optional().describe(
+        "Include GPUs that are currently out of stock. Default is false"
+      )
+    },
+    async (params) => {
+      const data = await graphqlRequest(
+        `
+        query {
+          gpuTypes {
+            id
+            displayName
+            memoryInGb
+            secureCloud
+            communityCloud
+            lowestPrice(input: { gpuCount: 1 }) {
+              stockStatus
+            }
+          }
+        }
+      `,
+        tracking
+      );
+      const stockPriority = {
+        High: 3,
+        Medium: 2,
+        Low: 1
+      };
+      const isAvailable = (gpu) => {
+        const status = gpu.lowestPrice?.stockStatus;
+        return !!status && status !== "Out";
+      };
+      let gpuTypes = data.gpuTypes.filter((gpu) => gpu.id !== "unknown");
+      if (!params.includeUnavailable) {
+        gpuTypes = gpuTypes.filter(isAvailable);
+      }
+      if (params.minMemoryGb) {
+        gpuTypes = gpuTypes.filter(
+          (gpu) => gpu.memoryInGb >= params.minMemoryGb
+        );
+      }
+      if (params.secureCloudOnly) {
+        gpuTypes = gpuTypes.filter((gpu) => gpu.secureCloud);
+      }
+      if (params.communityCloudOnly) {
+        gpuTypes = gpuTypes.filter((gpu) => gpu.communityCloud);
+      }
+      if (params.searchTerm) {
+        const term = params.searchTerm.toLowerCase();
+        gpuTypes = gpuTypes.filter(
+          (gpu) => gpu.id.toLowerCase().includes(term) || gpu.displayName.toLowerCase().includes(term)
+        );
+      }
+      gpuTypes.sort((a, b) => {
+        const aP = stockPriority[a.lowestPrice?.stockStatus || ""] || 0;
+        const bP = stockPriority[b.lowestPrice?.stockStatus || ""] || 0;
+        if (bP !== aP) return bP - aP;
+        return b.memoryInGb - a.memoryInGb;
+      });
+      const result = gpuTypes.map((gpu) => ({
+        id: gpu.id,
+        displayName: gpu.displayName,
+        memoryGb: gpu.memoryInGb,
+        secureCloud: gpu.secureCloud,
+        communityCloud: gpu.communityCloud,
+        stockStatus: gpu.lowestPrice?.stockStatus || "unavailable",
+        available: isAvailable(gpu)
+      }));
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "list-data-centers",
+    {
+      region: import_zod2.z.string().optional().describe(
+        "Filter by region/location (e.g., 'United States', 'Europe', 'Canada')"
+      )
+    },
+    async (params) => {
+      const data = await graphqlRequest(
+        `
+        query {
+          dataCenters {
+            id
+            name
+            location
+          }
+        }
+      `,
+        tracking
+      );
+      let dataCenters = data.dataCenters;
+      if (params.region) {
+        const term = params.region.toLowerCase();
+        dataCenters = dataCenters.filter(
+          (dc) => dc.location.toLowerCase().includes(term)
+        );
+      }
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(dataCenters, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "list-pods",
+    {
+      ...listPaginationParams,
+      computeType: import_zod2.z.enum(["GPU", "CPU"]).optional().describe("Filter to only GPU or only CPU Pods"),
+      gpuTypeId: import_zod2.z.array(import_zod2.z.string()).optional().describe("Filter to Pods with any of the listed GPU types"),
+      dataCenterId: import_zod2.z.array(import_zod2.z.string()).optional().describe("Filter to Pods in any of the provided data centers"),
+      name: import_zod2.z.string().optional().describe("Filter to Pods with the provided name"),
+      includeMachine: import_zod2.z.boolean().optional().describe("Include information about the machine"),
+      includeNetworkVolume: import_zod2.z.boolean().optional().describe("Include information about attached network volumes")
+    },
+    async (params) => {
+      const queryParams = new URLSearchParams();
+      if (params.computeType)
+        queryParams.append("computeType", params.computeType);
+      if (params.gpuTypeId)
+        params.gpuTypeId.forEach(
+          (type) => queryParams.append("gpuTypeId", type)
+        );
+      if (params.dataCenterId)
+        params.dataCenterId.forEach(
+          (dc) => queryParams.append("dataCenterId", dc)
+        );
+      if (params.name) queryParams.append("name", params.name);
+      if (params.includeMachine)
+        queryParams.append("includeMachine", params.includeMachine.toString());
+      if (params.includeNetworkVolume)
+        queryParams.append(
+          "includeNetworkVolume",
+          params.includeNetworkVolume.toString()
+        );
+      const queryString = queryParams.toString() ? `?${queryParams.toString()}` : "";
+      const result = await runpodRequest(`/pods${queryString}`);
+      return capListResult(result, {
+        limit: params.limit,
+        cursor: params.cursor
+      });
+    }
+  );
+  server.tool(
+    "get-pod",
+    {
+      podId: import_zod2.z.string().describe("ID of the pod to retrieve"),
+      includeMachine: import_zod2.z.boolean().optional().describe("Include information about the machine"),
+      includeNetworkVolume: import_zod2.z.boolean().optional().describe("Include information about attached network volumes")
+    },
+    async (params) => {
+      const queryParams = new URLSearchParams();
+      if (params.includeMachine)
+        queryParams.append("includeMachine", params.includeMachine.toString());
+      if (params.includeNetworkVolume)
+        queryParams.append(
+          "includeNetworkVolume",
+          params.includeNetworkVolume.toString()
+        );
+      const queryString = queryParams.toString() ? `?${queryParams.toString()}` : "";
+      const result = await runpodRequest(`/pods/${params.podId}${queryString}`);
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  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: import_zod2.z.string().optional().describe("Name for the pod"),
+      imageName: import_zod2.z.string().describe("Docker image to use"),
+      cloudType: import_zod2.z.enum(["SECURE", "COMMUNITY"]).optional().describe("SECURE or COMMUNITY cloud"),
+      gpuTypeIds: import_zod2.z.array(import_zod2.z.string()).optional().describe("List of acceptable GPU types"),
+      gpuCount: import_zod2.z.number().optional().describe("Number of GPUs"),
+      containerDiskInGb: import_zod2.z.number().optional().describe("Container disk size in GB"),
+      volumeInGb: import_zod2.z.number().optional().describe("Volume size in GB"),
+      volumeMountPath: import_zod2.z.string().optional().describe("Path to mount the volume"),
+      ports: import_zod2.z.array(import_zod2.z.string()).optional().describe("Ports to expose (e.g., '8888/http', '22/tcp')"),
+      env: import_zod2.z.record(import_zod2.z.string()).optional().describe("Environment variables"),
+      dataCenterIds: import_zod2.z.array(import_zod2.z.string()).optional().describe("List of data centers")
+    },
+    async (params) => {
+      const result = await runpodRequest("/pods", "POST", params);
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "update-pod",
+    {
+      podId: import_zod2.z.string().describe("ID of the pod to update"),
+      name: import_zod2.z.string().optional().describe("New name for the pod"),
+      imageName: import_zod2.z.string().optional().describe("New Docker image"),
+      containerDiskInGb: import_zod2.z.number().optional().describe("New container disk size in GB"),
+      volumeInGb: import_zod2.z.number().optional().describe("New volume size in GB"),
+      volumeMountPath: import_zod2.z.string().optional().describe("New path to mount the volume"),
+      ports: import_zod2.z.array(import_zod2.z.string()).optional().describe("New ports to expose"),
+      env: import_zod2.z.record(import_zod2.z.string()).optional().describe("New environment variables")
+    },
+    async (params) => {
+      const { podId, ...updateParams } = params;
+      const result = await runpodRequest(
+        `/pods/${podId}`,
+        "PATCH",
+        updateParams
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "start-pod",
+    {
+      podId: import_zod2.z.string().describe("ID of the pod to start")
+    },
+    async (params) => {
+      const result = await runpodRequest(`/pods/${params.podId}/start`, "POST");
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "stop-pod",
+    {
+      podId: import_zod2.z.string().describe("ID of the pod to stop")
+    },
+    async (params) => {
+      const result = await runpodRequest(`/pods/${params.podId}/stop`, "POST");
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "delete-pod",
+    {
+      podId: import_zod2.z.string().describe("ID of the pod to delete")
+    },
+    async (params) => {
+      const result = await runpodRequest(`/pods/${params.podId}`, "DELETE");
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "list-endpoints",
+    {
+      ...listPaginationParams,
+      includeTemplate: import_zod2.z.boolean().optional().describe("Include template information"),
+      includeWorkers: import_zod2.z.boolean().optional().describe("Include information about workers")
+    },
+    async (params) => {
+      const queryParams = new URLSearchParams();
+      if (params.includeTemplate)
+        queryParams.append(
+          "includeTemplate",
+          params.includeTemplate.toString()
+        );
+      if (params.includeWorkers)
+        queryParams.append("includeWorkers", params.includeWorkers.toString());
+      const queryString = queryParams.toString() ? `?${queryParams.toString()}` : "";
+      const result = await runpodRequest(`/endpoints${queryString}`);
+      return capListResult(result, {
+        limit: params.limit,
+        cursor: params.cursor
+      });
+    }
+  );
+  server.tool(
+    "get-endpoint",
+    {
+      endpointId: import_zod2.z.string().describe("ID of the endpoint to retrieve"),
+      includeTemplate: import_zod2.z.boolean().optional().describe("Include template information"),
+      includeWorkers: import_zod2.z.boolean().optional().describe("Include information about workers")
+    },
+    async (params) => {
+      const queryParams = new URLSearchParams();
+      if (params.includeTemplate)
+        queryParams.append(
+          "includeTemplate",
+          params.includeTemplate.toString()
+        );
+      if (params.includeWorkers)
+        queryParams.append("includeWorkers", params.includeWorkers.toString());
+      const queryString = queryParams.toString() ? `?${queryParams.toString()}` : "";
+      const result = await runpodRequest(
+        `/endpoints/${params.endpointId}${queryString}`
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "create-endpoint",
+    {
+      name: import_zod2.z.string().optional().describe("Name for the endpoint"),
+      templateId: import_zod2.z.string().describe("Template ID to use"),
+      computeType: import_zod2.z.enum(["GPU", "CPU"]).optional().describe("GPU or CPU endpoint"),
+      gpuTypeIds: import_zod2.z.array(import_zod2.z.string()).optional().describe("List of acceptable GPU types"),
+      gpuCount: import_zod2.z.number().optional().describe("Number of GPUs per worker"),
+      workersMin: import_zod2.z.number().optional().describe("Minimum number of workers"),
+      workersMax: import_zod2.z.number().optional().describe("Maximum number of workers"),
+      dataCenterIds: import_zod2.z.array(import_zod2.z.string()).optional().describe("List of data centers")
+    },
+    async (params) => {
+      const result = await runpodRequest("/endpoints", "POST", params);
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "update-endpoint",
+    {
+      endpointId: import_zod2.z.string().describe("ID of the endpoint to update"),
+      name: import_zod2.z.string().optional().describe("New name for the endpoint"),
+      workersMin: import_zod2.z.number().optional().describe("New minimum number of workers"),
+      workersMax: import_zod2.z.number().optional().describe("New maximum number of workers"),
+      idleTimeout: import_zod2.z.number().optional().describe("New idle timeout in seconds"),
+      scalerType: import_zod2.z.enum(["QUEUE_DELAY", "REQUEST_COUNT"]).optional().describe("Scaler type"),
+      scalerValue: import_zod2.z.number().optional().describe("Scaler value")
+    },
+    async (params) => {
+      const { endpointId, ...updateParams } = params;
+      const result = await runpodRequest(
+        `/endpoints/${endpointId}`,
+        "PATCH",
+        updateParams
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "delete-endpoint",
+    {
+      endpointId: import_zod2.z.string().describe("ID of the endpoint to delete")
+    },
+    async (params) => {
+      const result = await runpodRequest(
+        `/endpoints/${params.endpointId}`,
+        "DELETE"
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  const endpointIdSchema = import_zod2.z.string().regex(/^[a-zA-Z0-9_-]+$/, "Invalid endpoint ID format");
+  const jobIdSchema = import_zod2.z.string().regex(/^[a-zA-Z0-9_-]+$/, "Invalid job ID format");
+  const inputSchema = import_zod2.z.record(import_zod2.z.unknown()).describe(
+    "Input payload for the worker handler. The expected fields depend on the deployed model or worker."
+  );
+  const webhookSchema = import_zod2.z.string().url().optional().describe(
+    "Webhook URL to receive job completion notifications instead of polling"
+  );
+  const policySchema = import_zod2.z.object({
+    executionTimeout: import_zod2.z.number().optional().describe("Maximum execution time in milliseconds"),
+    lowPriority: import_zod2.z.boolean().optional().describe("Submit as a low-priority job"),
+    ttl: import_zod2.z.number().optional().describe("Time-to-live for the job result in milliseconds")
+  }).optional().describe("Execution policy options");
+  const s3ConfigSchema = import_zod2.z.object({
+    accessId: import_zod2.z.string().describe("S3 access key ID"),
+    accessSecret: import_zod2.z.string().describe("S3 secret access key"),
+    bucketName: import_zod2.z.string().describe("S3 bucket name"),
+    endpointUrl: import_zod2.z.string().describe("S3 endpoint URL")
+  }).optional().describe("S3-compatible storage config for large outputs");
+  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.",
+    {
+      endpointId: endpointIdSchema.describe(
+        "ID of the Serverless endpoint to run"
+      ),
+      input: inputSchema,
+      webhook: webhookSchema,
+      policy: policySchema,
+      s3Config: s3ConfigSchema
+    },
+    async (params) => {
+      const { endpointId, ...body } = params;
+      const result = await serverlessRequest(
+        endpointId,
+        "/run",
+        "POST",
+        body
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  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).",
+    {
+      endpointId: endpointIdSchema.describe(
+        "ID of the Serverless endpoint to run synchronously"
+      ),
+      input: inputSchema,
+      wait: import_zod2.z.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,
+      policy: policySchema,
+      s3Config: s3ConfigSchema
+    },
+    async (params) => {
+      const { endpointId, wait, ...body } = params;
+      const path = wait ? `/runsync?wait=${wait}` : "/runsync";
+      const result = await serverlessRequest(
+        endpointId,
+        path,
+        "POST",
+        body
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  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.",
+    {
+      endpointId: endpointIdSchema.describe(
+        "ID of the Serverless endpoint the job belongs to"
+      ),
+      jobId: jobIdSchema.describe("ID of the job to check")
+    },
+    async (params) => {
+      const result = await serverlessRequest(
+        params.endpointId,
+        `/status/${params.jobId}`
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  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.",
+    {
+      endpointId: endpointIdSchema.describe(
+        "ID of the Serverless endpoint the job belongs to"
+      ),
+      jobId: jobIdSchema.describe("ID of the job to stream results from")
+    },
+    async (params) => {
+      const TERMINAL_STATUSES = /* @__PURE__ */ new Set([
+        "COMPLETED",
+        "FAILED",
+        "CANCELLED",
+        "TIMED_OUT"
+      ]);
+      const MAX_POLL_TIME_MS = 5 * 60 * 1e3;
+      const POLL_INTERVAL_MS = 1e3;
+      const MAX_CONSECUTIVE_ERRORS = 5;
+      const allChunks = [];
+      let finalResult = {};
+      let consecutiveErrors = 0;
+      let lastError;
+      const startTime = Date.now();
+      while (true) {
+        try {
+          const result = await serverlessRequest(
+            params.endpointId,
+            `/stream/${params.jobId}`
+          );
+          consecutiveErrors = 0;
+          if (Array.isArray(result.stream)) {
+            allChunks.push(...result.stream);
+          }
+          finalResult = result;
+          if (TERMINAL_STATUSES.has(result.status)) {
+            break;
+          }
+        } catch (error) {
+          consecutiveErrors++;
+          lastError = error instanceof Error ? error.message : String(error);
+          if (consecutiveErrors >= MAX_CONSECUTIVE_ERRORS) {
+            finalResult.error = `Polling aborted after ${MAX_CONSECUTIVE_ERRORS} consecutive errors: ${lastError}`;
+            break;
+          }
+        }
+        if (Date.now() - startTime > MAX_POLL_TIME_MS) {
+          finalResult.pollingTimedOut = true;
+          finalResult.note = "Polling timed out after 5 minutes. Use get-job-status to check the job later.";
+          if (lastError) finalResult.lastError = lastError;
+          break;
+        }
+        await new Promise((resolve) => setTimeout(resolve, POLL_INTERVAL_MS));
+      }
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(
+              { ...finalResult, stream: allChunks },
+              null,
+              2
+            )
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "cancel-job",
+    "Cancel a Serverless job that is queued or in progress.",
+    {
+      endpointId: endpointIdSchema.describe(
+        "ID of the Serverless endpoint the job belongs to"
+      ),
+      jobId: jobIdSchema.describe("ID of the job to cancel")
+    },
+    async (params) => {
+      const result = await serverlessRequest(
+        params.endpointId,
+        `/cancel/${params.jobId}`,
+        "POST"
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  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.",
+    {
+      endpointId: endpointIdSchema.describe(
+        "ID of the Serverless endpoint the job belongs to"
+      ),
+      jobId: jobIdSchema.describe("ID of the job to retry")
+    },
+    async (params) => {
+      const result = await serverlessRequest(
+        params.endpointId,
+        `/retry/${params.jobId}`,
+        "POST"
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "endpoint-health",
+    "Get the health and operational status of a Serverless endpoint, including worker counts and job statistics.",
+    {
+      endpointId: endpointIdSchema.describe(
+        "ID of the Serverless endpoint to check health for"
+      )
+    },
+    async (params) => {
+      const result = await serverlessRequest(params.endpointId, "/health");
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  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.",
+    {
+      endpointId: endpointIdSchema.describe(
+        "ID of the Serverless endpoint to purge the queue for"
+      )
+    },
+    async (params) => {
+      const result = await serverlessRequest(
+        params.endpointId,
+        "/purge-queue",
+        "POST"
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  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.`,
+    {
+      ...listPaginationParams,
+      includeRunpodTemplates: import_zod2.z.boolean().optional().describe("Include official Runpod templates in the response"),
+      includePublicTemplates: import_zod2.z.boolean().optional().describe("Include community-made public templates in the response"),
+      includeEndpointBoundTemplates: import_zod2.z.boolean().optional().describe(
+        "Include templates bound to Serverless endpoints in the response"
+      )
+    },
+    async (params) => {
+      const queryParams = new URLSearchParams();
+      if (params.includeRunpodTemplates)
+        queryParams.set("includeRunpodTemplates", "true");
+      if (params.includePublicTemplates)
+        queryParams.set("includePublicTemplates", "true");
+      if (params.includeEndpointBoundTemplates)
+        queryParams.set("includeEndpointBoundTemplates", "true");
+      const query = queryParams.toString();
+      const result = await runpodRequest(
+        `/templates${query ? `?${query}` : ""}`
+      );
+      return capListResult(result, {
+        limit: params.limit,
+        cursor: params.cursor
+      });
+    }
+  );
+  server.tool(
+    "get-template",
+    {
+      templateId: import_zod2.z.string().describe("ID of the template to retrieve")
+    },
+    async (params) => {
+      const result = await runpodRequest(`/templates/${params.templateId}`);
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "create-template",
+    {
+      name: import_zod2.z.string().describe("Name for the template"),
+      imageName: import_zod2.z.string().describe("Docker image to use"),
+      isServerless: import_zod2.z.boolean().optional().describe("Is this a serverless template"),
+      ports: import_zod2.z.array(import_zod2.z.string()).optional().describe("Ports to expose"),
+      dockerEntrypoint: import_zod2.z.array(import_zod2.z.string()).optional().describe("Docker entrypoint commands"),
+      dockerStartCmd: import_zod2.z.array(import_zod2.z.string()).optional().describe("Docker start commands"),
+      env: import_zod2.z.record(import_zod2.z.string()).optional().describe("Environment variables"),
+      containerDiskInGb: import_zod2.z.number().optional().describe("Container disk size in GB"),
+      volumeInGb: import_zod2.z.number().optional().describe("Volume size in GB"),
+      volumeMountPath: import_zod2.z.string().optional().describe("Path to mount the volume"),
+      readme: import_zod2.z.string().optional().describe("README content in markdown format"),
+      containerRegistryAuthId: import_zod2.z.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);
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "update-template",
+    {
+      templateId: import_zod2.z.string().describe("ID of the template to update"),
+      name: import_zod2.z.string().optional().describe("New name for the template"),
+      imageName: import_zod2.z.string().optional().describe("New Docker image"),
+      ports: import_zod2.z.array(import_zod2.z.string()).optional().describe("New ports to expose"),
+      env: import_zod2.z.record(import_zod2.z.string()).optional().describe("New environment variables"),
+      readme: import_zod2.z.string().optional().describe("New README content in markdown format"),
+      containerRegistryAuthId: import_zod2.z.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;
+      const result = await runpodRequest(
+        `/templates/${templateId}`,
+        "PATCH",
+        updateParams
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "delete-template",
+    {
+      templateId: import_zod2.z.string().describe("ID of the template to delete")
+    },
+    async (params) => {
+      const result = await runpodRequest(
+        `/templates/${params.templateId}`,
+        "DELETE"
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool("list-network-volumes", listPaginationParams, async (params) => {
+    const result = await runpodRequest("/networkvolumes");
+    return capListResult(result, {
+      limit: params.limit,
+      cursor: params.cursor
+    });
+  });
+  server.tool(
+    "get-network-volume",
+    {
+      networkVolumeId: import_zod2.z.string().describe("ID of the network volume to retrieve")
+    },
+    async (params) => {
+      const result = await runpodRequest(
+        `/networkvolumes/${params.networkVolumeId}`
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "create-network-volume",
+    {
+      name: import_zod2.z.string().describe("Name for the network volume"),
+      size: import_zod2.z.number().describe("Size in GB (1-4000)"),
+      dataCenterId: import_zod2.z.string().describe("Data center ID")
+    },
+    async (params) => {
+      const result = await runpodRequest("/networkvolumes", "POST", params);
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "update-network-volume",
+    {
+      networkVolumeId: import_zod2.z.string().describe("ID of the network volume to update"),
+      name: import_zod2.z.string().optional().describe("New name for the network volume"),
+      size: import_zod2.z.number().optional().describe("New size in GB (must be larger than current)")
+    },
+    async (params) => {
+      const { networkVolumeId, ...updateParams } = params;
+      const result = await runpodRequest(
+        `/networkvolumes/${networkVolumeId}`,
+        "PATCH",
+        updateParams
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "delete-network-volume",
+    {
+      networkVolumeId: import_zod2.z.string().describe("ID of the network volume to delete")
+    },
+    async (params) => {
+      const result = await runpodRequest(
+        `/networkvolumes/${params.networkVolumeId}`,
+        "DELETE"
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  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: import_zod2.z.string().describe("ID of the container registry auth to retrieve")
+    },
+    async (params) => {
+      const result = await runpodRequest(
+        `/containerregistryauth/${params.containerRegistryAuthId}`
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "create-container-registry-auth",
+    {
+      name: import_zod2.z.string().describe("Name for the container registry auth"),
+      username: import_zod2.z.string().describe("Registry username"),
+      password: import_zod2.z.string().describe("Registry password")
+    },
+    async (params) => {
+      const result = await runpodRequest(
+        "/containerregistryauth",
+        "POST",
+        params
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+  server.tool(
+    "delete-container-registry-auth",
+    {
+      containerRegistryAuthId: import_zod2.z.string().describe("ID of the container registry auth to delete")
+    },
+    async (params) => {
+      const result = await runpodRequest(
+        `/containerregistryauth/${params.containerRegistryAuthId}`,
+        "DELETE"
+      );
+      return {
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify(result, null, 2)
+          }
+        ]
+      };
+    }
+  );
+}
+
+// src/http.ts
+function extractBearerToken(req) {
+  const authHeader = req.headers.authorization;
+  if (!authHeader) return null;
+  const parts = authHeader.split(" ");
+  if (parts.length !== 2 || parts[0].toLowerCase() !== "bearer") return null;
+  return parts[1];
+}
+function getBaseUrl(req) {
+  const proto = req.headers["x-forwarded-proto"]?.split(",")[0] ?? "https";
+  const host = req.headers.host;
+  if (!host) {
+    throw new Error("Missing Host header");
+  }
+  return `${proto}://${host}`;
+}
+function writeUnauthorized(req, res, error) {
+  const headers = {
+    "Content-Type": "application/json",
+    "WWW-Authenticate": `Bearer realm="mcp", resource_metadata="${getBaseUrl(
+      req
+    )}/.well-known/oauth-protected-resource"`
+  };
+  res.writeHead(401, headers);
+  res.end(JSON.stringify({ error }));
+}
+async function handleMcpRequest(req, res, opts = {}) {
+  const bearerToken = extractBearerToken(req);
+  console.log("mcp_request", {
+    method: req.method,
+    url: req.url,
+    hasBearerToken: !!bearerToken
+  });
+  if (!bearerToken) {
+    writeUnauthorized(
+      req,
+      res,
+      "Missing or invalid Authorization header. Provide a Bearer token."
+    );
+    return;
+  }
+  const server = createServer(opts.serverVersion);
+  registerTools(server, {
+    apiKey: bearerToken,
+    transport: "http",
+    clientUserAgent: req.headers["user-agent"],
+    serverVersion: opts.serverVersion
+  });
+  const transport = new import_streamableHttp.StreamableHTTPServerTransport({
+    sessionIdGenerator: void 0
+    // stateless — no session persistence
+  });
+  res.on("close", () => {
+    transport.close().catch(() => {
+    });
+    server.close().catch(() => {
+    });
+  });
+  await server.connect(transport);
+  await transport.handleRequest(req, res, req.body);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  handleMcpRequest,
+  registerTools
+});
+//# sourceMappingURL=http.cjs.map