replicate-client 0.1.9 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e9834788e800bcbdd19fe6e813b8c19c30b4d62ca0fe5035fe05d32b95126ac7
-  data.tar.gz: da4c94e22f0f3e884a690ab7c4d4701c812b17ec304efd2e8e956ea557aebf9f
+  metadata.gz: fad1f0756031bf4b2d7847c59836b4fcc570ed2354ccd8e771056f2f3d26fa0e
+  data.tar.gz: e8c163b483dba74ce529a8a7af809807706fcdd45d7ae3edaec37124b7259765
 SHA512:
-  metadata.gz: 60b067804aae73acf3bdf193936fa504bc748f023b3a55d2a168fc3014fd8a920b3b6a276ef2f5fa2c76d51c57ad2354cb324bd2e6495d7ba95b7d7283ab39e2
-  data.tar.gz: e8bba1499966efe2ec8deb92bcf9cc352368586b39f06309a151dbfb18d3f6de85ed69c36be86340d18b820989f36efde8666cc2d14718026713cd9c00fc9437
+  metadata.gz: f4292065e85be45396415204d7898b76a4aecc498e44b05354178ef59103adbffc27c453429dd976371ca9c17893b443b119d3faa672a518cdf853e1438a84f6
+  data.tar.gz: 55194df7b411c76e99ff369e4186fe3ed6d8cd25c79a1b2d6a63f53319c7baafde517057e61e7d643f6795679b1ca57e1490fff7dd63fecd1266d53d73493606

data/Gemfile.lock

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

data/README.md

@@ -1,16 +1,14 @@
 # ReplicateClient
 
-**🚧 This gem is still under development 🚧**
-
 ## Installation
 
 Install the gem and add to the application"s Gemfile by executing:
 
-    $ bundle add replicate
+    $ bundle add replicate-client
 
 If bundler is not being used to manage dependencies, install the gem by executing:
 
-    $ gem install replicate
+    $ gem install replicate-client
 
 ## Usage
 
@@ -199,4 +197,3 @@ Official models will not have vesions. The version id will be nil.
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
 
 To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
-

data/Rakefile

@@ -6,7 +6,7 @@ require "rake/testtask"
 Rake::TestTask.new(:test) do |t|
   t.libs << "test"
   t.libs << "lib"
-  t.test_files = FileList["test/**/test_*.rb"]
+  t.test_files = FileList["test/**/*_test.rb"]
 end
 
 require "rubocop/rake_task"

data/lib/replicate-client/client.rb

@@ -2,6 +2,11 @@
 
 module ReplicateClient
   class Client
+    # The configuration for the client.
+    #
+    # @return [ReplicateClient::Configuration]
+    attr_accessor :configuration
+
     # Initialize the client.
     #
     # @param configuration [ReplicateClient::Configuration] The configuration for the client.

data/lib/replicate-client/deployment.rb

@@ -7,17 +7,19 @@ module ReplicateClient
     class << self
       # List all deployments.
       #
+      # @param client [ReplicateClient::Client] The client to use for requests.
+      #
       # @yield [ReplicateClient::Deployment] Yields a deployment.
       #
       # @return [void]
-      def auto_paging_each(&block)
+      def auto_paging_each(client: ReplicateClient.client, &block)
         cursor = nil
 
         loop do
           url_params = cursor ? "?cursor=#{cursor}" : ""
-          attributes = ReplicateClient.client.get("#{INDEX_PATH}#{url_params}")
+          attributes = client.get("#{INDEX_PATH}#{url_params}")
 
-          deployments = attributes["results"].map { |deployment| new(deployment) }
+          deployments = attributes["results"].map { |deployment| new(deployment, client: client) }
 
           deployments.each(&block)
 
@@ -34,9 +36,11 @@ module ReplicateClient
       # @param hardware [ReplicateClient::Hardware, String] The hardware SKU.
       # @param min_instances [Integer] The minimum number of instances.
       # @param max_instances [Integer] The maximum number of instances.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Deployment]
-      def create!(name:, model:, hardware:, min_instances:, max_instances:, version_id: nil)
+      def create!(name:, model:, hardware:, min_instances:, max_instances:, version_id: nil,
+                  client: ReplicateClient.client)
         model_full_name = model.is_a?(Model) ? model.full_name : model
         hardware_sku = hardware.is_a?(Hardware) ? hardware.sku : hardware
         version = if version_id
@@ -44,7 +48,7 @@ module ReplicateClient
                   elsif model.is_a?(Model)
                     model.version_id
                   else
-                    Model.find(model).latest_version.id
+                    Model.find(model, client: client).latest_version.id
                   end
 
         body = {
@@ -56,41 +60,44 @@ module ReplicateClient
           max_instances: max_instances
         }
 
-        attributes = ReplicateClient.client.post(INDEX_PATH, body)
-        new(attributes)
+        attributes = client.post(INDEX_PATH, body)
+        new(attributes, client: client)
       end
 
       # Find a deployment by owner and name.
       #
       # @param full_name [String] The full name of the deployment in "owner/name" format.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Deployment]
-      def find(full_name)
+      def find(full_name, client: ReplicateClient.client)
         path = build_path(**parse_full_name(full_name))
-        attributes = ReplicateClient.client.get(path)
-        new(attributes)
+        attributes = client.get(path)
+        new(attributes, client: client)
       end
 
       # Find a deployment by owner and name.
       #
       # @param owner [String] The owner of the deployment.
       # @param name [String] The name of the deployment.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Deployment]
-      def find_by!(owner:, name:)
+      def find_by!(owner:, name:, client: ReplicateClient.client)
         path = build_path(owner: owner, name: name)
-        attributes = ReplicateClient.client.get(path)
-        new(attributes)
+        attributes = client.get(path)
+        new(attributes, client: client)
       end
 
       # Find a deployment by owner and name.
       #
       # @param owner [String] The owner of the deployment.
       # @param name [String] The name of the deployment.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Deployment, nil]
-      def find_by(owner:, name:)
-        find_by!(owner: owner, name: name)
+      def find_by(owner:, name:, client: ReplicateClient.client)
+        find_by!(owner: owner, name: name, client: client)
       rescue ReplicateClient::NotFoundError
         nil
       end
@@ -99,11 +106,12 @@ module ReplicateClient
       #
       # @param owner [String] The owner of the deployment.
       # @param name [String] The name of the deployment.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [void]
-      def destroy!(owner:, name:)
+      def destroy!(owner:, name:, client: ReplicateClient.client)
         path = build_path(owner: owner, name: name)
-        ReplicateClient.client.delete(path)
+        client.delete(path)
       end
 
       # Build the path for a specific deployment.
@@ -130,12 +138,19 @@ module ReplicateClient
     # Attributes for deployment.
     attr_accessor :owner, :name, :current_release
 
+    # The client used to make API requests for this deployment.
+    #
+    # @return [ReplicateClient::Client]
+    attr_accessor :client
+
     # Initialize a new deployment instance.
     #
     # @param attributes [Hash] The attributes of the deployment.
+    # @param client [ReplicateClient::Client] The client to use for requests.
     #
     # @return [ReplicateClient::Deployment]
-    def initialize(attributes)
+    def initialize(attributes, client: ReplicateClient.client)
+      @client = client
       reset_attributes(attributes)
     end
 
@@ -143,7 +158,7 @@ module ReplicateClient
     #
     # @return [void]
     def destroy!
-      self.class.destroy!(owner: owner, name: name)
+      self.class.destroy!(owner: owner, name: name, client: @client)
     end
 
     # Update the deployment.
@@ -155,8 +170,8 @@ module ReplicateClient
     #
     # @return [void]
     def update!(hardware: nil, min_instances: nil, max_instances: nil, version: nil)
-      version_id = version.is_a?(Version) ? version.id : version
-      path = build_path(owner: owner, name: name)
+      version_id = version.is_a?(ReplicateClient::Model::Version) ? version.id : version
+      path = self.class.build_path(owner: owner, name: name)
       body = {
         hardware: hardware,
         min_instances: min_instances,
@@ -164,7 +179,7 @@ module ReplicateClient
         version: version_id
       }.compact
 
-      attributes = ReplicateClient.client.patch(path, body)
+      attributes = @client.patch(path, body)
       reset_attributes(attributes)
     end
 
@@ -172,7 +187,7 @@ module ReplicateClient
     #
     # @return [void]
     def reload!
-      attributes = ReplicateClient.client.get(path)
+      attributes = @client.get(path)
       reset_attributes(attributes)
     end
 
@@ -195,7 +210,8 @@ module ReplicateClient
         deployment: self,
         input: input,
         webhook_url: webhook_url,
-        webhook_events_filter: webhook_events_filter
+        webhook_events_filter: webhook_events_filter,
+        client: @client
       )
     end
 

data/lib/replicate-client/hardware.rb

@@ -7,19 +7,22 @@ module ReplicateClient
     class << self
       # List all available hardware.
       #
+      # @param client [ReplicateClient::Client] The client to use for requests.
+      #
       # @return [Array<ReplicateClient::Hardware>]
-      def all
-        response = ReplicateClient.client.get(INDEX_PATH)
+      def all(client: ReplicateClient.client)
+        response = client.get(INDEX_PATH)
         response.map { |attributes| new(attributes) }
       end
 
       # Find hardware by SKU.
       #
       # @param sku [String] The SKU of the hardware.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Hardware, nil]
-      def find_by(sku:)
-        all.find { |hardware| hardware.sku == sku }
+      def find_by(sku:, client: ReplicateClient.client)
+        all(client: client).find { |hardware| hardware.sku == sku }
       end
     end
 

data/lib/replicate-client/model/version.rb

@@ -11,12 +11,13 @@ module ReplicateClient
         # @param owner [String] The owner of the model.
         # @param name [String] The name of the model.
         # @param version_id [String] The version id of the model.
+        # @param client [ReplicateClient::Client] The client to use for requests.
         #
         # @return [ReplicateClient::Model::Version]
-        def find_by!(owner:, name:, version_id:)
+        def find_by!(owner:, name:, version_id:, client: ReplicateClient.client)
           path = build_path(owner: owner, name: name, version_id: version_id)
-          response = ReplicateClient.client.get(path)
-          new(response)
+          response = client.get(path)
+          new(response, client: client)
         end
 
         # Find a version of a model.
@@ -24,10 +25,11 @@ module ReplicateClient
         # @param owner [String] The owner of the model.
         # @param name [String] The name of the model.
         # @param version_id [String] The version id of the model.
+        # @param client [ReplicateClient::Client] The client to use for requests.
         #
         # @return [ReplicateClient::Model::Version]
-        def find_by(owner:, name:, version_id:)
-          find_by!(owner: owner, name: name, version_id: version_id)
+        def find_by(owner:, name:, version_id:, client: ReplicateClient.client)
+          find_by!(owner: owner, name: name, version_id: version_id, client: client)
         rescue ReplicateClient::NotFoundError
           nil
         end
@@ -36,12 +38,13 @@ module ReplicateClient
         #
         # @param owner [String] The owner of the model.
         # @param name [String] The name of the model.
+        # @param client [ReplicateClient::Client] The client to use for requests.
         #
         # @return [Array<ReplicateClient::Model::Version>]
-        def where(owner:, name:)
+        def where(owner:, name:, client: ReplicateClient.client)
           versions = []
 
-          auto_paging_each(owner: owner, name: name) do |version|
+          auto_paging_each(owner: owner, name: name, client: client) do |version|
             versions << version
           end
 
@@ -52,18 +55,19 @@ module ReplicateClient
         #
         # @param name [String] The name of the model.
         # @param owner [String] The owner of the model.
+        # @param client [ReplicateClient::Client] The client to use for requests.
         # @yield [ReplicateClient::Model] Yields a model.
         #
         # @return [void]
-        def auto_paging_each(owner:, name:, &block)
+        def auto_paging_each(owner:, name:, client: ReplicateClient.client, &block)
           cursor = nil
           model_path = Model.build_path(owner: owner, name: name)
 
           loop do
             url_params = cursor ? "?cursor=#{cursor}" : ""
-            attributes = ReplicateClient.client.get("#{model_path}#{INDEX_PATH}#{url_params}")
+            attributes = client.get("#{model_path}#{INDEX_PATH}#{url_params}")
 
-            versions = attributes["results"].map { |version| new(version) }
+            versions = attributes["results"].map { |version| new(version, client: client) }
 
             versions.each(&block)
 
@@ -105,7 +109,13 @@ module ReplicateClient
       # @return [Hash]
       attr_accessor :openapi_schema
 
-      def initialize(attributes)
+      # The client used to make API requests for this model version.
+      #
+      # @return [ReplicateClient::Client]
+      attr_accessor :client
+
+      def initialize(attributes, client: ReplicateClient.client)
+        @client = client
         @id = attributes["id"]
         @created_at = Time.parse(attributes["created_at"])
         @cog_version = attributes["cog_version"]
@@ -124,7 +134,8 @@ module ReplicateClient
           version: self,
           input: input,
           webhook_url: webhook_url,
-          webhook_events_filter: webhook_events_filter
+          webhook_events_filter: webhook_events_filter,
+          client: @client
         )
       end
 

data/lib/replicate-client/model.rb

@@ -16,12 +16,13 @@ module ReplicateClient
       #
       # @param owner [String] The owner of the model.
       # @param name [String] The name of the model.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Model]
-      def find_by!(owner:, name:, version_id: nil)
+      def find_by!(owner:, name:, version_id: nil, client: ReplicateClient.client)
         path = build_path(owner: owner, name: name)
-        response = ReplicateClient.client.get(path)
-        new(response, version_id: version_id)
+        response = client.get(path)
+        new(response, version_id: version_id, client: client)
       end
 
       # Find a model.
@@ -29,10 +30,11 @@ module ReplicateClient
       # @param owner [String] The owner of the model.
       # @param name [String] The name of the model.
       # @param version_id [String] The version id of the model to use.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Model]
-      def find_by(owner:, name:, version_id: nil)
-        find_by!(owner: owner, name: name, version_id: version_id)
+      def find_by(owner:, name:, version_id: nil, client: ReplicateClient.client)
+        find_by!(owner: owner, name: name, version_id: version_id, client: client)
       rescue ReplicateClient::NotFoundError
         nil
       end
@@ -42,10 +44,11 @@ module ReplicateClient
       #
       # @param name [String] The name of the model.
       # @param version_id [String] The version id of the model to use.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Model]
-      def find(name, version_id: nil)
-        find_by!(**parse_model_name(name), version_id: version_id)
+      def find(name, version_id: nil, client: ReplicateClient.client)
+        find_by!(**parse_model_name(name), version_id: version_id, client: client)
       end
 
       # Build the path for the model.
@@ -60,17 +63,19 @@ module ReplicateClient
 
       # Paginate through all models.
       #
+      # @param client [ReplicateClient::Client] The client to use for requests.
+      #
       # @yield [ReplicateClient::Model] Yields a model.
       #
       # @return [void]
-      def auto_paging_each(&block)
+      def auto_paging_each(client: ReplicateClient.client, &block)
         cursor = nil
 
         loop do
           url_params = cursor ? "?cursor=#{cursor}" : ""
-          attributes = ReplicateClient.client.get("#{INDEX_PATH}#{url_params}")
+          attributes = client.get("#{INDEX_PATH}#{url_params}")
 
-          models = attributes["results"].map { |model| new(model) }
+          models = attributes["results"].map { |model| new(model, client: client) }
 
           models.each(&block)
 
@@ -90,6 +95,7 @@ module ReplicateClient
       # @param paper_url [String, nil] A URL for the model’s paper.
       # @param license_url [String, nil] A URL for the model’s license.
       # @param cover_image_url [String, nil] A URL for the model’s cover image.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Model]
       def create!(
@@ -101,7 +107,8 @@ module ReplicateClient
         github_url: nil,
         paper_url: nil,
         license_url: nil,
-        cover_image_url: nil
+        cover_image_url: nil,
+        client: ReplicateClient.client
       )
         new_attributes = {
           owner: owner,
@@ -115,9 +122,9 @@ module ReplicateClient
           cover_image_url: cover_image_url
         }
 
-        attributes = ReplicateClient.client.post("/models", new_attributes)
+        attributes = client.post("/models", new_attributes)
 
-        new(attributes)
+        new(attributes, client: client)
       end
 
       # Parse the model name.
@@ -201,13 +208,20 @@ module ReplicateClient
     # @return [Hash]
     attr_accessor :latest_version_id
 
+    # The client used to make API requests for this model.
+    #
+    # @return [ReplicateClient::Client]
+    attr_accessor :client
+
     # Initialize a new model.
     #
     # @param attributes [Hash] The attributes of the model.
     # @param version_id [String] The version of the model to use.
+    # @param client [ReplicateClient::Client] The client to use for requests.
     #
     # @return [ReplicateClient::Model]
-    def initialize(attributes, version_id: nil)
+    def initialize(attributes, version_id: nil, client: ReplicateClient.client)
+      @client = client
       reset_attributes(attributes, version_id: version_id)
     end
 
@@ -222,7 +236,7 @@ module ReplicateClient
     #
     # @return [void]
     def destroy!
-      ReplicateClient.client.delete(path)
+      @client.delete(path)
     end
 
     # The path of the current version.
@@ -236,21 +250,21 @@ module ReplicateClient
     #
     # @return [ReplicateClient::Model::Version]
     def version
-      @version ||= Version.find_by!(owner: owner, name: name, version_id: version_id)
+      @version ||= Version.find_by!(owner: owner, name: name, version_id: version_id, client: @client)
     end
 
     # The latest version of the model.
     #
     # @return [ReplicateClient::Model::Version]
     def latest_version
-      @latest_version ||= Version.find_by!(owner: owner, name: name, version_id: latest_version_id)
+      @latest_version ||= Version.find_by!(owner: owner, name: name, version_id: latest_version_id, client: @client)
     end
 
     # The versions of the model.
     #
     # @return [Array<ReplicateClient::Model::Version>]
     def versions
-      @versions ||= Version.where(owner: owner, name: name)
+      @versions ||= Version.where(owner: owner, name: name, client: @client)
     end
 
     # Create a new prediction for the model.
@@ -264,14 +278,16 @@ module ReplicateClient
           model: self,
           input: input,
           webhook_url: webhook_url,
-          webhook_events_filter: webhook_events_filter
+          webhook_events_filter: webhook_events_filter,
+          client: @client
         )
       else
         Prediction.create!(
           version: version_id,
           input: input,
           webhook_url: webhook_url,
-          webhook_events_filter: webhook_events_filter
+          webhook_events_filter: webhook_events_filter,
+          client: @client
         )
       end
     end
@@ -280,7 +296,7 @@ module ReplicateClient
     #
     # @return [void]
     def reload!
-      attributes = ReplicateClient.client.get(path)
+      attributes = @client.get(path)
       reset_attributes(attributes, version_id: version_id)
     end
 

data/lib/replicate-client/prediction.rb

@@ -20,21 +20,23 @@ module ReplicateClient
       # @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.
+      # @param client [ReplicateClient::Client] The client to use for the prediction.
       #
       # @return [ReplicateClient::Prediction]
-      def create!(version:, input:, webhook_url: nil, webhook_events_filter: nil, sync: false)
+      def create!(version:, input:, webhook_url: nil, webhook_events_filter: nil, sync: false,
+                  client: ReplicateClient.client)
         args = {
           version: version.is_a?(Model::Version) ? version.id : version,
           input: input,
-          webhook: webhook_url || ReplicateClient.configuration.webhook_url,
+          webhook: webhook_url || client.configuration.webhook_url,
           webhook_events_filter: webhook_events_filter&.map(&:to_s)
         }
 
         headers = sync ? { "Prefer" => "wait" } : {}
 
-        prediction = ReplicateClient.client.post(INDEX_PATH, args, headers:)
+        prediction = client.post(INDEX_PATH, args, headers:)
 
-        new(prediction)
+        new(prediction, client: client)
       end
 
       # Create a new prediction for a deployment.
@@ -44,12 +46,14 @@ module ReplicateClient
       # @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.
+      # @param client [ReplicateClient::Client] The client to use for the prediction.
       #
       # @return [ReplicateClient::Prediction]
-      def create_for_deployment!(deployment:, input:, webhook_url: nil, webhook_events_filter: nil, sync: false)
+      def create_for_deployment!(deployment:, input:, webhook_url: nil, webhook_events_filter: nil, sync: false,
+                                 client: ReplicateClient.client)
         args = {
           input: input,
-          webhook: webhook_url || ReplicateClient.configuration.webhook_url,
+          webhook: webhook_url || client.configuration.webhook_url,
           webhook_events_filter: webhook_events_filter&.map(&:to_s)
         }
 
@@ -57,9 +61,9 @@ module ReplicateClient
 
         deployment_path = deployment.is_a?(Deployment) ? deployment.path : "#{Deployment::INDEX_PATH}/#{deployment}"
 
-        prediction = ReplicateClient.client.post("#{deployment_path}#{INDEX_PATH}", args, headers:)
+        prediction = client.post("#{deployment_path}#{INDEX_PATH}", args, headers:)
 
-        new(prediction)
+        new(prediction, client: client)
       end
 
       # Create a new prediction for a model.
@@ -69,50 +73,55 @@ module ReplicateClient
       # @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.
+      # @param client [ReplicateClient::Client] The client to use for the prediction.
       #
       # @return [ReplicateClient::Prediction]
-      def create_for_official_model!(model:, input:, webhook_url: nil, webhook_events_filter: nil, sync: false)
+      def create_for_official_model!(model:, input:, webhook_url: nil, webhook_events_filter: nil, sync: false,
+                                     client: ReplicateClient.client)
         model_path = model.is_a?(Model) ? model.path : Model.build_path(**Model.parse_model_name(model))
 
         args = {
           input: input,
-          webhook: webhook_url || ReplicateClient.configuration.webhook_url,
+          webhook: webhook_url || client.configuration.webhook_url,
           webhook_events_filter: webhook_events_filter&.map(&:to_s)
         }
 
         headers = sync ? { "Prefer" => "wait" } : {}
 
-        prediction = ReplicateClient.client.post("#{model_path}#{INDEX_PATH}", args, headers:)
+        prediction = client.post("#{model_path}#{INDEX_PATH}", args, headers:)
 
-        new(prediction)
+        new(prediction, client: client)
       end
 
       # Find a prediction.
       #
       # @param id [String] The ID of the prediction.
+      # @param client [ReplicateClient::Client] The client to use for the prediction.
       #
       # @return [ReplicateClient::Prediction]
-      def find(id)
-        attributes = ReplicateClient.client.get(build_path(id))
-        new(attributes)
+      def find(id, client: ReplicateClient.client)
+        attributes = client.get(build_path(id))
+        new(attributes, client: client)
       end
 
       # Find a prediction.
       #
       # @param id [String] The ID of the prediction.
+      # @param client [ReplicateClient::Client] The client to use for the prediction.
       #
       # @return [ReplicateClient::Prediction]
-      def find_by!(id:)
-        find(id)
+      def find_by!(id:, client: ReplicateClient.client)
+        find(id, client: client)
       end
 
       # Find a prediction.
       #
       # @param id [String] The ID of the prediction.
+      # @param client [ReplicateClient::Client] The client to use for the prediction.
       #
       # @return [ReplicateClient::Prediction]
-      def find_by(id:)
-        find_by!(id: id)
+      def find_by(id:, client: ReplicateClient.client)
+        find_by!(id: id, client: client)
       rescue ReplicateClient::NotFoundError
         nil
       end
@@ -129,10 +138,11 @@ module ReplicateClient
       # Cancel a prediction.
       #
       # @param id [String] The ID of the prediction.
+      # @param client [ReplicateClient::Client] The client to use for the prediction.
       #
       # @return [void]
-      def cancel!(id)
-        ReplicateClient.client.post("#{build_path(id)}/cancel")
+      def cancel!(id, client: ReplicateClient.client)
+        client.post("#{build_path(id)}/cancel")
       end
     end
 
@@ -206,7 +216,13 @@ module ReplicateClient
     # @return [String]
     attr_accessor :logs
 
-    def initialize(attributes)
+    # The client for the prediction.
+    #
+    # @return [ReplicateClient::Client]
+    attr_accessor :client
+
+    def initialize(attributes, client: ReplicateClient.client)
+      @client = client
       reset_attributes(attributes)
     end
 
@@ -214,7 +230,7 @@ module ReplicateClient
     #
     # @return [ReplicateClient::Prediction]
     def reload!
-      attributes = ReplicateClient.client.get(Prediction.build_path(@id))
+      attributes = @client.get(Prediction.build_path(@id))
       reset_attributes(attributes)
     end
 
@@ -236,7 +252,7 @@ module ReplicateClient
     #
     # @return [void]
     def cancel!
-      Prediction.cancel!(id)
+      Prediction.cancel!(id, client: @client)
     end
 
     # Check if the prediction is succeeded.

data/lib/replicate-client/training.rb

@@ -15,17 +15,19 @@ module ReplicateClient
     class << self
       # List all trainings.
       #
+      # @param client [ReplicateClient::Client] The client to use for requests.
+      #
       # @yield [ReplicateClient::Training] Yields a training.
       #
       # @return [void]
-      def auto_paging_each(&block)
+      def auto_paging_each(client: ReplicateClient.client, &block)
         cursor = nil
 
         loop do
           url_params = cursor ? "?cursor=#{cursor}" : ""
-          attributes = ReplicateClient.client.get("#{INDEX_PATH}#{url_params}")
+          attributes = client.get("#{INDEX_PATH}#{url_params}")
 
-          trainings = attributes["results"].map { |training| new(training) }
+          trainings = attributes["results"].map { |training| new(training, client: client) }
 
           trainings.each(&block)
 
@@ -44,9 +46,11 @@ module ReplicateClient
       # @param input [Hash] The input data for the training.
       # @param webhook_url [String, nil] A URL to receive webhook notifications.
       # @param webhook_events_filter [Array, nil] The events to trigger webhook requests.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Training]
-      def create!(owner:, name:, version:, destination:, input:, webhook_url: nil, webhook_events_filter: nil)
+      def create!(owner:, name:, version:, destination:, input:, webhook_url: nil, webhook_events_filter: nil,
+                  client: ReplicateClient.client)
         destination_str = destination.is_a?(ReplicateClient::Model) ? destination.full_name : destination
         version_id = version.is_a?(ReplicateClient::Model::Version) ? version.id : version
 
@@ -54,12 +58,12 @@ module ReplicateClient
         body = {
           destination: destination_str,
           input: input,
-          webhook: webhook_url || ReplicateClient.configuration.webhook_url,
+          webhook: webhook_url || client.configuration.webhook_url,
           webhook_events_filter: webhook_events_filter
         }
 
-        attributes = ReplicateClient.client.post(path, body)
-        new(attributes)
+        attributes = client.post(path, body)
+        new(attributes, client: client)
       end
 
       # Create a new training for a specific model.
@@ -69,10 +73,17 @@ module ReplicateClient
       # @param input [Hash] The input data for the training.
       # @param webhook_url [String, nil] A URL to receive webhook notifications.
       # @param webhook_events_filter [Array, nil] The events to trigger webhook requests.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Training]
-      def create_for_model!(model:, destination:, input:, webhook_url: nil, webhook_events_filter: nil)
-        model_instance = model.is_a?(ReplicateClient::Model) ? model : ReplicateClient::Model.find(model)
+      def create_for_model!(model:, destination:, input:, webhook_url: nil, webhook_events_filter: nil,
+                            client: ReplicateClient.client)
+        model_instance = if model.is_a?(ReplicateClient::Model)
+                           model
+                         else
+                           ReplicateClient::Model.find(model,
+                                                       client: client)
+                         end
         raise ArgumentError, "Invalid model" unless model_instance
 
         create!(
@@ -81,30 +92,33 @@ module ReplicateClient
           version: model_instance.version_id,
           destination: destination,
           input: input,
-          webhook_url: webhook_url || ReplicateClient.configuration.webhook_url,
-          webhook_events_filter: webhook_events_filter
+          webhook_url: webhook_url || client.configuration.webhook_url,
+          webhook_events_filter: webhook_events_filter,
+          client: client
         )
       end
 
       # Find a training by id.
       #
       # @param id [String] The id of the training.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [ReplicateClient::Training]
-      def find(id)
+      def find(id, client: ReplicateClient.client)
         path = build_path(id: id)
-        attributes = ReplicateClient.client.get(path)
-        new(attributes)
+        attributes = client.get(path)
+        new(attributes, client: client)
       end
 
       # Cancel a training.
       #
       # @param id [String] The id of the training.
+      # @param client [ReplicateClient::Client] The client to use for requests.
       #
       # @return [void]
-      def cancel!(id)
+      def cancel!(id, client: ReplicateClient.client)
         path = "#{build_path(id: id)}/cancel"
-        ReplicateClient.client.post(path)
+        client.post(path, {})
       end
 
       # Build the path for a specific training.
@@ -183,12 +197,19 @@ module ReplicateClient
     # @return [Hash, nil]
     attr_accessor :metrics
 
+    # The client used to make API requests for this training.
+    #
+    # @return [ReplicateClient::Client]
+    attr_accessor :client
+
     # Initialize a new training instance.
     #
     # @param attributes [Hash] The attributes of the training.
+    # @param client [ReplicateClient::Client] The client to use for requests.
     #
     # @return [ReplicateClient::Training]
-    def initialize(attributes)
+    def initialize(attributes, client: ReplicateClient.client)
+      @client = client
       reset_attributes(attributes)
     end
 
@@ -231,14 +252,14 @@ module ReplicateClient
     #
     # @return [void]
     def cancel!
-      ReplicateClient::Training.cancel!(id)
+      ReplicateClient::Training.cancel!(id, client: @client)
     end
 
     # Reload the training.
     #
     # @return [void]
     def reload!
-      attributes = ReplicateClient.client.get(Training.build_path(id: id))
+      attributes = @client.get(Training.build_path(id: id))
       reset_attributes(attributes)
     end
 
@@ -246,7 +267,7 @@ module ReplicateClient
     #
     # @return [ReplicateClient::Model]
     def model
-      @model ||= ReplicateClient::Model.find(model_full_name, version_id: version_id)
+      @model ||= ReplicateClient::Model.find(model_full_name, version_id: version_id, client: @client)
     end
 
     # The version instance of the training.

data/lib/replicate-client/version.rb

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

data/replicate-client.gemspec

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require_relative "lib/replicate-client/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "replicate-client"
+  spec.version = ReplicateClient::VERSION
+  spec.authors = ["Dylan Player"]
+  spec.email = ["dylan@851.sh"]
+
+  spec.summary = "Ruby client for Replicate API."
+  spec.homepage = "https://github.com/851-labs/replicate"
+  spec.required_ruby_version = ">= 3.3.0"
+
+  spec.metadata["allowed_push_host"] = "https://rubygems.org"
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = spec.homepage
+  spec.license = "MIT"
+
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      (f == __FILE__) || f.match(%r{\A(?:(?:bin|test|spec|features)/|\.(?:git|travis|circleci)|appveyor)})
+    end
+  end
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency("faraday", ">= 1")
+  spec.metadata["rubygems_mfa_required"] = "true"
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: replicate-client
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.10
 platform: ruby
 authors:
 - Dylan Player
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-01-24 00:00:00.000000000 Z
+date: 2025-09-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -48,6 +48,7 @@ files:
 - lib/replicate-client/training.rb
 - lib/replicate-client/version.rb
 - lib/replicate-client/webhook.rb
+- replicate-client.gemspec
 homepage: https://github.com/851-labs/replicate
 licenses:
 - MIT