@runpod/mcp-server 1.1.0 → 1.3.0

package/dist/index.js

@@ -1,638 +0,0 @@
-#!/usr/bin/env node
-"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 __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
-));
-
-// src/index.ts
-var import_mcp = require("@modelcontextprotocol/sdk/server/mcp.js");
-var import_stdio = require("@modelcontextprotocol/sdk/server/stdio.js");
-var import_zod = require("zod");
-var import_node_fetch = __toESM(require("node-fetch"), 1);
-var API_BASE_URL = "https://rest.runpod.io/v1";
-var API_KEY = process.env.RUNPOD_API_KEY;
-if (!API_KEY) {
-  console.error("RUNPOD_API_KEY environment variable is required");
-  process.exit(1);
-}
-var server = new import_mcp.McpServer({
-  name: "RunPod API Server",
-  version: "1.0.0",
-  capabilities: {
-    resources: {},
-    tools: {}
-  }
-});
-async function runpodRequest(endpoint, method = "GET", body) {
-  const url = `${API_BASE_URL}${endpoint}`;
-  const headers = {
-    Authorization: `Bearer ${API_KEY}`,
-    "Content-Type": "application/json"
-  };
-  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;
-  }
-}
-server.tool(
-  "list-pods",
-  {
-    computeType: import_zod.z.enum(["GPU", "CPU"]).optional().describe("Filter to only GPU or only CPU Pods"),
-    gpuTypeId: import_zod.z.array(import_zod.z.string()).optional().describe("Filter to Pods with any of the listed GPU types"),
-    dataCenterId: import_zod.z.array(import_zod.z.string()).optional().describe("Filter to Pods in any of the provided data centers"),
-    name: import_zod.z.string().optional().describe("Filter to Pods with the provided name"),
-    includeMachine: import_zod.z.boolean().optional().describe("Include information about the machine"),
-    includeNetworkVolume: import_zod.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 {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify(result, null, 2)
-        }
-      ]
-    };
-  }
-);
-server.tool(
-  "get-pod",
-  {
-    podId: import_zod.z.string().describe("ID of the pod to retrieve"),
-    includeMachine: import_zod.z.boolean().optional().describe("Include information about the machine"),
-    includeNetworkVolume: import_zod.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",
-  {
-    name: import_zod.z.string().optional().describe("Name for the pod"),
-    imageName: import_zod.z.string().describe("Docker image to use"),
-    cloudType: import_zod.z.enum(["SECURE", "COMMUNITY"]).optional().describe("SECURE or COMMUNITY cloud"),
-    gpuTypeIds: import_zod.z.array(import_zod.z.string()).optional().describe("List of acceptable GPU types"),
-    gpuCount: import_zod.z.number().optional().describe("Number of GPUs"),
-    containerDiskInGb: import_zod.z.number().optional().describe("Container disk size in GB"),
-    volumeInGb: import_zod.z.number().optional().describe("Volume size in GB"),
-    volumeMountPath: import_zod.z.string().optional().describe("Path to mount the volume"),
-    ports: import_zod.z.array(import_zod.z.string()).optional().describe("Ports to expose (e.g., '8888/http', '22/tcp')"),
-    env: import_zod.z.record(import_zod.z.string()).optional().describe("Environment variables"),
-    dataCenterIds: import_zod.z.array(import_zod.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_zod.z.string().describe("ID of the pod to update"),
-    name: import_zod.z.string().optional().describe("New name for the pod"),
-    imageName: import_zod.z.string().optional().describe("New Docker image"),
-    containerDiskInGb: import_zod.z.number().optional().describe("New container disk size in GB"),
-    volumeInGb: import_zod.z.number().optional().describe("New volume size in GB"),
-    volumeMountPath: import_zod.z.string().optional().describe("New path to mount the volume"),
-    ports: import_zod.z.array(import_zod.z.string()).optional().describe("New ports to expose"),
-    env: import_zod.z.record(import_zod.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_zod.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_zod.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_zod.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",
-  {
-    includeTemplate: import_zod.z.boolean().optional().describe("Include template information"),
-    includeWorkers: import_zod.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 {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify(result, null, 2)
-        }
-      ]
-    };
-  }
-);
-server.tool(
-  "get-endpoint",
-  {
-    endpointId: import_zod.z.string().describe("ID of the endpoint to retrieve"),
-    includeTemplate: import_zod.z.boolean().optional().describe("Include template information"),
-    includeWorkers: import_zod.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_zod.z.string().optional().describe("Name for the endpoint"),
-    templateId: import_zod.z.string().describe("Template ID to use"),
-    computeType: import_zod.z.enum(["GPU", "CPU"]).optional().describe("GPU or CPU endpoint"),
-    gpuTypeIds: import_zod.z.array(import_zod.z.string()).optional().describe("List of acceptable GPU types"),
-    gpuCount: import_zod.z.number().optional().describe("Number of GPUs per worker"),
-    workersMin: import_zod.z.number().optional().describe("Minimum number of workers"),
-    workersMax: import_zod.z.number().optional().describe("Maximum number of workers"),
-    dataCenterIds: import_zod.z.array(import_zod.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_zod.z.string().describe("ID of the endpoint to update"),
-    name: import_zod.z.string().optional().describe("New name for the endpoint"),
-    workersMin: import_zod.z.number().optional().describe("New minimum number of workers"),
-    workersMax: import_zod.z.number().optional().describe("New maximum number of workers"),
-    idleTimeout: import_zod.z.number().optional().describe("New idle timeout in seconds"),
-    scalerType: import_zod.z.enum(["QUEUE_DELAY", "REQUEST_COUNT"]).optional().describe("Scaler type"),
-    scalerValue: import_zod.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_zod.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)
-        }
-      ]
-    };
-  }
-);
-server.tool("list-templates", {}, async () => {
-  const result = await runpodRequest("/templates");
-  return {
-    content: [
-      {
-        type: "text",
-        text: JSON.stringify(result, null, 2)
-      }
-    ]
-  };
-});
-server.tool(
-  "get-template",
-  {
-    templateId: import_zod.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_zod.z.string().describe("Name for the template"),
-    imageName: import_zod.z.string().describe("Docker image to use"),
-    isServerless: import_zod.z.boolean().optional().describe("Is this a serverless template"),
-    ports: import_zod.z.array(import_zod.z.string()).optional().describe("Ports to expose"),
-    dockerEntrypoint: import_zod.z.array(import_zod.z.string()).optional().describe("Docker entrypoint commands"),
-    dockerStartCmd: import_zod.z.array(import_zod.z.string()).optional().describe("Docker start commands"),
-    env: import_zod.z.record(import_zod.z.string()).optional().describe("Environment variables"),
-    containerDiskInGb: import_zod.z.number().optional().describe("Container disk size in GB"),
-    volumeInGb: import_zod.z.number().optional().describe("Volume size in GB"),
-    volumeMountPath: import_zod.z.string().optional().describe("Path to mount the volume"),
-    readme: import_zod.z.string().optional().describe("README content in markdown format")
-  },
-  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_zod.z.string().describe("ID of the template to update"),
-    name: import_zod.z.string().optional().describe("New name for the template"),
-    imageName: import_zod.z.string().optional().describe("New Docker image"),
-    ports: import_zod.z.array(import_zod.z.string()).optional().describe("New ports to expose"),
-    env: import_zod.z.record(import_zod.z.string()).optional().describe("New environment variables"),
-    readme: import_zod.z.string().optional().describe("New README content in markdown format")
-  },
-  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_zod.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", {}, async () => {
-  const result = await runpodRequest("/networkvolumes");
-  return {
-    content: [
-      {
-        type: "text",
-        text: JSON.stringify(result, null, 2)
-      }
-    ]
-  };
-});
-server.tool(
-  "get-network-volume",
-  {
-    networkVolumeId: import_zod.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_zod.z.string().describe("Name for the network volume"),
-    size: import_zod.z.number().describe("Size in GB (1-4000)"),
-    dataCenterId: import_zod.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_zod.z.string().describe("ID of the network volume to update"),
-    name: import_zod.z.string().optional().describe("New name for the network volume"),
-    size: import_zod.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_zod.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", {}, async () => {
-  const result = await runpodRequest("/containerregistryauth");
-  return {
-    content: [
-      {
-        type: "text",
-        text: JSON.stringify(result, null, 2)
-      }
-    ]
-  };
-});
-server.tool(
-  "get-container-registry-auth",
-  {
-    containerRegistryAuthId: import_zod.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_zod.z.string().describe("Name for the container registry auth"),
-    username: import_zod.z.string().describe("Registry username"),
-    password: import_zod.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_zod.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)
-        }
-      ]
-    };
-  }
-);
-var transport = new import_stdio.StdioServerTransport();
-server.connect(transport);
-//# sourceMappingURL=index.js.map