aws-sdk-sagemakerruntimehttp2 1.0.0

data/lib/aws-sdk-sagemakerruntimehttp2/plugins/endpoints.rb

@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+
+module Aws::SageMakerRuntimeHTTP2
+  module Plugins
+    class Endpoints < Seahorse::Client::Plugin
+      option(
+        :endpoint_provider,
+        doc_type: 'Aws::SageMakerRuntimeHTTP2::EndpointProvider',
+        rbs_type: 'untyped',
+        docstring: <<~DOCS) do |_cfg|
+The endpoint provider used to resolve endpoints. Any object that responds to
+`#resolve_endpoint(parameters)` where `parameters` is a Struct similar to
+`Aws::SageMakerRuntimeHTTP2::EndpointParameters`.
+        DOCS
+        Aws::SageMakerRuntimeHTTP2::EndpointProvider.new
+      end
+
+      # @api private
+      class Handler < Seahorse::Client::Handler
+        def call(context)
+          unless context[:discovered_endpoint]
+            params = Aws::SageMakerRuntimeHTTP2::Endpoints.parameters_for_operation(context)
+            endpoint = context.config.endpoint_provider.resolve_endpoint(params)
+
+            context.http_request.endpoint = endpoint.url
+            apply_endpoint_headers(context, endpoint.headers)
+
+            context[:endpoint_params] = params
+            context[:endpoint_properties] = endpoint.properties
+          end
+
+          context[:auth_scheme] =
+            Aws::Endpoints.resolve_auth_scheme(context, endpoint)
+
+          with_metrics(context) { @handler.call(context) }
+        end
+
+        private
+
+        def with_metrics(context, &block)
+          metrics = []
+          metrics << 'ENDPOINT_OVERRIDE' unless context.config.regional_endpoint
+          if context[:auth_scheme] && context[:auth_scheme]['name'] == 'sigv4a'
+            metrics << 'SIGV4A_SIGNING'
+          end
+          if context.config.credentials&.credentials&.account_id
+            metrics << 'RESOLVED_ACCOUNT_ID'
+          end
+          Aws::Plugins::UserAgent.metric(*metrics, &block)
+        end
+
+        def apply_endpoint_headers(context, headers)
+          headers.each do |key, values|
+            value = values
+              .compact
+              .map { |s| Seahorse::Util.escape_header_list_string(s.to_s) }
+              .join(',')
+
+            context.http_request.headers[key] = value
+          end
+        end
+      end
+
+      def add_handlers(handlers, _config)
+        handlers.add(Handler, step: :build, priority: 75)
+      end
+    end
+  end
+end

data/lib/aws-sdk-sagemakerruntimehttp2/types.rb

@@ -0,0 +1,292 @@
+# frozen_string_literal: true
+
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws::SageMakerRuntimeHTTP2
+  module Types
+
+    # The input fails to satisfy the constraints specified by an AWS
+    # service.
+    #
+    # @!attribute [rw] message
+    #   Error message.
+    #   @return [String]
+    #
+    # @!attribute [rw] error_code
+    #   Error code.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/InputValidationError AWS API Documentation
+    #
+    class InputValidationError < Struct.new(
+      :message,
+      :error_code)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The request processing has failed because of an unknown error,
+    # exception or failure.
+    #
+    # @!attribute [rw] message
+    #   Error message.
+    #   @return [String]
+    #
+    # @!attribute [rw] error_code
+    #   Error code.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/InternalServerError AWS API Documentation
+    #
+    class InternalServerError < Struct.new(
+      :message,
+      :error_code)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Internal stream failure that occurs during streaming.
+    #
+    # @!attribute [rw] message
+    #   Error message.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/InternalStreamFailure AWS API Documentation
+    #
+    class InternalStreamFailure < Struct.new(
+      :message,
+      :event_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] endpoint_name
+    #   The name of the endpoint to invoke.
+    #   @return [String]
+    #
+    # @!attribute [rw] body
+    #   The request payload stream.
+    #   @return [Types::RequestStreamEvent]
+    #
+    # @!attribute [rw] target_variant
+    #   Target variant for the request.
+    #   @return [String]
+    #
+    # @!attribute [rw] model_invocation_path
+    #   Model invocation path.
+    #   @return [String]
+    #
+    # @!attribute [rw] model_query_string
+    #   Model query string.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/InvokeEndpointWithBidirectionalStreamInput AWS API Documentation
+    #
+    class InvokeEndpointWithBidirectionalStreamInput < Struct.new(
+      :endpoint_name,
+      :body,
+      :target_variant,
+      :model_invocation_path,
+      :model_query_string)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # @!attribute [rw] body
+    #   The response payload stream.
+    #   @return [Types::ResponseStreamEvent]
+    #
+    # @!attribute [rw] invoked_production_variant
+    #   The invoked production variant.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/InvokeEndpointWithBidirectionalStreamOutput AWS API Documentation
+    #
+    class InvokeEndpointWithBidirectionalStreamOutput < Struct.new(
+      :body,
+      :invoked_production_variant)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # An error occurred while processing the model.
+    #
+    # @!attribute [rw] message
+    #   Error message.
+    #   @return [String]
+    #
+    # @!attribute [rw] original_status_code
+    #   HTTP status code returned by model.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] original_message
+    #   Original error message from the model.
+    #   @return [String]
+    #
+    # @!attribute [rw] log_stream_arn
+    #   CloudWatch log stream ARN.
+    #   @return [String]
+    #
+    # @!attribute [rw] error_code
+    #   Error code.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/ModelError AWS API Documentation
+    #
+    class ModelError < Struct.new(
+      :message,
+      :original_status_code,
+      :original_message,
+      :log_stream_arn,
+      :error_code)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Model stream error that occurs during streaming.
+    #
+    # @!attribute [rw] message
+    #   Error message.
+    #   @return [String]
+    #
+    # @!attribute [rw] error_code
+    #   Error code.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/ModelStreamError AWS API Documentation
+    #
+    class ModelStreamError < Struct.new(
+      :message,
+      :error_code,
+      :event_type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Request payload part structure.
+    #
+    # @!attribute [rw] bytes
+    #   The payload bytes.
+    #   @return [String]
+    #
+    # @!attribute [rw] data_type
+    #   Data type header. Can be one of these possible values: "UTF8",
+    #   "BINARY".
+    #   @return [String]
+    #
+    # @!attribute [rw] completion_state
+    #   Completion state header. Can be one of these possible values:
+    #   "PARTIAL", "COMPLETE".
+    #   @return [String]
+    #
+    # @!attribute [rw] p
+    #   Padding string for alignment.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/RequestPayloadPart AWS API Documentation
+    #
+    class RequestPayloadPart < Struct.new(
+      :bytes,
+      :data_type,
+      :completion_state,
+      :p,
+      :event_type)
+      SENSITIVE = [:bytes]
+      include Aws::Structure
+    end
+
+    # Response payload part structure.
+    #
+    # @!attribute [rw] bytes
+    #   The payload bytes.
+    #   @return [String]
+    #
+    # @!attribute [rw] data_type
+    #   Data type header. Can be one of these possible values: "UTF8",
+    #   "BINARY".
+    #   @return [String]
+    #
+    # @!attribute [rw] completion_state
+    #   Completion state header. Can be one of these possible values:
+    #   "PARTIAL", "COMPLETE".
+    #   @return [String]
+    #
+    # @!attribute [rw] p
+    #   Padding string for alignment.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/ResponsePayloadPart AWS API Documentation
+    #
+    class ResponsePayloadPart < Struct.new(
+      :bytes,
+      :data_type,
+      :completion_state,
+      :p,
+      :event_type)
+      SENSITIVE = [:bytes]
+      include Aws::Structure
+    end
+
+    # The request has failed due to a temporary failure of the server.
+    #
+    # @!attribute [rw] message
+    #   Error message.
+    #   @return [String]
+    #
+    # @!attribute [rw] error_code
+    #   Error code.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/ServiceUnavailableError AWS API Documentation
+    #
+    class ServiceUnavailableError < Struct.new(
+      :message,
+      :error_code)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Request stream event union.
+    #
+    # EventStream is an Enumerator of Events.
+    #  #event_types #=> Array, returns all modeled event types in the stream
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/RequestStreamEvent AWS API Documentation
+    #
+    class RequestStreamEvent < Enumerator
+
+      def event_types
+        [
+          :payload_part
+        ]
+      end
+
+    end
+
+    # Response stream event union.
+    #
+    # EventStream is an Enumerator of Events.
+    #  #event_types #=> Array, returns all modeled event types in the stream
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/sagemaker-runtime-http2-2025-10-01/ResponseStreamEvent AWS API Documentation
+    #
+    class ResponseStreamEvent < Enumerator
+
+      def event_types
+        [
+          :payload_part,
+          :model_stream_error,
+          :internal_stream_failure
+        ]
+      end
+
+    end
+
+  end
+end
+

data/lib/aws-sdk-sagemakerruntimehttp2.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+
+require 'aws-sdk-core'
+require 'aws-sigv4'
+
+Aws::Plugins::GlobalConfiguration.add_identifier(:sagemakerruntimehttp2)
+
+# This module provides support for Amazon SageMaker Runtime HTTP2. This module is available in the
+# `aws-sdk-sagemakerruntimehttp2` gem.
+#
+# # Async Client
+#
+# The {AsyncClient} class provides one asynchronous method for each API operation.
+# Operation methods each accept a hash of request parameters and return an async
+# response. For streaming operations, you can signal input events and register
+# output event callbacks before waiting on the response.
+#
+#     sage_maker_runtime_http2 = Aws::SageMakerRuntimeHTTP2::AsyncClient.new
+#     resp = sage_maker_runtime_http2.invoke_endpoint_with_bidirectional_stream(
+#       # params and input stream
+#      ) do |output_stream|
+#       # register callbacks for events
+#     end
+#
+# See {AsyncClient} for more information.
+#
+# # Errors
+#
+# Errors returned from Amazon SageMaker Runtime HTTP2 are defined in the
+# {Errors} module and all extend {Errors::ServiceError}.
+#
+#     begin
+#       # do stuff
+#     rescue Aws::SageMakerRuntimeHTTP2::Errors::ServiceError
+#       # rescues all Amazon SageMaker Runtime HTTP2 API errors
+#     end
+#
+# See {Errors} for more information.
+#
+# @!group service
+module Aws::SageMakerRuntimeHTTP2
+  autoload :Types, 'aws-sdk-sagemakerruntimehttp2/types'
+  autoload :ClientApi, 'aws-sdk-sagemakerruntimehttp2/client_api'
+  module Plugins
+    autoload :Endpoints, 'aws-sdk-sagemakerruntimehttp2/plugins/endpoints.rb'
+  end
+  autoload :Errors, 'aws-sdk-sagemakerruntimehttp2/errors'
+  autoload :EndpointParameters, 'aws-sdk-sagemakerruntimehttp2/endpoint_parameters'
+  autoload :EndpointProvider, 'aws-sdk-sagemakerruntimehttp2/endpoint_provider'
+  autoload :Endpoints, 'aws-sdk-sagemakerruntimehttp2/endpoints'
+  autoload :AsyncClient, 'aws-sdk-sagemakerruntimehttp2/async_client'
+  autoload :EventStreams, 'aws-sdk-sagemakerruntimehttp2/event_streams'
+
+  GEM_VERSION = '1.0.0'
+
+end
+
+require_relative 'aws-sdk-sagemakerruntimehttp2/customizations'

data/sig/async_client.rbs

@@ -0,0 +1,87 @@
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws
+  module SageMakerRuntimeHTTP2
+    class AsyncClient < ::Seahorse::Client::AsyncBase
+      include ::Aws::AsyncClientStubs
+
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMakerRuntimeHTTP2/AsyncClient.html#initialize-instance_method
+      def self.new: (
+                      ?credentials: untyped,
+                      ?region: String,
+                      ?access_key_id: String,
+                      ?account_id: String,
+                      ?adaptive_retry_wait_to_fill: bool,
+                      ?auth_scheme_preference: Array[String],
+                      ?convert_params: bool,
+                      ?correct_clock_skew: bool,
+                      ?defaults_mode: String,
+                      ?disable_request_compression: bool,
+                      ?endpoint: String,
+                      ?event_stream_handler: Proc,
+                      ?ignore_configured_endpoint_urls: bool,
+                      ?input_event_stream_handler: Proc,
+                      ?log_formatter: untyped,
+                      ?log_level: Symbol,
+                      ?logger: untyped,
+                      ?max_attempts: Integer,
+                      ?output_event_stream_handler: Proc,
+                      ?profile: String,
+                      ?request_checksum_calculation: String,
+                      ?request_min_compression_size_bytes: Integer,
+                      ?response_checksum_validation: String,
+                      ?retry_backoff: Proc,
+                      ?retry_base_delay: Float,
+                      ?retry_jitter: (:none | :equal | :full | ^(Integer) -> Integer),
+                      ?retry_limit: Integer,
+                      ?retry_max_delay: Integer,
+                      ?retry_mode: ("legacy" | "standard" | "adaptive"),
+                      ?sdk_ua_app_id: String,
+                      ?secret_access_key: String,
+                      ?session_token: String,
+                      ?sigv4a_signing_region_set: Array[String],
+                      ?stub_responses: untyped,
+                      ?telemetry_provider: Aws::Telemetry::TelemetryProviderBase,
+                      ?token_provider: untyped,
+                      ?use_dualstack_endpoint: bool,
+                      ?use_fips_endpoint: bool,
+                      ?validate_params: bool,
+                      ?endpoint_provider: untyped,
+                      ?connection_read_timeout: (Float | Integer),
+                      ?connection_timeout: (Float | Integer),
+                      ?enable_alpn: bool,
+                      ?max_concurrent_streams: (Float | Integer),
+                      ?read_chunk_size: (Float | Integer),
+                      ?http_wire_trace: bool,
+                      ?ssl_verify_peer: bool,
+                      ?ssl_ca_bundle: String,
+                      ?ssl_ca_directory: String,
+                      ?ssl_ca_store: String,
+                      ?raise_response_errors: bool
+                    ) -> instance
+                  | (?Hash[Symbol, untyped]) -> instance
+
+
+      interface _InvokeEndpointWithBidirectionalStreamResponseSuccess
+        include ::Seahorse::Client::_ResponseSuccess[Types::InvokeEndpointWithBidirectionalStreamOutput]
+        def body: () -> Types::ResponseStreamEvent
+        def invoked_production_variant: () -> ::String
+      end
+      # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/SageMakerRuntimeHTTP2/AsyncClient.html#invoke_endpoint_with_bidirectional_stream-instance_method
+      def invoke_endpoint_with_bidirectional_stream: (
+                                                       endpoint_name: ::String,
+                                                       input_event_stream_handler: untyped,
+                                                       output_event_stream_handler: untyped,
+                                                       ?target_variant: ::String,
+                                                       ?model_invocation_path: ::String,
+                                                       ?model_query_string: ::String
+                                                     ) ?{ (*untyped) -> void } -> _InvokeEndpointWithBidirectionalStreamResponseSuccess
+                                                   | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) ?{ (*untyped) -> void } -> _InvokeEndpointWithBidirectionalStreamResponseSuccess
+    end
+  end
+end

data/sig/errors.rbs

@@ -0,0 +1,42 @@
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws
+  module SageMakerRuntimeHTTP2
+    module Errors
+      class ServiceError < ::Aws::Errors::ServiceError
+      end
+
+      class InputValidationError < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+        def error_code: () -> ::String
+      end
+      class InternalServerError < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+        def error_code: () -> ::String
+      end
+      class InternalStreamFailure < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+      end
+      class ModelError < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+        def original_status_code: () -> ::String
+        def original_message: () -> ::String
+        def log_stream_arn: () -> ::String
+        def error_code: () -> ::String
+      end
+      class ModelStreamError < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+        def error_code: () -> ::String
+      end
+      class ServiceUnavailableError < ::Aws::Errors::ServiceError
+        def message: () -> ::String
+        def error_code: () -> ::String
+      end
+    end
+  end
+end

data/sig/types.rbs

@@ -0,0 +1,92 @@
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws::SageMakerRuntimeHTTP2
+  module Types
+
+    class InputValidationError
+      attr_accessor message: ::String
+      attr_accessor error_code: ::String
+      SENSITIVE: []
+    end
+
+    class InternalServerError
+      attr_accessor message: ::String
+      attr_accessor error_code: ::String
+      SENSITIVE: []
+    end
+
+    class InternalStreamFailure
+      attr_accessor message: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class InvokeEndpointWithBidirectionalStreamInput
+      attr_accessor endpoint_name: ::String
+      attr_accessor body: Types::RequestStreamEvent
+      attr_accessor target_variant: ::String
+      attr_accessor model_invocation_path: ::String
+      attr_accessor model_query_string: ::String
+      SENSITIVE: []
+    end
+
+    class InvokeEndpointWithBidirectionalStreamOutput
+      attr_accessor body: Types::ResponseStreamEvent
+      attr_accessor invoked_production_variant: ::String
+      SENSITIVE: []
+    end
+
+    class ModelError
+      attr_accessor message: ::String
+      attr_accessor original_status_code: ::Integer
+      attr_accessor original_message: ::String
+      attr_accessor log_stream_arn: ::String
+      attr_accessor error_code: ::String
+      SENSITIVE: []
+    end
+
+    class ModelStreamError
+      attr_accessor message: ::String
+      attr_accessor error_code: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: []
+    end
+
+    class RequestPayloadPart
+      attr_accessor bytes: ::String
+      attr_accessor data_type: ::String
+      attr_accessor completion_state: ::String
+      attr_accessor p: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: [:bytes]
+    end
+
+    class ResponsePayloadPart
+      attr_accessor bytes: ::String
+      attr_accessor data_type: ::String
+      attr_accessor completion_state: ::String
+      attr_accessor p: ::String
+      attr_accessor event_type: untyped
+      SENSITIVE: [:bytes]
+    end
+
+    class ServiceUnavailableError
+      attr_accessor message: ::String
+      attr_accessor error_code: ::String
+      SENSITIVE: []
+    end
+
+    class RequestStreamEvent < Enumerator[untyped, untyped]
+      def event_types: () -> [:payload_part]
+    end
+
+    class ResponseStreamEvent < Enumerator[untyped, untyped]
+      def event_types: () -> [:payload_part, :model_stream_error, :internal_stream_failure]
+    end
+  end
+end

data/sig/waiters.rbs

@@ -0,0 +1,13 @@
+# WARNING ABOUT GENERATED CODE
+#
+# This file is generated. See the contributing guide for more information:
+# https://github.com/aws/aws-sdk-ruby/blob/version-3/CONTRIBUTING.md
+#
+# WARNING ABOUT GENERATED CODE
+
+module Aws
+  module SageMakerRuntimeHTTP2
+    module Waiters
+    end
+  end
+end