roseflow-stabilityai 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 52d0eb1ead98627dee65b61a5af00bb0f705ef83b1744c7db3e144f2649a16ed
+  data.tar.gz: eab68fcb891590911d69d3fb3e2f1ec12f9bb0ae36f3b196d7ec9527b68f9fed
+SHA512:
+  metadata.gz: 85894910204e152bb7ed3cd3179777b7e9e6d5a604a53fca7692893b1a3f4bb5048efeb094173b496a0a8be196a63bfc220edcabf030364ceba1493d8a8cd01a
+  data.tar.gz: e124b655082293c97b034512023e2061189cdcd9f90a5686b4a2fcbc22a27ded8f2da1862615d5cb4b81b845c22ffaea0a730b6ea9c5332cedd21ae3a40c1d42

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/.standard.yml

@@ -0,0 +1,3 @@
+# For available configuration options, see:
+#   https://github.com/testdouble/standard
+ruby_version: 2.6

data/CHANGELOG.md

@@ -0,0 +1,4 @@
+## [0.1.0] - 2023-08-09
+
+- Initial release
+- Implements text-to-image, image-to-image, SDX4 upscaling, and masking

data/CODE_OF_CONDUCT.md

@@ -0,0 +1,7 @@
+# Contributor Code of Conduct
+
+The Roseflow team is committed to fostering a welcoming community.
+
+**Our Code of Conduct can be found here**:
+
+https://roseflow.ai/conduct

data/Gemfile

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in roseflow-stabilityai.gemspec
+gemspec
+
+gem "rake", "~> 13.0"
+gem "rspec", "~> 3.0"
+gem "standard", "~> 1.3"
+gem "ulid-ruby"
+
+eval_gemfile "Gemfile.local" if File.exist?("Gemfile.local")

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2023 Lauri Jutila
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,47 @@
+# Roseflow Stability-AI integration
+
+This gem adds Stability-AI API support and integration for Roseflow, a framework for interacting with AI in Ruby.
+
+## Prerequisites
+
+To use this gem effectively, you need the [core Roseflow gem](https://github.com/roseflow-ai/roseflow).
+
+## Installation
+
+Install the gem and add to the application's Gemfile by executing:
+
+    $ bundle add roseflow-stabilityai
+
+If bundler is not being used to manage dependencies, install the gem by executing:
+
+    $ gem install roseflow-stabilityai
+
+## Usage
+
+See full documentation how to configure and use Roseflow with StabilityAI at [docs.roseflow.ai](https://docs.roseflow.ai/stabilityai).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/roseflow-ai/roseflow-stabilityai.
+
+## Community
+
+### Discord
+
+Join us in our [Discord](https://discord.gg/roseflow).
+
+### Twitter
+
+Connect with the core team on Twitter.
+
+<a href="https://twitter.com/ljuti" target="_blank">
+  <img alt="Twitter Follow" src="https://img.shields.io/twitter/follow/ljuti?logo=twitter&style=social">
+</a>
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Code of Conduct
+
+Everyone interacting in the Roseflow OpenAI project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/roseflow-ai/roseflow-stabilityai/blob/main/CODE_OF_CONDUCT.md).

data/Rakefile

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+require "standard/rake"
+
+task default: %i[spec standard]

data/lib/roseflow/ai/models/stabilityai_adapter.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require "roseflow/ai/model_interface"
+require "roseflow/ai/models/base_adapter"
+
+module Roseflow
+  module AI
+    module Models
+      class StabilityAIAdapter < BaseAdapter
+        include ModelInterface
+
+        def configuration
+          @configuration ||= StabilityAI::Model::Configuration.new(engine_id: @model.name)
+        end
+
+        def call(operation, options, &block)
+          @model.call(operation, options, &block)
+        end
+
+        def text_to_image(options = {})
+          @model.call(:text_to_image, options)
+        end
+
+        def image_to_image(options = {})
+          @model.call(:image_to_image, options)
+        end
+
+        def upscale(options = {})
+          @model.call(:upscale, options)
+        end
+
+        def masking(options = {})
+          @model.call(:masking, options)
+        end
+
+        def operations
+          @model.operations
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/client.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require "faraday"
+require "faraday/multipart"
+require "faraday/retry"
+require "faraday/typhoeus"
+require "roseflow/types"
+require "roseflow/stabilityai/config"
+require "roseflow/stabilityai/model"
+
+module Roseflow
+  module StabilityAI
+    class Client
+      FARADAY_RETRY_OPTIONS = {
+        max: 3,
+        interval: 0.05,
+        interval_randomness: 0.5,
+        backoff_factor: 2,
+      }
+
+      def initialize(config = Config.new, provider = nil)
+        @config = config
+        @provider = provider
+      end
+
+      def models
+        response = connection.get("/v1/engines/list")
+        body = JSON.parse(response.body)
+      end
+
+      def post(operation)
+        if operation.multipart?
+          response = multipart_connection.post(operation.path) do |request|
+            request.body = operation.body
+          end
+        else
+          response = connection.post(operation.path) do |request|
+            request.body = operation.body
+          end
+        end
+        response
+      end
+
+      private
+
+      attr_reader :config, :provider
+
+      def connection
+        @connection ||= Faraday.new(
+          url: config.api_url,
+          headers: {
+            "Stability-Client-ID" => config.client_id,
+            "Stability-Client-Version" => config.client_version,
+          },
+        ) do |faraday|
+          faraday.request :authorization, "Bearer", -> { config.api_key }
+          faraday.request :json
+          faraday.request :retry, FARADAY_RETRY_OPTIONS
+          faraday.adapter :typhoeus
+        end
+      end
+
+      def multipart_connection
+        @multipart_connection ||= Faraday.new(
+          url: config.api_url,
+          headers: {
+            "Stability-Client-ID" => config.client_id,
+            "Stability-Client-Version" => config.client_version,
+          },
+        ) do |faraday|
+          faraday.request :authorization, "Bearer", -> { config.api_key }
+          faraday.request :multipart
+          faraday.adapter :typhoeus
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/config.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require "anyway_config"
+
+module Roseflow
+  module StabilityAI
+    class Config < Anyway::Config
+      config_name :stabilityai
+
+      attr_config :api_key
+      attr_config api_url: "https://api.stability.ai"
+      attr_config client_id: "Roseflow Stability-AI Ruby Client"
+      attr_config client_version: "0.1.0"
+
+      required :api_key
+    end
+  end
+end

data/lib/roseflow/stabilityai/model.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require "roseflow/stabilityai/operation_handler"
+require "roseflow/stabilityai/response_handler"
+
+module Roseflow
+  module StabilityAI
+    class Model
+      attr_reader :name
+
+      def initialize(model_config, provider = nil)
+        @model_config = model_config
+        @provider = provider
+        assign_attributes
+      end
+
+      def call(operation, options)
+        operation = OperationHandler.new(operation, options).call
+        ResponseHandler.new(operation, client.post(operation)).call
+      end
+
+      private
+
+      attr_reader :model_config
+      attr_reader :provider
+
+      def client
+        @client ||= Client.new(provider.send(:config), provider)
+      end
+
+      def assign_attributes
+        @name = model_config.fetch("id")
+        @description = model_config.fetch("description")
+        @descriptive_name = model_config.fetch("name")
+        @type = model_config.fetch("type")
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/model_repository.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require "active_support/core_ext/module/delegation"
+
+module Roseflow
+  module StabilityAI
+    class ModelRepository
+      attr_reader :models
+
+      delegate :each, :all, :first, :last, to: :models
+
+      def initialize(provider)
+        @provider = provider
+        @models = load_models
+      end
+
+      def find(name)
+        @models.select { |model| model.name == name }.first
+      end
+
+      private
+
+      attr_reader :provider
+
+      def load_models
+        provider.client.models.map do |model|
+          Model.new(model, provider)
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/operation_handler.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require_relative "operations/text_to_image"
+require_relative "operations/image_to_image"
+require_relative "operations/upscale"
+require_relative "operations/masking"
+
+module Roseflow
+  module StabilityAI
+    class OperationHandler
+      OPERATION_CLASSES = {
+        text_to_image: Operations::TextToImage,
+        image_to_image: Operations::ImageToImage,
+        upscale: Operations::Upscale,
+        masking: Operations::Masking,
+      }
+
+      def initialize(operation, options = {})
+        @operation = operation
+        @options = options
+      end
+
+      def call
+        operation_class.new(@options)
+      end
+
+      private
+
+      def operation_class
+        OPERATION_CLASSES.fetch(@operation) do
+          raise ArgumentError, "Invalid operation: #{@operation}"
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/operations/base.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Roseflow
+  module StabilityAI
+    module Operations
+      class Base < Dry::Struct
+        transform_keys(&:to_sym)
+
+        attribute :engine_id, Types::String
+
+        def excluded_keys
+          [:path]
+        end
+
+        def multipart?
+          true
+        end
+
+        def body
+          to_h.except(*excluded_keys)
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/operations/image_to_image.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require "roseflow/stabilityai/operations/base"
+
+module Roseflow
+  module StabilityAI
+    module Operations
+      class ImageToImage < Base
+        attribute :text_prompts, Types::Array do
+          attribute :text, Types::String
+          attribute :weight, Types::Float.default(1.0)
+        end
+        attribute :init_image, Types::String
+        attribute :init_image_mode, Types::String.default("IMAGE_STRENGTH")
+        attribute :image_strength, Types::Float.default(0.35)
+        attribute :cfg_scale, Types::Integer.default(7)
+        attribute :clip_guidance_preset, Types::String.default("NONE")
+        attribute? :sampler, Types::String
+        attribute :samples, Types::Integer.default(1)
+        attribute :seed, Types::Integer.default(0)
+        attribute :style_preset, Types::String.default("photographic")
+
+        def path
+          "/v1/generation/#{engine_id}/image-to-image"
+        end
+
+        def type
+          :image_to_image
+        end
+
+        def excluded_keys
+          [:path, :engine_id, :text_prompts]
+        end
+
+        def body
+          initial = to_h.except(*excluded_keys)
+          initial.merge(
+            init_image: Faraday::Multipart::ParamPart.new(
+              init_image,
+              "image/png",
+            ),
+            text_prompts: text_prompts.each_with_index.map do |item, index|
+              [index, item.to_h]
+            end.to_h,
+          )
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/operations/masking.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require "roseflow/stabilityai/operations/base"
+
+module Roseflow
+  module StabilityAI
+    module Operations
+      class Masking < Base
+        attribute :text_prompts, Types::Array do
+          attribute :text, Types::String
+          attribute :weight, Types::Float.default(1.0)
+        end
+        attribute :init_image, Types::String
+        attribute :mask_image, Types::String
+        attribute :mask_source, Types::String.default("INIT_IMAGE_ALPHA")
+        attribute :cfg_scale, Types::Integer.default(7)
+        attribute :clip_guidance_preset, Types::String.default("NONE")
+        attribute? :sampler, Types::String
+        attribute :samples, Types::Integer.default(1)
+        attribute :seed, Types::Integer.default(0)
+        attribute :steps, Types::Integer.default(50)
+        attribute :style_preset, Types::String.default("photographic")
+
+        def path
+          "/v1/generation/#{engine_id}/image-to-image/masking"
+        end
+
+        def excluded_keys
+          [:path, :engine_id, :text_prompts, :mask_image]
+        end
+
+        def body
+          to_h.except(*excluded_keys).merge(
+            init_image: Faraday::Multipart::ParamPart.new(
+              init_image,
+              "image/png",
+            ),
+            # mask_image: Faraday::Multipart::ParamPart.new(
+            #   mask_image,
+            #   "image/png",
+            # ),
+            text_prompts: text_prompts.each_with_index.map do |item, index|
+              [index, item.to_h]
+            end.to_h,
+          )
+        end
+
+        def type
+          :masking
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/operations/text_to_image.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Roseflow
+  module StabilityAI
+    module Operations
+      class TextToImage < Base
+        attribute :height, Types::Integer.default(512)
+        attribute :width, Types::Integer.default(512)
+        attribute :text_prompts, Types::Array do
+          attribute :text, Types::String
+          attribute :weight, Types::Float.default(1.0)
+        end
+        attribute :cfg_scale, Types::Integer.default(7)
+        attribute :clip_guidance_preset, Types::String.default("NONE")
+        attribute? :sampler, Types::String
+        attribute :samples, Types::Integer.default(1)
+        attribute :seed, Types::Integer.default(0)
+        attribute :steps, Types::Integer.default(50)
+        attribute? :style_preset, Types::String.default("photographic")
+
+        def type
+          :text_to_image
+        end
+
+        def multipart?
+          false
+        end
+
+        def path
+          "/v1/generation/#{engine_id}/text-to-image"
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/operations/upscale.rb

@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+require "roseflow/stabilityai/operations/base"
+
+# TODO: Very sharp operation class currently, needs to handle edge cases better.
+module Roseflow
+  module StabilityAI
+    module Operations
+      class Upscale < Base
+        attribute :upscale_type, Types::String
+
+        attribute? :esrgan do
+          attribute :image, Types::String
+          attribute? :width, Types::Integer
+          attribute? :height, Types::Integer
+        end
+
+        attribute? :sdx4 do
+          attribute :image, Types::String
+          attribute? :width, Types::Integer
+          attribute? :height, Types::Integer
+          attribute :text_prompts, Types::Array do
+            attribute :text, Types::String
+            attribute :weight, Types::Float.default(1.0)
+          end
+          attribute :seed, Types::Integer.default(0)
+          attribute :steps, Types::Integer.default(50)
+          attribute :cfg_scale, Types::Integer.default(7)
+        end
+
+        def path
+          "/v1/generation/#{engine_id}/image-to-image/upscale"
+        end
+
+        def excluded_keys
+          [:path, :engine_id, :text_prompts, :upscale_type, :esrgan, :sdx4]
+        end
+
+        def body
+          case upscale_type.to_sym
+          when :sdx4
+            sdx4_body
+          when :esrgan
+            esrgan_body
+          end
+        end
+
+        def sdx4_body
+          initial = to_h.except(*excluded_keys)
+          sdx4.to_h.merge(
+            image: Faraday::Multipart::ParamPart.new(
+              sdx4.image,
+              "image/png",
+            ),
+            text_prompts: sdx4.text_prompts.each_with_index.map do |item, index|
+              [index, item.to_h]
+            end.to_h,
+          ).merge(initial)
+        end
+
+        def esrgan_body
+          initial = to_h.except(*excluded_keys)
+          esrgan.to_h.merge(
+            image: Faraday::Multipart::ParamPart.new(
+              esrgan.image,
+              "image/png",
+            ),
+          ).merge(initial)
+        end
+
+        def type
+          :upscale
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/provider.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Roseflow
+  module StabilityAI
+    class Provider
+      def initialize(config = Config.new)
+        @config = config
+      end
+
+      def models
+        @models ||= ModelRepository.new(self)
+      end
+
+      def client
+        @client ||= Client.new(config, self)
+      end
+
+      def call(operation, options, &block)
+        models.find(options.fetch(:engine_id)).call(operation, options, &block)
+      end
+
+      private
+
+      attr_reader :config
+    end
+  end
+end

data/lib/roseflow/stabilityai/response_handler.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require "roseflow/stabilityai/responses/text_to_image_response"
+require "roseflow/stabilityai/responses/image_to_image_response"
+require "roseflow/stabilityai/responses/upscale_response"
+require "roseflow/stabilityai/responses/masking_response"
+require "roseflow/stabilityai/responses/error_response"
+
+module Roseflow
+  module StabilityAI
+    class ResponseHandler
+      def initialize(operation, response)
+        @operation = operation
+        @response = response
+      end
+
+      def call
+        if response.success?
+          handle_successful_response
+        else
+          handle_error_response
+        end
+      end
+
+      private
+
+      attr_reader :operation, :response
+
+      def handle_successful_response
+        json = JSON.parse(response.body)
+
+        case operation.type
+        when :text_to_image
+          Responses::TextToImageResponse.new(json)
+        when :image_to_image
+          Responses::ImageToImageResponse.new(json)
+        when :upscale
+          Responses::UpscaleResponse.new(json)
+        when :masking
+          Responses::MaskingResponse.new(json)
+        else
+          raise ArgumentError, "Invalid operation: #{operation.type}"
+        end
+      end
+
+      def handle_error_response
+        puts response.status
+        puts JSON.parse(response.body)
+        Responses::ErrorResponse.from(
+          code: response.status,
+          operation: operation.type,
+          body: JSON.parse(response.body),
+        )
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/responses/base_response.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require "dry/struct"
+
+module Roseflow
+  module StabilityAI
+    module Responses
+      class BaseResponse < Dry::Struct
+        transform_keys(&:to_sym)
+
+        def success?
+          true
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/responses/error_response.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require "dry/struct"
+
+module Roseflow
+  module StabilityAI
+    module Responses
+      class ErrorResponse < Dry::Struct
+        transform_keys(&:to_sym)
+
+        attribute :code, Types::Integer
+        attribute :operation, Types::String
+
+        def success?
+          false
+        end
+
+        # TODO: Implement this method
+        def self.from(code: nil, operation: nil, body: nil)
+        end
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/responses/image_to_image_response.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require_relative "./base_response"
+
+module Roseflow
+  module StabilityAI
+    module Responses
+      class ImageToImageResponse < BaseResponse
+        transform_keys { |key| key.to_s.underscore.to_sym }
+
+        attribute :artifacts, Types::Array do
+          attribute :base64, Types::String
+          attribute :finish_reason, Types::String
+          attribute :seed, Types::Integer
+        end
+
+        alias_method :images, :artifacts
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/responses/masking_response.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require_relative "./base_response"
+
+module Roseflow
+  module StabilityAI
+    module Responses
+      class MaskingResponse < BaseResponse
+        transform_keys { |key| key.to_s.underscore.to_sym }
+
+        attribute :artifacts, Types::Array do
+          attribute :base64, Types::String
+          attribute :finish_reason, Types::String
+          attribute :seed, Types::Integer
+        end
+
+        alias_method :images, :artifacts
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/responses/text_to_image_response.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require_relative "./base_response"
+
+module Roseflow
+  module StabilityAI
+    module Responses
+      class TextToImageResponse < BaseResponse
+        transform_keys { |key| key.to_s.underscore.to_sym }
+
+        attribute :artifacts, Types::Array do
+          attribute :base64, Types::String
+          attribute :finish_reason, Types::String
+          attribute :seed, Types::Integer
+        end
+
+        alias_method :images, :artifacts
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/responses/upscale_response.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require_relative "./base_response"
+
+module Roseflow
+  module StabilityAI
+    module Responses
+      class UpscaleResponse < BaseResponse
+        transform_keys { |key| key.to_s.underscore.to_sym }
+
+        attribute :artifacts, Types::Array do
+          attribute :base64, Types::String
+          attribute :finish_reason, Types::String
+          attribute :seed, Types::Integer
+        end
+
+        alias_method :images, :artifacts
+      end
+    end
+  end
+end

data/lib/roseflow/stabilityai/version.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Roseflow
+  module StabilityAI
+    def self.gem_version
+      Gem::Version.new VERSION::STRING
+    end
+
+    module VERSION
+      MAJOR = 0
+      MINOR = 1
+      PATCH = 0
+      PRE = nil
+
+      STRING = [MAJOR, MINOR, PATCH, PRE].compact.join(".")
+    end
+  end
+end

data/lib/roseflow/stabilityai.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require "roseflow/stabilityai/version"
+require "roseflow/stabilityai/config"
+require "roseflow/stabilityai/client"
+require "roseflow/stabilityai/model_repository"
+require "roseflow/stabilityai/model"
+require "roseflow/stabilityai/operation_handler"
+require "roseflow/stabilityai/provider"
+require "roseflow/stabilityai/response_handler"
+
+module Roseflow
+  module StabilityAI
+    class Error < StandardError; end
+  end
+end

data/roseflow-stabilityai.gemspec

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require_relative "lib/roseflow/stabilityai/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "roseflow-stabilityai"
+  spec.version = Roseflow::StabilityAI.gem_version
+  spec.authors = ["Lauri Jutila"]
+  spec.email = ["ljuti@users.noreply.github.com"]
+
+  spec.summary = "Roseflow meets StabilityAI"
+  spec.description = "StabilityAI integration and models for Roseflow"
+  spec.homepage = "https://github.com/roseflow-ai/roseflow-stabilityai"
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 3.2.0"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/roseflow-ai/roseflow-stabilityai"
+  spec.metadata["changelog_uri"] = "https://github.com/roseflow-ai/roseflow-stabilityai/blob/main/CHANGELOG.md"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(__dir__) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      (File.expand_path(f) == __FILE__) || f.start_with?(*%w[bin/ test/ spec/ features/ .git .circleci appveyor])
+    end
+  end
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "activesupport"
+  spec.add_dependency "anyway_config", "~> 2.0"
+  spec.add_dependency "dry-struct"
+  spec.add_dependency "faraday"
+  spec.add_dependency "faraday-retry"
+  spec.add_dependency "faraday-typhoeus"
+  spec.add_dependency "faraday-multipart"
+
+  spec.add_development_dependency "awesome_print"
+  spec.add_development_dependency "pry"
+  spec.add_development_dependency "pry-byebug"
+  spec.add_development_dependency "roseflow"
+  spec.add_development_dependency "webmock"
+  spec.add_development_dependency "vcr"
+end

data/sig/roseflow/stabilityai.rbs

@@ -0,0 +1,6 @@
+module Roseflow
+  module Stabilityai
+    VERSION: String
+    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+  end
+end

metadata

@@ -0,0 +1,259 @@
+--- !ruby/object:Gem::Specification
+name: roseflow-stabilityai
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Lauri Jutila
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2023-08-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: anyway_config
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: dry-struct
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: faraday-retry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: faraday-typhoeus
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: faraday-multipart
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: awesome_print
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: roseflow
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: vcr
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: StabilityAI integration and models for Roseflow
+email:
+- ljuti@users.noreply.github.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- ".standard.yml"
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/roseflow/ai/models/stabilityai_adapter.rb
+- lib/roseflow/stabilityai.rb
+- lib/roseflow/stabilityai/client.rb
+- lib/roseflow/stabilityai/config.rb
+- lib/roseflow/stabilityai/model.rb
+- lib/roseflow/stabilityai/model_repository.rb
+- lib/roseflow/stabilityai/operation_handler.rb
+- lib/roseflow/stabilityai/operations/base.rb
+- lib/roseflow/stabilityai/operations/image_to_image.rb
+- lib/roseflow/stabilityai/operations/masking.rb
+- lib/roseflow/stabilityai/operations/text_to_image.rb
+- lib/roseflow/stabilityai/operations/upscale.rb
+- lib/roseflow/stabilityai/provider.rb
+- lib/roseflow/stabilityai/response_handler.rb
+- lib/roseflow/stabilityai/responses/base_response.rb
+- lib/roseflow/stabilityai/responses/error_response.rb
+- lib/roseflow/stabilityai/responses/image_to_image_response.rb
+- lib/roseflow/stabilityai/responses/masking_response.rb
+- lib/roseflow/stabilityai/responses/text_to_image_response.rb
+- lib/roseflow/stabilityai/responses/upscale_response.rb
+- lib/roseflow/stabilityai/version.rb
+- roseflow-stabilityai.gemspec
+- sig/roseflow/stabilityai.rbs
+homepage: https://github.com/roseflow-ai/roseflow-stabilityai
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/roseflow-ai/roseflow-stabilityai
+  source_code_uri: https://github.com/roseflow-ai/roseflow-stabilityai
+  changelog_uri: https://github.com/roseflow-ai/roseflow-stabilityai/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.2.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.1
+signing_key:
+specification_version: 4
+summary: Roseflow meets StabilityAI
+test_files: []