ruby-leonardoai 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d108ed0e36199b2e7841810b31b77e951b1173082f97aa75410de1050aaa6080
-  data.tar.gz: c31edd35e33bc9c5a0c9ba2df992f8ce9b8f38698cdd9c48d0789a441f0eb8cd
+  metadata.gz: 177f0c64e72ef9ede48bf722a6a8886be330abcb1f9ab10af719395cb27046c2
+  data.tar.gz: 4c1ed2cb485ada7871c810c6908194bbfe33876af0d44d983a2044494d81324c
 SHA512:
-  metadata.gz: 9fce8ddb89a37f921531baecab3ee91d95314bc94757acdbab8d140c871b64caa65d961bb1c8929242c1e0d40e075cf64f088a81c73d8e6185969ae3e2ebb159
-  data.tar.gz: 2acadd28592ca2cb8131892b853add8193afef93f22f74d00af9060df6c7cbeed8d11cfd761106ab2cb4677a4c2f6e12e552eae51086b97a1df2757d0a878375
+  metadata.gz: 06d681a7ed63d0d19df5506856d9f62c2254853b031dffd72710cd9988e6050d1c14797c8d5ec5508b323065557255a451d532d8bc322ab278a40607b7132095
+  data.tar.gz: b801221868605de96484b84fad6be03c20653b1046bb9099be052276abfe6b8b5a1b97d01d7d4eaeaf4c8b6331f8a9103333303341d115ca33590999c4a14f8e

data/README.md

@@ -134,7 +134,29 @@ puts response.dig("sdGenerationJob", "generationId")
 ```
 
 ### Model
-UNDER CONSTRUCTION
+```ruby
+# https://cloud.leonardo.ai/api/rest/v1/models-3d/upload
+
+params = {
+    :modelExtension=>"this-is-an-example", 
+    :name=>"some string"
+}
+
+response = client.models.upload_3d_model(parameters: params) 
+```
+
+```ruby
+# https://cloud.leonardo.ai/api/rest/v1/models/{id}
+
+params = {
+    :modelExtension=>"this-is-an-example", 
+    :name=>"some string"
+}
+
+response = client.models.get_custom_models_by_id(id: "your-custom-model-id-here") 
+```
+
+For more parameters, please check the API found [here](https://docs.leonardo.ai/reference/post_models-3d-upload)
 
 ### Unzoom
 UNDER CONSTRUCTION

data/lib/leonardoai/client.rb

@@ -27,5 +27,9 @@ module LeonardoAI
     def generations
       @generations ||= LeonardoAI::Generations.new(client: self)
     end
+
+    def models
+      @models ||= LeonardoAI::Models.new(client: self)
+    end
   end
 end

data/lib/leonardoai/generations.rb

@@ -8,16 +8,32 @@ module LeonardoAI
       @client.json_post(path: "/generations", parameters: parameters)
     end
 
-    def edit(parameters: {})
-      @client.multipart_post(path: "/edits", parameters: open_files(parameters))
+    def get_generation(id:)
+      @client.get(path: "/generations/#{id}")
     end
 
-    private
+    def delete_generation(id:)
+      @client.delete(path: "/generations/#{id}")
+    end
+
+    def get_user_generations(user_id:)
+      @client.get(path: "/generations/user/#{user_id}")
+    end
+
+    def generate_texture_generation(parameters: {})
+      @client.json_post(path: "/generations-texture", parameters: parameters)
+    end
+
+    def get_texture_generation_by_model_id(model_id:)
+      @client.get(path: "/generations-texture/model/#{model_id}")
+    end
+
+    def get_texture_generation_by_id(id:)
+      @client.get(path: "/generations-texture/model/#{id}")
+    end
 
-    def open_files(parameters)
-      parameters = parameters.merge(image: File.open(parameters[:image]))
-      parameters = parameters.merge(mask: File.open(parameters[:mask])) if parameters[:mask]
-      parameters
+    def delete_texture_generation(id:)
+      @client.delete(path: "/generations-texture/#{id}")
     end
   end
 end

data/lib/leonardoai/models.rb

@@ -0,0 +1,39 @@
+module LeonardoAI
+  class Models
+    def initialize(client: nil)
+      @client = client
+    end
+
+    def upload_3d_model(parameters: {})
+      @client.json_post(path: "/models-3d/upload", parameters: parameters)
+    end
+
+    def get_3d_models_by_user_id(user_id:, offset: 0, limit: 10)
+      @client.get(path: "/models-3d/user/#{user_id}?offset=#{offset}&limit=#{limit}")
+    end
+
+    def get_3d_models_by_id(id:, offset: 0, limit: 10)
+      @client.get(path: "/models-3d/#{id}?offset=#{offset}&limit=#{limit}")
+    end
+
+    def delete_3d_model(id:)
+      @client.delete(path: "/models-3d/#{id}")
+    end
+
+    def train(parameters: {})
+      @client.json_post(path: "/models", parameters: parameters)
+    end
+
+    def get_custom_models_by_id(id:)
+      @client.get(path: "/models/#{id}")
+    end
+
+    def delete_models_by_id(id:)
+      @client.delete(path: "/models/#{id}")
+    end
+
+    def get_platform_models(offset: 0, limit: 10)
+      @client.get(path: "/platformModels?offset=#{offset}&limit=#{limit}")
+    end
+  end
+end

data/lib/leonardoai/version.rb

@@ -1,3 +1,3 @@
 module LeonardoAI
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

data/lib/leonardoai.rb

@@ -3,6 +3,7 @@ require "faraday/multipart"
 require_relative "leonardoai/http"
 require_relative "leonardoai/client"
 require_relative "leonardoai/generations"
+require_relative "leonardoai/models"
 require_relative "leonardoai/version"
 
 module LeonardoAI

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-leonardoai
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Donald Lee
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-10-11 00:00:00.000000000 Z
+date: 2023-10-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -60,6 +60,7 @@ files:
 - lib/leonardoai/compatibility.rb
 - lib/leonardoai/generations.rb
 - lib/leonardoai/http.rb
+- lib/leonardoai/models.rb
 - lib/leonardoai/version.rb
 - lib/ruby/leonardoai.rb
 - ruby-leonardoai.gemspec