stability_sdk 0.2.0 → 0.3.0

data/lib/generation_pb.rb

@@ -21,6 +21,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :index, :uint32, 8
       optional :finish_reason, :enum, 9, "gooseai.FinishReason"
       optional :seed, :uint32, 10
+      optional :uuid, :string, 12
+      optional :size, :uint64, 13
       oneof :data do
         optional :binary, :bytes, 5
         optional :text, :string, 6
@@ -40,20 +42,6 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
         optional :artifact, :message, 4, "gooseai.Artifact"
       end
     end
-    add_message "gooseai.AnswerMeta" do
-      proto3_optional :gpu_id, :string, 1
-      proto3_optional :cpu_id, :string, 2
-      proto3_optional :node_id, :string, 3
-      proto3_optional :engine_id, :string, 4
-    end
-    add_message "gooseai.Answer" do
-      optional :answer_id, :string, 1
-      optional :request_id, :string, 2
-      optional :received, :uint64, 3
-      optional :created, :uint64, 4
-      proto3_optional :meta, :message, 6, "gooseai.AnswerMeta"
-      repeated :artifacts, :message, 7, "gooseai.Artifact"
-    end
     add_message "gooseai.SamplerParameters" do
       proto3_optional :eta, :float, 1
       proto3_optional :sampling_steps, :uint64, 2
@@ -63,15 +51,48 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     end
     add_message "gooseai.ConditionerParameters" do
       proto3_optional :vector_adjust_prior, :string, 1
+      proto3_optional :conditioner, :message, 2, "gooseai.Model"
     end
     add_message "gooseai.ScheduleParameters" do
-      optional :start, :float, 1
-      optional :end, :float, 2
+      proto3_optional :start, :float, 1
+      proto3_optional :end, :float, 2
+      proto3_optional :value, :float, 3
     end
     add_message "gooseai.StepParameter" do
       optional :scaled_step, :float, 1
       proto3_optional :sampler, :message, 2, "gooseai.SamplerParameters"
       proto3_optional :schedule, :message, 3, "gooseai.ScheduleParameters"
+      proto3_optional :guidance, :message, 4, "gooseai.GuidanceParameters"
+    end
+    add_message "gooseai.Model" do
+      optional :architecture, :enum, 1, "gooseai.ModelArchitecture"
+      optional :publisher, :string, 2
+      optional :dataset, :string, 3
+      optional :version, :float, 4
+      optional :semantic_version, :string, 5
+      optional :alias, :string, 6
+    end
+    add_message "gooseai.CutoutParameters" do
+      repeated :cutouts, :message, 1, "gooseai.CutoutParameters"
+      proto3_optional :count, :uint32, 2
+      proto3_optional :gray, :float, 3
+      proto3_optional :blur, :float, 4
+      proto3_optional :size_power, :float, 5
+    end
+    add_message "gooseai.GuidanceScheduleParameters" do
+      optional :duration, :float, 1
+      optional :value, :float, 2
+    end
+    add_message "gooseai.GuidanceInstanceParameters" do
+      repeated :models, :message, 2, "gooseai.Model"
+      proto3_optional :guidance_strength, :float, 3
+      repeated :schedule, :message, 4, "gooseai.GuidanceScheduleParameters"
+      proto3_optional :cutouts, :message, 5, "gooseai.CutoutParameters"
+      proto3_optional :prompt, :message, 6, "gooseai.Prompt"
+    end
+    add_message "gooseai.GuidanceParameters" do
+      optional :guidance_preset, :enum, 1, "gooseai.GuidancePreset"
+      repeated :instances, :message, 2, "gooseai.GuidanceInstanceParameters"
     end
     add_message "gooseai.TransformType" do
       oneof :type do
@@ -104,17 +125,51 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       repeated :exceeds, :message, 2, "gooseai.ClassifierCategory"
       proto3_optional :realized_action, :enum, 3, "gooseai.Action"
     end
+    add_message "gooseai.AssetParameters" do
+      optional :action, :enum, 1, "gooseai.AssetAction"
+      optional :project_id, :string, 2
+      optional :use, :enum, 3, "gooseai.AssetUse"
+    end
+    add_message "gooseai.AnswerMeta" do
+      proto3_optional :gpu_id, :string, 1
+      proto3_optional :cpu_id, :string, 2
+      proto3_optional :node_id, :string, 3
+      proto3_optional :engine_id, :string, 4
+    end
+    add_message "gooseai.Answer" do
+      optional :answer_id, :string, 1
+      optional :request_id, :string, 2
+      optional :received, :uint64, 3
+      optional :created, :uint64, 4
+      proto3_optional :meta, :message, 6, "gooseai.AnswerMeta"
+      repeated :artifacts, :message, 7, "gooseai.Artifact"
+    end
     add_message "gooseai.Request" do
       optional :engine_id, :string, 1
       optional :request_id, :string, 2
       optional :requested_type, :enum, 3, "gooseai.ArtifactType"
       repeated :prompt, :message, 4, "gooseai.Prompt"
       proto3_optional :conditioner, :message, 6, "gooseai.ConditionerParameters"
-      proto3_optional :classifier, :message, 7, "gooseai.ClassifierParameters"
       oneof :params do
         optional :image, :message, 5, "gooseai.ImageParameters"
+        optional :classifier, :message, 7, "gooseai.ClassifierParameters"
+        optional :asset, :message, 8, "gooseai.AssetParameters"
       end
     end
+    add_message "gooseai.OnStatus" do
+      repeated :reason, :enum, 1, "gooseai.FinishReason"
+      proto3_optional :target, :string, 2
+      repeated :action, :enum, 3, "gooseai.StageAction"
+    end
+    add_message "gooseai.Stage" do
+      optional :id, :string, 1
+      optional :request, :message, 2, "gooseai.Request"
+      repeated :on_status, :message, 3, "gooseai.OnStatus"
+    end
+    add_message "gooseai.ChainRequest" do
+      optional :request_id, :string, 1
+      repeated :stage, :message, 2, "gooseai.Stage"
+    end
     add_enum "gooseai.FinishReason" do
       value :NULL, 0
       value :LENGTH, 1
@@ -147,6 +202,21 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :UPSCALER_GFPGAN, 1
       value :UPSCALER_ESRGAN, 2
     end
+    add_enum "gooseai.GuidancePreset" do
+      value :GUIDANCE_PRESET_NONE, 0
+      value :GUIDANCE_PRESET_SIMPLE, 1
+      value :GUIDANCE_PRESET_FAST_BLUE, 2
+      value :GUIDANCE_PRESET_FAST_GREEN, 3
+      value :GUIDANCE_PRESET_SLOW, 4
+      value :GUIDANCE_PRESET_SLOWER, 5
+      value :GUIDANCE_PRESET_SLOWEST, 6
+    end
+    add_enum "gooseai.ModelArchitecture" do
+      value :MODEL_ARCHITECTURE_NONE, 0
+      value :MODEL_ARCHITECTURE_CLIP_VIT, 1
+      value :MODEL_ARCHITECTURE_CLIP_RESNET, 2
+      value :MODEL_ARCHITECTURE_LDM, 3
+    end
     add_enum "gooseai.Action" do
       value :ACTION_PASSTHROUGH, 0
       value :ACTION_REGENERATE_DUPLICATE, 1
@@ -159,6 +229,23 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :CLSFR_MODE_ZEROSHOT, 0
       value :CLSFR_MODE_MULTICLASS, 1
     end
+    add_enum "gooseai.AssetAction" do
+      value :ASSET_PUT, 0
+      value :ASSET_GET, 1
+      value :ASSET_DELETE, 2
+    end
+    add_enum "gooseai.AssetUse" do
+      value :ASSET_USE_UNDEFINED, 0
+      value :ASSET_USE_INPUT, 1
+      value :ASSET_USE_OUTPUT, 2
+      value :ASSET_USE_INTERMEDIATE, 3
+      value :ASSET_USE_PROJECT, 4
+    end
+    add_enum "gooseai.StageAction" do
+      value :STAGE_ACTION_PASS, 0
+      value :STAGE_ACTION_DISCARD, 1
+      value :STAGE_ACTION_RETURN, 2
+    end
   end
 end
 
@@ -168,22 +255,36 @@ module Gooseai
   Artifact = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Artifact").msgclass
   PromptParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.PromptParameters").msgclass
   Prompt = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Prompt").msgclass
-  AnswerMeta = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AnswerMeta").msgclass
-  Answer = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Answer").msgclass
   SamplerParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.SamplerParameters").msgclass
   ConditionerParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ConditionerParameters").msgclass
   ScheduleParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ScheduleParameters").msgclass
   StepParameter = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.StepParameter").msgclass
+  Model = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Model").msgclass
+  CutoutParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CutoutParameters").msgclass
+  GuidanceScheduleParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GuidanceScheduleParameters").msgclass
+  GuidanceInstanceParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GuidanceInstanceParameters").msgclass
+  GuidanceParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GuidanceParameters").msgclass
   TransformType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.TransformType").msgclass
   ImageParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ImageParameters").msgclass
   ClassifierConcept = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierConcept").msgclass
   ClassifierCategory = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierCategory").msgclass
   ClassifierParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierParameters").msgclass
+  AssetParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AssetParameters").msgclass
+  AnswerMeta = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AnswerMeta").msgclass
+  Answer = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Answer").msgclass
   Request = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Request").msgclass
+  OnStatus = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OnStatus").msgclass
+  Stage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Stage").msgclass
+  ChainRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ChainRequest").msgclass
   FinishReason = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.FinishReason").enummodule
   ArtifactType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ArtifactType").enummodule
   DiffusionSampler = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.DiffusionSampler").enummodule
   Upscaler = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Upscaler").enummodule
+  GuidancePreset = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GuidancePreset").enummodule
+  ModelArchitecture = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ModelArchitecture").enummodule
   Action = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Action").enummodule
   ClassifierMode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierMode").enummodule
+  AssetAction = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AssetAction").enummodule
+  AssetUse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AssetUse").enummodule
+  StageAction = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.StageAction").enummodule
 end

data/lib/generation_services_pb.rb

@@ -6,6 +6,9 @@ require 'generation_pb'
 
 module Gooseai
   module GenerationService
+    #
+    # gRPC services
+    #
     class Service
 
       include ::GRPC::GenericService
@@ -15,6 +18,7 @@ module Gooseai
       self.service_name = 'gooseai.GenerationService'
 
       rpc :Generate, ::Gooseai::Request, stream(::Gooseai::Answer)
+      rpc :ChainGenerate, ::Gooseai::ChainRequest, stream(::Gooseai::Answer)
     end
 
     Stub = Service.rpc_stub_class

data/lib/stability_sdk/cli.rb

@@ -23,10 +23,15 @@ module StabilitySDK
           logger.warn "not implemented for ArtifactType #{artifact.type}"
         end
 
+        if artifact.finish_reason == :FILTER
+          logger.debug "the generated image is filtered"
+        end
+
         next if filename == "" || contents == ""
 
         File.open(filename, "wb") do |f|
           f.write(contents)
+          logger.debug "wrote #{artifact.type} to #{filename}"
         end
       end
     end

data/lib/stability_sdk/client.rb

@@ -1,5 +1,6 @@
 require "grpc"
 require "generation_services_pb"
+require "logger"
 
 module StabilitySDK
   class Client
@@ -14,7 +15,7 @@ module StabilitySDK
     DEFAULT_START_SCHEDULE = 1.0
     DEFAULT_END_SCHEDULE = 0.01
 
-    sampler_algorithms = {
+    SAMPLER_ALGORITHMS = {
       "ddim": Gooseai::DiffusionSampler::SAMPLER_DDIM,
       "plms": Gooseai::DiffusionSampler::SAMPLER_DDPM,
       "k_euler": Gooseai::DiffusionSampler::SAMPLER_K_EULER,
@@ -31,24 +32,42 @@ module StabilitySDK
       call_creds = GRPC::Core::CallCredentials.new(proc { { "authorization" => "Bearer #{options[:api_key]}" } })
       creds = channel_creds.compose(call_creds)
 
-      @stub = Gooseai::GenerationService::Stub.new(host, creds)
+      stub_params = {}
+      [:channel_override, :timeout, :propagate_mask, :channel_args, :interceptors].each do |kw|
+        stub_params[kw] = options[kw] if options.has_key?(kw)
+      end
+
+      if options.has_key?(:logger)
+        @logger = options[:logger]
+      else
+        logger = Logger.new(STDOUT)
+        logger.level = Logger::WARN
+        @logger = logger
+      end
+
+      @stub = Gooseai::GenerationService::Stub.new(host, creds, **stub_params)
     end
 
     def generate(prompt, options, &block)
-      width = options.has_key?(:width) ? options[:width].to_i : DEFAULT_IMAGE_WIDTH
-      height = options.has_key?(:height) ? options[:height].to_i : DEFAULT_IMAGE_HEIGHT
-      samples = options.has_key?(:num_samples) ? [:num_samples].to_i : DEFAULT_SAMPLE_SIZE
-      steps = options.has_key?(:steps) ? options[:steps].to_i : DEFAULT_STEPS
+      width = options.has_key?(:width) ? options[:width] : DEFAULT_IMAGE_WIDTH
+      height = options.has_key?(:height) ? options[:height] : DEFAULT_IMAGE_HEIGHT
+
+      if width % 64 != 0 || height % 64 != 0
+        raise InvalidParameter, "width and height must be a multiple of 64"
+      end
+
+      samples = options.has_key?(:num_samples) ? options[:num_samples] : DEFAULT_SAMPLE_SIZE
+      steps = options.has_key?(:steps) ? options[:steps] : DEFAULT_STEPS
       seed = options.has_key?(:seed) ? [options[:seed]] : [rand(4294967295)]
       transform = Gooseai::TransformType.new(
-        diffusion: options.has_key?(:sampler) ? sampler_algorithms[options[:sampler]] : DEFAULT_SAMPLER_ALGORITHM,
+        diffusion: options.has_key?(:sampler) ? SAMPLER_ALGORITHMS[options[:sampler].to_sym] : DEFAULT_SAMPLER_ALGORITHM,
       )
-      parameters = [Gooseai::StepParameter.new(
+      step_parameter = Gooseai::StepParameter.new(
         scaled_step: 0,
         sampler: Gooseai::SamplerParameters.new(
-          cfg_scale: options.has_key?(:cfg_scale) ? options[:cfg_scale].to_f : DEFAULT_CFG_SCALE,
+          cfg_scale: options.has_key?(:cfg_scale) ? options[:cfg_scale] : DEFAULT_CFG_SCALE,
         ),
-      )]
+      )
 
       prompt_param = []
       if prompt != ""
@@ -56,21 +75,65 @@ module StabilitySDK
       end
       if options.has_key?(:init_image)
         prompt_param << init_image_to_prompt(options[:init_image])
-        parameters = [Gooseai::StepParameter.new(
-          scaled_step: 0,
-          sampler: Gooseai::SamplerParameters.new(
-            cfg_scale: options.has_key?(:cfg_scale) ? options[:cfg_scale].to_f : DEFAULT_CFG_SCALE,
-          ),
-          schedule: Gooseai::ScheduleParameters.new(
-            start: options.has_key?(:start_schedule) ? options[:start_schedule].to_f : DEFAULT_START_SCHEDULE,
-            end: options.has_key?(:end_schedule) ? options[:end_schedule].to_f : DEFAULT_END_SCHEDULE,
-          ),
-        )]
+        step_parameter.scaled_step = 0
+        step_parameter.sampler = Gooseai::SamplerParameters.new(
+          cfg_scale: options.has_key?(:cfg_scale) ? options[:cfg_scale] : DEFAULT_CFG_SCALE,
+        )
+        step_parameter.schedule = Gooseai::ScheduleParameters.new(
+          start: options.has_key?(:start_schedule) ? options[:start_schedule] : DEFAULT_START_SCHEDULE,
+          end: options.has_key?(:end_schedule) ? options[:end_schedule] : DEFAULT_END_SCHEDULE,
+        )
       end
       if options.has_key?(:mask_image)
         prompt_param << mask_image_to_prompt(options[:mask_image])
       end
 
+      # CLIP guidance
+      if options.has_key?(:guidance_preset) && options[:guidance_preset] != "GUIDANCE_PRESET_NONE"
+        step_parameter.sampler = nil
+
+        guidance_prompt =
+          if options.has_key?(:guidance_prompt)
+            Gooseai::Prompt.new(text: options[:guidance_prompt])
+          else
+            Gooseai::Prompt.new(text: prompt)
+          end
+
+        guidance_strength = nil
+        if options.has_key?(:guidance_strength) && options[:guidance_strength] != 0
+          guidance_strength = options[:guidance_strength]
+        end
+
+        models = nil
+        if options.has_key?(:guidance_models)
+          models = options[:guidance_models].map { |m| Gooseai::Model.new(alias: m) }
+        end
+
+        cutouts = nil
+        if options.has_key?(:guidance_cuts)
+          cutouts = Gooseai::CutoutParameters.new(count: options[:guidance_cuts])
+        end
+
+        step_parameter.guidance = Gooseai::GuidanceParameters.new(
+          guidance_preset: Gooseai::GuidancePreset.const_get(options[:guidance_preset].to_sym),
+          instances: [
+            Gooseai::GuidanceInstanceParameters.new(
+              guidance_strength: guidance_strength,
+              models: models,
+              cutouts: cutouts,
+              prompt: guidance_prompt,
+            ),
+          ],
+        )
+
+        if transform.diffusion != Gooseai::DiffusionSampler::SAMPLER_K_DPM_2_ANCESTRAL && transform.diffusion != Gooseai::DiffusionSampler::SAMPLER_K_EULER_ANCESTRAL
+          transform = Gooseai::TransformType.new(
+            diffusion: Gooseai::DiffusionSampler::SAMPLER_K_DPM_2_ANCESTRAL
+          )
+          @logger.warn "CLIP guidance is only supported with ancestral samplers. So override it with SAMPLER_K_DPM_2_ANCESTRAL."
+        end
+      end
+
       image_param = Gooseai::ImageParameters.new(
         width: width,
         height: height,
@@ -78,7 +141,7 @@ module StabilitySDK
         steps: steps,
         seed: seed,
         transform: transform,
-        parameters: parameters,
+        parameters: [step_parameter],
       )
 
       req = Gooseai::Request.new(
@@ -87,8 +150,20 @@ module StabilitySDK
         image: image_param
       )
 
+      @logger.debug "sending request."
+      start = Time.now
       @stub.generate(req).each do |answer|
+        duration = Time.now - start
+        if answer.artifacts.size > 0
+          artifact_types = answer.artifacts.map { |a| a.type }
+          @logger.debug "got #{answer.answer_id} with #{artifact_types} in #{duration.round(2)}s"
+        else
+          @logger.debug "got keepalive #{answer.answer_id} in #{duration.round(2)}s"
+        end
+
         block.call(answer)
+
+        start = Time.now
       end
     end
 

data/lib/stability_sdk/dashboard_client.rb

@@ -0,0 +1,31 @@
+require "grpc"
+require "dashboard_services_pb"
+
+module StabilitySDK
+  class DashboardClient
+    DEFAULT_API_HOST = "grpc.stability.ai:443"
+
+    def initialize(options={})
+      host = options[:api_host] || DEFAULT_API_HOST
+      channel_creds = options.has_key?(:ca_cert) ? GRPC::Core::ChannelCredentials.new(options[:ca_cert]) : GRPC::Core::ChannelCredentials.new
+      call_creds = GRPC::Core::CallCredentials.new(proc { { "authorization" => "Bearer #{options[:api_key]}" } })
+      creds = channel_creds.compose(call_creds)
+
+      stub_params = {}
+      [:channel_override, :timeout, :propagate_mask, :channel_args, :interceptors].each do |kw|
+        stub_params[kw] = options[kw] if options.has_key?(kw)
+      end
+
+      @stub = Gooseai::DashboardService::Stub.new(host, creds, **stub_params)
+    end
+
+    def get_me
+      @stub.get_me(Gooseai::EmptyRequest.new)
+    end
+
+    def get_organization
+      me = self.get_me
+      @stub.get_organization(Gooseai::GetChargesRequest.new(organization_id: me.organizations[0].organization.id))
+    end
+  end
+end

data/lib/stability_sdk/version.rb

@@ -1,3 +1,3 @@
 module StabilitySDK
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

data/lib/stability_sdk.rb

@@ -1,8 +1,10 @@
 require "stability_sdk/version"
 require "stability_sdk/client"
 require "stability_sdk/cli"
+require "stability_sdk/dashboard_client"
 
 module StabilitySDK
   class Error < StandardError; end
   class InsufficientParameter < StandardError; end
+  class InvalidParameter < StandardError; end
 end

data/stability_sdk.gemspec

@@ -10,6 +10,7 @@ Gem::Specification.new do |spec|
   spec.description   = "Interacting with stability.ai APIs (e.g. stable diffusion inference). Ruby client of https://github.com/Stability-AI/stability-sdk ."
   spec.homepage      = "https://github.com/cou929/stability-sdk-ruby"
   spec.required_ruby_version = Gem::Requirement.new(">= 2.3.0")
+  spec.license       = "MIT"
 
   spec.metadata["homepage_uri"] = spec.homepage
   spec.metadata["source_code_uri"] = "https://github.com/cou929/stability-sdk-ruby"
@@ -24,7 +25,9 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
-  spec.add_dependency "grpc"
-  spec.add_dependency "mime-types"
+  spec.add_dependency "grpc", ">= 1.41.1"
+  spec.add_dependency "mime-types", ">= 3.0.0"
+  spec.add_dependency "fastimage", "~> 2.2", ">= 2.2.6"
+  spec.add_dependency "thor", ">= 1.2.1"
   spec.add_development_dependency "grpc-tools"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stability_sdk
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-09-08 00:00:00.000000000 Z
+date: 2022-11-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc
@@ -16,28 +16,62 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 1.41.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 1.41.1
 - !ruby/object:Gem::Dependency
   name: mime-types
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 3.0.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 3.0.0
+- !ruby/object:Gem::Dependency
+  name: fastimage
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.2.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.2.6
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.2.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.2.1
 - !ruby/object:Gem::Dependency
   name: grpc-tools
   requirement: !ruby/object:Gem::Requirement
@@ -58,28 +92,34 @@ email:
 - cou929@gmail.com
 executables:
 - stability-client
+- stability-dashboard-client
 extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".gitmodules"
 - ".travis.yml"
 - Gemfile
-- Gemfile.lock
+- LICENSE
 - README.md
 - Rakefile
 - bin/console
 - bin/setup
 - exe/stability-client
+- exe/stability-dashboard-client
+- lib/dashboard_pb.rb
+- lib/dashboard_services_pb.rb
 - lib/generation_pb.rb
 - lib/generation_services_pb.rb
 - lib/stability_sdk.rb
 - lib/stability_sdk/cli.rb
 - lib/stability_sdk/client.rb
+- lib/stability_sdk/dashboard_client.rb
 - lib/stability_sdk/version.rb
-- proto/generation.proto
 - stability_sdk.gemspec
 homepage: https://github.com/cou929/stability-sdk-ruby
-licenses: []
+licenses:
+- MIT
 metadata:
   homepage_uri: https://github.com/cou929/stability-sdk-ruby
   source_code_uri: https://github.com/cou929/stability-sdk-ruby
@@ -99,7 +139,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
+rubygems_version: 3.1.6
 signing_key: 
 specification_version: 4
 summary: Ruby client for interacting with stability.ai APIs (e.g. stable diffusion

data/Gemfile.lock

@@ -1,34 +0,0 @@
-PATH
-  remote: .
-  specs:
-    stability_sdk (0.2.0)
-      grpc
-      mime-types
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    google-protobuf (3.21.5)
-    googleapis-common-protos-types (1.4.0)
-      google-protobuf (~> 3.14)
-    grpc (1.48.0)
-      google-protobuf (~> 3.19)
-      googleapis-common-protos-types (~> 1.0)
-    grpc-tools (1.48.0)
-    mime-types (3.4.1)
-      mime-types-data (~> 3.2015)
-    mime-types-data (3.2022.0105)
-    minitest (5.16.3)
-    rake (12.3.3)
-
-PLATFORMS
-  arm64-darwin-20
-
-DEPENDENCIES
-  grpc-tools
-  minitest (~> 5.0)
-  rake (~> 12.0)
-  stability_sdk!
-
-BUNDLED WITH
-   2.2.2