ragie_ruby_sdk 1.0.20 → 1.0.22

data/docs/UpdatePartitionParams.md

@@ -0,0 +1,22 @@
+# RagieRubySdk::UpdatePartitionParams
+
+## Properties
+
+| Name | Type | Description | Notes |
+| ---- | ---- | ----------- | ----- |
+| **context_aware** | **Boolean** |  | [optional] |
+| **description** | **String** |  | [optional] |
+| **metadata_schema** | [**Hash<String, CreatePartitionParamsMetadataSchemaValue>**](CreatePartitionParamsMetadataSchemaValue.md) |  | [optional] |
+
+## Example
+
+```ruby
+require 'ragie_ruby_sdk'
+
+instance = RagieRubySdk::UpdatePartitionParams.new(
+  context_aware: null,
+  description: null,
+  metadata_schema: null
+)
+```
+

data/lib/ragie_ruby_sdk/api/beta_api.rb

@@ -292,132 +292,6 @@ module RagieRubySdk
       return data, status_code, headers
     end
 
-    # Disable Mcp
-    # Disables context-aware descriptions for a partition. This will stop automatically generating descriptions for the partition.
-    # @param partition_id [String] 
-    # @param [Hash] opts the optional parameters
-    # @return [Object]
-    def disable_mcp_partitions_partition_id_mcp_delete(partition_id, opts = {})
-      data, _status_code, _headers = disable_mcp_partitions_partition_id_mcp_delete_with_http_info(partition_id, opts)
-      data
-    end
-
-    # Disable Mcp
-    # Disables context-aware descriptions for a partition. This will stop automatically generating descriptions for the partition.
-    # @param partition_id [String] 
-    # @param [Hash] opts the optional parameters
-    # @return [Array<(Object, Integer, Hash)>] Object data, response status code and response headers
-    def disable_mcp_partitions_partition_id_mcp_delete_with_http_info(partition_id, opts = {})
-      if @api_client.config.debugging
-        @api_client.config.logger.debug 'Calling API: BetaApi.disable_mcp_partitions_partition_id_mcp_delete ...'
-      end
-      # verify the required parameter 'partition_id' is set
-      if @api_client.config.client_side_validation && partition_id.nil?
-        fail ArgumentError, "Missing the required parameter 'partition_id' when calling BetaApi.disable_mcp_partitions_partition_id_mcp_delete"
-      end
-      # resource path
-      local_var_path = '/partitions/{partition_id}/mcp'.sub('{' + 'partition_id' + '}', CGI.escape(partition_id.to_s))
-
-      # query parameters
-      query_params = opts[:query_params] || {}
-
-      # header parameters
-      header_params = opts[:header_params] || {}
-      # HTTP header 'Accept' (if needed)
-      header_params['Accept'] = @api_client.select_header_accept(['application/json']) unless header_params['Accept']
-
-      # form parameters
-      form_params = opts[:form_params] || {}
-
-      # http body (model)
-      post_body = opts[:debug_body]
-
-      # return_type
-      return_type = opts[:debug_return_type] || 'Object'
-
-      # auth_names
-      auth_names = opts[:debug_auth_names] || ['auth']
-
-      new_options = opts.merge(
-        :operation => :"BetaApi.disable_mcp_partitions_partition_id_mcp_delete",
-        :header_params => header_params,
-        :query_params => query_params,
-        :form_params => form_params,
-        :body => post_body,
-        :auth_names => auth_names,
-        :return_type => return_type
-      )
-
-      data, status_code, headers = @api_client.call_api(:DELETE, local_var_path, new_options)
-      if @api_client.config.debugging
-        @api_client.config.logger.debug "API called: BetaApi#disable_mcp_partitions_partition_id_mcp_delete\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
-      end
-      return data, status_code, headers
-    end
-
-    # Enable Mcp
-    # Enables context-aware descriptions for a partition. This will allow the automatically generate a desccription for based on the documents in the partition.
-    # @param partition_id [String] 
-    # @param [Hash] opts the optional parameters
-    # @return [Object]
-    def enable_mcp_partitions_partition_id_mcp_post(partition_id, opts = {})
-      data, _status_code, _headers = enable_mcp_partitions_partition_id_mcp_post_with_http_info(partition_id, opts)
-      data
-    end
-
-    # Enable Mcp
-    # Enables context-aware descriptions for a partition. This will allow the automatically generate a desccription for based on the documents in the partition.
-    # @param partition_id [String] 
-    # @param [Hash] opts the optional parameters
-    # @return [Array<(Object, Integer, Hash)>] Object data, response status code and response headers
-    def enable_mcp_partitions_partition_id_mcp_post_with_http_info(partition_id, opts = {})
-      if @api_client.config.debugging
-        @api_client.config.logger.debug 'Calling API: BetaApi.enable_mcp_partitions_partition_id_mcp_post ...'
-      end
-      # verify the required parameter 'partition_id' is set
-      if @api_client.config.client_side_validation && partition_id.nil?
-        fail ArgumentError, "Missing the required parameter 'partition_id' when calling BetaApi.enable_mcp_partitions_partition_id_mcp_post"
-      end
-      # resource path
-      local_var_path = '/partitions/{partition_id}/mcp'.sub('{' + 'partition_id' + '}', CGI.escape(partition_id.to_s))
-
-      # query parameters
-      query_params = opts[:query_params] || {}
-
-      # header parameters
-      header_params = opts[:header_params] || {}
-      # HTTP header 'Accept' (if needed)
-      header_params['Accept'] = @api_client.select_header_accept(['application/json']) unless header_params['Accept']
-
-      # form parameters
-      form_params = opts[:form_params] || {}
-
-      # http body (model)
-      post_body = opts[:debug_body]
-
-      # return_type
-      return_type = opts[:debug_return_type] || 'Object'
-
-      # auth_names
-      auth_names = opts[:debug_auth_names] || ['auth']
-
-      new_options = opts.merge(
-        :operation => :"BetaApi.enable_mcp_partitions_partition_id_mcp_post",
-        :header_params => header_params,
-        :query_params => query_params,
-        :form_params => form_params,
-        :body => post_body,
-        :auth_names => auth_names,
-        :return_type => return_type
-      )
-
-      data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
-      if @api_client.config.debugging
-        @api_client.config.logger.debug "API called: BetaApi#enable_mcp_partitions_partition_id_mcp_post\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
-      end
-      return data, status_code, headers
-    end
-
     # List Authenticators
     # List all authenticators sorted by created_at in descending order. Results are paginated with a max limit of 100. When more authenticators are available, a `cursor` will be provided. Use the `cursor` parameter to retrieve the subsequent page.
     # @param [Hash] opts the optional parameters

data/lib/ragie_ruby_sdk/api/partitions_api.rb

@@ -150,132 +150,6 @@ module RagieRubySdk
       return data, status_code, headers
     end
 
-    # Disable Mcp
-    # Disables context-aware descriptions for a partition. This will stop automatically generating descriptions for the partition.
-    # @param partition_id [String] 
-    # @param [Hash] opts the optional parameters
-    # @return [Object]
-    def disable_mcp_partitions_partition_id_mcp_delete(partition_id, opts = {})
-      data, _status_code, _headers = disable_mcp_partitions_partition_id_mcp_delete_with_http_info(partition_id, opts)
-      data
-    end
-
-    # Disable Mcp
-    # Disables context-aware descriptions for a partition. This will stop automatically generating descriptions for the partition.
-    # @param partition_id [String] 
-    # @param [Hash] opts the optional parameters
-    # @return [Array<(Object, Integer, Hash)>] Object data, response status code and response headers
-    def disable_mcp_partitions_partition_id_mcp_delete_with_http_info(partition_id, opts = {})
-      if @api_client.config.debugging
-        @api_client.config.logger.debug 'Calling API: PartitionsApi.disable_mcp_partitions_partition_id_mcp_delete ...'
-      end
-      # verify the required parameter 'partition_id' is set
-      if @api_client.config.client_side_validation && partition_id.nil?
-        fail ArgumentError, "Missing the required parameter 'partition_id' when calling PartitionsApi.disable_mcp_partitions_partition_id_mcp_delete"
-      end
-      # resource path
-      local_var_path = '/partitions/{partition_id}/mcp'.sub('{' + 'partition_id' + '}', CGI.escape(partition_id.to_s))
-
-      # query parameters
-      query_params = opts[:query_params] || {}
-
-      # header parameters
-      header_params = opts[:header_params] || {}
-      # HTTP header 'Accept' (if needed)
-      header_params['Accept'] = @api_client.select_header_accept(['application/json']) unless header_params['Accept']
-
-      # form parameters
-      form_params = opts[:form_params] || {}
-
-      # http body (model)
-      post_body = opts[:debug_body]
-
-      # return_type
-      return_type = opts[:debug_return_type] || 'Object'
-
-      # auth_names
-      auth_names = opts[:debug_auth_names] || ['auth']
-
-      new_options = opts.merge(
-        :operation => :"PartitionsApi.disable_mcp_partitions_partition_id_mcp_delete",
-        :header_params => header_params,
-        :query_params => query_params,
-        :form_params => form_params,
-        :body => post_body,
-        :auth_names => auth_names,
-        :return_type => return_type
-      )
-
-      data, status_code, headers = @api_client.call_api(:DELETE, local_var_path, new_options)
-      if @api_client.config.debugging
-        @api_client.config.logger.debug "API called: PartitionsApi#disable_mcp_partitions_partition_id_mcp_delete\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
-      end
-      return data, status_code, headers
-    end
-
-    # Enable Mcp
-    # Enables context-aware descriptions for a partition. This will allow the automatically generate a desccription for based on the documents in the partition.
-    # @param partition_id [String] 
-    # @param [Hash] opts the optional parameters
-    # @return [Object]
-    def enable_mcp_partitions_partition_id_mcp_post(partition_id, opts = {})
-      data, _status_code, _headers = enable_mcp_partitions_partition_id_mcp_post_with_http_info(partition_id, opts)
-      data
-    end
-
-    # Enable Mcp
-    # Enables context-aware descriptions for a partition. This will allow the automatically generate a desccription for based on the documents in the partition.
-    # @param partition_id [String] 
-    # @param [Hash] opts the optional parameters
-    # @return [Array<(Object, Integer, Hash)>] Object data, response status code and response headers
-    def enable_mcp_partitions_partition_id_mcp_post_with_http_info(partition_id, opts = {})
-      if @api_client.config.debugging
-        @api_client.config.logger.debug 'Calling API: PartitionsApi.enable_mcp_partitions_partition_id_mcp_post ...'
-      end
-      # verify the required parameter 'partition_id' is set
-      if @api_client.config.client_side_validation && partition_id.nil?
-        fail ArgumentError, "Missing the required parameter 'partition_id' when calling PartitionsApi.enable_mcp_partitions_partition_id_mcp_post"
-      end
-      # resource path
-      local_var_path = '/partitions/{partition_id}/mcp'.sub('{' + 'partition_id' + '}', CGI.escape(partition_id.to_s))
-
-      # query parameters
-      query_params = opts[:query_params] || {}
-
-      # header parameters
-      header_params = opts[:header_params] || {}
-      # HTTP header 'Accept' (if needed)
-      header_params['Accept'] = @api_client.select_header_accept(['application/json']) unless header_params['Accept']
-
-      # form parameters
-      form_params = opts[:form_params] || {}
-
-      # http body (model)
-      post_body = opts[:debug_body]
-
-      # return_type
-      return_type = opts[:debug_return_type] || 'Object'
-
-      # auth_names
-      auth_names = opts[:debug_auth_names] || ['auth']
-
-      new_options = opts.merge(
-        :operation => :"PartitionsApi.enable_mcp_partitions_partition_id_mcp_post",
-        :header_params => header_params,
-        :query_params => query_params,
-        :form_params => form_params,
-        :body => post_body,
-        :auth_names => auth_names,
-        :return_type => return_type
-      )
-
-      data, status_code, headers = @api_client.call_api(:POST, local_var_path, new_options)
-      if @api_client.config.debugging
-        @api_client.config.logger.debug "API called: PartitionsApi#enable_mcp_partitions_partition_id_mcp_post\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
-      end
-      return data, status_code, headers
-    end
-
     # Get Partition
     # Get a partition by its ID. Includes usage information such as the number of documents and pages hosted and processed. The partition's limits are also included.
     # @param partition_id [String] 
@@ -483,5 +357,79 @@ module RagieRubySdk
       end
       return data, status_code, headers
     end
+
+    # Update Partition
+    # Updates a partition. This includes the partition's description and metadata schema.
+    # @param partition_id [String] 
+    # @param update_partition_params [UpdatePartitionParams] 
+    # @param [Hash] opts the optional parameters
+    # @return [PartitionDetail]
+    def update_partition_partitions_partition_id_patch(partition_id, update_partition_params, opts = {})
+      data, _status_code, _headers = update_partition_partitions_partition_id_patch_with_http_info(partition_id, update_partition_params, opts)
+      data
+    end
+
+    # Update Partition
+    # Updates a partition. This includes the partition&#39;s description and metadata schema.
+    # @param partition_id [String] 
+    # @param update_partition_params [UpdatePartitionParams] 
+    # @param [Hash] opts the optional parameters
+    # @return [Array<(PartitionDetail, Integer, Hash)>] PartitionDetail data, response status code and response headers
+    def update_partition_partitions_partition_id_patch_with_http_info(partition_id, update_partition_params, opts = {})
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: PartitionsApi.update_partition_partitions_partition_id_patch ...'
+      end
+      # verify the required parameter 'partition_id' is set
+      if @api_client.config.client_side_validation && partition_id.nil?
+        fail ArgumentError, "Missing the required parameter 'partition_id' when calling PartitionsApi.update_partition_partitions_partition_id_patch"
+      end
+      # verify the required parameter 'update_partition_params' is set
+      if @api_client.config.client_side_validation && update_partition_params.nil?
+        fail ArgumentError, "Missing the required parameter 'update_partition_params' when calling PartitionsApi.update_partition_partitions_partition_id_patch"
+      end
+      # resource path
+      local_var_path = '/partitions/{partition_id}'.sub('{' + 'partition_id' + '}', CGI.escape(partition_id.to_s))
+
+      # query parameters
+      query_params = opts[:query_params] || {}
+
+      # header parameters
+      header_params = opts[:header_params] || {}
+      # HTTP header 'Accept' (if needed)
+      header_params['Accept'] = @api_client.select_header_accept(['application/json']) unless header_params['Accept']
+      # HTTP header 'Content-Type'
+      content_type = @api_client.select_header_content_type(['application/json'])
+      if !content_type.nil?
+          header_params['Content-Type'] = content_type
+      end
+
+      # form parameters
+      form_params = opts[:form_params] || {}
+
+      # http body (model)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(update_partition_params)
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'PartitionDetail'
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || ['auth']
+
+      new_options = opts.merge(
+        :operation => :"PartitionsApi.update_partition_partitions_partition_id_patch",
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type
+      )
+
+      data, status_code, headers = @api_client.call_api(:PATCH, local_var_path, new_options)
+      if @api_client.config.debugging
+        @api_client.config.logger.debug "API called: PartitionsApi#update_partition_partitions_partition_id_patch\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    end
   end
 end

data/lib/ragie_ruby_sdk/models/create_partition_params.rb

@@ -17,6 +17,8 @@ module RagieRubySdk
   class CreatePartitionParams
     attr_accessor :name
 
+    attr_accessor :description
+
     attr_accessor :pages_hosted_limit_monthly
 
     attr_accessor :pages_processed_limit_monthly
@@ -47,6 +49,7 @@ module RagieRubySdk
     def self.attribute_map
       {
         :'name' => :'name',
+        :'description' => :'description',
         :'pages_hosted_limit_monthly' => :'pages_hosted_limit_monthly',
         :'pages_processed_limit_monthly' => :'pages_processed_limit_monthly',
         :'pages_hosted_limit_max' => :'pages_hosted_limit_max',
@@ -77,6 +80,7 @@ module RagieRubySdk
     def self.openapi_types
       {
         :'name' => :'String',
+        :'description' => :'String',
         :'pages_hosted_limit_monthly' => :'Integer',
         :'pages_processed_limit_monthly' => :'Integer',
         :'pages_hosted_limit_max' => :'Integer',
@@ -96,6 +100,7 @@ module RagieRubySdk
     # List of attributes with nullable: true
     def self.openapi_nullable
       Set.new([
+        :'description',
         :'pages_hosted_limit_monthly',
         :'pages_processed_limit_monthly',
         :'pages_hosted_limit_max',
@@ -134,6 +139,10 @@ module RagieRubySdk
         self.name = nil
       end
 
+      if attributes.key?(:'description')
+        self.description = attributes[:'description']
+      end
+
       if attributes.key?(:'pages_hosted_limit_monthly')
         self.pages_hosted_limit_monthly = attributes[:'pages_hosted_limit_monthly']
       end
@@ -405,6 +414,7 @@ module RagieRubySdk
       return true if self.equal?(o)
       self.class == o.class &&
           name == o.name &&
+          description == o.description &&
           pages_hosted_limit_monthly == o.pages_hosted_limit_monthly &&
           pages_processed_limit_monthly == o.pages_processed_limit_monthly &&
           pages_hosted_limit_max == o.pages_hosted_limit_max &&
@@ -429,7 +439,7 @@ module RagieRubySdk
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [name, pages_hosted_limit_monthly, pages_processed_limit_monthly, pages_hosted_limit_max, pages_processed_limit_max, audio_processed_limit_monthly, audio_processed_limit_max, video_processed_limit_monthly, video_processed_limit_max, media_streamed_limit_monthly, media_streamed_limit_max, media_hosted_limit_monthly, media_hosted_limit_max, metadata_schema].hash
+      [name, description, pages_hosted_limit_monthly, pages_processed_limit_monthly, pages_hosted_limit_max, pages_processed_limit_max, audio_processed_limit_monthly, audio_processed_limit_max, video_processed_limit_monthly, video_processed_limit_max, media_streamed_limit_monthly, media_streamed_limit_max, media_hosted_limit_monthly, media_hosted_limit_max, metadata_schema].hash
     end
 
     # Builds the object from hash

data/lib/ragie_ruby_sdk/models/partition.rb

@@ -21,6 +21,12 @@ module RagieRubySdk
 
     attr_accessor :limit_exceeded_at
 
+    attr_accessor :description
+
+    attr_accessor :context_aware
+
+    attr_accessor :metadata_schema
+
     attr_accessor :limits
 
     # Attribute mapping from ruby-style variable name to JSON key.
@@ -29,6 +35,9 @@ module RagieRubySdk
         :'name' => :'name',
         :'is_default' => :'is_default',
         :'limit_exceeded_at' => :'limit_exceeded_at',
+        :'description' => :'description',
+        :'context_aware' => :'context_aware',
+        :'metadata_schema' => :'metadata_schema',
         :'limits' => :'limits'
       }
     end
@@ -49,6 +58,9 @@ module RagieRubySdk
         :'name' => :'String',
         :'is_default' => :'Boolean',
         :'limit_exceeded_at' => :'Time',
+        :'description' => :'String',
+        :'context_aware' => :'Boolean',
+        :'metadata_schema' => :'Hash<String, CreatePartitionParamsMetadataSchemaValue>',
         :'limits' => :'PartitionLimits'
       }
     end
@@ -57,6 +69,8 @@ module RagieRubySdk
     def self.openapi_nullable
       Set.new([
         :'limit_exceeded_at',
+        :'description',
+        :'metadata_schema',
       ])
     end
 
@@ -92,6 +106,26 @@ module RagieRubySdk
         self.limit_exceeded_at = attributes[:'limit_exceeded_at']
       end
 
+      if attributes.key?(:'description')
+        self.description = attributes[:'description']
+      else
+        self.description = nil
+      end
+
+      if attributes.key?(:'context_aware')
+        self.context_aware = attributes[:'context_aware']
+      else
+        self.context_aware = nil
+      end
+
+      if attributes.key?(:'metadata_schema')
+        if (value = attributes[:'metadata_schema']).is_a?(Hash)
+          self.metadata_schema = value
+        end
+      else
+        self.metadata_schema = nil
+      end
+
       if attributes.key?(:'limits')
         self.limits = attributes[:'limits']
       else
@@ -112,6 +146,10 @@ module RagieRubySdk
         invalid_properties.push('invalid value for "is_default", is_default cannot be nil.')
       end
 
+      if @context_aware.nil?
+        invalid_properties.push('invalid value for "context_aware", context_aware cannot be nil.')
+      end
+
       if @limits.nil?
         invalid_properties.push('invalid value for "limits", limits cannot be nil.')
       end
@@ -125,6 +163,7 @@ module RagieRubySdk
       warn '[DEPRECATED] the `valid?` method is obsolete'
       return false if @name.nil?
       return false if @is_default.nil?
+      return false if @context_aware.nil?
       return false if @limits.nil?
       true
     end
@@ -149,6 +188,16 @@ module RagieRubySdk
       @is_default = is_default
     end
 
+    # Custom attribute writer method with validation
+    # @param [Object] context_aware Value to be assigned
+    def context_aware=(context_aware)
+      if context_aware.nil?
+        fail ArgumentError, 'context_aware cannot be nil'
+      end
+
+      @context_aware = context_aware
+    end
+
     # Custom attribute writer method with validation
     # @param [Object] limits Value to be assigned
     def limits=(limits)
@@ -167,6 +216,9 @@ module RagieRubySdk
           name == o.name &&
           is_default == o.is_default &&
           limit_exceeded_at == o.limit_exceeded_at &&
+          description == o.description &&
+          context_aware == o.context_aware &&
+          metadata_schema == o.metadata_schema &&
           limits == o.limits
     end
 
@@ -179,7 +231,7 @@ module RagieRubySdk
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [name, is_default, limit_exceeded_at, limits].hash
+      [name, is_default, limit_exceeded_at, description, context_aware, metadata_schema, limits].hash
     end
 
     # Builds the object from hash

data/lib/ragie_ruby_sdk/models/partition_detail.rb

@@ -21,6 +21,12 @@ module RagieRubySdk
 
     attr_accessor :limit_exceeded_at
 
+    attr_accessor :description
+
+    attr_accessor :context_aware
+
+    attr_accessor :metadata_schema
+
     attr_accessor :limits
 
     attr_accessor :stats
@@ -31,6 +37,9 @@ module RagieRubySdk
         :'name' => :'name',
         :'is_default' => :'is_default',
         :'limit_exceeded_at' => :'limit_exceeded_at',
+        :'description' => :'description',
+        :'context_aware' => :'context_aware',
+        :'metadata_schema' => :'metadata_schema',
         :'limits' => :'limits',
         :'stats' => :'stats'
       }
@@ -52,6 +61,9 @@ module RagieRubySdk
         :'name' => :'String',
         :'is_default' => :'Boolean',
         :'limit_exceeded_at' => :'Time',
+        :'description' => :'String',
+        :'context_aware' => :'Boolean',
+        :'metadata_schema' => :'Hash<String, CreatePartitionParamsMetadataSchemaValue>',
         :'limits' => :'PartitionLimits',
         :'stats' => :'PartitionStats'
       }
@@ -61,6 +73,8 @@ module RagieRubySdk
     def self.openapi_nullable
       Set.new([
         :'limit_exceeded_at',
+        :'description',
+        :'metadata_schema',
       ])
     end
 
@@ -96,6 +110,26 @@ module RagieRubySdk
         self.limit_exceeded_at = attributes[:'limit_exceeded_at']
       end
 
+      if attributes.key?(:'description')
+        self.description = attributes[:'description']
+      else
+        self.description = nil
+      end
+
+      if attributes.key?(:'context_aware')
+        self.context_aware = attributes[:'context_aware']
+      else
+        self.context_aware = nil
+      end
+
+      if attributes.key?(:'metadata_schema')
+        if (value = attributes[:'metadata_schema']).is_a?(Hash)
+          self.metadata_schema = value
+        end
+      else
+        self.metadata_schema = nil
+      end
+
       if attributes.key?(:'limits')
         self.limits = attributes[:'limits']
       else
@@ -122,6 +156,10 @@ module RagieRubySdk
         invalid_properties.push('invalid value for "is_default", is_default cannot be nil.')
       end
 
+      if @context_aware.nil?
+        invalid_properties.push('invalid value for "context_aware", context_aware cannot be nil.')
+      end
+
       if @limits.nil?
         invalid_properties.push('invalid value for "limits", limits cannot be nil.')
       end
@@ -139,6 +177,7 @@ module RagieRubySdk
       warn '[DEPRECATED] the `valid?` method is obsolete'
       return false if @name.nil?
       return false if @is_default.nil?
+      return false if @context_aware.nil?
       return false if @limits.nil?
       return false if @stats.nil?
       true
@@ -164,6 +203,16 @@ module RagieRubySdk
       @is_default = is_default
     end
 
+    # Custom attribute writer method with validation
+    # @param [Object] context_aware Value to be assigned
+    def context_aware=(context_aware)
+      if context_aware.nil?
+        fail ArgumentError, 'context_aware cannot be nil'
+      end
+
+      @context_aware = context_aware
+    end
+
     # Custom attribute writer method with validation
     # @param [Object] limits Value to be assigned
     def limits=(limits)
@@ -192,6 +241,9 @@ module RagieRubySdk
           name == o.name &&
           is_default == o.is_default &&
           limit_exceeded_at == o.limit_exceeded_at &&
+          description == o.description &&
+          context_aware == o.context_aware &&
+          metadata_schema == o.metadata_schema &&
           limits == o.limits &&
           stats == o.stats
     end
@@ -205,7 +257,7 @@ module RagieRubySdk
     # Calculates hash code according to all attributes.
     # @return [Integer] Hash code
     def hash
-      [name, is_default, limit_exceeded_at, limits, stats].hash
+      [name, is_default, limit_exceeded_at, description, context_aware, metadata_schema, limits, stats].hash
     end
 
     # Builds the object from hash