stability_sdk 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2b707c869873f4d9ff68de1d281a70b498fb9d0509f83e6e3b3550e8f1d86bff
-  data.tar.gz: 04454b501ae8d8d4ed62140812b2f0482e9178fb54a7d4fd42b1e7bf2817f1fb
+  metadata.gz: 3e707d41b381676dfbc958975bae028b82a52a7828529c52db7809db125791e9
+  data.tar.gz: 9e7aafbd1c0e5638ad2bc860312e55a0b2973ec3c8423fa050730cfa15887efa
 SHA512:
-  metadata.gz: efef7f4abc46b1664a4545514598c888472e7828a67d9ca93f5597c6cf3f1709cbc58bd1508936696626985689762b8d9100f773b1175f1a38fde8e507961df0
-  data.tar.gz: ca8e9fbc0ab70bb2f05b5171985c8159a4c423fe32c6a1903c42548366c1d1d17c40a172c85921e884291c64ef34e398bfda8f7e8e08af1632894504e292ede8
+  metadata.gz: 37cb8a18a1accb3cf30f6dc780781f71dfe75ad648713a2c335d1b2be41f7546c0450d88ec0ae833354db24a3d3659fe0ac8163d1f3a1a076f08e8e344282c9c
+  data.tar.gz: aa66ce05d53d3d5bd6aa661f752a5d71f61c9835804d69f38834e07bc4e5259d971816eb4d4061264cfadc9a825a425b0469c794369eb48c1080cdbf1d1342d3

data/.gitignore

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

data/.gitmodules

@@ -0,0 +1,3 @@
+[submodule "api-interfaces"]
+	path = api-interfaces
+	url = git@github.com:Stability-AI/api-interfaces.git

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2022 Kosei Moriyama
+
+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

@@ -19,8 +19,7 @@ Or install it yourself as:
     $ gem install stability_sdk
 
 ## Usage
-
-First you need to create a [DreamStudio](https://beta.dreamstudio.ai/home)'s account and get an API Key of it.
+First, you need to create a [DreamStudio](https://beta.dreamstudio.ai/home)'s account and get an API Key of it.
 
 - Access [DreamStudio](https://beta.dreamstudio.ai/dream) and create an account if you have not had it
 - Go to the [membership page](https://beta.dreamstudio.ai/membership)
@@ -62,9 +61,25 @@ Options:
     -m, --mask_image=VAL             path to mask image
         --start_schedule=VAL         start schedule for init image (must be greater than 0, 1 is full strength text prompt, no trace of image). default 1.0
         --end_schedule=VAL           end schedule for init image. default 0.01
+        --guidance_preset=VAL        Guidance preset to use. See generation.GuidancePreset for supported values. default `GUIDANCE_PRESET_NONE`
+        --guidance_cuts=VAL          Number of cuts to use for guidance. default 0
+        --guidance_strength=VAL      Strength of the guidance. We recommend values in range [0.0,1.0]. A good default is 0.25. default nil
+        --guidance_prompt=VAL        Prompt to use for guidance, defaults to `YOUR_PROMPT_TEXT` argument (above) if not specified.
+        --guidance_models=VAL        Models to use for guidance. default nil
     -v, --verbose
 ```
 
+#### CLIP guidance
+
+By specifying `--guidance_preset` and other `--guidance_*` options, you can use CLIP guidance, which is enabled by default on DreamStudio's web interface.
+See also [this notebook](https://github.com/Stability-AI/stability-sdk/blob/c04381f960008f37c7392467cfaabfdf8f763e6a/nbs/demo_colab.ipynb) for more details.
+
+```sh
+# example
+# [note] --steps=35 is recommended. which is also the default value of the web interface
+STABILITY_SDK_API_KEY=xxx stability-client --guidance_preset=GUIDANCE_PRESET_FAST_BLUE --guidance_strength=0.25 --steps=35 --sampler=k_dpm_2_ancestral "A dream of a distant galaxy, by Caspar David Friedrich, matte painting trending on artstation HQ"
+```
+
 ### SDK usage
 
 This sample code saves a generated image as `result.png`.
@@ -72,7 +87,7 @@ This sample code saves a generated image as `result.png`.
 ```ruby
 require "stability_sdk"
 
-client = StabilitySDK::Client.new(api_key: "YOUR_API_KEY")
+client = StabilitySDK::Client.new(api_key: "YOUR_API_KEY", timeout: 600)
 
 prompt = "your prompot here"
 options = {}
@@ -92,12 +107,63 @@ client.generate(prompt, options) do |answer|
 end
 ```
 
+### Dashboard API
+
+Dashboard API is a way to interact with DreamStudio Web UIs, such as getting user info, payment info, etc.
+
+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 get_me
+
+# get organization info
+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 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
 
 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.
 
 To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
 
+### update submodule to try a new feature on Stability-AI/api-interfaces
+
+- update submodule
+
+```sh
+git submodule update --init  # if you haven't fetched the content of the submodule yet
+cd api-interfaces
+
+# checkout some branch/commit you need
+git fetch
+git reset --hard origin/some_branch
+
+cd ..
+```
+
+- build
+
+```sh
+bundle exec rake protoc
+
+git diff
+# now you may be able to confirm that the diff is created in lib/generation_pb.rb
+```
+
+- modify the `lib/stability_sdk/client.rb` to try some new features
+
 ## Contributing
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/cou929/stability-sdk-ruby.

data/Rakefile

@@ -8,3 +8,13 @@ Rake::TestTask.new(:test) do |t|
 end
 
 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"
+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"
 
@@ -17,20 +18,25 @@ opt.banner = "Usage: stability-client [options] YOUR_PROMPT_TEXT"
 opt.separator ""
 opt.separator "Options:"
 opt.on("--api_key=VAL", "api key of DreamStudio account. You can also specify by a STABILITY_SDK_API_KEY environment variable") {|v| options[:api_key] = v }
-opt.on("-H", "--height=VAL", "height of image in pixel. default 512") {|v| options[:height] = v }
-opt.on("-W", "--width=VAL", "width of image in pixel. default 512") {|v| options[:width] = v }
-opt.on("-C", "--cfg_scale=VAL", "CFG scale factor. default 7.0") {|v| options[:cfg_scale] = v }
-opt.on("-A", "--sampler=VAL", "ddim, plms, k_euler, k_euler_ancestral, k_heun, k_dpm_2, k_dpm_2_ancestral, k_lms. default k_lms") {|v| options[:sampler] = v }
-opt.on("-s", "--steps=VAL", "number of steps. default 50") {|v| options[:steps] = v }
-opt.on("-S", "--seed=VAL", "random seed to use in integer") {|v| options[:seed] = v }
-opt.on("-p", "--prefix=VAL", "output prefixes for artifacts. default `generation`") {|v| options[:prefix] = v }
+opt.on("-H", "--height=VAL", Integer, "height of image in pixel. default 512") {|v| options[:height] = v }
+opt.on("-W", "--width=VAL", Integer, "width of image in pixel. default 512") {|v| options[:width] = v }
+opt.on("-C", "--cfg_scale=VAL", Float, "CFG scale factor. default 7.0") {|v| options[:cfg_scale] = v }
+opt.on("-A", "--sampler=VAL", String, "ddim, plms, k_euler, k_euler_ancestral, k_heun, k_dpm_2, k_dpm_2_ancestral, k_lms. default k_lms") {|v| options[:sampler] = v }
+opt.on("-s", "--steps=VAL", Integer, "number of steps. default 50") {|v| options[:steps] = v }
+opt.on("-S", "--seed=VAL", Integer, "random seed to use in integer") {|v| options[:seed] = v }
+opt.on("-p", "--prefix=VAL", String, "output prefixes for artifacts. default `generation`") {|v| options[:prefix] = v }
 opt.on("--no-store", "do not write out artifacts") {|v| options[:no_store] = v }
-opt.on("-n", "--num_samples=VAL", "number of samples to generate") {|v| options[:num_samples] = v }
-opt.on("-e", "--engine=VAL", "engine to use for inference. default `stable-diffusion-v1`") {|v| options[:engine_id] = v }
-opt.on("-i", "--init_image=VAL", "path to init image") {|v| options[:init_image] = v }
-opt.on("-m", "--mask_image=VAL", "path to mask image") {|v| options[:mask_image] = v }
-opt.on("--start_schedule=VAL", "start schedule for init image (must be greater than 0, 1 is full strength text prompt, no trace of image). default 1.0") {|v| options[:start_schedule] = v }
-opt.on("--end_schedule=VAL", "end schedule for init image. default 0.01") {|v| options[:end_schedule] = v }
+opt.on("-n", "--num_samples=VAL", Integer, "number of samples to generate. default 1") {|v| options[:num_samples] = v }
+opt.on("-e", "--engine=VAL", String, "engine to use for inference. default `stable-diffusion-v1-5`") {|v| options[:engine_id] = v }
+opt.on("-i", "--init_image=VAL", String, "path to init image") {|v| options[:init_image] = v }
+opt.on("-m", "--mask_image=VAL", String, "path to mask image") {|v| options[:mask_image] = v }
+opt.on("--start_schedule=VAL", Float, "start schedule for init image (must be greater than 0, 1 is full strength text prompt, no trace of image). default 1.0") {|v| options[:start_schedule] = v }
+opt.on("--end_schedule=VAL", Float, "end schedule for init image. default 0.01") {|v| options[:end_schedule] = v }
+opt.on("--guidance_preset=VAL", String,"Guidance preset to use. See generation.GuidancePreset for supported values. default `GUIDANCE_PRESET_NONE`") {|v| options[:guidance_preset] = v }
+opt.on("--guidance_cuts=VAL", Integer, "Number of cuts to use for guidance. default 0") {|v| options[:guidance_cuts] = v }
+opt.on("--guidance_strength=VAL", Float, "Strength of the guidance. We recommend values in range [0.0,1.0]. A good default is 0.25. default nil") {|v| options[:guidance_strength] = v }
+opt.on("--guidance_prompt=VAL", String, "Prompt to use for guidance, defaults to `YOUR_PROMPT_TEXT` argument (above) if not specified.") {|v| options[:guidance_prompt] = v }
+opt.on("--guidance_models=VAL", Array, "Models to use for guidance. default nil") {|v| options[:guidance_models] = v }
 opt.on("-v", "--verbose") { logger.level = Logger::DEBUG }
 opt.parse!(ARGV)
 
@@ -42,7 +48,12 @@ 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)
 
-client = StabilitySDK::Client.new(api_key: options[:api_key])
+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], logger: logger)
 
 client.generate(prompt, options) do |answer|
   StabilitySDK::CLI.save_answer(answer, options, logger)

data/exe/stability-dashboard-client

@@ -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::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