fix: Fixed.

Author: HavenDV

src/libs/Replicate/PredictionResponseExtensions.cs

@@ -11,11 +11,14 @@ public static class PredictionResponseExtensions
     /// <param name="response"></param>
     /// <returns></returns>
     /// <exception cref="ArgumentNullException"></exception>
-    public static bool IsSuccessful(this PredictionResponse response)
+    public static bool IsCompleted(this SchemasPredictionResponse response)
     {
         response = response ?? throw new ArgumentNullException(nameof(response));
 
-        return response.Status == "succeeded";
+        return response.Status is 
+            SchemasPredictionResponseStatus.Succeeded or
+            SchemasPredictionResponseStatus.Canceled or
+            SchemasPredictionResponseStatus.Failed;
     }
 
     /// <summary>
@@ -27,22 +30,23 @@ public static bool IsSuccessful(this PredictionResponse response)
     /// <returns></returns>
     /// <exception cref="ArgumentNullException"></exception>
     /// <exception cref="ArgumentException"></exception>
-    public static async Task<PredictionResponse> WaitUntilSuccessfulAsync(
-        this PredictionResponse response,
+    public static async Task<PredictionResponse?> WaitUntilSuccessfulAsync(
+        this SchemasPredictionResponse response,
         ReplicateApi api,
         CancellationToken cancellationToken = default)
     {
         response = response ?? throw new ArgumentNullException(nameof(response));
         api = api ?? throw new ArgumentNullException(nameof(api));
         var id = response.Id ?? throw new ArgumentException(nameof(response.Id));
 
-        while (!response.IsSuccessful())
+        PredictionResponse? predictionResponse = null;
+        while (!response.IsCompleted())
         {
             await Task.Delay(1000, cancellationToken).ConfigureAwait(false);
 
-            response = await api.PredictionsGetAsync(id, cancellationToken).ConfigureAwait(false);
+            predictionResponse = await api.PredictionsGetAsync(id, cancellationToken).ConfigureAwait(false);
         }
 
-        return response;
+        return predictionResponse;
     }
 }

src/tests/Replicate.IntegrationTests/Tests.Account.cs

@@ -10,6 +10,6 @@ public async Task Account()
         var response = await api.AccountGetAsync();
 
         response.Should().NotBeNull();
-        response.Type.HasValue.Should().BeTrue();
+        response.Type.Should().Be(AccountGetResponseType.User);
     }
 }

src/tests/Replicate.IntegrationTests/Tests.CreatePredictionForFluxPro.cs

@@ -28,9 +28,9 @@ public async Task CreatePredictionForFluxPro()
 
         var endResponse = await response.WaitUntilSuccessfulAsync(api);
 
-        Console.WriteLine($"Seed: {endResponse.Input?.Seed}.");
+        Console.WriteLine($"Seed: {endResponse?.Input?.Seed}.");
         Console.WriteLine("Image available at:");
-        Console.WriteLine(endResponse.Output);
+        Console.WriteLine(endResponse?.Output);
 
         //// ![output](https://raw.githubusercontent.com/tryAGI/Replicate/dd1e5c2cbebc53e9b343f1372e5a660159e79ef3/assets/output.webp)
     }