Add cancel prediction endpoint.

replicate · Feb 22, 2024 · 0e52d08 · 0e52d08
1 parent b0335b6
commit 0e52d08
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/prediction.go b/prediction.go
@@ -141,3 +141,12 @@ func (r *Client) GetPrediction(ctx context.Context, id string) (*Prediction, err
 	}
 	return prediction, nil
 }
+
+func (r *Client) CancelPrediction(ctx context.Context, id string) (*Prediction, error) {
+	prediction := &Prediction{}
+	err := r.fetch(ctx, "POST", fmt.Sprintf("/predictions/%s/cancel", id), nil, prediction)
+	if err != nil {
+		return nil, fmt.Errorf("failed to cancel prediction: %w", err)
+	}
+	return prediction, nil
+}