ruby-leonardoai 0.1.4 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d108ed0e36199b2e7841810b31b77e951b1173082f97aa75410de1050aaa6080
-  data.tar.gz: c31edd35e33bc9c5a0c9ba2df992f8ce9b8f38698cdd9c48d0789a441f0eb8cd
+  metadata.gz: de55b3e839a008371101dab19fefc4157d9290c42a3e2de665493362648feea3
+  data.tar.gz: a6d79f4179c88c9c5cdede664203ea30cf5cb1e273e9c3dfb8f3f110384c37ed
 SHA512:
-  metadata.gz: 9fce8ddb89a37f921531baecab3ee91d95314bc94757acdbab8d140c871b64caa65d961bb1c8929242c1e0d40e075cf64f088a81c73d8e6185969ae3e2ebb159
-  data.tar.gz: 2acadd28592ca2cb8131892b853add8193afef93f22f74d00af9060df6c7cbeed8d11cfd761106ab2cb4677a4c2f6e12e552eae51086b97a1df2757d0a878375
+  metadata.gz: d5669c7773cf2df4e6a7c65309fe2631df857321535c7ee9e52bc361444c6e2606e0ed0960a4c920d0bee765dfea050137a45d5bbbf4f372fd8a910667df7d0e
+  data.tar.gz: b171eee19f9d2e9300b79c663ab7a5de9949834472039f40c46789536ff098678e8e4ca0633de3e330c54ca4ce1667010d375739d6ebadbb7d79b81acdcda41a

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ruby-leonardoai (0.1.3)
+    ruby-leonardoai (0.1.6)
       faraday (>= 1)
       faraday-multipart (>= 1)
 

data/README.md

@@ -109,12 +109,40 @@ LeonardoAI.configure do |config|
     } # Optional
 end
 ```
+### Running in Console
+Easily run tests in console with ```bin/console```
+
+```ruby
+# Example
+
+client = LeonardoAI::Client.new(access_token: "your-access-token-here")
+
+params = {
+    :height=>1024, 
+    :prompt=>"A ad flyer of a ninja cat, heavy texture, moonlit background, circle design, tshirt design, pen and ink style", 
+    :width=>512, :num_images=>1, 
+    :photoReal=>false, 
+    :presetStyle=>"LEONARDO", 
+    :promptMagic=>true, 
+    :promptMagicVersion=>"v2", 
+    :public=>false, 
+    :init_strength=>0.4, 
+    :sd_version=>"v2", 
+}
+
+response = client.generations.generate(parameters: params)
+puts response
+```
+
+
 
 ### Generation
 
 ChatGPT is a model that can be used to generate text in a conversational style. You can use it to [generate a response](https://platform.openai.com/docs/api-reference/chat/create) to a sequence of [messages](https://platform.openai.com/docs/guides/chat/introduction):
 
 ```ruby
+# https://docs.leonardo.ai/reference/creategeneration
+
 params = {
     :height=>1024, 
     :prompt=>"A ad flyer of a ninja cat, heavy texture, moonlit background, circle design, tshirt design, pen and ink style", 
@@ -134,13 +162,65 @@ puts response.dig("sdGenerationJob", "generationId")
 ```
 
 ### Model
-UNDER CONSTRUCTION
+```ruby
+# https://docs.leonardo.ai/reference/post_models-3d-upload
+
+params = {
+    :modelExtension=>"this-is-an-example", 
+    :name=>"some string"
+}
 
-### Unzoom
-UNDER CONSTRUCTION
+response = client.models.upload_3d_model(parameters: params) 
+```
+
+```ruby
+# https://docs.leonardo.ai/reference/getmodelbyid
+
+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)
+
+### Variation
+```ruby
+# https://docs.leonardo.ai/reference/post_variations-unzoom
+
+params = {
+    :id=>"this-is-some-id", 
+    :isVariation=>"boolean: true|false"
+}
+
+response = client.variations.create_unzoom(parameters: params) 
+```
+
+```ruby
+# https://docs.leonardo.ai/reference/getvariationbyid
+
+response = client.variations.get_variation_by_id(id: "your-variation-id-here") 
+```
 
 ### Dataset
-UNDER CONSTRUCTION
+```ruby
+# https://docs.leonardo.ai/reference/createdataset
+
+params = {
+    :name=>"this-is-some-string-id", 
+    :description=>"this-is-some-string"
+}
+
+response = client.datasets.create_dataset(parameters: params) 
+```
+
+```ruby
+# https://docs.leonardo.ai/reference/getdatasetbyid
+
+response = client.datasets.get_dataset_by_id(id: "your-dataset-id-here") 
+```
 
 ## Development
 

data/lib/leonardoai/client.rb

@@ -27,5 +27,17 @@ module LeonardoAI
     def generations
       @generations ||= LeonardoAI::Generations.new(client: self)
     end
+
+    def models
+      @models ||= LeonardoAI::Models.new(client: self)
+    end
+
+    def variations
+      @variations ||= LeonardoAI::Variations.new(client: self)
+    end
+
+    def datasets
+      @datasets ||= LeonardoAI::Datasets.new(client: self)
+    end
   end
 end

data/lib/leonardoai/datasets.rb

@@ -0,0 +1,32 @@
+module LeonardoAI
+  class Datasets
+    def initialize(client: nil)
+      @client = client
+    end
+
+    # params: {name: string|null, description: string|null}
+    def create_dataset(parameters: {})
+      @client.json_post(path: "/datasets", parameters: parameters)
+    end
+
+    # id: string
+    def get_dataset_by_id(id:)
+      @client.get(path: "/datasets/#{id}")
+    end
+
+    # id: string
+    def delete_dataset(id:)
+      @client.delete(path: "/datasets/#{id}")
+    end
+
+    # dataset_id: string (req'd0, params: {extension: string|null}
+    def upload_dataset_image(dataset_id:, parameters: {})
+      @client.json_post(path: "/datasets/#{dataset_id}/upload", parameters: parameters)
+    end
+
+    # dataset_id: string (req'd0, params: {generatedImageId: string|null}
+    def upload_dataset_image(dataset_id:, parameters: {})
+      @client.json_post(path: "/datasets/#{dataset_id}/upload/gen", parameters: parameters)
+    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/variations.rb

@@ -0,0 +1,21 @@
+module LeonardoAI
+  class Variations
+    def initialize(client: nil)
+      @client = client
+    end
+
+    # params: {id: string|null, isVariation: boolean|null}
+    def create_unzoom(parameters: {})
+      @client.json_post(path: "/variations/unzoom", parameters: parameters)
+    end
+
+    # params: {id: string|null}
+    def create_upscale(parameters: {})
+      @client.json_post(path: "/variations/upscale", parameters: parameters)
+    end
+
+    def get_variation_by_id(id:)
+      @client.get(path: "/variations/#{id}")
+    end
+  end
+end

data/lib/leonardoai/version.rb

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

data/lib/leonardoai.rb

@@ -3,6 +3,9 @@ require "faraday/multipart"
 require_relative "leonardoai/http"
 require_relative "leonardoai/client"
 require_relative "leonardoai/generations"
+require_relative "leonardoai/models"
+require_relative "leonardoai/variations"
+require_relative "leonardoai/datasets"
 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.6
 platform: ruby
 authors:
 - Donald Lee
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-10-11 00:00:00.000000000 Z
+date: 2023-10-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -58,8 +58,11 @@ files:
 - lib/leonardoai.rb
 - lib/leonardoai/client.rb
 - lib/leonardoai/compatibility.rb
+- lib/leonardoai/datasets.rb
 - lib/leonardoai/generations.rb
 - lib/leonardoai/http.rb
+- lib/leonardoai/models.rb
+- lib/leonardoai/variations.rb
 - lib/leonardoai/version.rb
 - lib/ruby/leonardoai.rb
 - ruby-leonardoai.gemspec