replicate-client 0.1.7 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f6798a521201c15a08cd456167052ac6de82b43f60487ea454dabda1865a28ce
-  data.tar.gz: cd44ea8758209f771645f2d9755438d6ad890758f83ef2ce8dd03d39f1162d4e
+  metadata.gz: e9834788e800bcbdd19fe6e813b8c19c30b4d62ca0fe5035fe05d32b95126ac7
+  data.tar.gz: da4c94e22f0f3e884a690ab7c4d4701c812b17ec304efd2e8e956ea557aebf9f
 SHA512:
-  metadata.gz: 7b0f59e6ae5803a0cefc0ec8a0d2312cf45b4205a7052cfae84bc70b13c062ebd533fed18c92971513139235418cb1a08662eff816cfa211fdc854451133df29
-  data.tar.gz: 18e5cb1d7eb54ceddb10eac736da1e56baa8245eb558b4670da2fcd4e99a1d5519bc473313bee0bb5ba7a191cf1f3aeeafccbedea24a202a474df6a716684dba
+  metadata.gz: 60b067804aae73acf3bdf193936fa504bc748f023b3a55d2a168fc3014fd8a920b3b6a276ef2f5fa2c76d51c57ad2354cb324bd2e6495d7ba95b7d7283ab39e2
+  data.tar.gz: e8bba1499966efe2ec8deb92bcf9cc352368586b39f06309a151dbfb18d3f6de85ed69c36be86340d18b820989f36efde8666cc2d14718026713cd9c00fc9437

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    replicate-client (0.1.6)
+    replicate-client (0.1.8)
       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,21 @@ 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" } : {}
+
+        deployment_path = deployment.is_a?(Deployment) ? deployment.path : "#{Deployment::INDEX_PATH}/#{deployment}"
+
+        prediction = ReplicateClient.client.post("#{deployment_path}#{INDEX_PATH}", args, headers:)
 
         new(prediction)
       end
@@ -60,9 +68,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 +80,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.9"
 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.9
 platform: ruby
 authors:
 - Dylan Player
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-09-02 00:00:00.000000000 Z
+date: 2025-01-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday