Update deployments.createPrediction to use wait

A first draft of this interface used `block`, but we ended up going with `wait`
(as either a boolean or a number) for the predictions.createPrediction method.
This commit brings the two implementations inline, and deprecates the `block`
parameter.

Author: evilstreak

lib/deployments.js

@@ -9,11 +9,12 @@ const { transformFileInputs } = require("./util");
  * @param {object} options.input - Required. An object with the model inputs
  * @param {string} [options.webhook] - An HTTPS URL for receiving a webhook when the prediction has new output
  * @param {string[]} [options.webhook_events_filter] - You can change which events trigger webhook requests by specifying webhook events (`start`|`output`|`logs`|`completed`)
- * @param {boolean} [options.block] - Whether to wait until the prediction is completed before returning. Defaults to false
+ * @param {boolean|integer} [options.wait] - Whether to wait until the prediction is completed before returning. If an integer is provided, it will wait for that many seconds. Defaults to false
+ * @param {boolean} [options.block] - DEPRECATED: Whether to wait until the prediction is completed before returning. Defaults to false
  * @returns {Promise<object>} Resolves with the created prediction data
  */
 async function createPrediction(deployment_owner, deployment_name, options) {
-  const { input, block, ...data } = options;
+  const { input, wait, block, ...data } = options;
 
   if (data.webhook) {
     try {
@@ -25,7 +26,14 @@ async function createPrediction(deployment_owner, deployment_name, options) {
   }
 
   const headers = {};
-  if (block) {
+  if (wait) {
+    if (typeof wait === "number") {
+      const n = Math.max(1, Math.ceil(Number(wait)) || 1);
+      headers["Prefer"] = `wait=${n}`;
+    } else {
+      headers["Prefer"] = "wait";
+    }
+  } else if (block) {
     headers["Prefer"] = "wait";
   }