replicate-client 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: efa5152767827d0b6d95d1b83cf211c9dfb82d11885401c74394d6dbd068949e
-  data.tar.gz: '040619b058d796d577bb7d1b75e21d84af4e1999b3ccebcc5b168ba5469241f6'
+  metadata.gz: f066662aa759a5881284f5008568d092d3ec28b8cbc1d87a2221c42a6c6fd4d4
+  data.tar.gz: ced72f79a1601dd5c0702e726d508c7226130de8c92b890df6e55fbaeeeb5552
 SHA512:
-  metadata.gz: 706867512328515a36e3af092bdd9bfb39a633e476cd37b7c4afc9f3e492be7f427c61eedc58e4ccaa85f151c5a4e17291bdcbbe1e7dc5a627c366c85dbf3125
-  data.tar.gz: 9a0c11660772e9f85f2e59676ce854cf9e2f4d79e2d8d701d6df6340bed17484a0297a13d1115eaed5d2b10a815be06d41feed24dce049ebdb7f2afd89ed6930
+  metadata.gz: 0ba457621a0fdfc737c83a5e632abe315735fa45bcf792dab468a2d620aed526012a4f49c06bc43451c1d7702064b596d1ae32c9af07765295a54bd3b78f7017
+  data.tar.gz: a3dcf6a0153b84b7c44b0137dee34072e3d0ca89b5127cf12d96ba8e4edac14488a178c931034879b6444fc21fb8862c2472fad9a9a3f40071fca59559a7e727

data/Gemfile.lock

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

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 851 Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/lib/replicate-client/training.rb

@@ -42,11 +42,11 @@ module ReplicateClient
       # @param destination [ReplicateClient::Model, String] The destination model instance or string in "owner/name"
       # format.
       # @param input [Hash] The input data for the training.
-      # @param webhook [String, nil] A URL to receive webhook notifications.
+      # @param webhook_url [String, nil] A URL to receive webhook notifications.
       # @param webhook_events_filter [Array, nil] The events to trigger webhook requests.
       #
       # @return [ReplicateClient::Training]
-      def create!(owner:, name:, version:, destination:, input:, webhook: nil, webhook_events_filter: nil)
+      def create!(owner:, name:, version:, destination:, input:, webhook_url: nil, webhook_events_filter: nil)
         destination_str = destination.is_a?(ReplicateClient::Model) ? destination.full_name : destination
         version_id = version.is_a?(ReplicateClient::Model::Version) ? version.id : version
 
@@ -54,7 +54,7 @@ module ReplicateClient
         body = {
           destination: destination_str,
           input: input,
-          webhook: webhook,
+          webhook: webhook_url || ReplicateClient.configuration.webhook_url,
           webhook_events_filter: webhook_events_filter
         }
 
@@ -67,11 +67,11 @@ module ReplicateClient
       # @param model [ReplicateClient::Model, String] The model instance or a string representing the model ID.
       # @param destination [ReplicateClient::Model, String] The destination model or full name in "owner/name" format.
       # @param input [Hash] The input data for the training.
-      # @param webhook [String, nil] A URL to receive webhook notifications.
+      # @param webhook_url [String, nil] A URL to receive webhook notifications.
       # @param webhook_events_filter [Array, nil] The events to trigger webhook requests.
       #
       # @return [ReplicateClient::Training]
-      def create_for_model!(model:, destination:, input:, webhook: nil, webhook_events_filter: nil)
+      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)
         raise ArgumentError, "Invalid model" unless model_instance
 
@@ -81,7 +81,7 @@ module ReplicateClient
           version: model_instance.version_id,
           destination: destination,
           input: input,
-          webhook: webhook,
+          webhook: webhook_url || ReplicateClient.configuration.webhook_url,
           webhook_events_filter: webhook_events_filter
         )
       end
@@ -125,12 +125,12 @@ module ReplicateClient
     # The full model name in the format "owner/name".
     #
     # @return [String]
-    attr_accessor :model
+    attr_accessor :model_full_name
 
     # The version ID of the model being trained.
     #
     # @return [String]
-    attr_accessor :version
+    attr_accessor :version_id
 
     # The input data provided for the training.
     #
@@ -150,9 +150,14 @@ module ReplicateClient
 
     # The timestamp when the training was completed.
     #
-    # @return [String, nil]
+    # @return [Time, nil]
     attr_accessor :completed_at
 
+    # The timestamp when the training was started.
+    #
+    # @return [Time, nil]
+    attr_accessor :started_at
+
     # The logs generated during the training process.
     #
     # @return [String]
@@ -173,6 +178,11 @@ module ReplicateClient
     # @return [Hash, nil]
     attr_accessor :output
 
+    # The metrics generated during the training process.
+    #
+    # @return [Hash, nil]
+    attr_accessor :metrics
+
     # Initialize a new training instance.
     #
     # @param attributes [Hash] The attributes of the training.
@@ -232,6 +242,20 @@ module ReplicateClient
       reset_attributes(attributes)
     end
 
+    # The model instance of the training.
+    #
+    # @return [ReplicateClient::Model]
+    def model
+      @model ||= ReplicateClient::Model.find(model_full_name, version_id: version_id)
+    end
+
+    # The version instance of the training.
+    #
+    # @return [ReplicateClient::Model::Version]
+    def version
+      @version ||= model.version
+    end
+
     private
 
     # Set the attributes of the training.
@@ -241,8 +265,8 @@ module ReplicateClient
     # @return [void]
     def reset_attributes(attributes)
       @id = attributes["id"]
-      @model = attributes["model"]
-      @version = attributes["version"]
+      @model_full_name = attributes["model"]
+      @version_id = attributes["version"]
       @input = attributes["input"]
       @status = attributes["status"]
       @created_at = attributes["created_at"]
@@ -251,6 +275,11 @@ module ReplicateClient
       @error = attributes["error"]
       @urls = attributes["urls"]
       @output = attributes["output"]
+      @started_at = attributes["started_at"]
+      @metrics = attributes["metrics"]
+
+      @model = nil
+      @version = nil
     end
   end
 end

data/lib/replicate-client/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: replicate-client
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Dylan Player
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-08-24 00:00:00.000000000 Z
+date: 2024-08-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -35,6 +35,7 @@ files:
 - ".ruby-version"
 - Gemfile
 - Gemfile.lock
+- LICENSE
 - README.md
 - Rakefile
 - lib/replicate-client.rb