replicate-ruby 0.1.4 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c4b445afdad565b11f6456385d9c947b20fc257330f1c81454780cfa0a95b960
-  data.tar.gz: 70a6fc73b854cffd4b4daff1cae5913d3a783b04163ea31dcf40a5e7ebbbed5f
+  metadata.gz: 1c51346f40acaa0f9e2f0c1ac97630fa2908d5e87d94de2d27196f2b33ec6135
+  data.tar.gz: 7559f35badbfd904058c70f3561d45793b6f3290e2905500902b755d2c45aeb2
 SHA512:
-  metadata.gz: 2eee8935e75ded9fc9c143859d666bf0a39d6c3416609ea5563e2da915c49c62f2dc97129b264a209a59b7b2d8e966f218193e8c27dc72120a06dc3bcc138c17
-  data.tar.gz: 4898ebb685097d47b724210ca4094a420bb94c728b22b18bbec079884010e2a6b20795ca32eba4c98bc31c134636cba2426b9c1f2bfdf40d772414f9963a5b2c
+  metadata.gz: fb6c205ce5e7cd48decf8ea79067fe4eaa6942ee00c50b85a8efb82d549ae2deec969591b8ddee302b424d583d5e20f3d2c9f860e10bde8d51d633b6c9564036
+  data.tar.gz: 6d4255676a72b959ddb9ba3280adf181e6e6bbdd9012ac0085e0173bedb63e97a3458f2e8201dd22854fda4f10967ecb075f428759b67b31fd9550ad689e0fba

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    replicate-ruby (0.1.4)
+    replicate-ruby (0.1.6)
       addressable
       faraday (>= 2.0)
       faraday-retry
@@ -15,7 +15,7 @@ GEM
     coderay (1.1.3)
     crack (0.4.5)
       rexml
-    faraday (2.7.1)
+    faraday (2.7.2)
       faraday-net_http (>= 2.0, < 3.1)
       ruby2_keywords (>= 0.0.4)
     faraday-net_http (3.0.2)

data/lib/replicate/client/prediction.rb

@@ -21,6 +21,7 @@ module Replicate
       # Create a prediction
       # @see https://replicate.com/docs/reference/http#create-prediction
       def create_prediction(params)
+        params[:webhook_completed] ||= webhook_url
         Replicate::Record::Prediction.new(self, post("predictions", params))
       end
 

data/lib/replicate/configurable.rb

@@ -2,14 +2,14 @@
 
 module Replicate
   module Configurable
-    attr_accessor :api_token
+    attr_accessor :api_token, :webhook_url
     attr_writer :api_endpoint
 
     class << self
       # List of configurable keys for {Datatrans::Client}
       # @return [Array] of option keys
       def keys
-        @keys ||= %i[api_token api_endpoint]
+        @keys ||= %i[api_token api_endpoint webhook_url]
       end
     end
 

data/lib/replicate/record/prediction.rb

@@ -10,6 +10,15 @@ module Replicate
       def cancel
         @data = client.cancel_prediction(id).data
       end
+
+      def finished?
+        case status
+        when "succeeded", "failed", "canceled"
+          true
+        else
+          false
+        end
+      end
     end
   end
 end

data/lib/replicate/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Replicate
-  VERSION = "0.1.4"
+  VERSION = "0.1.6"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: replicate-ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.6
 platform: ruby
 authors:
 - Dreaming Tulpa
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-12-07 00:00:00.000000000 Z
+date: 2022-12-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday