@aws-sdk/client-sagemaker-runtime 3.395.0 → 3.404.0

package/README.md

@@ -219,3 +219,11 @@ InvokeEndpointAsync
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/clients/client-sagemaker-runtime/classes/invokeendpointasynccommand.html) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/clients/client-sagemaker-runtime/interfaces/invokeendpointasynccommandinput.html) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/clients/client-sagemaker-runtime/interfaces/invokeendpointasynccommandoutput.html)
 
 </details>
+<details>
+<summary>
+InvokeEndpointWithResponseStream
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/clients/client-sagemaker-runtime/classes/invokeendpointwithresponsestreamcommand.html) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/clients/client-sagemaker-runtime/interfaces/invokeendpointwithresponsestreamcommandinput.html) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/clients/client-sagemaker-runtime/interfaces/invokeendpointwithresponsestreamcommandoutput.html)
+
+</details>

package/dist-cjs/SageMakerRuntime.js

@@ -4,10 +4,12 @@ exports.SageMakerRuntime = void 0;
 const smithy_client_1 = require("@smithy/smithy-client");
 const InvokeEndpointAsyncCommand_1 = require("./commands/InvokeEndpointAsyncCommand");
 const InvokeEndpointCommand_1 = require("./commands/InvokeEndpointCommand");
+const InvokeEndpointWithResponseStreamCommand_1 = require("./commands/InvokeEndpointWithResponseStreamCommand");
 const SageMakerRuntimeClient_1 = require("./SageMakerRuntimeClient");
 const commands = {
     InvokeEndpointCommand: InvokeEndpointCommand_1.InvokeEndpointCommand,
     InvokeEndpointAsyncCommand: InvokeEndpointAsyncCommand_1.InvokeEndpointAsyncCommand,
+    InvokeEndpointWithResponseStreamCommand: InvokeEndpointWithResponseStreamCommand_1.InvokeEndpointWithResponseStreamCommand,
 };
 class SageMakerRuntime extends SageMakerRuntimeClient_1.SageMakerRuntimeClient {
 }

package/dist-cjs/SageMakerRuntimeClient.js

@@ -7,6 +7,7 @@ const middleware_recursion_detection_1 = require("@aws-sdk/middleware-recursion-
 const middleware_signing_1 = require("@aws-sdk/middleware-signing");
 const middleware_user_agent_1 = require("@aws-sdk/middleware-user-agent");
 const config_resolver_1 = require("@smithy/config-resolver");
+const eventstream_serde_config_resolver_1 = require("@smithy/eventstream-serde-config-resolver");
 const middleware_content_length_1 = require("@smithy/middleware-content-length");
 const middleware_endpoint_1 = require("@smithy/middleware-endpoint");
 const middleware_retry_1 = require("@smithy/middleware-retry");
@@ -25,9 +26,10 @@ class SageMakerRuntimeClient extends smithy_client_1.Client {
         const _config_5 = (0, middleware_host_header_1.resolveHostHeaderConfig)(_config_4);
         const _config_6 = (0, middleware_signing_1.resolveAwsAuthConfig)(_config_5);
         const _config_7 = (0, middleware_user_agent_1.resolveUserAgentConfig)(_config_6);
-        const _config_8 = (0, runtimeExtensions_1.resolveRuntimeExtensions)(_config_7, configuration?.extensions || []);
-        super(_config_8);
-        this.config = _config_8;
+        const _config_8 = (0, eventstream_serde_config_resolver_1.resolveEventStreamSerdeConfig)(_config_7);
+        const _config_9 = (0, runtimeExtensions_1.resolveRuntimeExtensions)(_config_8, configuration?.extensions || []);
+        super(_config_9);
+        this.config = _config_9;
         this.middlewareStack.use((0, middleware_retry_1.getRetryPlugin)(this.config));
         this.middlewareStack.use((0, middleware_content_length_1.getContentLengthPlugin)(this.config));
         this.middlewareStack.use((0, middleware_host_header_1.getHostHeaderPlugin)(this.config));

package/dist-cjs/commands/InvokeEndpointWithResponseStreamCommand.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InvokeEndpointWithResponseStreamCommand = exports.$Command = void 0;
+const middleware_endpoint_1 = require("@smithy/middleware-endpoint");
+const middleware_serde_1 = require("@smithy/middleware-serde");
+const smithy_client_1 = require("@smithy/smithy-client");
+Object.defineProperty(exports, "$Command", { enumerable: true, get: function () { return smithy_client_1.Command; } });
+const models_0_1 = require("../models/models_0");
+const Aws_restJson1_1 = require("../protocols/Aws_restJson1");
+class InvokeEndpointWithResponseStreamCommand extends smithy_client_1.Command {
+    static getEndpointParameterInstructions() {
+        return {
+            UseFIPS: { type: "builtInParams", name: "useFipsEndpoint" },
+            Endpoint: { type: "builtInParams", name: "endpoint" },
+            Region: { type: "builtInParams", name: "region" },
+            UseDualStack: { type: "builtInParams", name: "useDualstackEndpoint" },
+        };
+    }
+    constructor(input) {
+        super();
+        this.input = input;
+    }
+    resolveMiddleware(clientStack, configuration, options) {
+        this.middlewareStack.use((0, middleware_serde_1.getSerdePlugin)(configuration, this.serialize, this.deserialize));
+        this.middlewareStack.use((0, middleware_endpoint_1.getEndpointPlugin)(configuration, InvokeEndpointWithResponseStreamCommand.getEndpointParameterInstructions()));
+        const stack = clientStack.concat(this.middlewareStack);
+        const { logger } = configuration;
+        const clientName = "SageMakerRuntimeClient";
+        const commandName = "InvokeEndpointWithResponseStreamCommand";
+        const handlerExecutionContext = {
+            logger,
+            clientName,
+            commandName,
+            inputFilterSensitiveLog: models_0_1.InvokeEndpointWithResponseStreamInputFilterSensitiveLog,
+            outputFilterSensitiveLog: models_0_1.InvokeEndpointWithResponseStreamOutputFilterSensitiveLog,
+        };
+        const { requestHandler } = configuration;
+        return stack.resolve((request) => requestHandler.handle(request.request, options || {}), handlerExecutionContext);
+    }
+    serialize(input, context) {
+        return (0, Aws_restJson1_1.se_InvokeEndpointWithResponseStreamCommand)(input, context);
+    }
+    deserialize(output, context) {
+        return (0, Aws_restJson1_1.de_InvokeEndpointWithResponseStreamCommand)(output, context);
+    }
+}
+exports.InvokeEndpointWithResponseStreamCommand = InvokeEndpointWithResponseStreamCommand;

package/dist-cjs/commands/index.js

@@ -3,3 +3,4 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const tslib_1 = require("tslib");
 tslib_1.__exportStar(require("./InvokeEndpointAsyncCommand"), exports);
 tslib_1.__exportStar(require("./InvokeEndpointCommand"), exports);
+tslib_1.__exportStar(require("./InvokeEndpointWithResponseStreamCommand"), exports);

package/dist-cjs/endpoint/ruleset.js

@@ -3,5 +3,5 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ruleSet = void 0;
 const s = "required", t = "fn", u = "argv", v = "ref";
 const a = "isSet", b = "tree", c = "error", d = "endpoint", e = "PartitionResult", f = "stringEquals", g = { [s]: false, "type": "String" }, h = { [s]: true, "default": false, "type": "Boolean" }, i = { [v]: "Endpoint" }, j = { [t]: "booleanEquals", [u]: [{ [v]: "UseFIPS" }, true] }, k = { [t]: "booleanEquals", [u]: [{ [v]: "UseDualStack" }, true] }, l = {}, m = { [t]: "booleanEquals", [u]: [true, { [t]: "getAttr", [u]: [{ [v]: e }, "supportsFIPS"] }] }, n = { [t]: "booleanEquals", [u]: [true, { [t]: "getAttr", [u]: [{ [v]: e }, "supportsDualStack"] }] }, o = { [t]: "getAttr", [u]: [{ [v]: e }, "name"] }, p = [j], q = [k], r = [{ [v]: "Region" }];
-const _data = { version: "1.0", parameters: { Region: g, UseDualStack: h, UseFIPS: h, Endpoint: g }, rules: [{ conditions: [{ [t]: a, [u]: [i] }], type: b, rules: [{ conditions: p, error: "Invalid Configuration: FIPS and custom endpoint are not supported", type: c }, { type: b, rules: [{ conditions: q, error: "Invalid Configuration: Dualstack and custom endpoint are not supported", type: c }, { endpoint: { url: i, properties: l, headers: l }, type: d }] }] }, { type: b, rules: [{ conditions: [{ [t]: a, [u]: r }], type: b, rules: [{ conditions: [{ [t]: "aws.partition", [u]: r, assign: e }], type: b, rules: [{ conditions: [j, k], type: b, rules: [{ conditions: [m, n], type: b, rules: [{ type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker-fips.{Region}.{PartitionResult#dualStackDnsSuffix}", properties: l, headers: l }, type: d }] }] }, { error: "FIPS and DualStack are enabled, but this partition does not support one or both", type: c }] }, { conditions: p, type: b, rules: [{ conditions: [m], type: b, rules: [{ type: b, rules: [{ conditions: [{ [t]: f, [u]: ["aws", o] }], endpoint: { url: "https://runtime-fips.sagemaker.{Region}.amazonaws.com", properties: l, headers: l }, type: d }, { conditions: [{ [t]: f, [u]: ["aws-us-gov", o] }], endpoint: { url: "https://runtime.sagemaker.{Region}.amazonaws.com", properties: l, headers: l }, type: d }, { endpoint: { url: "https://runtime.sagemaker-fips.{Region}.{PartitionResult#dnsSuffix}", properties: l, headers: l }, type: d }] }] }, { error: "FIPS is enabled but this partition does not support FIPS", type: c }] }, { conditions: q, type: b, rules: [{ conditions: [n], type: b, rules: [{ type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker.{Region}.{PartitionResult#dualStackDnsSuffix}", properties: l, headers: l }, type: d }] }] }, { error: "DualStack is enabled but this partition does not support DualStack", type: c }] }, { type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker.{Region}.{PartitionResult#dnsSuffix}", properties: l, headers: l }, type: d }] }] }] }, { error: "Invalid Configuration: Missing Region", type: c }] }] };
+const _data = { version: "1.0", parameters: { Region: g, UseDualStack: h, UseFIPS: h, Endpoint: g }, rules: [{ conditions: [{ [t]: a, [u]: [i] }], type: b, rules: [{ conditions: p, error: "Invalid Configuration: FIPS and custom endpoint are not supported", type: c }, { conditions: q, error: "Invalid Configuration: Dualstack and custom endpoint are not supported", type: c }, { endpoint: { url: i, properties: l, headers: l }, type: d }] }, { conditions: [{ [t]: a, [u]: r }], type: b, rules: [{ conditions: [{ [t]: "aws.partition", [u]: r, assign: e }], type: b, rules: [{ conditions: [j, k], type: b, rules: [{ conditions: [m, n], type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker-fips.{Region}.{PartitionResult#dualStackDnsSuffix}", properties: l, headers: l }, type: d }] }, { error: "FIPS and DualStack are enabled, but this partition does not support one or both", type: c }] }, { conditions: p, type: b, rules: [{ conditions: [m], type: b, rules: [{ conditions: [{ [t]: f, [u]: ["aws", o] }], endpoint: { url: "https://runtime-fips.sagemaker.{Region}.amazonaws.com", properties: l, headers: l }, type: d }, { conditions: [{ [t]: f, [u]: ["aws-us-gov", o] }], endpoint: { url: "https://runtime.sagemaker.{Region}.amazonaws.com", properties: l, headers: l }, type: d }, { endpoint: { url: "https://runtime.sagemaker-fips.{Region}.{PartitionResult#dnsSuffix}", properties: l, headers: l }, type: d }] }, { error: "FIPS is enabled but this partition does not support FIPS", type: c }] }, { conditions: q, type: b, rules: [{ conditions: [n], type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker.{Region}.{PartitionResult#dualStackDnsSuffix}", properties: l, headers: l }, type: d }] }, { error: "DualStack is enabled but this partition does not support DualStack", type: c }] }, { endpoint: { url: "https://runtime.sagemaker.{Region}.{PartitionResult#dnsSuffix}", properties: l, headers: l }, type: d }] }] }, { error: "Invalid Configuration: Missing Region", type: c }] };
 exports.ruleSet = _data;

package/dist-cjs/models/models_0.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.InvokeEndpointAsyncInputFilterSensitiveLog = exports.InvokeEndpointOutputFilterSensitiveLog = exports.InvokeEndpointInputFilterSensitiveLog = exports.ValidationError = exports.ServiceUnavailable = exports.ModelNotReadyException = exports.ModelError = exports.InternalFailure = exports.InternalDependencyException = void 0;
+exports.InvokeEndpointWithResponseStreamOutputFilterSensitiveLog = exports.ResponseStreamFilterSensitiveLog = exports.PayloadPartFilterSensitiveLog = exports.InvokeEndpointWithResponseStreamInputFilterSensitiveLog = exports.InvokeEndpointAsyncInputFilterSensitiveLog = exports.InvokeEndpointOutputFilterSensitiveLog = exports.InvokeEndpointInputFilterSensitiveLog = exports.ResponseStream = exports.ModelStreamError = exports.InternalStreamFailure = exports.ValidationError = exports.ServiceUnavailable = exports.ModelNotReadyException = exports.ModelError = exports.InternalFailure = exports.InternalDependencyException = void 0;
 const smithy_client_1 = require("@smithy/smithy-client");
 const SageMakerRuntimeServiceException_1 = require("./SageMakerRuntimeServiceException");
 class InternalDependencyException extends SageMakerRuntimeServiceException_1.SageMakerRuntimeServiceException {
@@ -90,6 +90,47 @@ class ValidationError extends SageMakerRuntimeServiceException_1.SageMakerRuntim
     }
 }
 exports.ValidationError = ValidationError;
+class InternalStreamFailure extends SageMakerRuntimeServiceException_1.SageMakerRuntimeServiceException {
+    constructor(opts) {
+        super({
+            name: "InternalStreamFailure",
+            $fault: "server",
+            ...opts,
+        });
+        this.name = "InternalStreamFailure";
+        this.$fault = "server";
+        Object.setPrototypeOf(this, InternalStreamFailure.prototype);
+        this.Message = opts.Message;
+    }
+}
+exports.InternalStreamFailure = InternalStreamFailure;
+class ModelStreamError extends SageMakerRuntimeServiceException_1.SageMakerRuntimeServiceException {
+    constructor(opts) {
+        super({
+            name: "ModelStreamError",
+            $fault: "client",
+            ...opts,
+        });
+        this.name = "ModelStreamError";
+        this.$fault = "client";
+        Object.setPrototypeOf(this, ModelStreamError.prototype);
+        this.Message = opts.Message;
+        this.ErrorCode = opts.ErrorCode;
+    }
+}
+exports.ModelStreamError = ModelStreamError;
+var ResponseStream;
+(function (ResponseStream) {
+    ResponseStream.visit = (value, visitor) => {
+        if (value.PayloadPart !== undefined)
+            return visitor.PayloadPart(value.PayloadPart);
+        if (value.ModelStreamError !== undefined)
+            return visitor.ModelStreamError(value.ModelStreamError);
+        if (value.InternalStreamFailure !== undefined)
+            return visitor.InternalStreamFailure(value.InternalStreamFailure);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(ResponseStream = exports.ResponseStream || (exports.ResponseStream = {}));
 const InvokeEndpointInputFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.Body && { Body: smithy_client_1.SENSITIVE_STRING }),
@@ -107,3 +148,31 @@ const InvokeEndpointAsyncInputFilterSensitiveLog = (obj) => ({
     ...(obj.CustomAttributes && { CustomAttributes: smithy_client_1.SENSITIVE_STRING }),
 });
 exports.InvokeEndpointAsyncInputFilterSensitiveLog = InvokeEndpointAsyncInputFilterSensitiveLog;
+const InvokeEndpointWithResponseStreamInputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.Body && { Body: smithy_client_1.SENSITIVE_STRING }),
+    ...(obj.CustomAttributes && { CustomAttributes: smithy_client_1.SENSITIVE_STRING }),
+});
+exports.InvokeEndpointWithResponseStreamInputFilterSensitiveLog = InvokeEndpointWithResponseStreamInputFilterSensitiveLog;
+const PayloadPartFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.Bytes && { Bytes: smithy_client_1.SENSITIVE_STRING }),
+});
+exports.PayloadPartFilterSensitiveLog = PayloadPartFilterSensitiveLog;
+const ResponseStreamFilterSensitiveLog = (obj) => {
+    if (obj.PayloadPart !== undefined)
+        return { PayloadPart: (0, exports.PayloadPartFilterSensitiveLog)(obj.PayloadPart) };
+    if (obj.ModelStreamError !== undefined)
+        return { ModelStreamError: obj.ModelStreamError };
+    if (obj.InternalStreamFailure !== undefined)
+        return { InternalStreamFailure: obj.InternalStreamFailure };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+exports.ResponseStreamFilterSensitiveLog = ResponseStreamFilterSensitiveLog;
+const InvokeEndpointWithResponseStreamOutputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.Body && { Body: "STREAMING_CONTENT" }),
+    ...(obj.CustomAttributes && { CustomAttributes: smithy_client_1.SENSITIVE_STRING }),
+});
+exports.InvokeEndpointWithResponseStreamOutputFilterSensitiveLog = InvokeEndpointWithResponseStreamOutputFilterSensitiveLog;

package/dist-cjs/protocols/Aws_restJson1.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.de_InvokeEndpointAsyncCommand = exports.de_InvokeEndpointCommand = exports.se_InvokeEndpointAsyncCommand = exports.se_InvokeEndpointCommand = void 0;
+exports.de_InvokeEndpointWithResponseStreamCommand = exports.de_InvokeEndpointAsyncCommand = exports.de_InvokeEndpointCommand = exports.se_InvokeEndpointWithResponseStreamCommand = exports.se_InvokeEndpointAsyncCommand = exports.se_InvokeEndpointCommand = void 0;
 const protocol_http_1 = require("@smithy/protocol-http");
 const smithy_client_1 = require("@smithy/smithy-client");
 const models_0_1 = require("../models/models_0");
@@ -66,6 +66,34 @@ const se_InvokeEndpointAsyncCommand = async (input, context) => {
     });
 };
 exports.se_InvokeEndpointAsyncCommand = se_InvokeEndpointAsyncCommand;
+const se_InvokeEndpointWithResponseStreamCommand = async (input, context) => {
+    const { hostname, protocol = "https", port, path: basePath } = await context.endpoint();
+    const headers = (0, smithy_client_1.map)({}, isSerializableHeaderValue, {
+        "content-type": input.ContentType || "application/octet-stream",
+        "x-amzn-sagemaker-accept": input.Accept,
+        "x-amzn-sagemaker-custom-attributes": input.CustomAttributes,
+        "x-amzn-sagemaker-target-variant": input.TargetVariant,
+        "x-amzn-sagemaker-target-container-hostname": input.TargetContainerHostname,
+        "x-amzn-sagemaker-inference-id": input.InferenceId,
+    });
+    let resolvedPath = `${basePath?.endsWith("/") ? basePath.slice(0, -1) : basePath || ""}` +
+        "/endpoints/{EndpointName}/invocations-response-stream";
+    resolvedPath = (0, smithy_client_1.resolvedPath)(resolvedPath, input, "EndpointName", () => input.EndpointName, "{EndpointName}", false);
+    let body;
+    if (input.Body !== undefined) {
+        body = input.Body;
+    }
+    return new protocol_http_1.HttpRequest({
+        protocol,
+        hostname,
+        port,
+        method: "POST",
+        headers,
+        path: resolvedPath,
+        body,
+    });
+};
+exports.se_InvokeEndpointWithResponseStreamCommand = se_InvokeEndpointWithResponseStreamCommand;
 const de_InvokeEndpointCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_InvokeEndpointCommandError(output, context);
@@ -157,6 +185,55 @@ const de_InvokeEndpointAsyncCommandError = async (output, context) => {
             });
     }
 };
+const de_InvokeEndpointWithResponseStreamCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_InvokeEndpointWithResponseStreamCommandError(output, context);
+    }
+    const contents = (0, smithy_client_1.map)({
+        $metadata: deserializeMetadata(output),
+        ContentType: [, output.headers["x-amzn-sagemaker-content-type"]],
+        InvokedProductionVariant: [, output.headers["x-amzn-invoked-production-variant"]],
+        CustomAttributes: [, output.headers["x-amzn-sagemaker-custom-attributes"]],
+    });
+    const data = output.body;
+    contents.Body = de_ResponseStream(data, context);
+    return contents;
+};
+exports.de_InvokeEndpointWithResponseStreamCommand = de_InvokeEndpointWithResponseStreamCommand;
+const de_InvokeEndpointWithResponseStreamCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "InternalFailure":
+        case "com.amazonaws.sagemakerruntime#InternalFailure":
+            throw await de_InternalFailureRes(parsedOutput, context);
+        case "InternalStreamFailure":
+        case "com.amazonaws.sagemakerruntime#InternalStreamFailure":
+            throw await de_InternalStreamFailureRes(parsedOutput, context);
+        case "ModelError":
+        case "com.amazonaws.sagemakerruntime#ModelError":
+            throw await de_ModelErrorRes(parsedOutput, context);
+        case "ModelStreamError":
+        case "com.amazonaws.sagemakerruntime#ModelStreamError":
+            throw await de_ModelStreamErrorRes(parsedOutput, context);
+        case "ServiceUnavailable":
+        case "com.amazonaws.sagemakerruntime#ServiceUnavailable":
+            throw await de_ServiceUnavailableRes(parsedOutput, context);
+        case "ValidationError":
+        case "com.amazonaws.sagemakerruntime#ValidationError":
+            throw await de_ValidationErrorRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
 const throwDefaultError = (0, smithy_client_1.withBaseException)(SageMakerRuntimeServiceException_1.SageMakerRuntimeServiceException);
 const de_InternalDependencyExceptionRes = async (parsedOutput, context) => {
     const contents = (0, smithy_client_1.map)({});
@@ -184,6 +261,19 @@ const de_InternalFailureRes = async (parsedOutput, context) => {
     });
     return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
 };
+const de_InternalStreamFailureRes = async (parsedOutput, context) => {
+    const contents = (0, smithy_client_1.map)({});
+    const data = parsedOutput.body;
+    const doc = (0, smithy_client_1.take)(data, {
+        Message: smithy_client_1.expectString,
+    });
+    Object.assign(contents, doc);
+    const exception = new models_0_1.InternalStreamFailure({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
 const de_ModelErrorRes = async (parsedOutput, context) => {
     const contents = (0, smithy_client_1.map)({});
     const data = parsedOutput.body;
@@ -213,6 +303,20 @@ const de_ModelNotReadyExceptionRes = async (parsedOutput, context) => {
     });
     return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
 };
+const de_ModelStreamErrorRes = async (parsedOutput, context) => {
+    const contents = (0, smithy_client_1.map)({});
+    const data = parsedOutput.body;
+    const doc = (0, smithy_client_1.take)(data, {
+        ErrorCode: smithy_client_1.expectString,
+        Message: smithy_client_1.expectString,
+    });
+    Object.assign(contents, doc);
+    const exception = new models_0_1.ModelStreamError({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
 const de_ServiceUnavailableRes = async (parsedOutput, context) => {
     const contents = (0, smithy_client_1.map)({});
     const data = parsedOutput.body;
@@ -239,6 +343,45 @@ const de_ValidationErrorRes = async (parsedOutput, context) => {
     });
     return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
 };
+const de_ResponseStream = (output, context) => {
+    return context.eventStreamMarshaller.deserialize(output, async (event) => {
+        if (event["PayloadPart"] != null) {
+            return {
+                PayloadPart: await de_PayloadPart_event(event["PayloadPart"], context),
+            };
+        }
+        if (event["ModelStreamError"] != null) {
+            return {
+                ModelStreamError: await de_ModelStreamError_event(event["ModelStreamError"], context),
+            };
+        }
+        if (event["InternalStreamFailure"] != null) {
+            return {
+                InternalStreamFailure: await de_InternalStreamFailure_event(event["InternalStreamFailure"], context),
+            };
+        }
+        return { $unknown: output };
+    });
+};
+const de_InternalStreamFailure_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_InternalStreamFailureRes(parsedOutput, context);
+};
+const de_ModelStreamError_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_ModelStreamErrorRes(parsedOutput, context);
+};
+const de_PayloadPart_event = async (output, context) => {
+    const contents = {};
+    contents.Bytes = output.body;
+    return contents;
+};
 const deserializeMetadata = (output) => ({
     httpStatusCode: output.statusCode,
     requestId: output.headers["x-amzn-requestid"] ?? output.headers["x-amzn-request-id"] ?? output.headers["x-amz-request-id"],

package/dist-cjs/runtimeConfig.browser.js

@@ -6,6 +6,7 @@ const package_json_1 = tslib_1.__importDefault(require("../package.json"));
 const sha256_browser_1 = require("@aws-crypto/sha256-browser");
 const util_user_agent_browser_1 = require("@aws-sdk/util-user-agent-browser");
 const config_resolver_1 = require("@smithy/config-resolver");
+const eventstream_serde_browser_1 = require("@smithy/eventstream-serde-browser");
 const fetch_http_handler_1 = require("@smithy/fetch-http-handler");
 const invalid_dependency_1 = require("@smithy/invalid-dependency");
 const util_body_length_browser_1 = require("@smithy/util-body-length-browser");
@@ -26,6 +27,7 @@ const getRuntimeConfig = (config) => {
         credentialDefaultProvider: config?.credentialDefaultProvider ?? ((_) => () => Promise.reject(new Error("Credential is missing"))),
         defaultUserAgentProvider: config?.defaultUserAgentProvider ??
             (0, util_user_agent_browser_1.defaultUserAgent)({ serviceId: clientSharedValues.serviceId, clientVersion: package_json_1.default.version }),
+        eventStreamSerdeProvider: config?.eventStreamSerdeProvider ?? eventstream_serde_browser_1.eventStreamSerdeProvider,
         maxAttempts: config?.maxAttempts ?? util_retry_1.DEFAULT_MAX_ATTEMPTS,
         region: config?.region ?? (0, invalid_dependency_1.invalidProvider)("Region is missing"),
         requestHandler: config?.requestHandler ?? new fetch_http_handler_1.FetchHttpHandler(defaultConfigProvider),

package/dist-cjs/runtimeConfig.js

@@ -7,6 +7,7 @@ const client_sts_1 = require("@aws-sdk/client-sts");
 const credential_provider_node_1 = require("@aws-sdk/credential-provider-node");
 const util_user_agent_node_1 = require("@aws-sdk/util-user-agent-node");
 const config_resolver_1 = require("@smithy/config-resolver");
+const eventstream_serde_node_1 = require("@smithy/eventstream-serde-node");
 const hash_node_1 = require("@smithy/hash-node");
 const middleware_retry_1 = require("@smithy/middleware-retry");
 const node_config_provider_1 = require("@smithy/node-config-provider");
@@ -31,6 +32,7 @@ const getRuntimeConfig = (config) => {
         credentialDefaultProvider: config?.credentialDefaultProvider ?? (0, client_sts_1.decorateDefaultCredentialProvider)(credential_provider_node_1.defaultProvider),
         defaultUserAgentProvider: config?.defaultUserAgentProvider ??
             (0, util_user_agent_node_1.defaultUserAgent)({ serviceId: clientSharedValues.serviceId, clientVersion: package_json_1.default.version }),
+        eventStreamSerdeProvider: config?.eventStreamSerdeProvider ?? eventstream_serde_node_1.eventStreamSerdeProvider,
         maxAttempts: config?.maxAttempts ?? (0, node_config_provider_1.loadConfig)(middleware_retry_1.NODE_MAX_ATTEMPT_CONFIG_OPTIONS),
         region: config?.region ?? (0, node_config_provider_1.loadConfig)(config_resolver_1.NODE_REGION_CONFIG_OPTIONS, config_resolver_1.NODE_REGION_CONFIG_FILE_OPTIONS),
         requestHandler: config?.requestHandler ?? new node_http_handler_1.NodeHttpHandler(defaultConfigProvider),

package/dist-cjs/runtimeExtensions.js

@@ -1,16 +1,16 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.resolveRuntimeExtensions = void 0;
-const types_1 = require("@smithy/types");
+const smithy_client_1 = require("@smithy/smithy-client");
 const asPartial = (t) => t;
 const resolveRuntimeExtensions = (runtimeConfig, extensions) => {
-    const clientConfiguration = {
-        ...asPartial((0, types_1.getDefaultClientConfiguration)(runtimeConfig)),
+    const extensionConfiguration = {
+        ...asPartial((0, smithy_client_1.getDefaultExtensionConfiguration)(runtimeConfig)),
     };
-    extensions.forEach((extension) => extension.configureClient(clientConfiguration));
+    extensions.forEach((extension) => extension.configure(extensionConfiguration));
     return {
         ...runtimeConfig,
-        ...(0, types_1.resolveDefaultRuntimeConfig)(clientConfiguration),
+        ...(0, smithy_client_1.resolveDefaultRuntimeConfig)(extensionConfiguration),
     };
 };
 exports.resolveRuntimeExtensions = resolveRuntimeExtensions;

package/dist-es/SageMakerRuntime.js

@@ -1,10 +1,12 @@
 import { createAggregatedClient } from "@smithy/smithy-client";
 import { InvokeEndpointAsyncCommand, } from "./commands/InvokeEndpointAsyncCommand";
 import { InvokeEndpointCommand, } from "./commands/InvokeEndpointCommand";
+import { InvokeEndpointWithResponseStreamCommand, } from "./commands/InvokeEndpointWithResponseStreamCommand";
 import { SageMakerRuntimeClient } from "./SageMakerRuntimeClient";
 const commands = {
     InvokeEndpointCommand,
     InvokeEndpointAsyncCommand,
+    InvokeEndpointWithResponseStreamCommand,
 };
 export class SageMakerRuntime extends SageMakerRuntimeClient {
 }

package/dist-es/SageMakerRuntimeClient.js

@@ -4,6 +4,7 @@ import { getRecursionDetectionPlugin } from "@aws-sdk/middleware-recursion-detec
 import { getAwsAuthPlugin, resolveAwsAuthConfig, } from "@aws-sdk/middleware-signing";
 import { getUserAgentPlugin, resolveUserAgentConfig, } from "@aws-sdk/middleware-user-agent";
 import { resolveRegionConfig } from "@smithy/config-resolver";
+import { resolveEventStreamSerdeConfig, } from "@smithy/eventstream-serde-config-resolver";
 import { getContentLengthPlugin } from "@smithy/middleware-content-length";
 import { resolveEndpointConfig } from "@smithy/middleware-endpoint";
 import { getRetryPlugin, resolveRetryConfig } from "@smithy/middleware-retry";
@@ -22,9 +23,10 @@ export class SageMakerRuntimeClient extends __Client {
         const _config_5 = resolveHostHeaderConfig(_config_4);
         const _config_6 = resolveAwsAuthConfig(_config_5);
         const _config_7 = resolveUserAgentConfig(_config_6);
-        const _config_8 = resolveRuntimeExtensions(_config_7, configuration?.extensions || []);
-        super(_config_8);
-        this.config = _config_8;
+        const _config_8 = resolveEventStreamSerdeConfig(_config_7);
+        const _config_9 = resolveRuntimeExtensions(_config_8, configuration?.extensions || []);
+        super(_config_9);
+        this.config = _config_9;
         this.middlewareStack.use(getRetryPlugin(this.config));
         this.middlewareStack.use(getContentLengthPlugin(this.config));
         this.middlewareStack.use(getHostHeaderPlugin(this.config));

package/dist-es/commands/InvokeEndpointWithResponseStreamCommand.js

@@ -0,0 +1,43 @@
+import { getEndpointPlugin } from "@smithy/middleware-endpoint";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+import { InvokeEndpointWithResponseStreamInputFilterSensitiveLog, InvokeEndpointWithResponseStreamOutputFilterSensitiveLog, } from "../models/models_0";
+import { de_InvokeEndpointWithResponseStreamCommand, se_InvokeEndpointWithResponseStreamCommand, } from "../protocols/Aws_restJson1";
+export { $Command };
+export class InvokeEndpointWithResponseStreamCommand extends $Command {
+    static getEndpointParameterInstructions() {
+        return {
+            UseFIPS: { type: "builtInParams", name: "useFipsEndpoint" },
+            Endpoint: { type: "builtInParams", name: "endpoint" },
+            Region: { type: "builtInParams", name: "region" },
+            UseDualStack: { type: "builtInParams", name: "useDualstackEndpoint" },
+        };
+    }
+    constructor(input) {
+        super();
+        this.input = input;
+    }
+    resolveMiddleware(clientStack, configuration, options) {
+        this.middlewareStack.use(getSerdePlugin(configuration, this.serialize, this.deserialize));
+        this.middlewareStack.use(getEndpointPlugin(configuration, InvokeEndpointWithResponseStreamCommand.getEndpointParameterInstructions()));
+        const stack = clientStack.concat(this.middlewareStack);
+        const { logger } = configuration;
+        const clientName = "SageMakerRuntimeClient";
+        const commandName = "InvokeEndpointWithResponseStreamCommand";
+        const handlerExecutionContext = {
+            logger,
+            clientName,
+            commandName,
+            inputFilterSensitiveLog: InvokeEndpointWithResponseStreamInputFilterSensitiveLog,
+            outputFilterSensitiveLog: InvokeEndpointWithResponseStreamOutputFilterSensitiveLog,
+        };
+        const { requestHandler } = configuration;
+        return stack.resolve((request) => requestHandler.handle(request.request, options || {}), handlerExecutionContext);
+    }
+    serialize(input, context) {
+        return se_InvokeEndpointWithResponseStreamCommand(input, context);
+    }
+    deserialize(output, context) {
+        return de_InvokeEndpointWithResponseStreamCommand(output, context);
+    }
+}

package/dist-es/commands/index.js

@@ -1,2 +1,3 @@
 export * from "./InvokeEndpointAsyncCommand";
 export * from "./InvokeEndpointCommand";
+export * from "./InvokeEndpointWithResponseStreamCommand";

package/dist-es/endpoint/ruleset.js

@@ -1,4 +1,4 @@
 const s = "required", t = "fn", u = "argv", v = "ref";
 const a = "isSet", b = "tree", c = "error", d = "endpoint", e = "PartitionResult", f = "stringEquals", g = { [s]: false, "type": "String" }, h = { [s]: true, "default": false, "type": "Boolean" }, i = { [v]: "Endpoint" }, j = { [t]: "booleanEquals", [u]: [{ [v]: "UseFIPS" }, true] }, k = { [t]: "booleanEquals", [u]: [{ [v]: "UseDualStack" }, true] }, l = {}, m = { [t]: "booleanEquals", [u]: [true, { [t]: "getAttr", [u]: [{ [v]: e }, "supportsFIPS"] }] }, n = { [t]: "booleanEquals", [u]: [true, { [t]: "getAttr", [u]: [{ [v]: e }, "supportsDualStack"] }] }, o = { [t]: "getAttr", [u]: [{ [v]: e }, "name"] }, p = [j], q = [k], r = [{ [v]: "Region" }];
-const _data = { version: "1.0", parameters: { Region: g, UseDualStack: h, UseFIPS: h, Endpoint: g }, rules: [{ conditions: [{ [t]: a, [u]: [i] }], type: b, rules: [{ conditions: p, error: "Invalid Configuration: FIPS and custom endpoint are not supported", type: c }, { type: b, rules: [{ conditions: q, error: "Invalid Configuration: Dualstack and custom endpoint are not supported", type: c }, { endpoint: { url: i, properties: l, headers: l }, type: d }] }] }, { type: b, rules: [{ conditions: [{ [t]: a, [u]: r }], type: b, rules: [{ conditions: [{ [t]: "aws.partition", [u]: r, assign: e }], type: b, rules: [{ conditions: [j, k], type: b, rules: [{ conditions: [m, n], type: b, rules: [{ type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker-fips.{Region}.{PartitionResult#dualStackDnsSuffix}", properties: l, headers: l }, type: d }] }] }, { error: "FIPS and DualStack are enabled, but this partition does not support one or both", type: c }] }, { conditions: p, type: b, rules: [{ conditions: [m], type: b, rules: [{ type: b, rules: [{ conditions: [{ [t]: f, [u]: ["aws", o] }], endpoint: { url: "https://runtime-fips.sagemaker.{Region}.amazonaws.com", properties: l, headers: l }, type: d }, { conditions: [{ [t]: f, [u]: ["aws-us-gov", o] }], endpoint: { url: "https://runtime.sagemaker.{Region}.amazonaws.com", properties: l, headers: l }, type: d }, { endpoint: { url: "https://runtime.sagemaker-fips.{Region}.{PartitionResult#dnsSuffix}", properties: l, headers: l }, type: d }] }] }, { error: "FIPS is enabled but this partition does not support FIPS", type: c }] }, { conditions: q, type: b, rules: [{ conditions: [n], type: b, rules: [{ type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker.{Region}.{PartitionResult#dualStackDnsSuffix}", properties: l, headers: l }, type: d }] }] }, { error: "DualStack is enabled but this partition does not support DualStack", type: c }] }, { type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker.{Region}.{PartitionResult#dnsSuffix}", properties: l, headers: l }, type: d }] }] }] }, { error: "Invalid Configuration: Missing Region", type: c }] }] };
+const _data = { version: "1.0", parameters: { Region: g, UseDualStack: h, UseFIPS: h, Endpoint: g }, rules: [{ conditions: [{ [t]: a, [u]: [i] }], type: b, rules: [{ conditions: p, error: "Invalid Configuration: FIPS and custom endpoint are not supported", type: c }, { conditions: q, error: "Invalid Configuration: Dualstack and custom endpoint are not supported", type: c }, { endpoint: { url: i, properties: l, headers: l }, type: d }] }, { conditions: [{ [t]: a, [u]: r }], type: b, rules: [{ conditions: [{ [t]: "aws.partition", [u]: r, assign: e }], type: b, rules: [{ conditions: [j, k], type: b, rules: [{ conditions: [m, n], type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker-fips.{Region}.{PartitionResult#dualStackDnsSuffix}", properties: l, headers: l }, type: d }] }, { error: "FIPS and DualStack are enabled, but this partition does not support one or both", type: c }] }, { conditions: p, type: b, rules: [{ conditions: [m], type: b, rules: [{ conditions: [{ [t]: f, [u]: ["aws", o] }], endpoint: { url: "https://runtime-fips.sagemaker.{Region}.amazonaws.com", properties: l, headers: l }, type: d }, { conditions: [{ [t]: f, [u]: ["aws-us-gov", o] }], endpoint: { url: "https://runtime.sagemaker.{Region}.amazonaws.com", properties: l, headers: l }, type: d }, { endpoint: { url: "https://runtime.sagemaker-fips.{Region}.{PartitionResult#dnsSuffix}", properties: l, headers: l }, type: d }] }, { error: "FIPS is enabled but this partition does not support FIPS", type: c }] }, { conditions: q, type: b, rules: [{ conditions: [n], type: b, rules: [{ endpoint: { url: "https://runtime.sagemaker.{Region}.{PartitionResult#dualStackDnsSuffix}", properties: l, headers: l }, type: d }] }, { error: "DualStack is enabled but this partition does not support DualStack", type: c }] }, { endpoint: { url: "https://runtime.sagemaker.{Region}.{PartitionResult#dnsSuffix}", properties: l, headers: l }, type: d }] }] }, { error: "Invalid Configuration: Missing Region", type: c }] };
 export const ruleSet = _data;

package/dist-es/models/models_0.js

@@ -81,6 +81,45 @@ export class ValidationError extends __BaseException {
         this.Message = opts.Message;
     }
 }
+export class InternalStreamFailure extends __BaseException {
+    constructor(opts) {
+        super({
+            name: "InternalStreamFailure",
+            $fault: "server",
+            ...opts,
+        });
+        this.name = "InternalStreamFailure";
+        this.$fault = "server";
+        Object.setPrototypeOf(this, InternalStreamFailure.prototype);
+        this.Message = opts.Message;
+    }
+}
+export class ModelStreamError extends __BaseException {
+    constructor(opts) {
+        super({
+            name: "ModelStreamError",
+            $fault: "client",
+            ...opts,
+        });
+        this.name = "ModelStreamError";
+        this.$fault = "client";
+        Object.setPrototypeOf(this, ModelStreamError.prototype);
+        this.Message = opts.Message;
+        this.ErrorCode = opts.ErrorCode;
+    }
+}
+export var ResponseStream;
+(function (ResponseStream) {
+    ResponseStream.visit = (value, visitor) => {
+        if (value.PayloadPart !== undefined)
+            return visitor.PayloadPart(value.PayloadPart);
+        if (value.ModelStreamError !== undefined)
+            return visitor.ModelStreamError(value.ModelStreamError);
+        if (value.InternalStreamFailure !== undefined)
+            return visitor.InternalStreamFailure(value.InternalStreamFailure);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(ResponseStream || (ResponseStream = {}));
 export const InvokeEndpointInputFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.Body && { Body: SENSITIVE_STRING }),
@@ -95,3 +134,27 @@ export const InvokeEndpointAsyncInputFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.CustomAttributes && { CustomAttributes: SENSITIVE_STRING }),
 });
+export const InvokeEndpointWithResponseStreamInputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.Body && { Body: SENSITIVE_STRING }),
+    ...(obj.CustomAttributes && { CustomAttributes: SENSITIVE_STRING }),
+});
+export const PayloadPartFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.Bytes && { Bytes: SENSITIVE_STRING }),
+});
+export const ResponseStreamFilterSensitiveLog = (obj) => {
+    if (obj.PayloadPart !== undefined)
+        return { PayloadPart: PayloadPartFilterSensitiveLog(obj.PayloadPart) };
+    if (obj.ModelStreamError !== undefined)
+        return { ModelStreamError: obj.ModelStreamError };
+    if (obj.InternalStreamFailure !== undefined)
+        return { InternalStreamFailure: obj.InternalStreamFailure };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const InvokeEndpointWithResponseStreamOutputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.Body && { Body: "STREAMING_CONTENT" }),
+    ...(obj.CustomAttributes && { CustomAttributes: SENSITIVE_STRING }),
+});