schematichq 1.4.17 → 1.5.0

data/lib/schematic/credits/client.rb

@@ -53,7 +53,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ListBillingCreditsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ListBillingCreditsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -92,7 +92,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CreateBillingCreditResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CreateBillingCreditResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -127,7 +127,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::GetSingleBillingCreditResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::GetSingleBillingCreditResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -171,7 +171,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::UpdateBillingCreditResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::UpdateBillingCreditResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -206,7 +206,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::SoftDeleteBillingCreditResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::SoftDeleteBillingCreditResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -245,7 +245,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ListCompanyCreditBalancesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ListCompanyCreditBalancesResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -301,7 +301,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ListCreditBundlesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ListCreditBundlesResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -341,7 +341,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CreateCreditBundleResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CreateCreditBundleResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -376,7 +376,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::GetCreditBundleResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::GetCreditBundleResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -420,7 +420,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::UpdateCreditBundleDetailsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::UpdateCreditBundleDetailsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -455,7 +455,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::DeleteCreditBundleResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::DeleteCreditBundleResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -511,7 +511,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CountCreditBundlesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CountCreditBundlesResponse.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::Credits::Types::CountBillingCreditsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CountBillingCreditsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -601,7 +601,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ZeroOutGrantResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ZeroOutGrantResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -641,7 +641,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::GrantBillingCreditsToCompanyResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::GrantBillingCreditsToCompanyResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -694,7 +694,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CountCompanyGrantsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CountCompanyGrantsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -747,7 +747,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ListCompanyGrantsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ListCompanyGrantsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -797,7 +797,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CountBillingCreditsGrantsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CountBillingCreditsGrantsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -847,7 +847,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ListGrantsForCreditResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ListGrantsForCreditResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -886,7 +886,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::AcquireCreditLeaseResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::AcquireCreditLeaseResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -929,7 +929,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ExtendCreditLeaseResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ExtendCreditLeaseResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -937,7 +937,7 @@ module Schematic
       end
 
       # @param request_options [Hash]
-      # @param params [Schematic::Types::ReleaseCreditLeaseRequestBody]
+      # @param params [Hash]
       # @option request_options [String] :base_url
       # @option request_options [Hash{String => Object}] :additional_headers
       # @option request_options [Hash{String => Object}] :additional_query_parameters
@@ -946,24 +946,15 @@ module Schematic
       # @option params [String] :lease_id
       #
       # @example
-      #   client.credits.release_credit_lease(
-      #     lease_id: "lease_id",
-      #     request: {
-      #       key: "value"
-      #     }
-      #   )
+      #   client.credits.release_credit_lease(lease_id: "lease_id")
       #
       # @return [Schematic::Credits::Types::ReleaseCreditLeaseResponse]
       def release_credit_lease(request_options: {}, **params)
         params = Schematic::Internal::Types::Utils.normalize_keys(params)
-        path_param_names = %i[lease_id]
-        body_params = params.except(*path_param_names)
-
         request = Schematic::Internal::JSON::Request.new(
           base_url: request_options[:base_url],
           method: "PUT",
           path: "billing/credits/lease/#{URI.encode_uri_component(params[:lease_id].to_s)}/release",
-          body: body_params,
           request_options: request_options
         )
         begin
@@ -973,7 +964,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ReleaseCreditLeaseResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ReleaseCreditLeaseResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1035,7 +1026,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ListBillingPlanCreditGrantsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ListBillingPlanCreditGrantsResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1076,7 +1067,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CreateBillingPlanCreditGrantResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CreateBillingPlanCreditGrantResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1111,7 +1102,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::GetSingleBillingPlanCreditGrantResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::GetSingleBillingPlanCreditGrantResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1154,7 +1145,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::UpdateBillingPlanCreditGrantResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::UpdateBillingPlanCreditGrantResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1197,7 +1188,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::DeleteBillingPlanCreditGrantResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::DeleteBillingPlanCreditGrantResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1259,7 +1250,81 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CountBillingPlanCreditGrantsResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CountBillingPlanCreditGrantsResponse.load(response.body))
+        else
+          error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
+          raise error_class.new(response.body, code: code)
+        end
+      end
+
+      # @param request_options [Hash]
+      # @param params [Schematic::Credits::Types::ReserveCreditsRequestBody]
+      # @option request_options [String] :base_url
+      # @option request_options [Hash{String => Object}] :additional_headers
+      # @option request_options [Hash{String => Object}] :additional_query_parameters
+      # @option request_options [Hash{String => Object}] :additional_body_parameters
+      # @option request_options [Integer] :timeout_in_seconds
+      #
+      # @example
+      #   client.credits.reserve_credits(
+      #     amount: 1.1,
+      #     company_id: "company_id",
+      #     credit_type_id: "credit_type_id"
+      #   )
+      #
+      # @return [Schematic::Credits::Types::ReserveCreditsResponse]
+      def reserve_credits(request_options: {}, **params)
+        params = Schematic::Internal::Types::Utils.normalize_keys(params)
+        request = Schematic::Internal::JSON::Request.new(
+          base_url: request_options[:base_url],
+          method: "POST",
+          path: "billing/credits/reservations",
+          body: Schematic::Credits::Types::ReserveCreditsRequestBody.new(params).to_h,
+          request_options: request_options
+        )
+        begin
+          response = @client.send(request)
+        rescue Net::HTTPRequestTimeout
+          raise Schematic::Errors::TimeoutError
+        end
+        code = response.code.to_i
+        if code.between?(200, 299)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ReserveCreditsResponse.load(response.body))
+        else
+          error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
+          raise error_class.new(response.body, code: code)
+        end
+      end
+
+      # @param request_options [Hash]
+      # @param params [Hash]
+      # @option request_options [String] :base_url
+      # @option request_options [Hash{String => Object}] :additional_headers
+      # @option request_options [Hash{String => Object}] :additional_query_parameters
+      # @option request_options [Hash{String => Object}] :additional_body_parameters
+      # @option request_options [Integer] :timeout_in_seconds
+      # @option params [String] :reservation_id
+      #
+      # @example
+      #   client.credits.release_credit_reservation(reservation_id: "reservation_id")
+      #
+      # @return [Schematic::Credits::Types::ReleaseCreditReservationResponse]
+      def release_credit_reservation(request_options: {}, **params)
+        params = Schematic::Internal::Types::Utils.normalize_keys(params)
+        request = Schematic::Internal::JSON::Request.new(
+          base_url: request_options[:base_url],
+          method: "PUT",
+          path: "billing/credits/reservations/#{URI.encode_uri_component(params[:reservation_id].to_s)}/release",
+          request_options: request_options
+        )
+        begin
+          response = @client.send(request)
+        rescue Net::HTTPRequestTimeout
+          raise Schematic::Errors::TimeoutError
+        end
+        code = response.code.to_i
+        if code.between?(200, 299)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ReleaseCreditReservationResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1318,7 +1383,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ListCreditSpendPoliciesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ListCreditSpendPoliciesResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1356,7 +1421,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CreateCreditSpendPolicyResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CreateCreditSpendPolicyResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1391,7 +1456,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::GetCreditSpendPolicyResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::GetCreditSpendPolicyResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1431,7 +1496,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::UpdateCreditSpendPolicyResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::UpdateCreditSpendPolicyResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1466,7 +1531,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::DeleteCreditSpendPolicyResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::DeleteCreditSpendPolicyResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1525,7 +1590,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CountCreditSpendPoliciesResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CountCreditSpendPoliciesResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1587,7 +1652,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::ListCreditEventLedgerResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::ListCreditEventLedgerResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)
@@ -1649,7 +1714,7 @@ module Schematic
         end
         code = response.code.to_i
         if code.between?(200, 299)
-          Schematic::Credits::Types::CountCreditEventLedgerResponse.load(response.body)
+          (response.body.to_s.empty? ? nil : Schematic::Credits::Types::CountCreditEventLedgerResponse.load(response.body))
         else
           error_class = Schematic::Errors::ResponseError.subclass_for_code(code)
           raise error_class.new(response.body, code: code)

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/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/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/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.