Apply automatic formatting

Author: mattt

index.d.ts

@@ -1,7 +1,7 @@
-declare module 'replicate' {
-  type Status = 'starting' | 'processing' | 'succeeded' | 'failed' | 'canceled';
-  type Visibility = 'public' | 'private';
-  type WebhookEventType = 'start' | 'output' | 'logs' | 'completed';
+declare module "replicate" {
+  type Status = "starting" | "processing" | "succeeded" | "failed" | "canceled";
+  type Visibility = "public" | "private";
+  type WebhookEventType = "start" | "output" | "logs" | "completed";
 
   export interface ApiError extends Error {
     request: Request;
@@ -17,15 +17,15 @@ declare module 'replicate' {
 
   export interface Hardware {
     sku: string;
-    name: string
+    name: string;
   }
 
   export interface Model {
     url: string;
     owner: string;
     name: string;
     description?: string;
-    visibility: 'public' | 'private';
+    visibility: "public" | "private";
     github_url?: string;
     paper_url?: string;
     license_url?: string;
@@ -49,7 +49,7 @@ declare module 'replicate' {
     version: string;
     input: object;
     output?: any;
-    source: 'api' | 'web';
+    source: "api" | "web";
     error?: any;
     logs?: string;
     metrics?: {
@@ -80,13 +80,16 @@ declare module 'replicate' {
       auth?: string;
       userAgent?: string;
       baseUrl?: string;
-      fetch?: (input: Request | string, init?: RequestInit) => Promise<Response>
+      fetch?: (
+        input: Request | string,
+        init?: RequestInit
+      ) => Promise<Response>;
     });
 
     auth: string;
     userAgent?: string;
     baseUrl?: string;
-    fetch: (input: Request | string, init?: RequestInit) => Promise<Response>
+    fetch: (input: Request | string, init?: RequestInit) => Promise<Response>;
 
     run(
       identifier: `${string}/${string}` | `${string}/${string}:${string}`,
@@ -100,14 +103,17 @@ declare module 'replicate' {
       progress?: (prediction: Prediction) => void
     ): Promise<object>;
 
-    request(route: string | URL, options: {
-      method?: string;
-      headers?: object | Headers;
-      params?: object;
-      data?: object;
-    }): Promise<Response>;
+    request(
+      route: string | URL,
+      options: {
+        method?: string;
+        headers?: object | Headers;
+        params?: object;
+        data?: object;
+      }
+    ): Promise<Response>;
 
-    paginate<T>(endpoint: () => Promise<Page<T>>): AsyncGenerator<[ T ]>;
+    paginate<T>(endpoint: () => Promise<Page<T>>): AsyncGenerator<[T]>;
 
     wait(
       prediction: Prediction,
@@ -138,8 +144,8 @@ declare module 'replicate' {
     };
 
     hardware: {
-      list(): Promise<Hardware[]>
-    }
+      list(): Promise<Hardware[]>;
+    };
 
     models: {
       get(model_owner: string, model_name: string): Promise<Model>;
@@ -155,7 +161,8 @@ declare module 'replicate' {
           paper_url?: string;
           license_url?: string;
           cover_image_url?: string;
-        }): Promise<Model>;
+        }
+      ): Promise<Model>;
       versions: {
         list(model_owner: string, model_name: string): Promise<ModelVersion[]>;
         get(

index.js

@@ -1,15 +1,15 @@
-const ApiError = require('./lib/error');
-const ModelVersionIdentifier = require('./lib/identifier');
-const { withAutomaticRetries } = require('./lib/util');
+const ApiError = require("./lib/error");
+const ModelVersionIdentifier = require("./lib/identifier");
+const { withAutomaticRetries } = require("./lib/util");
 
-const collections = require('./lib/collections');
-const deployments = require('./lib/deployments');
-const hardware = require('./lib/hardware');
-const models = require('./lib/models');
-const predictions = require('./lib/predictions');
-const trainings = require('./lib/trainings');
+const collections = require("./lib/collections");
+const deployments = require("./lib/deployments");
+const hardware = require("./lib/hardware");
+const models = require("./lib/models");
+const predictions = require("./lib/predictions");
+const trainings = require("./lib/trainings");
 
-const packageJSON = require('./package.json');
+const packageJSON = require("./package.json");
 
 /**
  * Replicate API client library
@@ -43,7 +43,7 @@ class Replicate {
     this.auth = options.auth || process.env.REPLICATE_API_TOKEN;
     this.userAgent =
       options.userAgent || `replicate-javascript/${packageJSON.version}`;
-    this.baseUrl = options.baseUrl || 'https://api.replicate.com/v1';
+    this.baseUrl = options.baseUrl || "https://api.replicate.com/v1";
     this.fetch = options.fetch || globalThis.fetch;
 
     this.collections = {
@@ -54,7 +54,7 @@ class Replicate {
     this.deployments = {
       predictions: {
         create: deployments.predictions.create.bind(this),
-      }
+      },
     };
 
     this.hardware = {
@@ -131,26 +131,30 @@ class Replicate {
 
     const { signal } = options;
 
-    prediction = await this.wait(prediction, wait || {}, async (updatedPrediction) => {
-      // Call progress callback with the updated prediction object
-      if (progress) {
-        progress(updatedPrediction);
-      }
-
-      if (signal && signal.aborted) {
-        await this.predictions.cancel(updatedPrediction.id);
-        return true; // stop polling
+    prediction = await this.wait(
+      prediction,
+      wait || {},
+      async (updatedPrediction) => {
+        // Call progress callback with the updated prediction object
+        if (progress) {
+          progress(updatedPrediction);
+        }
+
+        if (signal && signal.aborted) {
+          await this.predictions.cancel(updatedPrediction.id);
+          return true; // stop polling
+        }
+
+        return false; // continue polling
       }
-
-      return false; // continue polling
-    });
+    );
 
     // Call progress callback with the completed prediction object
     if (progress) {
       progress(prediction);
     }
 
-    if (prediction.status === 'failed') {
+    if (prediction.status === "failed") {
       throw new Error(`Prediction failed: ${prediction.error}`);
     }
 
@@ -177,27 +181,23 @@ class Replicate {
       url = route;
     } else {
       url = new URL(
-        route.startsWith('/') ? route.slice(1) : route,
-        baseUrl.endsWith('/') ? baseUrl : `${baseUrl}/`
+        route.startsWith("/") ? route.slice(1) : route,
+        baseUrl.endsWith("/") ? baseUrl : `${baseUrl}/`
       );
     }
 
-    const {
-      method = 'GET',
-      params = {},
-      data,
-    } = options;
+    const { method = "GET", params = {}, data } = options;
 
     for (const [key, value] of Object.entries(params)) {
       url.searchParams.append(key, value);
     }
 
     const headers = new Headers();
     if (auth) {
-      headers.append('Authorization', `Token ${auth}`);
+      headers.append("Authorization", `Token ${auth}`);
     }
-    headers.append('Content-Type', 'application/json');
-    headers.append('User-Agent', userAgent);
+    headers.append("Content-Type", "application/json");
+    headers.append("User-Agent", userAgent);
     if (options.headers) {
       for (const [key, value] of options.headers.entries()) {
         headers.append(key, value);
@@ -210,22 +210,25 @@ class Replicate {
       body: data ? JSON.stringify(data) : undefined,
     };
 
-    const shouldRetry = method === 'GET' ?
-      (response) => (response.status === 429 || response.status >= 500) :
-      (response) => (response.status === 429);
+    const shouldRetry =
+      method === "GET"
+        ? (response) => response.status === 429 || response.status >= 500
+        : (response) => response.status === 429;
 
     // Workaround to fix `TypeError: Illegal invocation` error in Cloudflare Workers
     // https://github.com/replicate/replicate-javascript/issues/134
     const _fetch = this.fetch; // eslint-disable-line no-underscore-dangle
-    const response = await withAutomaticRetries(async () => _fetch(url, init), { shouldRetry });
+    const response = await withAutomaticRetries(async () => _fetch(url, init), {
+      shouldRetry,
+    });
 
     if (!response.ok) {
       const request = new Request(url, init);
       const responseText = await response.text();
       throw new ApiError(
         `Request to ${url} failed with status ${response.status} ${response.statusText}: ${responseText}.`,
         request,
-        response,
+        response
       );
     }
 
@@ -243,11 +246,12 @@ class Replicate {
    * @param {Function} endpoint - Function that returns a promise for the next page of results
    * @yields {object[]} Each page of results
    */
-  async * paginate(endpoint) {
+  async *paginate(endpoint) {
     const response = await endpoint();
     yield response.results;
     if (response.next) {
-      const nextPage = () => this.request(response.next, { method: 'GET' }).then((r) => r.json());
+      const nextPage = () =>
+        this.request(response.next, { method: "GET" }).then((r) => r.json());
       yield* this.paginate(nextPage);
     }
   }
@@ -271,13 +275,13 @@ class Replicate {
   async wait(prediction, options, stop) {
     const { id } = prediction;
     if (!id) {
-      throw new Error('Invalid prediction');
+      throw new Error("Invalid prediction");
     }
 
     if (
-      prediction.status === 'succeeded' ||
-      prediction.status === 'failed' ||
-      prediction.status === 'canceled'
+      prediction.status === "succeeded" ||
+      prediction.status === "failed" ||
+      prediction.status === "canceled"
     ) {
       return prediction;
     }
@@ -290,12 +294,12 @@ class Replicate {
     let updatedPrediction = await this.predictions.get(id);
 
     while (
-      updatedPrediction.status !== 'succeeded' &&
-      updatedPrediction.status !== 'failed' &&
-      updatedPrediction.status !== 'canceled'
+      updatedPrediction.status !== "succeeded" &&
+      updatedPrediction.status !== "failed" &&
+      updatedPrediction.status !== "canceled"
     ) {
       /* eslint-disable no-await-in-loop */
-      if (stop && await stop(updatedPrediction) === true) {
+      if (stop && (await stop(updatedPrediction)) === true) {
         break;
       }
 
@@ -304,7 +308,7 @@ class Replicate {
       /* eslint-enable no-await-in-loop */
     }
 
-    if (updatedPrediction.status === 'failed') {
+    if (updatedPrediction.status === "failed") {
       throw new Error(`Prediction failed: ${updatedPrediction.error}`);
     }