modelslab-ruby 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b8dff2850365cd3bb76c82eedd1b2789b81223f0ff9dfbe8c6defc61367c42f5
-  data.tar.gz: ed5130cae3fde7a041f4fb305bb656e09e9fe9b42bd905ef091cf61e942ee3e8
+  metadata.gz: ebece3b74c1f78c4e8d5d7f67b1d8963855655795878f947a972dd6b6fba7835
+  data.tar.gz: 33a9276ff845c7895bcd1d46adfe952fbacec34e9b35c479a37cd324b80af1c3
 SHA512:
-  metadata.gz: 4c1b9819d1ee03959e004a8fac1a19b7bf1baa470fbbc20d7eb63df033bb209a3586c1f2e7b273336ead0a7f08b34b3a9d1b67d3717bc4a7fe30d044021b945e
-  data.tar.gz: 141231b0013babee29f12c62fd31cb006f8a740da3f1e39b52a1e0161201c6fd9cb1d4c81da72a37614a39b72a99af5ec78443a4a5bc086acc44d2e4b819f428
+  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,7 @@ 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:)
@@ -22,7 +22,7 @@ module ModelsLab
         request_id: request_id
       }
 
-      @client.json_post(path: "images/fetch", parameters: parameters)
+      @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.2.0"
+  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.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Florian Felsing
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-09-06 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