daytona_api_client 0.182.0 → 0.183.0

data/lib/daytona_api_client/models/create_organization_region_quota.rb

@@ -0,0 +1,360 @@
+=begin
+#Daytona
+
+#Daytona AI platform API Docs
+
+The version of the OpenAPI document: 1.0
+Contact: support@daytona.com
+Generated by: https://openapi-generator.tech
+Generator version: 7.21.0
+
+=end
+
+require 'date'
+require 'time'
+
+module DaytonaApiClient
+  class CreateOrganizationRegionQuota < ApiModelBase
+    attr_accessor :sandbox_class
+
+    attr_accessor :total_cpu_quota
+
+    attr_accessor :total_memory_quota
+
+    attr_accessor :total_disk_quota
+
+    attr_accessor :total_gpu_quota
+
+    attr_accessor :max_cpu_per_sandbox
+
+    attr_accessor :max_memory_per_sandbox
+
+    attr_accessor :max_disk_per_sandbox
+
+    attr_accessor :max_disk_per_non_ephemeral_sandbox
+
+    attr_accessor :max_cpu_per_gpu_sandbox
+
+    attr_accessor :max_memory_per_gpu_sandbox
+
+    attr_accessor :max_disk_per_gpu_sandbox
+
+    class EnumAttributeValidator
+      attr_reader :datatype
+      attr_reader :allowable_values
+
+      def initialize(datatype, allowable_values)
+        @allowable_values = allowable_values.map do |value|
+          case datatype.to_s
+          when /Integer/i
+            value.to_i
+          when /Float/i
+            value.to_f
+          else
+            value
+          end
+        end
+      end
+
+      def valid?(value)
+        !value || allowable_values.include?(value)
+      end
+    end
+
+    # Attribute mapping from ruby-style variable name to JSON key.
+    def self.attribute_map
+      {
+        :'sandbox_class' => :'sandboxClass',
+        :'total_cpu_quota' => :'totalCpuQuota',
+        :'total_memory_quota' => :'totalMemoryQuota',
+        :'total_disk_quota' => :'totalDiskQuota',
+        :'total_gpu_quota' => :'totalGpuQuota',
+        :'max_cpu_per_sandbox' => :'maxCpuPerSandbox',
+        :'max_memory_per_sandbox' => :'maxMemoryPerSandbox',
+        :'max_disk_per_sandbox' => :'maxDiskPerSandbox',
+        :'max_disk_per_non_ephemeral_sandbox' => :'maxDiskPerNonEphemeralSandbox',
+        :'max_cpu_per_gpu_sandbox' => :'maxCpuPerGpuSandbox',
+        :'max_memory_per_gpu_sandbox' => :'maxMemoryPerGpuSandbox',
+        :'max_disk_per_gpu_sandbox' => :'maxDiskPerGpuSandbox'
+      }
+    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
+      {
+        :'sandbox_class' => :'SandboxClass',
+        :'total_cpu_quota' => :'Float',
+        :'total_memory_quota' => :'Float',
+        :'total_disk_quota' => :'Float',
+        :'total_gpu_quota' => :'Float',
+        :'max_cpu_per_sandbox' => :'Float',
+        :'max_memory_per_sandbox' => :'Float',
+        :'max_disk_per_sandbox' => :'Float',
+        :'max_disk_per_non_ephemeral_sandbox' => :'Float',
+        :'max_cpu_per_gpu_sandbox' => :'Float',
+        :'max_memory_per_gpu_sandbox' => :'Float',
+        :'max_disk_per_gpu_sandbox' => :'Float'
+      }
+    end
+
+    # List of attributes with nullable: true
+    def self.openapi_nullable
+      Set.new([
+        :'max_cpu_per_sandbox',
+        :'max_memory_per_sandbox',
+        :'max_disk_per_sandbox',
+        :'max_disk_per_non_ephemeral_sandbox',
+        :'max_cpu_per_gpu_sandbox',
+        :'max_memory_per_gpu_sandbox',
+        :'max_disk_per_gpu_sandbox'
+      ])
+    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 `DaytonaApiClient::CreateOrganizationRegionQuota` 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 `DaytonaApiClient::CreateOrganizationRegionQuota`. 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?(:'sandbox_class')
+        self.sandbox_class = attributes[:'sandbox_class']
+      else
+        self.sandbox_class = nil
+      end
+
+      if attributes.key?(:'total_cpu_quota')
+        self.total_cpu_quota = attributes[:'total_cpu_quota']
+      else
+        self.total_cpu_quota = nil
+      end
+
+      if attributes.key?(:'total_memory_quota')
+        self.total_memory_quota = attributes[:'total_memory_quota']
+      else
+        self.total_memory_quota = nil
+      end
+
+      if attributes.key?(:'total_disk_quota')
+        self.total_disk_quota = attributes[:'total_disk_quota']
+      else
+        self.total_disk_quota = nil
+      end
+
+      if attributes.key?(:'total_gpu_quota')
+        self.total_gpu_quota = attributes[:'total_gpu_quota']
+      else
+        self.total_gpu_quota = nil
+      end
+
+      if attributes.key?(:'max_cpu_per_sandbox')
+        self.max_cpu_per_sandbox = attributes[:'max_cpu_per_sandbox']
+      end
+
+      if attributes.key?(:'max_memory_per_sandbox')
+        self.max_memory_per_sandbox = attributes[:'max_memory_per_sandbox']
+      end
+
+      if attributes.key?(:'max_disk_per_sandbox')
+        self.max_disk_per_sandbox = attributes[:'max_disk_per_sandbox']
+      end
+
+      if attributes.key?(:'max_disk_per_non_ephemeral_sandbox')
+        self.max_disk_per_non_ephemeral_sandbox = attributes[:'max_disk_per_non_ephemeral_sandbox']
+      end
+
+      if attributes.key?(:'max_cpu_per_gpu_sandbox')
+        self.max_cpu_per_gpu_sandbox = attributes[:'max_cpu_per_gpu_sandbox']
+      end
+
+      if attributes.key?(:'max_memory_per_gpu_sandbox')
+        self.max_memory_per_gpu_sandbox = attributes[:'max_memory_per_gpu_sandbox']
+      end
+
+      if attributes.key?(:'max_disk_per_gpu_sandbox')
+        self.max_disk_per_gpu_sandbox = attributes[:'max_disk_per_gpu_sandbox']
+      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 @sandbox_class.nil?
+        invalid_properties.push('invalid value for "sandbox_class", sandbox_class cannot be nil.')
+      end
+
+      if @total_cpu_quota.nil?
+        invalid_properties.push('invalid value for "total_cpu_quota", total_cpu_quota cannot be nil.')
+      end
+
+      if @total_memory_quota.nil?
+        invalid_properties.push('invalid value for "total_memory_quota", total_memory_quota cannot be nil.')
+      end
+
+      if @total_disk_quota.nil?
+        invalid_properties.push('invalid value for "total_disk_quota", total_disk_quota cannot be nil.')
+      end
+
+      if @total_gpu_quota.nil?
+        invalid_properties.push('invalid value for "total_gpu_quota", total_gpu_quota cannot be nil.')
+      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 @sandbox_class.nil?
+      return false if @total_cpu_quota.nil?
+      return false if @total_memory_quota.nil?
+      return false if @total_disk_quota.nil?
+      return false if @total_gpu_quota.nil?
+      true
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] sandbox_class Value to be assigned
+    def sandbox_class=(sandbox_class)
+      if sandbox_class.nil?
+        fail ArgumentError, 'sandbox_class cannot be nil'
+      end
+
+      @sandbox_class = sandbox_class
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] total_cpu_quota Value to be assigned
+    def total_cpu_quota=(total_cpu_quota)
+      if total_cpu_quota.nil?
+        fail ArgumentError, 'total_cpu_quota cannot be nil'
+      end
+
+      @total_cpu_quota = total_cpu_quota
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] total_memory_quota Value to be assigned
+    def total_memory_quota=(total_memory_quota)
+      if total_memory_quota.nil?
+        fail ArgumentError, 'total_memory_quota cannot be nil'
+      end
+
+      @total_memory_quota = total_memory_quota
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] total_disk_quota Value to be assigned
+    def total_disk_quota=(total_disk_quota)
+      if total_disk_quota.nil?
+        fail ArgumentError, 'total_disk_quota cannot be nil'
+      end
+
+      @total_disk_quota = total_disk_quota
+    end
+
+    # Custom attribute writer method with validation
+    # @param [Object] total_gpu_quota Value to be assigned
+    def total_gpu_quota=(total_gpu_quota)
+      if total_gpu_quota.nil?
+        fail ArgumentError, 'total_gpu_quota cannot be nil'
+      end
+
+      @total_gpu_quota = total_gpu_quota
+    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 &&
+          sandbox_class == o.sandbox_class &&
+          total_cpu_quota == o.total_cpu_quota &&
+          total_memory_quota == o.total_memory_quota &&
+          total_disk_quota == o.total_disk_quota &&
+          total_gpu_quota == o.total_gpu_quota &&
+          max_cpu_per_sandbox == o.max_cpu_per_sandbox &&
+          max_memory_per_sandbox == o.max_memory_per_sandbox &&
+          max_disk_per_sandbox == o.max_disk_per_sandbox &&
+          max_disk_per_non_ephemeral_sandbox == o.max_disk_per_non_ephemeral_sandbox &&
+          max_cpu_per_gpu_sandbox == o.max_cpu_per_gpu_sandbox &&
+          max_memory_per_gpu_sandbox == o.max_memory_per_gpu_sandbox &&
+          max_disk_per_gpu_sandbox == o.max_disk_per_gpu_sandbox
+    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
+      [sandbox_class, total_cpu_quota, total_memory_quota, total_disk_quota, total_gpu_quota, max_cpu_per_sandbox, max_memory_per_sandbox, max_disk_per_sandbox, max_disk_per_non_ephemeral_sandbox, max_cpu_per_gpu_sandbox, max_memory_per_gpu_sandbox, max_disk_per_gpu_sandbox].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/daytona_api_client/models/create_sandbox.rb

@@ -39,9 +39,6 @@ module DaytonaApiClient
     # Comma-separated list of allowed CIDR network addresses for the sandbox
     attr_accessor :network_allow_list
 
-    # The sandbox class type
-    attr_accessor :_class
-
     # The target (region) where the sandbox will be created
     attr_accessor :target
 
@@ -72,27 +69,8 @@ module DaytonaApiClient
     # Build information for the sandbox
     attr_accessor :build_info
 
-    class EnumAttributeValidator
-      attr_reader :datatype
-      attr_reader :allowable_values
-
-      def initialize(datatype, allowable_values)
-        @allowable_values = allowable_values.map do |value|
-          case datatype.to_s
-          when /Integer/i
-            value.to_i
-          when /Float/i
-            value.to_f
-          else
-            value
-          end
-        end
-      end
-
-      def valid?(value)
-        !value || allowable_values.include?(value)
-      end
-    end
+    # ID or name of an existing sandbox to link the new sandbox to. The new sandbox will be scheduled on the same runner as the linked sandbox so a local network can be established between them. Linked sandboxes must be ephemeral (autoDeleteInterval=0) and cannot themselves be linked to another sandbox.
+    attr_accessor :linked_sandbox
 
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
@@ -105,7 +83,6 @@ module DaytonaApiClient
         :'public' => :'public',
         :'network_block_all' => :'networkBlockAll',
         :'network_allow_list' => :'networkAllowList',
-        :'_class' => :'class',
         :'target' => :'target',
         :'cpu' => :'cpu',
         :'gpu' => :'gpu',
@@ -115,7 +92,8 @@ module DaytonaApiClient
         :'auto_archive_interval' => :'autoArchiveInterval',
         :'auto_delete_interval' => :'autoDeleteInterval',
         :'volumes' => :'volumes',
-        :'build_info' => :'buildInfo'
+        :'build_info' => :'buildInfo',
+        :'linked_sandbox' => :'linkedSandbox'
       }
     end
 
@@ -140,7 +118,6 @@ module DaytonaApiClient
         :'public' => :'Boolean',
         :'network_block_all' => :'Boolean',
         :'network_allow_list' => :'String',
-        :'_class' => :'String',
         :'target' => :'String',
         :'cpu' => :'Integer',
         :'gpu' => :'Integer',
@@ -150,7 +127,8 @@ module DaytonaApiClient
         :'auto_archive_interval' => :'Integer',
         :'auto_delete_interval' => :'Integer',
         :'volumes' => :'Array<SandboxVolume>',
-        :'build_info' => :'CreateBuildInfo'
+        :'build_info' => :'CreateBuildInfo',
+        :'linked_sandbox' => :'String'
       }
     end
 
@@ -212,10 +190,6 @@ module DaytonaApiClient
         self.network_allow_list = attributes[:'network_allow_list']
       end
 
-      if attributes.key?(:'_class')
-        self._class = attributes[:'_class']
-      end
-
       if attributes.key?(:'target')
         self.target = attributes[:'target']
       end
@@ -257,6 +231,10 @@ module DaytonaApiClient
       if attributes.key?(:'build_info')
         self.build_info = attributes[:'build_info']
       end
+
+      if attributes.key?(:'linked_sandbox')
+        self.linked_sandbox = attributes[:'linked_sandbox']
+      end
     end
 
     # Show invalid properties with the reasons. Usually used together with valid?
@@ -271,21 +249,9 @@ module DaytonaApiClient
     # @return true if the model is valid
     def valid?
       warn '[DEPRECATED] the `valid?` method is obsolete'
-      _class_validator = EnumAttributeValidator.new('String', ["small", "medium", "large", "unknown_default_open_api"])
-      return false unless _class_validator.valid?(@_class)
       true
     end
 
-    # Custom attribute writer method checking allowed values (enum).
-    # @param [Object] _class Object to be assigned
-    def _class=(_class)
-      validator = EnumAttributeValidator.new('String', ["small", "medium", "large", "unknown_default_open_api"])
-      unless validator.valid?(_class)
-        fail ArgumentError, "invalid value for \"_class\", must be one of #{validator.allowable_values}."
-      end
-      @_class = _class
-    end
-
     # Checks equality by comparing each attribute.
     # @param [Object] Object to be compared
     def ==(o)
@@ -299,7 +265,6 @@ module DaytonaApiClient
           public == o.public &&
           network_block_all == o.network_block_all &&
           network_allow_list == o.network_allow_list &&
-          _class == o._class &&
           target == o.target &&
           cpu == o.cpu &&
           gpu == o.gpu &&
@@ -309,7 +274,8 @@ module DaytonaApiClient
           auto_archive_interval == o.auto_archive_interval &&
           auto_delete_interval == o.auto_delete_interval &&
           volumes == o.volumes &&
-          build_info == o.build_info
+          build_info == o.build_info &&
+          linked_sandbox == o.linked_sandbox
     end
 
     # @see the `==` method
@@ -321,7 +287,7 @@ module DaytonaApiClient
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [name, snapshot, user, env, labels, public, network_block_all, network_allow_list, _class, target, cpu, gpu, memory, disk, auto_stop_interval, auto_archive_interval, auto_delete_interval, volumes, build_info].hash
+      [name, snapshot, user, env, labels, public, network_block_all, network_allow_list, target, cpu, gpu, memory, disk, auto_stop_interval, auto_archive_interval, auto_delete_interval, volumes, build_info, linked_sandbox].hash
     end
 
     # Builds the object from hash

data/lib/daytona_api_client/models/create_snapshot.rb

@@ -42,6 +42,31 @@ module DaytonaApiClient
     # ID of the region where the snapshot will be available. Defaults to organization default region if not specified.
     attr_accessor :region_id
 
+    # Target sandbox class. Determines which runners can host sandboxes created from this snapshot.
+    attr_accessor :sandbox_class
+
+    class EnumAttributeValidator
+      attr_reader :datatype
+      attr_reader :allowable_values
+
+      def initialize(datatype, allowable_values)
+        @allowable_values = allowable_values.map do |value|
+          case datatype.to_s
+          when /Integer/i
+            value.to_i
+          when /Float/i
+            value.to_f
+          else
+            value
+          end
+        end
+      end
+
+      def valid?(value)
+        !value || allowable_values.include?(value)
+      end
+    end
+
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
       {
@@ -53,7 +78,8 @@ module DaytonaApiClient
         :'memory' => :'memory',
         :'disk' => :'disk',
         :'build_info' => :'buildInfo',
-        :'region_id' => :'regionId'
+        :'region_id' => :'regionId',
+        :'sandbox_class' => :'sandboxClass'
       }
     end
 
@@ -78,7 +104,8 @@ module DaytonaApiClient
         :'memory' => :'Integer',
         :'disk' => :'Integer',
         :'build_info' => :'CreateBuildInfo',
-        :'region_id' => :'String'
+        :'region_id' => :'String',
+        :'sandbox_class' => :'SandboxClass'
       }
     end
 
@@ -143,6 +170,10 @@ module DaytonaApiClient
       if attributes.key?(:'region_id')
         self.region_id = attributes[:'region_id']
       end
+
+      if attributes.key?(:'sandbox_class')
+        self.sandbox_class = attributes[:'sandbox_class']
+      end
     end
 
     # Show invalid properties with the reasons. Usually used together with valid?
@@ -188,7 +219,8 @@ module DaytonaApiClient
           memory == o.memory &&
           disk == o.disk &&
           build_info == o.build_info &&
-          region_id == o.region_id
+          region_id == o.region_id &&
+          sandbox_class == o.sandbox_class
     end
 
     # @see the `==` method
@@ -200,7 +232,7 @@ module DaytonaApiClient
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [name, image_name, entrypoint, cpu, gpu, memory, disk, build_info, region_id].hash
+      [name, image_name, entrypoint, cpu, gpu, memory, disk, build_info, region_id, sandbox_class].hash
     end
 
     # Builds the object from hash

data/lib/daytona_api_client/models/region_quota.rb

@@ -19,6 +19,8 @@ module DaytonaApiClient
 
     attr_accessor :region_id
 
+    attr_accessor :sandbox_class
+
     attr_accessor :total_cpu_quota
 
     attr_accessor :total_memory_quota
@@ -41,11 +43,34 @@ module DaytonaApiClient
 
     attr_accessor :max_disk_per_gpu_sandbox
 
+    class EnumAttributeValidator
+      attr_reader :datatype
+      attr_reader :allowable_values
+
+      def initialize(datatype, allowable_values)
+        @allowable_values = allowable_values.map do |value|
+          case datatype.to_s
+          when /Integer/i
+            value.to_i
+          when /Float/i
+            value.to_f
+          else
+            value
+          end
+        end
+      end
+
+      def valid?(value)
+        !value || allowable_values.include?(value)
+      end
+    end
+
     # Attribute mapping from ruby-style variable name to JSON key.
     def self.attribute_map
       {
         :'organization_id' => :'organizationId',
         :'region_id' => :'regionId',
+        :'sandbox_class' => :'sandboxClass',
         :'total_cpu_quota' => :'totalCpuQuota',
         :'total_memory_quota' => :'totalMemoryQuota',
         :'total_disk_quota' => :'totalDiskQuota',
@@ -75,6 +100,7 @@ module DaytonaApiClient
       {
         :'organization_id' => :'String',
         :'region_id' => :'String',
+        :'sandbox_class' => :'SandboxClass',
         :'total_cpu_quota' => :'Float',
         :'total_memory_quota' => :'Float',
         :'total_disk_quota' => :'Float',
@@ -130,6 +156,12 @@ module DaytonaApiClient
         self.region_id = nil
       end
 
+      if attributes.key?(:'sandbox_class')
+        self.sandbox_class = attributes[:'sandbox_class']
+      else
+        self.sandbox_class = nil
+      end
+
       if attributes.key?(:'total_cpu_quota')
         self.total_cpu_quota = attributes[:'total_cpu_quota']
       else
@@ -210,6 +242,10 @@ module DaytonaApiClient
         invalid_properties.push('invalid value for "region_id", region_id cannot be nil.')
       end
 
+      if @sandbox_class.nil?
+        invalid_properties.push('invalid value for "sandbox_class", sandbox_class cannot be nil.')
+      end
+
       if @total_cpu_quota.nil?
         invalid_properties.push('invalid value for "total_cpu_quota", total_cpu_quota cannot be nil.')
       end
@@ -235,6 +271,7 @@ module DaytonaApiClient
       warn '[DEPRECATED] the `valid?` method is obsolete'
       return false if @organization_id.nil?
       return false if @region_id.nil?
+      return false if @sandbox_class.nil?
       return false if @total_cpu_quota.nil?
       return false if @total_memory_quota.nil?
       return false if @total_disk_quota.nil?
@@ -262,6 +299,16 @@ module DaytonaApiClient
       @region_id = region_id
     end
 
+    # Custom attribute writer method with validation
+    # @param [Object] sandbox_class Value to be assigned
+    def sandbox_class=(sandbox_class)
+      if sandbox_class.nil?
+        fail ArgumentError, 'sandbox_class cannot be nil'
+      end
+
+      @sandbox_class = sandbox_class
+    end
+
     # Custom attribute writer method with validation
     # @param [Object] total_cpu_quota Value to be assigned
     def total_cpu_quota=(total_cpu_quota)
@@ -309,6 +356,7 @@ module DaytonaApiClient
       self.class == o.class &&
           organization_id == o.organization_id &&
           region_id == o.region_id &&
+          sandbox_class == o.sandbox_class &&
           total_cpu_quota == o.total_cpu_quota &&
           total_memory_quota == o.total_memory_quota &&
           total_disk_quota == o.total_disk_quota &&
@@ -331,7 +379,7 @@ module DaytonaApiClient
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [organization_id, region_id, total_cpu_quota, total_memory_quota, total_disk_quota, total_gpu_quota, max_cpu_per_sandbox, max_memory_per_sandbox, max_disk_per_sandbox, max_disk_per_non_ephemeral_sandbox, max_cpu_per_gpu_sandbox, max_memory_per_gpu_sandbox, max_disk_per_gpu_sandbox].hash
+      [organization_id, region_id, sandbox_class, total_cpu_quota, total_memory_quota, total_disk_quota, total_gpu_quota, max_cpu_per_sandbox, max_memory_per_sandbox, max_disk_per_sandbox, max_disk_per_non_ephemeral_sandbox, max_cpu_per_gpu_sandbox, max_memory_per_gpu_sandbox, max_disk_per_gpu_sandbox].hash
     end
 
     # Builds the object from hash