Fixup types

Author: aron

example.js

@@ -0,0 +1,34 @@
+const Replicate = require("./");
+const replicate = new Replicate({ useFileOutput: true });
+
+(async function () {
+  const input = {
+    prompt:
+      'black forest gateau cake spelling out the words "FLUX SCHNELL", tasty, food photography, dynamic shot',
+    go_fast: true,
+    megapixels: "1",
+    num_outputs: 1,
+    aspect_ratio: "1:1",
+    output_format: "webp",
+    output_quality: 80,
+  };
+
+  const start = performance.now();
+  const output = await replicate.run("black-forest-labs/flux-schnell", {
+    wait: 0.0001,
+    input,
+  });
+
+  const blob = await output[0].blob();
+  console.log(
+    "output",
+    blob,
+    "url",
+    output[0].url(),
+    "duration",
+    performance.now() - start
+  );
+})().catch((err) => {
+  console.error(err);
+  process.exit(1);
+});

index.d.ts

@@ -93,9 +93,9 @@ declare module "replicate" {
     model: string;
     version: string;
     input: object;
-    output?: any;
+    output?: any; // TODO: this should be `unknown`
     source: "api" | "web";
-    error?: any;
+    error?: unknown;
     logs?: string;
     metrics?: {
       predict_time?: number;
@@ -156,13 +156,13 @@ declare module "replicate" {
       identifier: `${string}/${string}` | `${string}/${string}:${string}`,
       options: {
         input: object;
-        wait?: boolean | number | { mode?: "poll"; interval?: number };
+        wait?: boolean | number | { interval?: number };
         webhook?: string;
         webhook_events_filter?: WebhookEventType[];
         signal?: AbortSignal;
       },
       progress?: (prediction: Prediction) => void
-    ): Promise<Prediction | any>;
+    ): Promise<unknown>;
 
     stream(
       identifier: `${string}/${string}` | `${string}/${string}:${string}`,
@@ -217,7 +217,7 @@ declare module "replicate" {
             webhook_events_filter?: WebhookEventType[];
             wait?: boolean | number | { mode?: "poll"; interval?: number };
           }
-        ): Promise<Prediction | any>;
+        ): Promise<Prediction>;
       };
       get(
         deployment_owner: string,
@@ -306,7 +306,7 @@ declare module "replicate" {
           webhook_events_filter?: WebhookEventType[];
           wait?: boolean | number | { mode?: "poll"; interval?: number };
         } & ({ version: string } | { model: string })
-      ): Promise<Prediction | any>;
+      ): Promise<Prediction>;
       get(prediction_id: string): Promise<Prediction>;
       cancel(prediction_id: string): Promise<Prediction>;
       list(): Promise<Page<Prediction>>;