modelslab-ruby 0.1.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7685a8b833ae136ac948c017244988abcfe8a7a6dd6286512eceeb28f53b9d94
-  data.tar.gz: 14e568c5a3c39d2047d88d33430075279432d4b448758e4edc47db9432f2f664
+  metadata.gz: ebece3b74c1f78c4e8d5d7f67b1d8963855655795878f947a972dd6b6fba7835
+  data.tar.gz: 33a9276ff845c7895bcd1d46adfe952fbacec34e9b35c479a37cd324b80af1c3
 SHA512:
-  metadata.gz: 9d5ddfd7994968f41620c9545f58d84256c79c4341ee71ff161344c797996736b8765dc5373b46fd7109deefc9de1a2cfad7ddeef053fe58a99eea3e26682f27
-  data.tar.gz: 5b1ba8645c9356ed898d90b536411a5c5cde07f2d3c12bdda53438437b65fd2594f476ebdc77169effed3f1435d160ca3fcc9bed46c408d27dc4868007daa7ed
+  metadata.gz: 875b1aa4b484c2ac927029118a6495f675297baeb92cf6595b9cc6a2af5e766e31e8362a810ad84b8fbb56b0e0197bc14b1681430516fe0e646404eec35ba61a
+  data.tar.gz: e2ebf832d6bb86469d907d98f5113df909fbe602f257537d7dace5deaa1dfb52c2683864fca65304db2fc53b7de41aa312610b7b55d5bdd158754ad98bf64867

data/lib/modelslab/client.rb

@@ -20,5 +20,9 @@ module ModelsLab
     def images
       @images ||= ModelsLab::Images.new(client: self)
     end
+
+    def enterprise
+      @enterprise ||= Enterprise.new(client: self)
+    end
   end
 end

data/lib/modelslab/enterprise.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module ModelsLab
+  class Enterprise
+    def initialize(client:)
+      @client = client
+    end
+
+    def text2img(prompt:, model_id:, options: {})
+      parameters = {
+        key: @client.api_key,
+        prompt: prompt,
+        model_id: model_id
+      }.merge(options)
+
+      @client.json_post(path: "v1/enterprise/text2img", parameters: parameters)
+    end
+
+    def fetch_queued_image(request_id:)
+      parameters = {
+        key: @client.api_key,
+        request_id: request_id
+      }
+
+      @client.json_post(path: "v1/enterprise/fetch", parameters: parameters)
+    end
+  end
+end

data/lib/modelslab/images.rb

@@ -13,7 +13,16 @@ module ModelsLab
         model_id: model_id
       }.merge(options)
 
-      @client.json_post(path: "images/text2img", parameters: parameters)
+      @client.json_post(path: "v6/images/text2img", parameters: parameters)
+    end
+
+    def fetch_queued_image(request_id:)
+      parameters = {
+        key: @client.api_key,
+        request_id: request_id
+      }
+
+      @client.json_post(path: "v6/images/fetch", parameters: parameters)
     end
   end
 end

data/lib/modelslab/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ModelsLab
-  VERSION = "0.1.1"
+  VERSION = "0.3.0"
 end

data/lib/modelslab.rb

@@ -7,6 +7,7 @@ require_relative "modelslab/version"
 require_relative "modelslab/http"
 require_relative "modelslab/client"
 require_relative "modelslab/images"
+require_relative "modelslab/enterprise"
 
 module ModelsLab
   class Error < StandardError; end
@@ -31,7 +32,7 @@ module ModelsLab
   class Configuration
     attr_accessor :api_key, :uri_base, :request_timeout, :extra_headers
 
-    DEFAULT_URI_BASE = "https://modelslab.com/api/v6"
+    DEFAULT_URI_BASE = "https://modelslab.com/api"
     DEFAULT_REQUEST_TIMEOUT = 120
 
     def initialize

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: modelslab-ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Florian Felsing
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-08-22 00:00:00.000000000 Z
+date: 2024-09-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -33,6 +33,7 @@ extra_rdoc_files: []
 files:
 - lib/modelslab.rb
 - lib/modelslab/client.rb
+- lib/modelslab/enterprise.rb
 - lib/modelslab/http.rb
 - lib/modelslab/images.rb
 - lib/modelslab/version.rb