Formatting

Author: mattt

lib/deployments.js

@@ -13,7 +13,7 @@ const { transformFileInputs } = require("./util");
  * @returns {Promise<object>} Resolves with the created prediction data
  */
 async function createPrediction(deployment_owner, deployment_name, options) {
-  const { stream, input: _input, ...data } = options;
+  const { stream, input, ...data } = options;
 
   if (data.webhook) {
     try {
@@ -24,14 +24,15 @@ async function createPrediction(deployment_owner, deployment_name, options) {
     }
   }
 
-  // Transform any file looking fields into strings (either data uri or urls).
-  const input = await transformFileInputs(_input);
-
   const response = await this.request(
     `/deployments/${deployment_owner}/${deployment_name}/predictions`,
     {
       method: "POST",
-      data: { ...data, input, stream },
+      data: {
+        ...data,
+        input: await transformFileInputs(input),
+        stream,
+      },
     }
   );
 

lib/predictions.js

@@ -13,7 +13,7 @@ const { transformFileInputs } = require("./util");
  * @returns {Promise<object>} Resolves with the created prediction
  */
 async function createPrediction(options) {
-  const { model, version, stream, input: _input, ...data } = options;
+  const { model, version, stream, input, ...data } = options;
 
   if (data.webhook) {
     try {
@@ -24,19 +24,25 @@ async function createPrediction(options) {
     }
   }
 
-  // Transform any file looking fields into strings (either data uri or urls).
-  const input = await transformFileInputs(_input);
-
   let response;
   if (version) {
     response = await this.request("/predictions", {
       method: "POST",
-      data: { ...data, stream, input, version },
+      data: {
+        ...data,
+        input: await transformFileInputs(input),
+        version,
+        stream,
+      },
     });
   } else if (model) {
     response = await this.request(`/models/${model}/predictions`, {
       method: "POST",
-      data: { ...data, stream, input },
+      data: {
+        ...data,
+        input: await transformFileInputs(input),
+        stream,
+      },
     });
   } else {
     throw new Error("Either model or version must be specified");