tensors-ruby 0.1.0

data/lib/tensors-ruby/models/download_request.rb

@@ -0,0 +1,179 @@
+=begin
+#tensors
+
+#API for CivitAI model management and image gallery
+
+The version of the OpenAPI document: 0.1.18
+
+Generated by: https://openapi-generator.tech
+Generator version: 7.21.0
+
+=end
+
+require 'date'
+require 'time'
+
+module TensorsApi
+  # Request body for downloading a model.
+  class DownloadRequest < ApiModelBase
+    attr_accessor :version_id
+
+    attr_accessor :model_id
+
+    attr_accessor :hash
+
+    attr_accessor :output_dir
+
+    # Attribute mapping from ruby-style variable name to JSON key.
+    def self.attribute_map
+      {
+        :'version_id' => :'version_id',
+        :'model_id' => :'model_id',
+        :'hash' => :'hash',
+        :'output_dir' => :'output_dir'
+      }
+    end
+
+    # Returns attribute mapping this model knows about
+    def self.acceptable_attribute_map
+      attribute_map
+    end
+
+    # Returns all the JSON keys this model knows about
+    def self.acceptable_attributes
+      acceptable_attribute_map.values
+    end
+
+    # Attribute type mapping.
+    def self.openapi_types
+      {
+        :'version_id' => :'Integer',
+        :'model_id' => :'Integer',
+        :'hash' => :'String',
+        :'output_dir' => :'String'
+      }
+    end
+
+    # List of attributes with nullable: true
+    def self.openapi_nullable
+      Set.new([
+        :'version_id',
+        :'model_id',
+        :'hash',
+        :'output_dir'
+      ])
+    end
+
+    # Initializes the object
+    # @param [Hash] attributes Model attributes in the form of hash
+    def initialize(attributes = {})
+      if (!attributes.is_a?(Hash))
+        fail ArgumentError, "The input argument (attributes) must be a hash in `TensorsApi::DownloadRequest` initialize method"
+      end
+
+      # check to see if the attribute exists and convert string to symbol for hash key
+      acceptable_attribute_map = self.class.acceptable_attribute_map
+      attributes = attributes.each_with_object({}) { |(k, v), h|
+        if (!acceptable_attribute_map.key?(k.to_sym))
+          fail ArgumentError, "`#{k}` is not a valid attribute in `TensorsApi::DownloadRequest`. Please check the name to make sure it's valid. List of attributes: " + acceptable_attribute_map.keys.inspect
+        end
+        h[k.to_sym] = v
+      }
+
+      if attributes.key?(:'version_id')
+        self.version_id = attributes[:'version_id']
+      end
+
+      if attributes.key?(:'model_id')
+        self.model_id = attributes[:'model_id']
+      end
+
+      if attributes.key?(:'hash')
+        self.hash = attributes[:'hash']
+      end
+
+      if attributes.key?(:'output_dir')
+        self.output_dir = attributes[:'output_dir']
+      end
+    end
+
+    # Show invalid properties with the reasons. Usually used together with valid?
+    # @return Array for valid properties with the reasons
+    def list_invalid_properties
+      warn '[DEPRECATED] the `list_invalid_properties` method is obsolete'
+      invalid_properties = Array.new
+      invalid_properties
+    end
+
+    # Check to see if the all the properties in the model are valid
+    # @return true if the model is valid
+    def valid?
+      warn '[DEPRECATED] the `valid?` method is obsolete'
+      true
+    end
+
+    # Checks equality by comparing each attribute.
+    # @param [Object] Object to be compared
+    def ==(o)
+      return true if self.equal?(o)
+      self.class == o.class &&
+          version_id == o.version_id &&
+          model_id == o.model_id &&
+          hash == o.hash &&
+          output_dir == o.output_dir
+    end
+
+    # @see the `==` method
+    # @param [Object] Object to be compared
+    def eql?(o)
+      self == o
+    end
+
+    # Calculates hash code according to all attributes.
+    # @return [Integer] Hash code
+    def hash
+      [version_id, model_id, hash, output_dir].hash
+    end
+
+    # Builds the object from hash
+    # @param [Hash] attributes Model attributes in the form of hash
+    # @return [Object] Returns the model itself
+    def self.build_from_hash(attributes)
+      return nil unless attributes.is_a?(Hash)
+      attributes = attributes.transform_keys(&:to_sym)
+      transformed_hash = {}
+      openapi_types.each_pair do |key, type|
+        if attributes.key?(attribute_map[key]) && attributes[attribute_map[key]].nil?
+          transformed_hash["#{key}"] = nil
+        elsif type =~ /\AArray<(.*)>/i
+          # check to ensure the input is an array given that the attribute
+          # is documented as an array but the input is not
+          if attributes[attribute_map[key]].is_a?(Array)
+            transformed_hash["#{key}"] = attributes[attribute_map[key]].map { |v| _deserialize($1, v) }
+          end
+        elsif !attributes[attribute_map[key]].nil?
+          transformed_hash["#{key}"] = _deserialize(type, attributes[attribute_map[key]])
+        end
+      end
+      new(transformed_hash)
+    end
+
+    # Returns the object in the form of hash
+    # @return [Hash] Returns the object in the form of hash
+    def to_hash
+      hash = {}
+      self.class.attribute_map.each_pair do |attr, param|
+        value = self.send(attr)
+        if value.nil?
+          is_nullable = self.class.openapi_nullable.include?(attr)
+          next if !is_nullable || (is_nullable && !instance_variable_defined?(:"@#{attr}"))
+        end
+
+        hash[param] = _to_hash(value)
+      end
+      hash
+    end
+
+  end
+
+end

data/lib/tensors-ruby/models/generate_request.rb

@@ -0,0 +1,444 @@
+=begin
+#tensors
+
+#API for CivitAI model management and image gallery
+
+The version of the OpenAPI document: 0.1.18
+
+Generated by: https://openapi-generator.tech
+Generator version: 7.21.0
+
+=end
+
+require 'date'
+require 'time'
+
+module TensorsApi
+  # Request body for text-to-image generation.
+  class GenerateRequest < ApiModelBase
+    # Positive prompt text
+    attr_accessor :prompt
+
+    # Negative prompt text
+    attr_accessor :negative_prompt
+
+    attr_accessor :model
+
+    # Image width
+    attr_accessor :width
+
+    # Image height
+    attr_accessor :height
+
+    # Sampling steps
+    attr_accessor :steps
+
+    # CFG scale
+    attr_accessor :cfg
+
+    # Random seed (-1 for random)
+    attr_accessor :seed
+
+    # Sampler name
+    attr_accessor :sampler
+
+    # Scheduler name
+    attr_accessor :scheduler
+
+    attr_accessor :vae
+
+    attr_accessor :lora_name
+
+    # LoRA strength
+    attr_accessor :lora_strength
+
+    # Attribute mapping from ruby-style variable name to JSON key.
+    def self.attribute_map
+      {
+        :'prompt' => :'prompt',
+        :'negative_prompt' => :'negative_prompt',
+        :'model' => :'model',
+        :'width' => :'width',
+        :'height' => :'height',
+        :'steps' => :'steps',
+        :'cfg' => :'cfg',
+        :'seed' => :'seed',
+        :'sampler' => :'sampler',
+        :'scheduler' => :'scheduler',
+        :'vae' => :'vae',
+        :'lora_name' => :'lora_name',
+        :'lora_strength' => :'lora_strength'
+      }
+    end
+
+    # Returns attribute mapping this model knows about
+    def self.acceptable_attribute_map
+      attribute_map
+    end
+
+    # Returns all the JSON keys this model knows about
+    def self.acceptable_attributes
+      acceptable_attribute_map.values
+    end
+
+    # Attribute type mapping.
+    def self.openapi_types
+      {
+        :'prompt' => :'String',
+        :'negative_prompt' => :'String',
+        :'model' => :'String',
+        :'width' => :'Integer',
+        :'height' => :'Integer',
+        :'steps' => :'Integer',
+        :'cfg' => :'Float',
+        :'seed' => :'Integer',
+        :'sampler' => :'String',
+        :'scheduler' => :'String',
+        :'vae' => :'String',
+        :'lora_name' => :'String',
+        :'lora_strength' => :'Float'
+      }
+    end
+
+    # List of attributes with nullable: true
+    def self.openapi_nullable
+      Set.new([
+        :'model',
+        :'vae',
+        :'lora_name',
+      ])
+    end
+
+    # Initializes the object
+    # @param [Hash] attributes Model attributes in the form of hash
+    def initialize(attributes = {})
+      if (!attributes.is_a?(Hash))
+        fail ArgumentError, "The input argument (attributes) must be a hash in `TensorsApi::GenerateRequest` initialize method"
+      end
+
+      # check to see if the attribute exists and convert string to symbol for hash key
+      acceptable_attribute_map = self.class.acceptable_attribute_map
+      attributes = attributes.each_with_object({}) { |(k, v), h|
+        if (!acceptable_attribute_map.key?(k.to_sym))
+          fail ArgumentError, "`#{k}` is not a valid attribute in `TensorsApi::GenerateRequest`. Please check the name to make sure it's valid. List of attributes: " + acceptable_attribute_map.keys.inspect
+        end
+        h[k.to_sym] = v
+      }
+
+      if attributes.key?(:'prompt')
+        self.prompt = attributes[:'prompt']
+      else
+        self.prompt = nil
+      end
+
+      if attributes.key?(:'negative_prompt')
+        self.negative_prompt = attributes[:'negative_prompt']
+      else
+        self.negative_prompt = ''
+      end
+
+      if attributes.key?(:'model')
+        self.model = attributes[:'model']
+      end
+
+      if attributes.key?(:'width')
+        self.width = attributes[:'width']
+      else
+        self.width = 1024
+      end
+
+      if attributes.key?(:'height')
+        self.height = attributes[:'height']
+      else
+        self.height = 1024
+      end
+
+      if attributes.key?(:'steps')
+        self.steps = attributes[:'steps']
+      else
+        self.steps = 20
+      end
+
+      if attributes.key?(:'cfg')
+        self.cfg = attributes[:'cfg']
+      else
+        self.cfg = 7.0
+      end
+
+      if attributes.key?(:'seed')
+        self.seed = attributes[:'seed']
+      else
+        self.seed = -1
+      end
+
+      if attributes.key?(:'sampler')
+        self.sampler = attributes[:'sampler']
+      else
+        self.sampler = 'euler'
+      end
+
+      if attributes.key?(:'scheduler')
+        self.scheduler = attributes[:'scheduler']
+      else
+        self.scheduler = 'normal'
+      end
+
+      if attributes.key?(:'vae')
+        self.vae = attributes[:'vae']
+      end
+
+      if attributes.key?(:'lora_name')
+        self.lora_name = attributes[:'lora_name']
+      end
+
+      if attributes.key?(:'lora_strength')
+        self.lora_strength = attributes[:'lora_strength']
+      else
+        self.lora_strength = 0.8
+      end
+    end
+
+    # Show invalid properties with the reasons. Usually used together with valid?
+    # @return Array for valid properties with the reasons
+    def list_invalid_properties
+      warn '[DEPRECATED] the `list_invalid_properties` method is obsolete'
+      invalid_properties = Array.new
+      if @prompt.nil?
+        invalid_properties.push('invalid value for "prompt", prompt cannot be nil.')
+      end
+
+      if !@width.nil? && @width > 4096
+        invalid_properties.push('invalid value for "width", must be smaller than or equal to 4096.')
+      end
+
+      if !@width.nil? && @width < 64
+        invalid_properties.push('invalid value for "width", must be greater than or equal to 64.')
+      end
+
+      if !@height.nil? && @height > 4096
+        invalid_properties.push('invalid value for "height", must be smaller than or equal to 4096.')
+      end
+
+      if !@height.nil? && @height < 64
+        invalid_properties.push('invalid value for "height", must be greater than or equal to 64.')
+      end
+
+      if !@steps.nil? && @steps > 150
+        invalid_properties.push('invalid value for "steps", must be smaller than or equal to 150.')
+      end
+
+      if !@steps.nil? && @steps < 1
+        invalid_properties.push('invalid value for "steps", must be greater than or equal to 1.')
+      end
+
+      if !@cfg.nil? && @cfg > 30.0
+        invalid_properties.push('invalid value for "cfg", must be smaller than or equal to 30.0.')
+      end
+
+      if !@cfg.nil? && @cfg < 1.0
+        invalid_properties.push('invalid value for "cfg", must be greater than or equal to 1.0.')
+      end
+
+      if !@lora_strength.nil? && @lora_strength > 2.0
+        invalid_properties.push('invalid value for "lora_strength", must be smaller than or equal to 2.0.')
+      end
+
+      if !@lora_strength.nil? && @lora_strength < 0.0
+        invalid_properties.push('invalid value for "lora_strength", must be greater than or equal to 0.0.')
+      end
+
+      invalid_properties
+    end
+
+    # Check to see if the all the properties in the model are valid
+    # @return true if the model is valid
+    def valid?
+      warn '[DEPRECATED] the `valid?` method is obsolete'
+      return false if @prompt.nil?
+      return false if !@width.nil? && @width > 4096
+      return false if !@width.nil? && @width < 64
+      return false if !@height.nil? && @height > 4096
+      return false if !@height.nil? && @height < 64
+      return false if !@steps.nil? && @steps > 150
+      return false if !@steps.nil? && @steps < 1
+      return false if !@cfg.nil? && @cfg > 30.0
+      return false if !@cfg.nil? && @cfg < 1.0
+      return false if !@lora_strength.nil? && @lora_strength > 2.0
+      return false if !@lora_strength.nil? && @lora_strength < 0.0
+      true
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] prompt Value to be assigned
+    def prompt=(prompt)
+      if prompt.nil?
+        fail ArgumentError, 'prompt cannot be nil'
+      end
+
+      @prompt = prompt
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] width Value to be assigned
+    def width=(width)
+      if width.nil?
+        fail ArgumentError, 'width cannot be nil'
+      end
+
+      if width > 4096
+        fail ArgumentError, 'invalid value for "width", must be smaller than or equal to 4096.'
+      end
+
+      if width < 64
+        fail ArgumentError, 'invalid value for "width", must be greater than or equal to 64.'
+      end
+
+      @width = width
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] height Value to be assigned
+    def height=(height)
+      if height.nil?
+        fail ArgumentError, 'height cannot be nil'
+      end
+
+      if height > 4096
+        fail ArgumentError, 'invalid value for "height", must be smaller than or equal to 4096.'
+      end
+
+      if height < 64
+        fail ArgumentError, 'invalid value for "height", must be greater than or equal to 64.'
+      end
+
+      @height = height
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] steps Value to be assigned
+    def steps=(steps)
+      if steps.nil?
+        fail ArgumentError, 'steps cannot be nil'
+      end
+
+      if steps > 150
+        fail ArgumentError, 'invalid value for "steps", must be smaller than or equal to 150.'
+      end
+
+      if steps < 1
+        fail ArgumentError, 'invalid value for "steps", must be greater than or equal to 1.'
+      end
+
+      @steps = steps
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] cfg Value to be assigned
+    def cfg=(cfg)
+      if cfg.nil?
+        fail ArgumentError, 'cfg cannot be nil'
+      end
+
+      if cfg > 30.0
+        fail ArgumentError, 'invalid value for "cfg", must be smaller than or equal to 30.0.'
+      end
+
+      if cfg < 1.0
+        fail ArgumentError, 'invalid value for "cfg", must be greater than or equal to 1.0.'
+      end
+
+      @cfg = cfg
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] lora_strength Value to be assigned
+    def lora_strength=(lora_strength)
+      if lora_strength.nil?
+        fail ArgumentError, 'lora_strength cannot be nil'
+      end
+
+      if lora_strength > 2.0
+        fail ArgumentError, 'invalid value for "lora_strength", must be smaller than or equal to 2.0.'
+      end
+
+      if lora_strength < 0.0
+        fail ArgumentError, 'invalid value for "lora_strength", must be greater than or equal to 0.0.'
+      end
+
+      @lora_strength = lora_strength
+    end
+
+    # Checks equality by comparing each attribute.
+    # @param [Object] Object to be compared
+    def ==(o)
+      return true if self.equal?(o)
+      self.class == o.class &&
+          prompt == o.prompt &&
+          negative_prompt == o.negative_prompt &&
+          model == o.model &&
+          width == o.width &&
+          height == o.height &&
+          steps == o.steps &&
+          cfg == o.cfg &&
+          seed == o.seed &&
+          sampler == o.sampler &&
+          scheduler == o.scheduler &&
+          vae == o.vae &&
+          lora_name == o.lora_name &&
+          lora_strength == o.lora_strength
+    end
+
+    # @see the `==` method
+    # @param [Object] Object to be compared
+    def eql?(o)
+      self == o
+    end
+
+    # Calculates hash code according to all attributes.
+    # @return [Integer] Hash code
+    def hash
+      [prompt, negative_prompt, model, width, height, steps, cfg, seed, sampler, scheduler, vae, lora_name, lora_strength].hash
+    end
+
+    # Builds the object from hash
+    # @param [Hash] attributes Model attributes in the form of hash
+    # @return [Object] Returns the model itself
+    def self.build_from_hash(attributes)
+      return nil unless attributes.is_a?(Hash)
+      attributes = attributes.transform_keys(&:to_sym)
+      transformed_hash = {}
+      openapi_types.each_pair do |key, type|
+        if attributes.key?(attribute_map[key]) && attributes[attribute_map[key]].nil?
+          transformed_hash["#{key}"] = nil
+        elsif type =~ /\AArray<(.*)>/i
+          # check to ensure the input is an array given that the attribute
+          # is documented as an array but the input is not
+          if attributes[attribute_map[key]].is_a?(Array)
+            transformed_hash["#{key}"] = attributes[attribute_map[key]].map { |v| _deserialize($1, v) }
+          end
+        elsif !attributes[attribute_map[key]].nil?
+          transformed_hash["#{key}"] = _deserialize(type, attributes[attribute_map[key]])
+        end
+      end
+      new(transformed_hash)
+    end
+
+    # Returns the object in the form of hash
+    # @return [Hash] Returns the object in the form of hash
+    def to_hash
+      hash = {}
+      self.class.attribute_map.each_pair do |attr, param|
+        value = self.send(attr)
+        if value.nil?
+          is_nullable = self.class.openapi_nullable.include?(attr)
+          next if !is_nullable || (is_nullable && !instance_variable_defined?(:"@#{attr}"))
+        end
+
+        hash[param] = _to_hash(value)
+      end
+      hash
+    end
+
+  end
+
+end