new actions

Author: michelle0927

components/mistral_ai/actions/download-batch-job-results/download-batch-job-results.mjs

@@ -0,0 +1,50 @@
+import mistralAI from "../../mistral_ai.app.mjs";
+import fs from "fs";
+
+export default {
+  key: "mistral_ai-download-batch-job-results",
+  name: "Download Batch Job Results",
+  description: "Download a batch job results file to the /tmp direcory. [See the Documentation](https://docs.mistral.ai/api/#tag/files/operation/files_api_routes_download_file)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    mistralAI,
+    fileId: {
+      propDefinition: [
+        mistralAI,
+        "fileIds",
+        () => ({
+          sampleType: "batch_result",
+        }),
+      ],
+      type: "string",
+      label: "File ID",
+      description: "The identifier of a batch result file to download",
+    },
+    filename: {
+      type: "string",
+      label: "File Name",
+      description: "The filename to save the results file in the /tmp directory",
+    },
+  },
+  async run({ $ }) {
+    const response = await this.mistralAI.downloadFile({
+      $,
+      fileId: this.fileId,
+      responseType: "arraybuffer",
+    });
+
+    const rawcontent = response.toString("base64");
+    const buffer = Buffer.from(rawcontent, "base64");
+    const filename = this.filename;
+    const filePath = `/tmp/${filename}`;
+    fs.writeFileSync(filePath, buffer);
+
+    $.export("$summary", "Successfully downloaded batch results file");
+
+    return [
+      filename,
+      filePath,
+    ];
+  },
+};

components/mistral_ai/actions/get-batch-job-details/get-batch-job-details.mjs

@@ -0,0 +1,28 @@
+import mistralAI from "../../mistral_ai.app.mjs";
+
+export default {
+  key: "mistral_ai-get-batch-job-details",
+  name: "Get Batch Job Details",
+  description: "Get the details of a batch job by its ID. [See the Documentation](https://docs.mistral.ai/api/#tag/batch/operation/jobs_api_routes_batch_get_batch_job)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    mistralAI,
+    batchJobId: {
+      propDefinition: [
+        mistralAI,
+        "batchJobId",
+      ],
+    },
+  },
+  async run({ $ }) {
+    const response = await this.mistralAI.getBatchJob({
+      $,
+      jobId: this.batchJobId,
+    });
+    if (response?.id) {
+      $.export("$summary", `Successfully retrieved details for batch job with ID: ${this.batchJobId}`);
+    }
+    return response;
+  },
+};

components/mistral_ai/actions/upload-file/upload-file.mjs

@@ -0,0 +1,59 @@
+import mistralAI from "../../mistral_ai.app.mjs";
+import { ConfigurationError } from "@pipedream/platform";
+import fs from "fs";
+import FormData from "form-data";
+
+export default {
+  key: "mistral_ai-upload-file",
+  name: "Upload File",
+  description: "Upload a file that can be used across various endpoints. [See the Documentation](https://docs.mistral.ai/api/#tag/files/operation/files_api_routes_upload_file)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    mistralAI,
+    filePath: {
+      type: "string",
+      label: "File Path",
+      description: "The path to a file in the `/tmp` directory. The size of individual files can be a maximum of 512 MB. The Fine-tuning API only supports .jsonl files. [See the Pipedream documentation on working with files](https://pipedream.com/docs/code/nodejs/working-with-files/#writing-a-file-to-tmp)",
+    },
+    purpose: {
+      type: "string",
+      label: "Purpose",
+      description: "The purpose of the file",
+      options: [
+        "fine-tune",
+        "batch",
+        "ocr",
+      ],
+      optional: true,
+    },
+  },
+  async run({ $ }) {
+    const filePath = this.filePath.startsWith("/tmp/")
+      ? this.filePath
+      : `/tmp/${this.filePath}`;
+
+    if (!fs.existsSync(filePath)) {
+      throw new ConfigurationError(`File \`${filePath}\` not found`);
+    }
+
+    const fileContent = fs.createReadStream(filePath);
+    const form = new FormData();
+    form.append("file", fileContent);
+    if (this.purpose) {
+      form.append("purpose", this.purpose);
+    }
+
+    const response = await this.mistralAI.uploadFile({
+      $,
+      data: form,
+      headers: form.getHeaders(),
+    });
+
+    if (response?.filename) {
+      $.export("$summary", `Successfully uploaded file: ${response.filename}`);
+    }
+
+    return response;
+  },
+};

components/mistral_ai/mistral_ai.app.mjs

@@ -9,11 +9,14 @@ export default {
       type: "string[]",
       label: "File IDs",
       description: "Array of input file UUIDs for batch processing",
-      async options({ page }) {
+      async options({
+        page, sampleType,
+      }) {
         const { data } = await this.listFiles({
           params: {
             page,
             page_size: constants.DEFAULT_PAGE_SIZE,
+            sample_type: sampleType,
           },
         });
         return data?.map(({
@@ -38,6 +41,20 @@ export default {
         })) || [];
       },
     },
+    batchJobId: {
+      type: "string",
+      label: "Batch Job ID",
+      description: "The identifier of the batch job to retrieve",
+      async options({ page }) {
+        const { data } = await this.listBatchJobs({
+          params: {
+            page,
+            page_size: constants.DEFAULT_PAGE_SIZE,
+          },
+        });
+        return data?.map(({ id }) => id) || [];
+      },
+    },
   },
   methods: {
     _baseUrl() {
@@ -46,6 +63,7 @@ export default {
     _makeRequest({
       $ = this,
       path,
+      headers,
       ...otherOpts
     }) {
       return axios($, {
@@ -54,6 +72,7 @@ export default {
         headers: {
           "Authorization": `Bearer ${this.$auth.api_key}`,
           "Content-Type": "application/json",
+          ...headers,
         },
       });
     },
@@ -75,6 +94,29 @@ export default {
         ...opts,
       });
     },
+    getBatchJobDetails({
+      jobId, ...opts
+    }) {
+      return this._makeRequest({
+        path: `/batch/jobs/${jobId}`,
+        ...opts,
+      });
+    },
+    downloadFile({
+      fileId, ...opts
+    }) {
+      return this._makeRequest({
+        path: `/files/${fileId}/content`,
+        ...opts,
+      });
+    },
+    uploadFile(opts = {}) {
+      return this._makeRequest({
+        method: "POST",
+        path: "/files",
+        ...opts,
+      });
+    },
     createEmbeddings(opts = {}) {
       return this._makeRequest({
         method: "POST",

components/mistral_ai/package.json

@@ -13,6 +13,7 @@
     "access": "public"
   },
   "dependencies": {
-    "@pipedream/platform": "^3.0.3"
+    "@pipedream/platform": "^3.0.3",
+    "form-data": "^4.0.2"
   }
 }