modelslab-ruby 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: aaa8cfecff49fc4cc6ef03c942b20151c8f691d82ec417a77c9554015b8e43f3
+  data.tar.gz: e0cf57fc6a3034a4c48403e897710fdfef0157350b3201c7a44a1230aa3c8df6
+SHA512:
+  metadata.gz: 14bbe9eef78bd8ca532b4cd36be51b0243067dbcbff9ddc42b3d9c0aad768800bb7fa76b2ce033cc4bc3b7f91a0ba7ede7532376309f6c6e037172d00d781f2b
+  data.tar.gz: e09e522e585fc169e74992bb5d2b27879aae73c1d45b5cd84cb39836dd5b3b3adac297294e1dfb152a0d646fc0c7ca5cf0796ad6effb7543f3851cbf173d8946

data/lib/modelslab/client.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module ModelsLab
+  class Client
+    include ModelsLab::HTTP
+
+    CONFIG_KEYS = %i[api_key uri_base request_timeout extra_headers].freeze
+    attr_reader(*CONFIG_KEYS, :faraday_middleware)
+
+    def initialize(config = {}, &faraday_middleware)
+      CONFIG_KEYS.each do |key|
+        instance_variable_set(
+          "@#{key}",
+          config[key].nil? ? ModelsLab.configuration.send(key) : config[key]
+        )
+      end
+      @faraday_middleware = faraday_middleware
+    end
+
+    def images
+      @images ||= ModelsLab::Images.new(client: self)
+    end
+  end
+end

data/lib/modelslab/http.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module ModelsLab
+  module HTTP
+    def json_post(path:, parameters:)
+      response = connection.post(path) do |req|
+        req.headers["Content-Type"] = "application/json"
+        req.body = parameters.to_json
+      end
+      JSON.parse(response.body)
+    rescue JSON::ParserError
+      response.body
+    end
+
+    def connection
+      @connection ||= Faraday.new(url: uri_base) do |f|
+        f.request :json
+        f.response :raise_error
+        f.use ModelsLab::MiddlewareErrors
+        f.options.timeout = request_timeout
+        extra_headers.each do |key, value|
+          f.headers[key] = value
+        end
+        faraday_middleware&.call(f)
+        f.adapter Faraday.default_adapter
+      end
+    end
+  end
+end

data/lib/modelslab/images.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module ModelsLab
+  class Images
+    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: "images/text2img", parameters: parameters)
+    end
+  end
+end

data/lib/modelslab/version.rb

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

data/lib/modelslab.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+require "faraday"
+require "faraday/multipart"
+
+require_relative "modelslab/version"
+require_relative "modelslab/http"
+require_relative "modelslab/client"
+require_relative "modelslab/images"
+
+module ModelsLab
+  class Error < StandardError; end
+  class ConfigurationError < Error; end
+
+  class MiddlewareErrors < Faraday::Middleware
+    def call(env)
+      @app.call(env)
+    rescue Faraday::Error => e
+      raise e unless e.response.is_a?(Hash)
+
+      logger = Logger.new($stdout)
+      logger.formatter = proc do |_severity, _datetime, _progname, msg|
+        "\033[31mModelsLab HTTP Error (spotted in modelslab-ruby #{VERSION}): #{msg}\n\033[0m"
+      end
+      logger.error(e.response[:body])
+
+      raise e
+    end
+  end
+
+  class Configuration
+    attr_accessor :api_key, :uri_base, :request_timeout, :extra_headers
+
+    DEFAULT_URI_BASE = "https://modelslab.com/api/v6"
+    DEFAULT_REQUEST_TIMEOUT = 120
+
+    def initialize
+      @api_key = nil
+      @uri_base = DEFAULT_URI_BASE
+      @request_timeout = DEFAULT_REQUEST_TIMEOUT
+      @extra_headers = {}
+    end
+  end
+
+  class << self
+    attr_writer :configuration
+  end
+
+  def self.configuration
+    @configuration ||= ModelsLab::Configuration.new
+  end
+
+  def self.configure
+    yield(configuration)
+  end
+end

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: modelslab-ruby
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Florian Felsing
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2024-08-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: faraday-multipart
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+description: A Ruby wrapper for the ModelsLab API, focusing on text-to-image generation.
+email:
+- florian.felsing@googlemail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/modelslab.rb
+- lib/modelslab/client.rb
+- lib/modelslab/http.rb
+- lib/modelslab/images.rb
+- lib/modelslab/version.rb
+homepage: https://github.com/florianfelsing/modelslab-ruby
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+  homepage_uri: https://github.com/florianfelsing/modelslab-ruby
+  source_code_uri: https://github.com/florianfelsing/modelslab-ruby
+  changelog_uri: https://github.com/florianfelsing/modelslab-ruby/blob/main/CHANGELOG.md
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.19
+signing_key: 
+specification_version: 4
+summary: Ruby client for ModelsLab API
+test_files: []