stability_sdk 0.2.12 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d5aac33cb8d504aab26b515dc2fa6bd9d76c70829a0aa706eabc6be39043b143
-  data.tar.gz: ca6a3cd1cd24763a9f9427132a4e8fde4c688d9bc9d3409ef921139fa62fa232
+  metadata.gz: eb4bcef5f20a0b7147814c3c3cd90df2b39a54dc162d207a6c7f91141888cdb3
+  data.tar.gz: 77aa1b30334169b9437ca8798334fb081b7ff1c4305d09c69128bc5348768ba6
 SHA512:
-  metadata.gz: 95b76f6caf8e5a32af792cc0d284f81b76c7e38c682b24a5fc91204b3373096a47a98ab1ee6f0e32cbbd2039ad6b57c920c262661cda221c54eac529bd54e51a
-  data.tar.gz: f196c8b879b36e69c84d40c2528fd9fbf7f07e1ae6eec9c4a996a66a8a73841ea50d73d8ca941b7813ca9d5fbada94140143619bdb841de4ce9f71316af1303e
+  metadata.gz: d82e9bd6fdc0cec8b74e6ef774a82bd0d922327d99adae5b8ecccb55e7c7b3111e59d99c46fba8576aacbfd65fd4b3ea0c69a6e691b4b9967319a67b8a778b3f
+  data.tar.gz: 11ea3112c72f6a4a76fa6cd21457bf3808fddfa6956ffa170e7eebfdbdaca58f2e91dc68accb813b91bda6d82874c73b5d3fc4cf59025acdf96b969f98305904

data/README.md

@@ -107,23 +107,29 @@ client.generate(prompt, options) do |answer|
 end
 ```
 
-### [Unstable] Dashboard API
-
-**This feature is in a very early stage of development.**
+### Dashboard API
 
 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
+Currently, the API key is only allowed as read-only. And this client supports only GetMe and GetOrganization actions. 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
+STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client get_me
 
 # get organization info
-STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client-unstable get_organization
+STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client get_organization
 
 # i.e, get remaining balance
-STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client-unstable get_organization | jq .paymentInfo.balance
+STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client get_organization | jq .paymentInfo.balance
+```
+
+```ruby
+require "stability_sdk"
+
+client = StabilitySDK::DashboardClient.new(api_key: "your api key")
+res = client.get_organization
+p res.payment_info.balance  # prints a remaining balance
 ```
 
 ## Development
@@ -142,9 +148,13 @@ cd api-interfaces
 
 # checkout some branch/commit you need
 git fetch
-git reset --hard origin/some_branch
+git checkout origin/some_branch # or `git reset --hard origin/some_branch`
 
 cd ..
+git add
+
+# if you need to follow the updates of submodules inside the Stability-AI/api-interfaces, you have to update submodules recursively
+git submodule update --init --recursive
 ```
 
 - build

data/Rakefile

@@ -11,7 +11,7 @@ task :default => :test
 
 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"
+  sh "grpc_tools_ruby_protoc -I api-interfaces/src/proto/ -I api-interfaces/src/tensorizer/proto/ --ruby_out=lib --grpc_out=lib api-interfaces/src/proto/generation.proto"
 end
 
 desc "Compile a dashboard proto file"

data/exe/{stability-dashboard-client-unstable → stability-dashboard-client}

@@ -33,7 +33,7 @@ class CLI < Thor
     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)
+    return StabilitySDK::DashboardClient.new(api_key: api_key)
   end
 end
 

data/lib/generation_pb.rb

@@ -3,6 +3,8 @@
 
 require 'google/protobuf'
 
+require 'tensors_pb'
+
 Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("generation.proto", :syntax => :proto3) do
     add_message "gooseai.Token" do
@@ -28,6 +30,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
         optional :text, :string, 6
         optional :tokens, :message, 7, "gooseai.Tokens"
         optional :classifier, :message, 11, "gooseai.ClassifierParameters"
+        optional :tensor, :message, 14, "tensors.Tensor"
       end
     end
     add_message "gooseai.PromptParameters" do
@@ -48,6 +51,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       proto3_optional :latent_channels, :uint64, 3
       proto3_optional :downsampling_factor, :uint64, 4
       proto3_optional :cfg_scale, :float, 5
+      proto3_optional :init_noise_scale, :float, 6
     end
     add_message "gooseai.ConditionerParameters" do
       proto3_optional :vector_adjust_prior, :string, 1
@@ -108,6 +112,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       proto3_optional :steps, :uint64, 5
       proto3_optional :transform, :message, 6, "gooseai.TransformType"
       repeated :parameters, :message, 7, "gooseai.StepParameter"
+      proto3_optional :masked_area_init, :enum, 8, "gooseai.MaskedAreaInit"
+      proto3_optional :weight_method, :enum, 9, "gooseai.WeightMethod"
     end
     add_message "gooseai.ClassifierConcept" do
       optional :concept, :string, 1
@@ -186,6 +192,17 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :ARTIFACT_EMBEDDING, 5
       value :ARTIFACT_CLASSIFICATIONS, 6
       value :ARTIFACT_MASK, 7
+      value :ARTIFACT_LATENT, 8
+      value :ARTIFACT_TENSOR, 9
+    end
+    add_enum "gooseai.MaskedAreaInit" do
+      value :MASKED_AREA_INIT_ZERO, 0
+      value :MASKED_AREA_INIT_RANDOM, 1
+      value :MASKED_AREA_INIT_ORIGINAL, 2
+    end
+    add_enum "gooseai.WeightMethod" do
+      value :TEXT_ENCODER, 0
+      value :CROSS_ATTENTION, 1
     end
     add_enum "gooseai.DiffusionSampler" do
       value :SAMPLER_DDIM, 0
@@ -196,6 +213,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :SAMPLER_K_DPM_2, 5
       value :SAMPLER_K_DPM_2_ANCESTRAL, 6
       value :SAMPLER_K_LMS, 7
+      value :SAMPLER_K_DPMPP_2S_ANCESTRAL, 8
+      value :SAMPLER_K_DPMPP_2M, 9
     end
     add_enum "gooseai.Upscaler" do
       value :UPSCALER_RGB, 0
@@ -278,6 +297,8 @@ module Gooseai
   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
+  MaskedAreaInit = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.MaskedAreaInit").enummodule
+  WeightMethod = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.WeightMethod").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

data/lib/stability_sdk/{unstable/dashboard_client.rb → dashboard_client.rb}

@@ -1,7 +1,7 @@
 require "grpc"
 require "dashboard_services_pb"
 
-module StabilitySDK::Unstable
+module StabilitySDK
   class DashboardClient
     DEFAULT_API_HOST = "grpc.stability.ai:443"
 

data/lib/stability_sdk/version.rb

@@ -1,3 +1,3 @@
 module StabilitySDK
-  VERSION = "0.2.12"
+  VERSION = "0.3.1"
 end

data/lib/stability_sdk.rb

@@ -1,7 +1,7 @@
 require "stability_sdk/version"
 require "stability_sdk/client"
 require "stability_sdk/cli"
-require "stability_sdk/unstable/dashboard_client"
+require "stability_sdk/dashboard_client"
 
 module StabilitySDK
   class Error < StandardError; end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stability_sdk
 version: !ruby/object:Gem::Version
-  version: 0.2.12
+  version: 0.3.1
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-10-31 00:00:00.000000000 Z
+date: 2022-12-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc
@@ -92,7 +92,7 @@ email:
 - cou929@gmail.com
 executables:
 - stability-client
-- stability-dashboard-client-unstable
+- stability-dashboard-client
 extensions: []
 extra_rdoc_files: []
 files:
@@ -106,7 +106,7 @@ files:
 - bin/console
 - bin/setup
 - exe/stability-client
-- exe/stability-dashboard-client-unstable
+- exe/stability-dashboard-client
 - lib/dashboard_pb.rb
 - lib/dashboard_services_pb.rb
 - lib/generation_pb.rb
@@ -114,7 +114,7 @@ files:
 - lib/stability_sdk.rb
 - lib/stability_sdk/cli.rb
 - lib/stability_sdk/client.rb
-- lib/stability_sdk/unstable/dashboard_client.rb
+- lib/stability_sdk/dashboard_client.rb
 - lib/stability_sdk/version.rb
 - stability_sdk.gemspec
 homepage: https://github.com/cou929/stability-sdk-ruby