replicate-client 0.1.6 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5556c24e737090c5fb8425f5e947822e4b43dee0fbda7703d3fb1bc0f2bf4dd7
-  data.tar.gz: 6340ba5606a808c35fa4bf9f181ea82aa85c6108906ec93aaa0c5387d6a1a11e
+  metadata.gz: 61acc30d7e6406c1aa6f797a45ab6927945d399f3ce49934a2ada361a818df18
+  data.tar.gz: 6e0023bde6ce5e0897314e3f1bde97e1b7d99b9dd1de776568a883822852a874
 SHA512:
-  metadata.gz: c934a95922c925246279b3c59f6412c309508d926e14f1a0734c63728580db0ca0323a148fde5596563f703d2c8ee5adf08bdc89a87b4d52d4aac95e997dcf3a
-  data.tar.gz: 47e28051e876e484f43f993dc78a9fe9f14c3ddab29241974dc35693713fce0a845bb4a2c3f744d36a6a20b6680a40230921a6f89600a2c19ba365aef907c1e8
+  metadata.gz: ee9a5f6e05fbf0b8af2a34c4890dae5ec925d47b5ed0c9afceb866e07c1be71a3d2c50f98c1ca569b84bb03d17abc2f665e48a6927b0243fab2180ff2f43df44
+  data.tar.gz: 692003bec1af63306d896535c434b888ce9162431839809ae140b811a5c40e9323027143096b5e8ea0a1270fd79f6f23a4b544ed33d2090f43f4d7c0d2d449c6

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    replicate-client (0.1.4)
+    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/model/version.rb

@@ -127,6 +127,37 @@ module ReplicateClient
           webhook_events_filter: webhook_events_filter
         )
       end
+
+      # Get the prediction input schema from the openapi schema.
+      #
+      # @return [Hash] The prediction input schema.
+      def prediction_input_schema
+        resolve_ref(openapi_schema.dig("components", "schemas", "PredictionRequest", "properties", "input"))
+      end
+
+      # Get the training input schema from the openapi schema.
+      #
+      # @return [Hash] The training input schema.
+      def training_input_schema
+        resolve_ref(openapi_schema.dig("components", "schemas", "TrainingRequest", "properties", "input"))
+      end
+
+      private
+
+      # Resolve a reference in the openapi schema.
+      #
+      # @param schema [Hash] The schema to resolve.
+      #
+      # @return [Hash] The resolved schema.
+      def resolve_ref(schema)
+        if schema.is_a?(Hash) && schema["$ref"]
+          ref_path = schema["$ref"].split("/").drop(1)
+          resolved_schema = openapi_schema.dig(*ref_path)
+          resolve_ref(resolved_schema)
+        else
+          schema
+        end
+      end
     end
   end
 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.6"
+  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.6
+  version: 0.1.8
 platform: ruby
 authors:
 - Dylan Player
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-09-01 00:00:00.000000000 Z
+date: 2025-01-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday