[Components] runpod

components/runpod/actions/create-pod/create-pod.mjs

@@ -0,0 +1,182 @@
+import app from "../../runpod.app.mjs";
+import mutations from "../../common/mutations.mjs";
+import utils from "../../common/utils.mjs";
+
+export default {
+  key: "runpod-create-pod",
+  name: "Create Pod",
+  description: "Creates a new pod with the specified parameters. [See the documentation](https://docs.runpod.io/sdks/graphql/manage-pods#create-spot-pod)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    bidPerGpu: {
+      propDefinition: [
+        app,
+        "bidPerGpu",
+      ],
+    },
+    cloudType: {
+      type: "string",
+      label: "Cloud Type",
+      description: "The type of cloud to use for the pod.",
+      options: [
+        "SECURE",
+        "COMMUNITY",
+        "ALL",
+      ],
+    },
+    gpuCount: {
+      propDefinition: [
+        app,
+        "gpuCount",
+      ],
+    },
+    volumeInGb: {
+      type: "integer",
+      label: "Volume In GB",
+      description: "The size of the volume in GB.",
+    },
+    containerDiskInGb: {
+      type: "integer",
+      label: "Container Disk In GB",
+      description: "The size of the container disk in GB.",
+    },
+    minVcpuCount: {
+      type: "integer",
+      label: "Minimum VCPU Count",
+      description: "The minimum number of VCPUs.",
+    },
+    minMemoryInGb: {
+      type: "integer",
+      label: "Minimum Memory In GB",
+      description: "The minimum memory size in GB.",
+    },
+    gpuTypeId: {
+      propDefinition: [
+        app,
+        "gpuTypeId",
+      ],
+    },
+    name: {
+      type: "string",
+      label: "Name",
+      description: "The name of the pod.",
+    },
+    imageName: {
+      type: "string",
+      label: "Image Name",
+      description: "The name of the image to use for the pod.",
+    },
+    ports: {
+      type: "string",
+      label: "Ports",
+      description: "The ports to use for the pod.",
+    },
+    volumeMountPath: {
+      type: "string",
+      label: "Volume Mount Path",
+      description: "The path where the volume will be mounted.",
+    },
+    env: {
+      type: "string[]",
+      label: "Environment Variables",
+      description: "The environment variables to set for the pod. Each row should be formated with JSON string like this `{\"key\": \"ENV_NAME\", \"value\": \"ENV_VALUE\"}`",
+      optional: true,
+    },
+    dockerArgs: {
+      type: "string",
+      label: "Docker Args",
+      description: "The arguments to pass to the docker command.",
+      optional: true,
+    },
+    startJupyter: {
+      type: "boolean",
+      label: "Start Jupyter",
+      description: "Whether to start Jupyter for the pod.",
+      optional: true,
+    },
+    startSsh: {
+      type: "boolean",
+      label: "Start SSH",
+      description: "Whether to start SSH for the pod.",
+      optional: true,
+    },
+    stopAfter: {
+      type: "string",
+      label: "Stop After",
+      description: "The duration after which the pod will be stopped.",
+      optional: true,
+    },
+    supportPublicIp: {
+      type: "boolean",
+      label: "Support Public IP",
+      description: "Whether to support public IP for the pod.",
+      optional: true,
+    },
+    templateId: {
+      type: "string",
+      label: "Template ID",
+      description: "The ID of the template to use for the pod.",
+      optional: true,
+    },
+  },
+  methods: {
+    createPod(variables) {
+      return this.app.makeRequest({
+        query: mutations.createPod,
+        variables,
+      });
+    },
+  },
+  async run({ $ }) {
+    const {
+      createPod,
+      bidPerGpu,
+      cloudType,
+      gpuCount,
+      volumeInGb,
+      containerDiskInGb,
+      minVcpuCount,
+      minMemoryInGb,
+      gpuTypeId,
+      name,
+      imageName,
+      ports,
+      volumeMountPath,
+      env,
+      dockerArgs,
+      startJupyter,
+      startSsh,
+      stopAfter,
+      supportPublicIp,
+      templateId,
+    } = this;
+
+    const response = await createPod({
+      input: utils.cleanInput({
+        bidPerGpu,
+        cloudType,
+        gpuCount,
+        volumeInGb,
+        containerDiskInGb,
+        minVcpuCount,
+        minMemoryInGb,
+        gpuTypeId,
+        name,
+        imageName,
+        ports,
+        volumeMountPath,
+        env: utils.parseArray(env),
+        dockerArgs,
+        startJupyter,
+        startSsh,
+        stopAfter,
+        supportPublicIp,
+        templateId,
+      }),
+    });
+    $.export("$summary", `Successfully created a new pod with ID \`${response.id}\`.`);
+    return response;
+  },
+};

components/runpod/actions/get-pod/get-pod.mjs

@@ -0,0 +1,42 @@
+import app from "../../runpod.app.mjs";
+import queries from "../../common/queries.mjs";
+
+export default {
+  key: "runpod-get-pod",
+  name: "Get Pod Details",
+  description: "Get details of a pod by ID. [See the documentation](https://docs.runpod.io/sdks/graphql/manage-pods#get-pod-by-id).",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    podId: {
+      description: "The ID of the pod to get details for.",
+      propDefinition: [
+        app,
+        "podId",
+      ],
+    },
+  },
+  methods: {
+    getPodDetails(variables) {
+      return this.app.makeRequest({
+        query: queries.getPod,
+        variables,
+      });
+    },
+  },
+  async run({ $ }) {
+    const {
+      getPodDetails,
+      podId,
+    } = this;
+
+    const response = await getPodDetails({
+      input: {
+        podId,
+      },
+    });
+    $.export("$summary", `Succesfully retrieved details for pod with ID \`${response.id}\`.`);
+    return response;
+  },
+};

components/runpod/actions/start-pod/start-pod.mjs

@@ -0,0 +1,58 @@
+import app from "../../runpod.app.mjs";
+import mutations from "../../common/mutations.mjs";
+
+export default {
+  key: "runpod-start-pod",
+  name: "Start Pod",
+  description: "Starts a stopped pod, making it available for use. [See the documentation](https://docs.runpod.io/sdks/graphql/manage-pods#start-spot-pod)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    podId: {
+      propDefinition: [
+        app,
+        "podId",
+      ],
+    },
+    gpuCount: {
+      propDefinition: [
+        app,
+        "gpuCount",
+      ],
+    },
+    bidPerGpu: {
+      propDefinition: [
+        app,
+        "bidPerGpu",
+      ],
+    },
+  },
+  methods: {
+    startPod(variables) {
+      return this.app.makeRequest({
+        query: mutations.startPod,
+        variables,
+      });
+    },
+  },
+  async run({ $ }) {
+    const {
+      startPod,
+      podId,
+      gpuCount,
+      bidPerGpu,
+    } = this;
+
+    const response = await startPod({
+      input: {
+        podId,
+        gpuCount,
+        bidPerGpu: parseFloat(bidPerGpu),
+      },
+    });
+
+    $.export("$summary", `Sucessfully started pod with ID \`${response.id}\`.`);
+    return response;
+  },
+};

components/runpod/actions/stop-pod/stop-pod.mjs

@@ -0,0 +1,52 @@
+import app from "../../runpod.app.mjs";
+import mutations from "../../common/mutations.mjs";
+
+export default {
+  key: "runpod-stop-pod",
+  name: "Stop Pod",
+  description: "Stops a running pod, freeing up resources and preventing further charges. [See the documentation](https://docs.runpod.io/sdks/graphql/manage-pods#stop-pods)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    podId: {
+      description: "The ID of the pod to stop.",
+      propDefinition: [
+        app,
+        "podId",
+      ],
+    },
+    incrementVersion: {
+      type: "boolean",
+      label: "Increment Version",
+      description: "Whether to increment the pod version after stopping it.",
+      optional: true,
+    },
+  },
+  methods: {
+    stopPod(variables) {
+      return this.app.makeRequest({
+        query: mutations.stopPod,
+        variables,
+      });
+    },
+  },
+  async run({ $ }) {
+    const {
+      stopPod,
+      podId,
+      incrementVersion,
+    } = this;
+
+    const response = await stopPod({
+      input: {
+        podId,
+        incrementVersion,
+      },
+    });
+
+    $.export("$summary", `Succesfully stopped pod with ID \`${response.id}\`.`);
+
+    return response;
+  },
+};

components/runpod/common/mutations.mjs

@@ -0,0 +1,25 @@
+import { gql } from "graphql-request";
+
+export default {
+  createPod: gql`
+    mutation createPod($input: PodRentInterruptableInput!) {
+      podRentInterruptable(input: $input) {
+        id
+      }
+    }
+  `,
+  startPod: gql`
+    mutation startPod($input: PodBidResumeInput!) {
+      podResume(input: $input) {
+        id
+      }
+    }
+  `,
+  stopPod: gql`
+    mutation stopPod($input: PodStopInput!) {
+      podStop(input: $input) {
+        id
+      }
+    }
+  `,
+};