replicate-client 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f6798a521201c15a08cd456167052ac6de82b43f60487ea454dabda1865a28ce
-  data.tar.gz: cd44ea8758209f771645f2d9755438d6ad890758f83ef2ce8dd03d39f1162d4e
+  metadata.gz: 61acc30d7e6406c1aa6f797a45ab6927945d399f3ce49934a2ada361a818df18
+  data.tar.gz: 6e0023bde6ce5e0897314e3f1bde97e1b7d99b9dd1de776568a883822852a874
 SHA512:
-  metadata.gz: 7b0f59e6ae5803a0cefc0ec8a0d2312cf45b4205a7052cfae84bc70b13c062ebd533fed18c92971513139235418cb1a08662eff816cfa211fdc854451133df29
-  data.tar.gz: 18e5cb1d7eb54ceddb10eac736da1e56baa8245eb558b4670da2fcd4e99a1d5519bc473313bee0bb5ba7a191cf1f3aeeafccbedea24a202a474df6a716684dba
+  metadata.gz: ee9a5f6e05fbf0b8af2a34c4890dae5ec925d47b5ed0c9afceb866e07c1be71a3d2c50f98c1ca569b84bb03d17abc2f665e48a6927b0243fab2180ff2f43df44
+  data.tar.gz: 692003bec1af63306d896535c434b888ce9162431839809ae140b811a5c40e9323027143096b5e8ea0a1270fd79f6f23a4b544ed33d2090f43f4d7c0d2d449c6

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    replicate-client (0.1.6)
+    replicate-client (0.1.7)
       faraday (>= 1)
 
 GEM

data/lib/replicate-client/client.rb

@@ -15,13 +15,15 @@ module ReplicateClient
     #
     # @param path [String] The path to the API endpoint.
     # @param payload [Hash] The payload to send to the API.
+    # @param headers [Hash] The headers to send to the API.
     #
     # @return [Hash] The response from the API.
-    def post(path, payload)
+    def post(path, payload, headers: {})
       response = connection.post(build_url(path)) do |request|
         request.headers["Authorization"] = "Bearer #{@configuration.access_token}"
         request.headers["Content-Type"] = "application/json"
         request.headers["Accept"] = "application/json"
+        request.headers.merge!(headers)
         request.body = payload.compact.to_json
       end
 

data/lib/replicate-client/prediction.rb

@@ -19,9 +19,10 @@ module ReplicateClient
       # @param input [Hash] The input data for the prediction.
       # @param webhook_url [String] The URL to send webhook events to.
       # @param webhook_events_filter [Array<Symbol>] The events to send to the webhook.
+      # @param sync [Boolean] Whether to wait for the prediction to complete.
       #
       # @return [ReplicateClient::Prediction]
-      def create!(version:, input:, webhook_url: nil, webhook_events_filter: nil)
+      def create!(version:, input:, webhook_url: nil, webhook_events_filter: nil, sync: false)
         args = {
           version: version.is_a?(Model::Version) ? version.id : version,
           input: input,
@@ -29,7 +30,9 @@ module ReplicateClient
           webhook_events_filter: webhook_events_filter&.map(&:to_s)
         }
 
-        prediction = ReplicateClient.client.post(INDEX_PATH, args)
+        headers = sync ? { "Prefer" => 'wait' } : {}
+
+        prediction = ReplicateClient.client.post(INDEX_PATH, args, headers:)
 
         new(prediction)
       end
@@ -40,16 +43,19 @@ module ReplicateClient
       # @param input [Hash] The input data for the prediction.
       # @param webhook_url [String] The URL to send webhook events to.
       # @param webhook_events_filter [Array<Symbol>] The events to send to the webhook.
+      # @param sync [Boolean] Whether to wait for the prediction to complete.
       #
       # @return [ReplicateClient::Prediction]
-      def create_for_deployment!(deployment:, input:, webhook_url: nil, webhook_events_filter: nil)
+      def create_for_deployment!(deployment:, input:, webhook_url: nil, webhook_events_filter: nil, sync: false)
         args = {
           input: input,
           webhook: webhook_url || ReplicateClient.configuration.webhook_url,
           webhook_events_filter: webhook_events_filter&.map(&:to_s)
         }
 
-        prediction = ReplicateClient.client.post("#{deployment.path}#{INDEX_PATH}", args)
+        headers = sync ? { "Prefer" => 'wait' } : {}
+
+        prediction = ReplicateClient.client.post("#{deployment.path}#{INDEX_PATH}", args, headers:)
 
         new(prediction)
       end
@@ -60,9 +66,10 @@ module ReplicateClient
       # @param input [Hash] The input data for the prediction.
       # @param webhook_url [String] The URL to send webhook events to.
       # @param webhook_events_filter [Array<Symbol>] The events to send to the webhook.
+      # @param sync [Boolean] Whether to wait for the prediction to complete.
       #
       # @return [ReplicateClient::Prediction]
-      def create_for_official_model!(model:, input:, webhook_url: nil, webhook_events_filter: nil)
+      def create_for_official_model!(model:, input:, webhook_url: nil, webhook_events_filter: nil, sync: false)
         model_path = model.is_a?(Model) ? model.path : Model.build_path(**Model.parse_model_name(model))
 
         args = {
@@ -71,7 +78,9 @@ module ReplicateClient
           webhook_events_filter: webhook_events_filter&.map(&:to_s)
         }
 
-        prediction = ReplicateClient.client.post("#{model_path}#{INDEX_PATH}", args)
+        headers = sync ? { "Prefer" => 'wait' } : {}
+
+        prediction = ReplicateClient.client.post("#{model_path}#{INDEX_PATH}", args, headers:)
 
         new(prediction)
       end

data/lib/replicate-client/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ReplicateClient
-  VERSION = "0.1.7"
+  VERSION = "0.1.8"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: replicate-client
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: ruby
 authors:
 - Dylan Player
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-09-02 00:00:00.000000000 Z
+date: 2025-01-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday