stability_sdk 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2b707c869873f4d9ff68de1d281a70b498fb9d0509f83e6e3b3550e8f1d86bff
-  data.tar.gz: 04454b501ae8d8d4ed62140812b2f0482e9178fb54a7d4fd42b1e7bf2817f1fb
+  metadata.gz: 401df5c1aa6c7c458a171cc6e7f437a420d56a17f394236c42afe6b7114986bc
+  data.tar.gz: 744723a641d79456d255a0120118f8d218137d0901c760ede079d9dc51301f50
 SHA512:
-  metadata.gz: efef7f4abc46b1664a4545514598c888472e7828a67d9ca93f5597c6cf3f1709cbc58bd1508936696626985689762b8d9100f773b1175f1a38fde8e507961df0
-  data.tar.gz: ca8e9fbc0ab70bb2f05b5171985c8159a4c423fe32c6a1903c42548366c1d1d17c40a172c85921e884291c64ef34e398bfda8f7e8e08af1632894504e292ede8
+  metadata.gz: f37b8db84a96fb1df551ef789e43138eb80adb8e8540a7b3619e44b66bed7b193aeb704ab8099daaf64f761c3e92bfd90088faf3285c9a1ffc07a06e457ae20a
+  data.tar.gz: 7b8b4bec82c39a5046bbe805e213fe5f4c5ddb0dd279352bd619768a711fc17a0254a958ddeac46d3835a76e75e7578ab11e82fdda9cb77d477d62ba5677c222

data/.gitmodules

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

data/Gemfile.lock

@@ -1,9 +1,9 @@
 PATH
   remote: .
   specs:
-    stability_sdk (0.2.0)
-      grpc
-      mime-types
+    stability_sdk (0.2.1)
+      grpc (>= 1.0.0)
+      mime-types (>= 3.0.0)
 
 GEM
   remote: https://rubygems.org/

data/Rakefile

@@ -8,3 +8,8 @@ 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

data/lib/generation_pb.rb

@@ -65,8 +65,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       proto3_optional :vector_adjust_prior, :string, 1
     end
     add_message "gooseai.ScheduleParameters" do
-      optional :start, :float, 1
-      optional :end, :float, 2
+      proto3_optional :start, :float, 1
+      proto3_optional :end, :float, 2
     end
     add_message "gooseai.StepParameter" do
       optional :scaled_step, :float, 1
@@ -110,11 +110,25 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       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"
       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
@@ -159,6 +173,11 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :CLSFR_MODE_ZEROSHOT, 0
       value :CLSFR_MODE_MULTICLASS, 1
     end
+    add_enum "gooseai.StageAction" do
+      value :STAGE_ACTION_PASS, 0
+      value :STAGE_ACTION_DISCARD, 1
+      value :STAGE_ACTION_RETURN, 2
+    end
   end
 end
 
@@ -180,10 +199,14 @@ module Gooseai
   ClassifierCategory = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierCategory").msgclass
   ClassifierParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierParameters").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
   Action = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Action").enummodule
   ClassifierMode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierMode").enummodule
+  StageAction = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.StageAction").enummodule
 end

data/lib/generation_services_pb.rb

@@ -15,6 +15,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/version.rb

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

data/stability_sdk.gemspec

@@ -24,7 +24,7 @@ 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.0.0"
+  spec.add_dependency "mime-types", ">= 3.0.0"
   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.2.1
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-09-08 00:00:00.000000000 Z
+date: 2022-09-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc
@@ -16,28 +16,28 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 1.0.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 1.0.0
 - !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: grpc-tools
   requirement: !ruby/object:Gem::Requirement
@@ -62,6 +62,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".gitmodules"
 - ".travis.yml"
 - Gemfile
 - Gemfile.lock
@@ -76,7 +77,6 @@ files:
 - lib/stability_sdk/cli.rb
 - lib/stability_sdk/client.rb
 - lib/stability_sdk/version.rb
-- proto/generation.proto
 - stability_sdk.gemspec
 homepage: https://github.com/cou929/stability-sdk-ruby
 licenses: []

data/proto/generation.proto

@@ -1,186 +0,0 @@
-syntax = 'proto3';
-package gooseai;
-option go_package = "./;generation";
-
-enum FinishReason {
-  NULL = 0;
-  LENGTH = 1;
-  STOP = 2;
-  ERROR = 3;
-  FILTER = 4;
-}
-
-
-enum ArtifactType {
-  ARTIFACT_NONE = 0;
-  ARTIFACT_IMAGE = 1;
-  ARTIFACT_VIDEO = 2;
-  ARTIFACT_TEXT = 3;
-  ARTIFACT_TOKENS = 4;
-  ARTIFACT_EMBEDDING = 5;
-  ARTIFACT_CLASSIFICATIONS = 6;
-  ARTIFACT_MASK = 7;
-}
-
-message Token {
-  optional string text = 1;
-  uint32 id = 2;
-}
-
-message Tokens {
-  repeated Token tokens = 1;
-  optional string tokenizer_id = 2;
-}
-
-message Artifact {
-  uint64 id = 1;
-  ArtifactType type = 2;
-  string mime = 3;
-  optional string magic = 4;
-  oneof data {
-    bytes binary = 5;
-    string text = 6;
-    Tokens tokens = 7;
-    ClassifierParameters classifier = 11;
-  }
-  uint32 index = 8;
-  FinishReason finish_reason = 9;
-  uint32 seed = 10;
-}
-
-message PromptParameters {
-  optional bool init = 1;
-  optional float weight = 2;
-}
-
-message Prompt {
-  optional PromptParameters parameters = 1;
-  oneof prompt {
-    string text = 2;
-    Tokens tokens = 3;
-    Artifact artifact = 4;
-  }
-}
-
-message AnswerMeta {
-  optional string gpu_id = 1;
-  optional string cpu_id = 2;
-  optional string node_id = 3;
-  optional string engine_id = 4;
-}
-
-message Answer {
-  string answer_id = 1;
-  string request_id = 2;
-  uint64 received = 3;
-  uint64 created = 4;
-  optional AnswerMeta meta = 6;
-  repeated Artifact artifacts = 7;
-}
-
-enum DiffusionSampler {
-  SAMPLER_DDIM = 0;
-  SAMPLER_DDPM = 1;
-  SAMPLER_K_EULER = 2;
-  SAMPLER_K_EULER_ANCESTRAL = 3;
-  SAMPLER_K_HEUN = 4;
-  SAMPLER_K_DPM_2 = 5;
-  SAMPLER_K_DPM_2_ANCESTRAL = 6;
-  SAMPLER_K_LMS = 7;
-}
-
-message SamplerParameters {
-  optional float eta = 1;
-  optional uint64 sampling_steps = 2;
-  optional uint64 latent_channels = 3;
-  optional uint64 downsampling_factor = 4;
-  optional float cfg_scale = 5;
-}
-
-message ConditionerParameters {
-  optional string vector_adjust_prior = 1;
-}
-
-enum Upscaler {
-  UPSCALER_RGB = 0;
-  UPSCALER_GFPGAN = 1;
-  UPSCALER_ESRGAN = 2;
-}
-
-message ScheduleParameters {
-  float start = 1;
-  float end = 2;
-}
-
-message StepParameter {
-  float scaled_step = 1;
-  optional SamplerParameters sampler = 2;
-  optional ScheduleParameters schedule = 3;
-}
-
-message TransformType {
-  oneof type {
-      DiffusionSampler diffusion = 1;
-      Upscaler upscaler = 2;
-  }
-}
-
-message ImageParameters {
-  optional uint64 height = 1;
-  optional uint64 width = 2;
-  repeated uint32 seed = 3;
-  optional uint64 samples = 4;
-  optional uint64 steps = 5;
-  optional TransformType transform = 6;
-  repeated StepParameter parameters = 7;
-}
-
-enum Action {
-  ACTION_PASSTHROUGH = 0;
-  ACTION_REGENERATE_DUPLICATE = 1;
-  ACTION_REGENERATE = 2;
-  ACTION_OBFUSCATE_DUPLICATE = 3;
-  ACTION_OBFUSCATE = 4;
-  ACTION_DISCARD = 5;
-}
-
-enum ClassifierMode {
-  CLSFR_MODE_ZEROSHOT = 0;
-  CLSFR_MODE_MULTICLASS = 1;
-  /*CLSFR_MODE_ODDSRATIO = 2;*/
-}
-
-message ClassifierConcept {
-  string concept = 1;
-  optional float threshold = 2;
-}
-
-message ClassifierCategory {
-  string name = 1;
-  repeated ClassifierConcept concepts = 2;
-  optional float adjustment = 3;
-  optional Action action = 4;
-  optional ClassifierMode classifier_mode = 5;
-}
-
-message ClassifierParameters {
-  repeated ClassifierCategory categories = 1;
-  repeated ClassifierCategory exceeds = 2;
-  optional Action realized_action = 3;
-}
-
-message Request {
-  string engine_id = 1;
-  string request_id = 2;
-  ArtifactType requested_type = 3;
-  repeated Prompt prompt = 4;
-  oneof params {
-    ImageParameters image = 5;
-  }
-  optional ConditionerParameters conditioner = 6;
-  optional ClassifierParameters classifier = 7;
-}
-
-service GenerationService {
-  rpc Generate (Request) returns (stream Answer) {};
-}