Call the onProgress handler with the canceled prediction

Previously when aborting a `run()` request we were dropping the final
canceled prediction object and calling the `onProgress` callback with a
stale "processing" object.

Author: aron

index.js

@@ -162,15 +162,19 @@ class Replicate {
           progress(updatedPrediction);
         }
 
-        if (signal && signal.aborted) {
-          await this.predictions.cancel(updatedPrediction.id);
+        // We handle the cancel later in the function.
+        if (signal?.aborted) {
           return true; // stop polling
         }
 
         return false; // continue polling
       }
     );
 
+    if (signal?.aborted) {
+      prediction = await this.predictions.cancel(prediction.id);
+    }
+
     // Call progress callback with the completed prediction object
     if (progress) {
       progress(prediction);

index.test.ts

@@ -1258,19 +1258,40 @@ describe("Replicate client", () => {
           status: "canceled",
         });
 
-      await client.run(
+      const onProgress = jest.fn();
+      const output = await client.run(
         "owner/model:5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa",
         {
           input: { text: "Hello, world!" },
           signal,
-        }
+        },
+        onProgress
       );
 
       expect(body).toBeDefined();
       expect(body?.["signal"]).toBeUndefined();
       expect(signal.aborted).toBe(true);
       expect(output).toBeUndefined();
 
+      expect(onProgress).toHaveBeenNthCalledWith(
+        1,
+        expect.objectContaining({
+          status: "processing",
+        })
+      );
+      expect(onProgress).toHaveBeenNthCalledWith(
+        2,
+        expect.objectContaining({
+          status: "processing",
+        })
+      );
+      expect(onProgress).toHaveBeenNthCalledWith(
+        3,
+        expect.objectContaining({
+          status: "canceled",
+        })
+      );
+
       scope.done();
     });
   });