Automatically upload any files and replace them with URLs

Author: mattt

lib/predictions.js

@@ -1,3 +1,5 @@
+const { createFile } = require("./files");
+
 /**
  * Create a new prediction
  *
@@ -11,7 +13,7 @@
  * @returns {Promise<object>} Resolves with the created prediction
  */
 async function createPrediction(options) {
-  const { model, version, stream, ...data } = options;
+  const { model, version, stream, input, ...data } = options;
 
   if (data.webhook) {
     try {
@@ -22,16 +24,27 @@ async function createPrediction(options) {
     }
   }
 
+  // Automatically upload any files and replace them with URLs
+  if (typeof input === "object" && input !== null) {
+    for (const key in input) {
+      if (input[key] instanceof File || input[key] instanceof Blob) {
+        input[key] = createFile(input[key]).then((file) => file.urls.get);
+      }
+    }
+  }
+  // Wait for all file uploads to complete
+  await Promise.all(Object.values(input));
+
   let response;
   if (version) {
     response = await this.request("/predictions", {
       method: "POST",
-      data: { ...data, stream, version },
+      data: { ...data, input, stream, version },
     });
   } else if (model) {
     response = await this.request(`/models/${model}/predictions`, {
       method: "POST",
-      data: { ...data, stream },
+      data: { ...data, input, stream },
     });
   } else {
     throw new Error("Either model or version must be specified");