schematichq 1.4.16 → 1.5.0

data/lib/schematic/credits/types/count_credit_spend_policies_request.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class CountCreditSpendPoliciesRequest < Internal::Types::Model
+        field :billing_credit_id, -> { String }, optional: true, nullable: false
+
+        field :company_id, -> { String }, optional: true, nullable: false
+
+        field :scope_type, -> { Schematic::Types::CreditSpendPolicyScope }, optional: true, nullable: false
+
+        field :user_id, -> { String }, optional: true, nullable: false
+
+        field :user_ids, -> { String }, optional: true, nullable: false
+
+        field :limit, -> { Integer }, optional: true, nullable: false
+
+        field :offset, -> { Integer }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/count_credit_spend_policies_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class CountCreditSpendPoliciesResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::CountResponse }, optional: false, nullable: false
+
+        field :params, -> { Schematic::Credits::Types::CountCreditSpendPoliciesParams }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/create_company_credit_grant.rb

@@ -8,6 +8,8 @@ module Schematic
 
         field :company_id, -> { String }, optional: false, nullable: false
 
+        field :credit_bundle_id, -> { String }, optional: true, nullable: false
+
         field :credit_id, -> { String }, optional: false, nullable: false
 
         field :currency, -> { String }, optional: true, nullable: false

data/lib/schematic/credits/types/create_credit_spend_policy_request_body.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class CreateCreditSpendPolicyRequestBody < Internal::Types::Model
+        field :billing_credit_id, -> { String }, optional: false, nullable: false
+
+        field :company_id, -> { String }, optional: true, nullable: false
+
+        field :label, -> { String }, optional: true, nullable: false
+
+        field :max_per_draw, -> { Integer }, optional: false, nullable: false
+
+        field :user_id, -> { String }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/create_credit_spend_policy_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class CreateCreditSpendPolicyResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::CreditSpendPolicyResponseData }, optional: false, nullable: false
+
+        field :params, -> { Internal::Types::Hash[String, Object] }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/delete_credit_spend_policy_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class DeleteCreditSpendPolicyResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::DeleteResponse }, optional: false, nullable: false
+
+        field :params, -> { Internal::Types::Hash[String, Object] }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/get_credit_spend_policy_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class GetCreditSpendPolicyResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::CreditSpendPolicyResponseData }, optional: false, nullable: false
+
+        field :params, -> { Internal::Types::Hash[String, Object] }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/list_credit_spend_policies_params.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      # Input parameters
+      class ListCreditSpendPoliciesParams < Internal::Types::Model
+        field :billing_credit_id, -> { String }, optional: true, nullable: false
+
+        field :company_id, -> { String }, optional: true, nullable: false
+
+        field :limit, -> { Integer }, optional: true, nullable: false
+
+        field :offset, -> { Integer }, optional: true, nullable: false
+
+        field :scope_type, -> { Schematic::Types::CreditSpendPolicyScope }, optional: true, nullable: false
+
+        field :user_id, -> { String }, optional: true, nullable: false
+
+        field :user_ids, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/list_credit_spend_policies_request.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class ListCreditSpendPoliciesRequest < Internal::Types::Model
+        field :billing_credit_id, -> { String }, optional: true, nullable: false
+
+        field :company_id, -> { String }, optional: true, nullable: false
+
+        field :scope_type, -> { Schematic::Types::CreditSpendPolicyScope }, optional: true, nullable: false
+
+        field :user_id, -> { String }, optional: true, nullable: false
+
+        field :user_ids, -> { String }, optional: true, nullable: false
+
+        field :limit, -> { Integer }, optional: true, nullable: false
+
+        field :offset, -> { Integer }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/list_credit_spend_policies_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class ListCreditSpendPoliciesResponse < Internal::Types::Model
+        field :data, -> { Internal::Types::Array[Schematic::Types::CreditSpendPolicyResponseData] }, optional: false, nullable: false
+
+        field :params, -> { Schematic::Credits::Types::ListCreditSpendPoliciesParams }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/release_credit_reservation_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class ReleaseCreditReservationResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::CreditReservationResponseData }, optional: false, nullable: false
+
+        field :params, -> { Internal::Types::Hash[String, Object] }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/reserve_credits_request_body.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class ReserveCreditsRequestBody < Internal::Types::Model
+        field :amount, -> { Integer }, optional: false, nullable: false
+
+        field :company_id, -> { String }, optional: false, nullable: false
+
+        field :credit_type_id, -> { String }, optional: false, nullable: false
+
+        field :expires_at, -> { String }, optional: true, nullable: false
+
+        field :idempotency_key, -> { String }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/reserve_credits_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class ReserveCreditsResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::CreditReservationResponseData }, optional: false, nullable: false
+
+        field :params, -> { Internal::Types::Hash[String, Object] }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/update_credit_spend_policy_request_body.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class UpdateCreditSpendPolicyRequestBody < Internal::Types::Model
+        field :spend_policy_id, -> { String }, optional: false, nullable: false
+
+        field :label, -> { String }, optional: true, nullable: false
+
+        field :max_per_draw, -> { Integer }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/credits/types/update_credit_spend_policy_response.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Schematic
+  module Credits
+    module Types
+      class UpdateCreditSpendPolicyResponse < Internal::Types::Model
+        field :data, -> { Schematic::Types::CreditSpendPolicyResponseData }, optional: false, nullable: false
+
+        field :params, -> { Internal::Types::Hash[String, Object] }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/schematic/dataexports/client.rb

@@ -53,7 +53,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Dataexports::Types::ListDataExportsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Dataexports::Types::ListDataExportsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -91,7 +91,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Dataexports::Types::CreateDataExportResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Dataexports::Types::CreateDataExportResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -126,7 +126,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Dataexports::Types::GetDataExportResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Dataexports::Types::GetDataExportResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)

data/lib/schematic/datastream/client.rb

@@ -30,7 +30,8 @@ module Schematic
         redis_key_prefix: "schematic:",
         replicator_mode: false,
         replicator_health_url: REPLICATOR_HEALTH_URL,
-        replicator_health_interval: REPLICATOR_HEALTH_INTERVAL
+        replicator_health_interval: REPLICATOR_HEALTH_INTERVAL,
+        max_frame_size: nil
       )
         @api_key = api_key
         @base_url = base_url
@@ -40,6 +41,7 @@ module Schematic
         @replicator_mode = replicator_mode
         @replicator_health_url = replicator_health_url
         @replicator_health_interval = replicator_health_interval
+        @max_frame_size = max_frame_size
         @replicator_ready = false
         @cache_version = ""
         @mutex = Mutex.new
@@ -120,7 +122,8 @@ module Schematic
             api_key: @api_key,
             logger: @logger,
             message_handler: method(:handle_message),
-            ready_handler: method(:handle_ready)
+            ready_handler: method(:handle_ready),
+            max_frame_size: @max_frame_size
           )
           @ws_client.start
         end

data/lib/schematic/datastream/resource_cache.rb

@@ -3,8 +3,6 @@
 module Schematic
   module DataStream
     class ResourceCache
-      CACHE_KEY_PREFIX = "schematic"
-
       def initialize(primary_cache:, lookup_cache:, key_prefix:, get_id:, get_keys:, cache_version: nil)
         @primary_cache = primary_cache
         @lookup_cache = lookup_cache
@@ -65,11 +63,11 @@ module Schematic
       private
 
       def build_id_key(id)
-        "#{CACHE_KEY_PREFIX}:#{@key_prefix}:#{@cache_version}:#{id}"
+        "#{@key_prefix}:#{@cache_version}:#{id}"
       end
 
       def build_lookup_key(key_name, key_value)
-        "#{CACHE_KEY_PREFIX}:#{@key_prefix}:#{@cache_version}:#{key_name.to_s.downcase}:#{key_value}"
+        "#{@key_prefix}:#{@cache_version}:#{key_name.to_s.downcase}:#{key_value.to_s.downcase}"
       end
     end
   end

data/lib/schematic/datastream/websocket_client.rb

@@ -37,15 +37,20 @@ module Schematic
     DATASTREAM_MODE_DIRECT = "direct"
     UNKNOWN_VERSION = "unknown"
 
+    # Symbol the websocket gem records on Frame::Incoming#error when an incoming
+    # frame declares a payload larger than WebSocket.max_frame_size.
+    FRAME_TOO_LONG = :frame_too_long
+
     class WebSocketClient
       attr_reader :url, :connected, :ready
 
-      def initialize(base_url:, api_key:, logger:, message_handler:, ready_handler: nil)
+      def initialize(base_url:, api_key:, logger:, message_handler:, ready_handler: nil, max_frame_size: nil)
         @base_url = base_url
         @api_key = api_key
         @logger = logger
         @message_handler = message_handler
         @ready_handler = ready_handler
+        @max_frame_size = max_frame_size
         @connected = false
         @ready = false
         @should_reconnect = true
@@ -60,6 +65,7 @@ module Schematic
         @stopped = false
         @url = build_ws_url(base_url)
         @last_pong = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+        apply_max_frame_size
       end
 
       def start
@@ -211,6 +217,22 @@ module Schematic
                   break
                 end
               end
+
+              # WebSocket::Frame::Incoming#next swallows protocol errors unless
+              # WebSocket.should_raise is set: it returns nil and records the
+              # reason on #error, which is indistinguishable from "no complete
+              # frame buffered yet". The error is also sticky — the offending
+              # header is never consumed — so without this check an oversized
+              # frame stalls the stream silently while the socket stays open.
+              if (parser_error = @frame_parser.error)
+                log_frame_parser_error(parser_error)
+                break
+              end
+            rescue WebSocket::Error::Frame::TooLong
+              # Reached only when the host process has set
+              # WebSocket.should_raise = true.
+              log_frame_parser_error(FRAME_TOO_LONG)
+              break
             rescue IOError, Errno::ECONNRESET, OpenSSL::SSL::SSLError => e
               @logger.warn("DataStream read error: #{e.message}")
               break
@@ -321,6 +343,42 @@ module Schematic
         @write_mutex.synchronize { @socket&.write(frame.to_s) }
       end
 
+      # The maximum incoming frame size the websocket gem will accept. The gem
+      # only exposes this as WebSocket.max_frame_size, a module-level global read
+      # at decode time; Frame::Incoming takes no per-instance limit (passing
+      # max_frame_size: to its constructor is silently ignored). A per-client
+      # option therefore cannot be fully isolated from the rest of the process.
+      #
+      # The compromise: do nothing unless the caller explicitly asks for a limit,
+      # and when they do, only ever raise the global — never lower it — so we
+      # cannot shrink the ceiling out from under another websocket user in the
+      # same process. The change is logged so it is visible rather than silent.
+      def apply_max_frame_size
+        return if @max_frame_size.nil?
+
+        raise ArgumentError, "max_frame_size must be a positive Integer, got #{@max_frame_size.inspect}" unless @max_frame_size.is_a?(Integer) && @max_frame_size.positive?
+
+        current = ::WebSocket.max_frame_size
+        if @max_frame_size <= current
+          @logger.debug("DataStream max_frame_size #{@max_frame_size} is at or below the current WebSocket.max_frame_size (#{current}); leaving it unchanged")
+          return
+        end
+
+        ::WebSocket.max_frame_size = @max_frame_size
+        @logger.info("Raised WebSocket.max_frame_size from #{current} to #{@max_frame_size} bytes; the websocket gem stores this limit globally, so it applies to every websocket in this process")
+      end
+
+      def log_frame_parser_error(parser_error)
+        if parser_error == FRAME_TOO_LONG
+          @logger.error(
+            "DataStream received a WebSocket frame larger than the #{::WebSocket.max_frame_size} byte limit and cannot decode it. " \
+            "Raise the limit with datastream_options: { max_frame_size: <bytes> } (note: the websocket gem applies this limit process-wide)."
+          )
+        else
+          @logger.error("DataStream frame decode error: #{parser_error}")
+        end
+      end
+
       # Headers attached to the WebSocket handshake. The mode/client headers let
       # the backend tell direct-SDK connections apart from the
       # schematic-datastream-replicator and correlate them to a release.

data/lib/schematic/entitlements/client.rb

@@ -68,7 +68,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::ListCompanyOverridesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::ListCompanyOverridesResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -107,7 +107,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::CreateCompanyOverrideResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::CreateCompanyOverrideResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -142,7 +142,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::GetCompanyOverrideResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::GetCompanyOverrideResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -185,7 +185,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::UpdateCompanyOverrideResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::UpdateCompanyOverrideResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -220,7 +220,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::DeleteCompanyOverrideResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::DeleteCompanyOverrideResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -285,7 +285,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::CountCompanyOverridesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::CountCompanyOverridesResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -335,7 +335,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::ListFeatureCompaniesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::ListFeatureCompaniesResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -385,7 +385,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::CountFeatureCompaniesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::CountFeatureCompaniesResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -449,7 +449,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::ListFeatureUsageResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::ListFeatureUsageResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -508,7 +508,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::ListFeatureUsageHistoryResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::ListFeatureUsageHistoryResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -561,7 +561,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::GetFeatureUsageTimeSeriesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::GetFeatureUsageTimeSeriesResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -625,7 +625,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::CountFeatureUsageResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::CountFeatureUsageResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -675,7 +675,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::ListFeatureUsersResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::ListFeatureUsersResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -725,7 +725,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::CountFeatureUsersResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::CountFeatureUsersResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -796,7 +796,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::ListPlanEntitlementsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::ListPlanEntitlementsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -835,7 +835,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::CreatePlanEntitlementResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::CreatePlanEntitlementResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -870,7 +870,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::GetPlanEntitlementResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::GetPlanEntitlementResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -913,7 +913,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::UpdatePlanEntitlementResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::UpdatePlanEntitlementResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -948,7 +948,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::DeletePlanEntitlementResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::DeletePlanEntitlementResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -989,7 +989,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::UpsertPlanEntitlementForBillingProductResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::UpsertPlanEntitlementForBillingProductResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1060,7 +1060,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::CountPlanEntitlementsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::CountPlanEntitlementsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1098,7 +1098,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::DuplicatePlanEntitlementsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::DuplicatePlanEntitlementsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1139,7 +1139,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::GetFeatureUsageByCompanyResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::GetFeatureUsageByCompanyResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1189,7 +1189,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::GetUserUsageByCompanyResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::GetUserUsageByCompanyResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1239,7 +1239,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Entitlements::Types::GetUserUsageDetailResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Entitlements::Types::GetUserUsageDetailResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)