stability_sdk 0.2.9 → 0.2.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 27bb9b00352e1cf979dedd17438a5dc2a3dae93ed0d74a747e30f297152e81a2
-  data.tar.gz: c125901e07cc2f6d0704a423e2dad2541c3e31efbeb1ff69993f65bf804dcabb
+  metadata.gz: 431d7a70dc0a5a538abdbe9a486bf5c632d1e6a8e7e58dd7fedf1fa89b15722b
+  data.tar.gz: a6d5c05cf4915e7c91ceeb92864b2cc84f852ec40a61ce174ba57b3a97ec0df9
 SHA512:
-  metadata.gz: 4b79d8290c5c958f10a00f103efe6959b8e1f408d8ca3393453afb0a7433eda250252edab93195a1dd7885b9a3860aeffccd27833289f38ea373ff3bf1a274f0
-  data.tar.gz: 4481b59d12990bf00e5b716611dbd8055df7d4d262a4f1647ea2b217c9122cd790b671e76d51f1c082e83f2a0cebb1d62e2a80d7cb08f1ae96755db299a89dcb
+  metadata.gz: 7c5ba5862fd6446e243cdf54760fe596a4298e2d2860183df4cef847cc75d4414e9ceb21e8bd927e71e0fbacca9ce742628107197ca1e7c380abbbabff1e7cdc
+  data.tar.gz: 69b3cef394cbd5bd784bfd8953b1bd8721e4921e96f3a7bda79c0de3a733692853005008e1946cad75b24889df5dc1c8f9479238baac488a37a64ef49dbe39b7

data/.gitignore

@@ -6,3 +6,5 @@
 /pkg/
 /spec/reports/
 /tmp/
+Gemfile.lock
+.DS_Store

data/README.md

@@ -91,6 +91,25 @@ client.generate(prompt, options) do |answer|
 end
 ```
 
+### [Unstable] Dashboard API
+
+**This feature is in a very early stage of development.**
+
+Dashboard API is a way to interact with DreamStudio Web UIs, such as getting user info, payment info, etc.
+
+Currently, there is no canonical way to get the API key for the Dashboard API. You can retrieve the key by logging in to the [DreamStudio Web page](https://beta.dreamstudio.ai/dream) and inspect request the authorization header with Chrome Developer Tool. Please be aware that how to refresh the key or an expiration period is unknown. See also https://github.com/Stability-AI/stability-sdk/issues/23
+
+```sh
+# get user info
+STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client-unstable get_me
+
+# get organization info
+STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client-unstable get_organization
+
+# i.e, get remaining balance
+STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client-unstable get_organization | jq .paymentInfo.balance
+```
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/Rakefile

@@ -13,3 +13,8 @@ desc "Parse proto file and generate output"
 task :protoc do
   sh "grpc_tools_ruby_protoc -I api-interfaces/src/proto/ --ruby_out=lib --grpc_out=lib api-interfaces/src/proto/generation.proto"
 end
+
+desc "Compile a dashboard proto file"
+task :protoc_dashboard do
+  sh "grpc_tools_ruby_protoc -I api-interfaces/src/proto/ --ruby_out=lib --grpc_out=lib api-interfaces/src/proto/dashboard.proto"
+end

data/exe/stability-client

@@ -2,6 +2,7 @@
 
 require "optparse"
 require "mime/types"
+require "fastimage"
 require "logger"
 require "stability_sdk"
 
@@ -42,6 +43,11 @@ raise StabilitySDK::InsufficientParameter, "api key is required" if !options.has
 
 raise StabilitySDK::InsufficientParameter, "init_image is required if mask_image is provided" if options.has_key?(:mask_image) && !options.has_key?(:init_image)
 
+if options.has_key?(:init_image)
+  size = FastImage.size(options[:init_image])
+  raise StabilitySDK::InvalidParameter, "width and height of init_image must be a multiple of 64" if size[0] % 64 != 0 || size[1] % 64 != 0
+end
+
 client = StabilitySDK::Client.new(api_key: options[:api_key])
 
 client.generate(prompt, options) do |answer|

data/exe/stability-dashboard-client-unstable

@@ -0,0 +1,40 @@
+#!/usr/bin/env ruby
+
+require "thor"
+require "logger"
+require "stability_sdk"
+
+logger = Logger.new(STDOUT)
+logger.level = Logger::WARN
+
+Version = StabilitySDK::VERSION
+
+class CLI < Thor
+  class_option :api_key
+
+  desc "get_me", "get user info"
+  def get_me
+    client = self.client(options)
+    res = client.get_me
+    puts res.to_json
+  end
+
+  desc "get_organization", "get organization info"
+  def get_organization
+    client = self.client(options)
+    res = client.get_organization
+    puts res.to_json
+  end
+
+  private
+
+  def client(options)
+    api_key = options[:api_key]
+    api_key = ENV["STABILITY_SDK_DASHBOARD_API_KEY"] if ENV["STABILITY_SDK_DASHBOARD_API_KEY"]
+    raise StabilitySDK::InsufficientParameter, "api key is required" if api_key.nil?
+
+    return StabilitySDK::Unstable::DashboardClient.new(api_key: api_key)
+  end
+end
+
+CLI.start(ARGV)

data/lib/dashboard_pb.rb

@@ -0,0 +1,182 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: dashboard.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("dashboard.proto", :syntax => :proto3) do
+    add_message "gooseai.OrganizationMember" do
+      optional :organization, :message, 1, "gooseai.Organization"
+      proto3_optional :user, :message, 2, "gooseai.User"
+      optional :role, :enum, 3, "gooseai.OrganizationRole"
+      optional :is_default, :bool, 4
+    end
+    add_message "gooseai.OrganizationGrant" do
+      optional :amount_granted, :double, 1
+      optional :amount_used, :double, 2
+      optional :expires_at, :uint64, 3
+      optional :granted_at, :uint64, 4
+    end
+    add_message "gooseai.OrganizationPaymentInfo" do
+      optional :balance, :double, 1
+      repeated :grants, :message, 2, "gooseai.OrganizationGrant"
+    end
+    add_message "gooseai.OrganizationAutoCharge" do
+      optional :enabled, :bool, 1
+      optional :id, :string, 2
+      optional :created_at, :uint64, 3
+    end
+    add_message "gooseai.Organization" do
+      optional :id, :string, 1
+      optional :name, :string, 2
+      optional :description, :string, 3
+      repeated :members, :message, 4, "gooseai.OrganizationMember"
+      proto3_optional :payment_info, :message, 5, "gooseai.OrganizationPaymentInfo"
+      proto3_optional :stripe_customer_id, :string, 6
+      proto3_optional :auto_charge, :message, 7, "gooseai.OrganizationAutoCharge"
+    end
+    add_message "gooseai.APIKey" do
+      optional :key, :string, 1
+      optional :is_secret, :bool, 2
+      optional :created_at, :uint64, 3
+    end
+    add_message "gooseai.User" do
+      optional :id, :string, 1
+      proto3_optional :auth_id, :string, 2
+      optional :profile_picture, :string, 3
+      optional :email, :string, 4
+      repeated :organizations, :message, 5, "gooseai.OrganizationMember"
+      repeated :api_keys, :message, 7, "gooseai.APIKey"
+      optional :created_at, :uint64, 8
+      proto3_optional :email_verified, :bool, 9
+    end
+    add_message "gooseai.CostData" do
+      optional :amount_tokens, :uint32, 1
+      optional :amount_credits, :double, 2
+    end
+    add_message "gooseai.UsageMetric" do
+      optional :operation, :string, 1
+      optional :engine, :string, 2
+      optional :input_cost, :message, 3, "gooseai.CostData"
+      optional :output_cost, :message, 4, "gooseai.CostData"
+      proto3_optional :user, :string, 5
+      optional :aggregation_timestamp, :uint64, 6
+    end
+    add_message "gooseai.CostTotal" do
+      optional :amount_tokens, :uint32, 1
+      optional :amount_credits, :double, 2
+    end
+    add_message "gooseai.TotalMetricsData" do
+      optional :input_total, :message, 1, "gooseai.CostTotal"
+      optional :output_total, :message, 2, "gooseai.CostTotal"
+    end
+    add_message "gooseai.Metrics" do
+      repeated :metrics, :message, 1, "gooseai.UsageMetric"
+      optional :total, :message, 2, "gooseai.TotalMetricsData"
+    end
+    add_message "gooseai.EmptyRequest" do
+    end
+    add_message "gooseai.GetOrganizationRequest" do
+      optional :id, :string, 1
+    end
+    add_message "gooseai.GetMetricsRequest" do
+      optional :organization_id, :string, 1
+      proto3_optional :user_id, :string, 2
+      optional :range_from, :uint64, 3
+      optional :range_to, :uint64, 4
+      optional :include_per_request_metrics, :bool, 5
+    end
+    add_message "gooseai.APIKeyRequest" do
+      optional :is_secret, :bool, 1
+    end
+    add_message "gooseai.APIKeyFindRequest" do
+      optional :id, :string, 1
+    end
+    add_message "gooseai.UpdateDefaultOrganizationRequest" do
+      optional :organization_id, :string, 1
+    end
+    add_message "gooseai.ClientSettings" do
+      optional :settings, :bytes, 1
+    end
+    add_message "gooseai.CreateAutoChargeIntentRequest" do
+      optional :organization_id, :string, 1
+      optional :monthly_maximum, :uint64, 2
+      optional :minimum_value, :uint64, 3
+      optional :amount_credits, :uint64, 4
+    end
+    add_message "gooseai.CreateChargeRequest" do
+      optional :amount, :uint64, 1
+      optional :organization_id, :string, 2
+    end
+    add_message "gooseai.GetChargesRequest" do
+      optional :organization_id, :string, 1
+      optional :range_from, :uint64, 2
+      optional :range_to, :uint64, 3
+    end
+    add_message "gooseai.Charge" do
+      optional :id, :string, 1
+      optional :paid, :bool, 2
+      optional :receipt_link, :string, 3
+      optional :payment_link, :string, 4
+      optional :created_at, :uint64, 5
+      optional :amount_credits, :uint64, 6
+    end
+    add_message "gooseai.Charges" do
+      repeated :charges, :message, 1, "gooseai.Charge"
+    end
+    add_message "gooseai.GetAutoChargeRequest" do
+      optional :organization_id, :string, 1
+    end
+    add_message "gooseai.AutoChargeIntent" do
+      optional :id, :string, 1
+      optional :payment_link, :string, 2
+      optional :created_at, :uint64, 3
+      optional :monthly_maximum, :uint64, 4
+      optional :minimum_value, :uint64, 5
+      optional :amount_credits, :uint64, 6
+    end
+    add_message "gooseai.UpdateUserInfoRequest" do
+      proto3_optional :email, :string, 1
+    end
+    add_message "gooseai.UserPasswordChangeTicket" do
+      optional :ticket, :string, 1
+    end
+    add_enum "gooseai.OrganizationRole" do
+      value :MEMBER, 0
+      value :ACCOUNTANT, 1
+      value :OWNER, 2
+    end
+  end
+end
+
+module Gooseai
+  OrganizationMember = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationMember").msgclass
+  OrganizationGrant = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationGrant").msgclass
+  OrganizationPaymentInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationPaymentInfo").msgclass
+  OrganizationAutoCharge = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationAutoCharge").msgclass
+  Organization = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Organization").msgclass
+  APIKey = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.APIKey").msgclass
+  User = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.User").msgclass
+  CostData = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CostData").msgclass
+  UsageMetric = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.UsageMetric").msgclass
+  CostTotal = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CostTotal").msgclass
+  TotalMetricsData = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.TotalMetricsData").msgclass
+  Metrics = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Metrics").msgclass
+  EmptyRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.EmptyRequest").msgclass
+  GetOrganizationRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GetOrganizationRequest").msgclass
+  GetMetricsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GetMetricsRequest").msgclass
+  APIKeyRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.APIKeyRequest").msgclass
+  APIKeyFindRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.APIKeyFindRequest").msgclass
+  UpdateDefaultOrganizationRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.UpdateDefaultOrganizationRequest").msgclass
+  ClientSettings = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClientSettings").msgclass
+  CreateAutoChargeIntentRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CreateAutoChargeIntentRequest").msgclass
+  CreateChargeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CreateChargeRequest").msgclass
+  GetChargesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GetChargesRequest").msgclass
+  Charge = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Charge").msgclass
+  Charges = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Charges").msgclass
+  GetAutoChargeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GetAutoChargeRequest").msgclass
+  AutoChargeIntent = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AutoChargeIntent").msgclass
+  UpdateUserInfoRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.UpdateUserInfoRequest").msgclass
+  UserPasswordChangeTicket = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.UserPasswordChangeTicket").msgclass
+  OrganizationRole = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationRole").enummodule
+end

data/lib/dashboard_services_pb.rb

@@ -0,0 +1,41 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# Source: dashboard.proto for package 'gooseai'
+
+require 'grpc'
+require 'dashboard_pb'
+
+module Gooseai
+  module DashboardService
+    class Service
+
+      include ::GRPC::GenericService
+
+      self.marshal_class_method = :encode
+      self.unmarshal_class_method = :decode
+      self.service_name = 'gooseai.DashboardService'
+
+      # Get info
+      rpc :GetMe, ::Gooseai::EmptyRequest, ::Gooseai::User
+      rpc :GetOrganization, ::Gooseai::GetOrganizationRequest, ::Gooseai::Organization
+      rpc :GetMetrics, ::Gooseai::GetMetricsRequest, ::Gooseai::Metrics
+      # API key management
+      rpc :CreateAPIKey, ::Gooseai::APIKeyRequest, ::Gooseai::APIKey
+      rpc :DeleteAPIKey, ::Gooseai::APIKeyFindRequest, ::Gooseai::APIKey
+      # User settings
+      rpc :UpdateDefaultOrganization, ::Gooseai::UpdateDefaultOrganizationRequest, ::Gooseai::User
+      rpc :GetClientSettings, ::Gooseai::EmptyRequest, ::Gooseai::ClientSettings
+      rpc :SetClientSettings, ::Gooseai::ClientSettings, ::Gooseai::ClientSettings
+      rpc :UpdateUserInfo, ::Gooseai::UpdateUserInfoRequest, ::Gooseai::User
+      rpc :CreatePasswordChangeTicket, ::Gooseai::EmptyRequest, ::Gooseai::UserPasswordChangeTicket
+      rpc :DeleteAccount, ::Gooseai::EmptyRequest, ::Gooseai::User
+      # Payment functions
+      rpc :CreateCharge, ::Gooseai::CreateChargeRequest, ::Gooseai::Charge
+      rpc :GetCharges, ::Gooseai::GetChargesRequest, ::Gooseai::Charges
+      rpc :CreateAutoChargeIntent, ::Gooseai::CreateAutoChargeIntentRequest, ::Gooseai::AutoChargeIntent
+      rpc :UpdateAutoChargeIntent, ::Gooseai::CreateAutoChargeIntentRequest, ::Gooseai::AutoChargeIntent
+      rpc :GetAutoChargeIntent, ::Gooseai::GetAutoChargeRequest, ::Gooseai::AutoChargeIntent
+    end
+
+    Stub = Service.rpc_stub_class
+  end
+end

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
       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,6 +125,25 @@ 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
@@ -113,6 +153,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       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
@@ -161,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
@@ -173,6 +229,18 @@ 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
@@ -187,17 +255,23 @@ 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
@@ -206,7 +280,11 @@ module Gooseai
   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

data/lib/stability_sdk/client.rb

@@ -42,6 +42,11 @@ module StabilitySDK
     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
+
+      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].to_i : DEFAULT_SAMPLE_SIZE
       steps = options.has_key?(:steps) ? options[:steps].to_i : DEFAULT_STEPS
       seed = options.has_key?(:seed) ? [options[:seed].to_i] : [rand(4294967295)]

data/lib/stability_sdk/unstable/dashboard_client.rb

@@ -0,0 +1,31 @@
+require "grpc"
+require "dashboard_services_pb"
+
+module StabilitySDK::Unstable
+  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.9"
+  VERSION = "0.2.11"
 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/unstable/dashboard_client"
 
 module StabilitySDK
   class Error < StandardError; end
   class InsufficientParameter < StandardError; end
+  class InvalidParameter < StandardError; end
 end

data/stability_sdk.gemspec

@@ -27,5 +27,7 @@ Gem::Specification.new do |spec|
 
   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.9
+  version: 0.2.11
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-10-03 00:00:00.000000000 Z
+date: 2022-10-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc
@@ -38,6 +38,40 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         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,6 +92,7 @@ email:
 - cou929@gmail.com
 executables:
 - stability-client
+- stability-dashboard-client-unstable
 extensions: []
 extra_rdoc_files: []
 files:
@@ -65,18 +100,21 @@ files:
 - ".gitmodules"
 - ".travis.yml"
 - Gemfile
-- Gemfile.lock
 - LICENSE
 - README.md
 - Rakefile
 - bin/console
 - bin/setup
 - exe/stability-client
+- exe/stability-dashboard-client-unstable
+- 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/unstable/dashboard_client.rb
 - lib/stability_sdk/version.rb
 - stability_sdk.gemspec
 homepage: https://github.com/cou929/stability-sdk-ruby
@@ -101,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,35 +0,0 @@
-PATH
-  remote: .
-  specs:
-    stability_sdk (0.2.9)
-      grpc (>= 1.41.1)
-      mime-types (>= 3.0.0)
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    google-protobuf (3.21.7)
-    googleapis-common-protos-types (1.4.0)
-      google-protobuf (~> 3.14)
-    grpc (1.49.1)
-      google-protobuf (~> 3.21)
-      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
-  ruby
-
-DEPENDENCIES
-  grpc-tools
-  minitest (~> 5.0)
-  rake (~> 12.0)
-  stability_sdk!
-
-BUNDLED WITH
-   2.2.2