replicate-client 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5556c24e737090c5fb8425f5e947822e4b43dee0fbda7703d3fb1bc0f2bf4dd7
-  data.tar.gz: 6340ba5606a808c35fa4bf9f181ea82aa85c6108906ec93aaa0c5387d6a1a11e
+  metadata.gz: f6798a521201c15a08cd456167052ac6de82b43f60487ea454dabda1865a28ce
+  data.tar.gz: cd44ea8758209f771645f2d9755438d6ad890758f83ef2ce8dd03d39f1162d4e
 SHA512:
-  metadata.gz: c934a95922c925246279b3c59f6412c309508d926e14f1a0734c63728580db0ca0323a148fde5596563f703d2c8ee5adf08bdc89a87b4d52d4aac95e997dcf3a
-  data.tar.gz: 47e28051e876e484f43f993dc78a9fe9f14c3ddab29241974dc35693713fce0a845bb4a2c3f744d36a6a20b6680a40230921a6f89600a2c19ba365aef907c1e8
+  metadata.gz: 7b0f59e6ae5803a0cefc0ec8a0d2312cf45b4205a7052cfae84bc70b13c062ebd533fed18c92971513139235418cb1a08662eff816cfa211fdc854451133df29
+  data.tar.gz: 18e5cb1d7eb54ceddb10eac736da1e56baa8245eb558b4670da2fcd4e99a1d5519bc473313bee0bb5ba7a191cf1f3aeeafccbedea24a202a474df6a716684dba

data/Gemfile.lock

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

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/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ReplicateClient
-  VERSION = "0.1.6"
+  VERSION = "0.1.7"
 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.7
 platform: ruby
 authors:
 - Dylan Player
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-09-01 00:00:00.000000000 Z
+date: 2024-09-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday