@aws-sdk/client-bedrock-agent-runtime 3.461.0

package/dist-cjs/protocols/Aws_restJson1.js

@@ -0,0 +1,689 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.de_RetrieveAndGenerateCommand = exports.de_RetrieveCommand = exports.de_InvokeAgentCommand = exports.se_RetrieveAndGenerateCommand = exports.se_RetrieveCommand = exports.se_InvokeAgentCommand = void 0;
+const core_1 = require("@aws-sdk/core");
+const protocol_http_1 = require("@smithy/protocol-http");
+const smithy_client_1 = require("@smithy/smithy-client");
+const BedrockAgentRuntimeServiceException_1 = require("../models/BedrockAgentRuntimeServiceException");
+const models_0_1 = require("../models/models_0");
+const se_InvokeAgentCommand = async (input, context) => {
+    const { hostname, protocol = "https", port, path: basePath } = await context.endpoint();
+    const headers = {
+        "content-type": "application/json",
+    };
+    let resolvedPath = `${basePath?.endsWith("/") ? basePath.slice(0, -1) : basePath || ""}` +
+        "/agents/{agentId}/agentAliases/{agentAliasId}/sessions/{sessionId}/text";
+    resolvedPath = (0, smithy_client_1.resolvedPath)(resolvedPath, input, "agentId", () => input.agentId, "{agentId}", false);
+    resolvedPath = (0, smithy_client_1.resolvedPath)(resolvedPath, input, "agentAliasId", () => input.agentAliasId, "{agentAliasId}", false);
+    resolvedPath = (0, smithy_client_1.resolvedPath)(resolvedPath, input, "sessionId", () => input.sessionId, "{sessionId}", false);
+    let body;
+    body = JSON.stringify((0, smithy_client_1.take)(input, {
+        enableTrace: [],
+        endSession: [],
+        inputText: [],
+        sessionState: (_) => (0, smithy_client_1._json)(_),
+    }));
+    return new protocol_http_1.HttpRequest({
+        protocol,
+        hostname,
+        port,
+        method: "POST",
+        headers,
+        path: resolvedPath,
+        body,
+    });
+};
+exports.se_InvokeAgentCommand = se_InvokeAgentCommand;
+const se_RetrieveCommand = async (input, context) => {
+    const { hostname, protocol = "https", port, path: basePath } = await context.endpoint();
+    const headers = {
+        "content-type": "application/json",
+    };
+    let resolvedPath = `${basePath?.endsWith("/") ? basePath.slice(0, -1) : basePath || ""}` +
+        "/knowledgebases/{knowledgeBaseId}/retrieve";
+    resolvedPath = (0, smithy_client_1.resolvedPath)(resolvedPath, input, "knowledgeBaseId", () => input.knowledgeBaseId, "{knowledgeBaseId}", false);
+    let body;
+    body = JSON.stringify((0, smithy_client_1.take)(input, {
+        nextToken: [],
+        retrievalConfiguration: (_) => (0, smithy_client_1._json)(_),
+        retrievalQuery: (_) => (0, smithy_client_1._json)(_),
+    }));
+    return new protocol_http_1.HttpRequest({
+        protocol,
+        hostname,
+        port,
+        method: "POST",
+        headers,
+        path: resolvedPath,
+        body,
+    });
+};
+exports.se_RetrieveCommand = se_RetrieveCommand;
+const se_RetrieveAndGenerateCommand = async (input, context) => {
+    const { hostname, protocol = "https", port, path: basePath } = await context.endpoint();
+    const headers = {
+        "content-type": "application/json",
+    };
+    const resolvedPath = `${basePath?.endsWith("/") ? basePath.slice(0, -1) : basePath || ""}` + "/retrieveAndGenerate";
+    let body;
+    body = JSON.stringify((0, smithy_client_1.take)(input, {
+        input: (_) => (0, smithy_client_1._json)(_),
+        retrieveAndGenerateConfiguration: (_) => (0, smithy_client_1._json)(_),
+        sessionConfiguration: (_) => (0, smithy_client_1._json)(_),
+        sessionId: [],
+    }));
+    return new protocol_http_1.HttpRequest({
+        protocol,
+        hostname,
+        port,
+        method: "POST",
+        headers,
+        path: resolvedPath,
+        body,
+    });
+};
+exports.se_RetrieveAndGenerateCommand = se_RetrieveAndGenerateCommand;
+const de_InvokeAgentCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_InvokeAgentCommandError(output, context);
+    }
+    const contents = (0, smithy_client_1.map)({
+        $metadata: deserializeMetadata(output),
+        contentType: [, output.headers["x-amzn-bedrock-agent-content-type"]],
+        sessionId: [, output.headers["x-amz-bedrock-agent-session-id"]],
+    });
+    const data = output.body;
+    contents.completion = de_ResponseStream(data, context);
+    return contents;
+};
+exports.de_InvokeAgentCommand = de_InvokeAgentCommand;
+const de_InvokeAgentCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "AccessDeniedException":
+        case "com.amazonaws.bedrockagentruntime#AccessDeniedException":
+            throw await de_AccessDeniedExceptionRes(parsedOutput, context);
+        case "BadGatewayException":
+        case "com.amazonaws.bedrockagentruntime#BadGatewayException":
+            throw await de_BadGatewayExceptionRes(parsedOutput, context);
+        case "ConflictException":
+        case "com.amazonaws.bedrockagentruntime#ConflictException":
+            throw await de_ConflictExceptionRes(parsedOutput, context);
+        case "DependencyFailedException":
+        case "com.amazonaws.bedrockagentruntime#DependencyFailedException":
+            throw await de_DependencyFailedExceptionRes(parsedOutput, context);
+        case "InternalServerException":
+        case "com.amazonaws.bedrockagentruntime#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "ResourceNotFoundException":
+        case "com.amazonaws.bedrockagentruntime#ResourceNotFoundException":
+            throw await de_ResourceNotFoundExceptionRes(parsedOutput, context);
+        case "ServiceQuotaExceededException":
+        case "com.amazonaws.bedrockagentruntime#ServiceQuotaExceededException":
+            throw await de_ServiceQuotaExceededExceptionRes(parsedOutput, context);
+        case "ThrottlingException":
+        case "com.amazonaws.bedrockagentruntime#ThrottlingException":
+            throw await de_ThrottlingExceptionRes(parsedOutput, context);
+        case "ValidationException":
+        case "com.amazonaws.bedrockagentruntime#ValidationException":
+            throw await de_ValidationExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
+const de_RetrieveCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_RetrieveCommandError(output, context);
+    }
+    const contents = (0, smithy_client_1.map)({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = (0, smithy_client_1.expectNonNull)((0, smithy_client_1.expectObject)(await parseBody(output.body, context)), "body");
+    const doc = (0, smithy_client_1.take)(data, {
+        nextToken: smithy_client_1.expectString,
+        retrievalResults: (_) => de_KnowledgeBaseRetrievalResults(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
+exports.de_RetrieveCommand = de_RetrieveCommand;
+const de_RetrieveCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "AccessDeniedException":
+        case "com.amazonaws.bedrockagentruntime#AccessDeniedException":
+            throw await de_AccessDeniedExceptionRes(parsedOutput, context);
+        case "BadGatewayException":
+        case "com.amazonaws.bedrockagentruntime#BadGatewayException":
+            throw await de_BadGatewayExceptionRes(parsedOutput, context);
+        case "ConflictException":
+        case "com.amazonaws.bedrockagentruntime#ConflictException":
+            throw await de_ConflictExceptionRes(parsedOutput, context);
+        case "DependencyFailedException":
+        case "com.amazonaws.bedrockagentruntime#DependencyFailedException":
+            throw await de_DependencyFailedExceptionRes(parsedOutput, context);
+        case "InternalServerException":
+        case "com.amazonaws.bedrockagentruntime#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "ResourceNotFoundException":
+        case "com.amazonaws.bedrockagentruntime#ResourceNotFoundException":
+            throw await de_ResourceNotFoundExceptionRes(parsedOutput, context);
+        case "ServiceQuotaExceededException":
+        case "com.amazonaws.bedrockagentruntime#ServiceQuotaExceededException":
+            throw await de_ServiceQuotaExceededExceptionRes(parsedOutput, context);
+        case "ThrottlingException":
+        case "com.amazonaws.bedrockagentruntime#ThrottlingException":
+            throw await de_ThrottlingExceptionRes(parsedOutput, context);
+        case "ValidationException":
+        case "com.amazonaws.bedrockagentruntime#ValidationException":
+            throw await de_ValidationExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
+const de_RetrieveAndGenerateCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_RetrieveAndGenerateCommandError(output, context);
+    }
+    const contents = (0, smithy_client_1.map)({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = (0, smithy_client_1.expectNonNull)((0, smithy_client_1.expectObject)(await parseBody(output.body, context)), "body");
+    const doc = (0, smithy_client_1.take)(data, {
+        citations: smithy_client_1._json,
+        output: smithy_client_1._json,
+        sessionId: smithy_client_1.expectString,
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
+exports.de_RetrieveAndGenerateCommand = de_RetrieveAndGenerateCommand;
+const de_RetrieveAndGenerateCommandError = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseErrorBody(output.body, context),
+    };
+    const errorCode = loadRestJsonErrorCode(output, parsedOutput.body);
+    switch (errorCode) {
+        case "AccessDeniedException":
+        case "com.amazonaws.bedrockagentruntime#AccessDeniedException":
+            throw await de_AccessDeniedExceptionRes(parsedOutput, context);
+        case "BadGatewayException":
+        case "com.amazonaws.bedrockagentruntime#BadGatewayException":
+            throw await de_BadGatewayExceptionRes(parsedOutput, context);
+        case "ConflictException":
+        case "com.amazonaws.bedrockagentruntime#ConflictException":
+            throw await de_ConflictExceptionRes(parsedOutput, context);
+        case "DependencyFailedException":
+        case "com.amazonaws.bedrockagentruntime#DependencyFailedException":
+            throw await de_DependencyFailedExceptionRes(parsedOutput, context);
+        case "InternalServerException":
+        case "com.amazonaws.bedrockagentruntime#InternalServerException":
+            throw await de_InternalServerExceptionRes(parsedOutput, context);
+        case "ResourceNotFoundException":
+        case "com.amazonaws.bedrockagentruntime#ResourceNotFoundException":
+            throw await de_ResourceNotFoundExceptionRes(parsedOutput, context);
+        case "ServiceQuotaExceededException":
+        case "com.amazonaws.bedrockagentruntime#ServiceQuotaExceededException":
+            throw await de_ServiceQuotaExceededExceptionRes(parsedOutput, context);
+        case "ThrottlingException":
+        case "com.amazonaws.bedrockagentruntime#ThrottlingException":
+            throw await de_ThrottlingExceptionRes(parsedOutput, context);
+        case "ValidationException":
+        case "com.amazonaws.bedrockagentruntime#ValidationException":
+            throw await de_ValidationExceptionRes(parsedOutput, context);
+        default:
+            const parsedBody = parsedOutput.body;
+            return throwDefaultError({
+                output,
+                parsedBody,
+                errorCode,
+            });
+    }
+};
+const throwDefaultError = (0, smithy_client_1.withBaseException)(BedrockAgentRuntimeServiceException_1.BedrockAgentRuntimeServiceException);
+const de_AccessDeniedExceptionRes = 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.AccessDeniedException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
+const de_BadGatewayExceptionRes = 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,
+        resourceName: smithy_client_1.expectString,
+    });
+    Object.assign(contents, doc);
+    const exception = new models_0_1.BadGatewayException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
+const de_ConflictExceptionRes = 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.ConflictException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
+const de_DependencyFailedExceptionRes = 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,
+        resourceName: smithy_client_1.expectString,
+    });
+    Object.assign(contents, doc);
+    const exception = new models_0_1.DependencyFailedException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
+const de_InternalServerExceptionRes = 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.InternalServerException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
+const de_ResourceNotFoundExceptionRes = 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.ResourceNotFoundException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
+const de_ServiceQuotaExceededExceptionRes = 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.ServiceQuotaExceededException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
+const de_ThrottlingExceptionRes = 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.ThrottlingException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
+const de_ValidationExceptionRes = 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.ValidationException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
+const de_ResponseStream = (output, context) => {
+    return context.eventStreamMarshaller.deserialize(output, async (event) => {
+        if (event["chunk"] != null) {
+            return {
+                chunk: await de_PayloadPart_event(event["chunk"], context),
+            };
+        }
+        if (event["trace"] != null) {
+            return {
+                trace: await de_TracePart_event(event["trace"], context),
+            };
+        }
+        if (event["internalServerException"] != null) {
+            return {
+                internalServerException: await de_InternalServerException_event(event["internalServerException"], context),
+            };
+        }
+        if (event["validationException"] != null) {
+            return {
+                validationException: await de_ValidationException_event(event["validationException"], context),
+            };
+        }
+        if (event["resourceNotFoundException"] != null) {
+            return {
+                resourceNotFoundException: await de_ResourceNotFoundException_event(event["resourceNotFoundException"], context),
+            };
+        }
+        if (event["serviceQuotaExceededException"] != null) {
+            return {
+                serviceQuotaExceededException: await de_ServiceQuotaExceededException_event(event["serviceQuotaExceededException"], context),
+            };
+        }
+        if (event["throttlingException"] != null) {
+            return {
+                throttlingException: await de_ThrottlingException_event(event["throttlingException"], context),
+            };
+        }
+        if (event["accessDeniedException"] != null) {
+            return {
+                accessDeniedException: await de_AccessDeniedException_event(event["accessDeniedException"], context),
+            };
+        }
+        if (event["conflictException"] != null) {
+            return {
+                conflictException: await de_ConflictException_event(event["conflictException"], context),
+            };
+        }
+        if (event["dependencyFailedException"] != null) {
+            return {
+                dependencyFailedException: await de_DependencyFailedException_event(event["dependencyFailedException"], context),
+            };
+        }
+        if (event["badGatewayException"] != null) {
+            return {
+                badGatewayException: await de_BadGatewayException_event(event["badGatewayException"], context),
+            };
+        }
+        return { $unknown: output };
+    });
+};
+const de_AccessDeniedException_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_AccessDeniedExceptionRes(parsedOutput, context);
+};
+const de_BadGatewayException_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_BadGatewayExceptionRes(parsedOutput, context);
+};
+const de_ConflictException_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_ConflictExceptionRes(parsedOutput, context);
+};
+const de_DependencyFailedException_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_DependencyFailedExceptionRes(parsedOutput, context);
+};
+const de_InternalServerException_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_InternalServerExceptionRes(parsedOutput, context);
+};
+const de_PayloadPart_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, de_PayloadPart(data, context));
+    return contents;
+};
+const de_ResourceNotFoundException_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_ResourceNotFoundExceptionRes(parsedOutput, context);
+};
+const de_ServiceQuotaExceededException_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_ServiceQuotaExceededExceptionRes(parsedOutput, context);
+};
+const de_ThrottlingException_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_ThrottlingExceptionRes(parsedOutput, context);
+};
+const de_TracePart_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, de_TracePart(data, context));
+    return contents;
+};
+const de_ValidationException_event = async (output, context) => {
+    const parsedOutput = {
+        ...output,
+        body: await parseBody(output.body, context),
+    };
+    return de_ValidationExceptionRes(parsedOutput, context);
+};
+const de_InferenceConfiguration = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        maximumLength: smithy_client_1.expectInt32,
+        stopSequences: smithy_client_1._json,
+        temperature: smithy_client_1.limitedParseFloat32,
+        topK: smithy_client_1.expectInt32,
+        topP: smithy_client_1.limitedParseFloat32,
+    });
+};
+const de_KnowledgeBaseRetrievalResult = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        content: smithy_client_1._json,
+        location: smithy_client_1._json,
+        score: smithy_client_1.limitedParseDouble,
+    });
+};
+const de_KnowledgeBaseRetrievalResults = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_KnowledgeBaseRetrievalResult(entry, context);
+    });
+    return retVal;
+};
+const de_ModelInvocationInput = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        inferenceConfiguration: (_) => de_InferenceConfiguration(_, context),
+        overrideLambda: smithy_client_1.expectString,
+        parserMode: smithy_client_1.expectString,
+        promptCreationMode: smithy_client_1.expectString,
+        text: smithy_client_1.expectString,
+        traceId: smithy_client_1.expectString,
+        type: smithy_client_1.expectString,
+    });
+};
+const de_OrchestrationTrace = (output, context) => {
+    if (output.invocationInput != null) {
+        return {
+            invocationInput: (0, smithy_client_1._json)(output.invocationInput),
+        };
+    }
+    if (output.modelInvocationInput != null) {
+        return {
+            modelInvocationInput: de_ModelInvocationInput(output.modelInvocationInput, context),
+        };
+    }
+    if (output.observation != null) {
+        return {
+            observation: (0, smithy_client_1._json)(output.observation),
+        };
+    }
+    if (output.rationale != null) {
+        return {
+            rationale: (0, smithy_client_1._json)(output.rationale),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
+const de_PayloadPart = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        attribution: smithy_client_1._json,
+        bytes: context.base64Decoder,
+    });
+};
+const de_PostProcessingTrace = (output, context) => {
+    if (output.modelInvocationInput != null) {
+        return {
+            modelInvocationInput: de_ModelInvocationInput(output.modelInvocationInput, context),
+        };
+    }
+    if (output.modelInvocationOutput != null) {
+        return {
+            modelInvocationOutput: (0, smithy_client_1._json)(output.modelInvocationOutput),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
+const de_PreProcessingTrace = (output, context) => {
+    if (output.modelInvocationInput != null) {
+        return {
+            modelInvocationInput: de_ModelInvocationInput(output.modelInvocationInput, context),
+        };
+    }
+    if (output.modelInvocationOutput != null) {
+        return {
+            modelInvocationOutput: (0, smithy_client_1._json)(output.modelInvocationOutput),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
+const de_Trace = (output, context) => {
+    if (output.failureTrace != null) {
+        return {
+            failureTrace: (0, smithy_client_1._json)(output.failureTrace),
+        };
+    }
+    if (output.orchestrationTrace != null) {
+        return {
+            orchestrationTrace: de_OrchestrationTrace((0, core_1.awsExpectUnion)(output.orchestrationTrace), context),
+        };
+    }
+    if (output.postProcessingTrace != null) {
+        return {
+            postProcessingTrace: de_PostProcessingTrace((0, core_1.awsExpectUnion)(output.postProcessingTrace), context),
+        };
+    }
+    if (output.preProcessingTrace != null) {
+        return {
+            preProcessingTrace: de_PreProcessingTrace((0, core_1.awsExpectUnion)(output.preProcessingTrace), context),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
+const de_TracePart = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        agentAliasId: smithy_client_1.expectString,
+        agentId: smithy_client_1.expectString,
+        sessionId: smithy_client_1.expectString,
+        trace: (_) => de_Trace((0, core_1.awsExpectUnion)(_), context),
+    });
+};
+const deserializeMetadata = (output) => ({
+    httpStatusCode: output.statusCode,
+    requestId: output.headers["x-amzn-requestid"] ?? output.headers["x-amzn-request-id"] ?? output.headers["x-amz-request-id"],
+    extendedRequestId: output.headers["x-amz-id-2"],
+    cfId: output.headers["x-amz-cf-id"],
+});
+const collectBodyString = (streamBody, context) => (0, smithy_client_1.collectBody)(streamBody, context).then((body) => context.utf8Encoder(body));
+const isSerializableHeaderValue = (value) => value !== undefined &&
+    value !== null &&
+    value !== "" &&
+    (!Object.getOwnPropertyNames(value).includes("length") || value.length != 0) &&
+    (!Object.getOwnPropertyNames(value).includes("size") || value.size != 0);
+const parseBody = (streamBody, context) => collectBodyString(streamBody, context).then((encoded) => {
+    if (encoded.length) {
+        return JSON.parse(encoded);
+    }
+    return {};
+});
+const parseErrorBody = async (errorBody, context) => {
+    const value = await parseBody(errorBody, context);
+    value.message = value.message ?? value.Message;
+    return value;
+};
+const loadRestJsonErrorCode = (output, data) => {
+    const findKey = (object, key) => Object.keys(object).find((k) => k.toLowerCase() === key.toLowerCase());
+    const sanitizeErrorCode = (rawValue) => {
+        let cleanValue = rawValue;
+        if (typeof cleanValue === "number") {
+            cleanValue = cleanValue.toString();
+        }
+        if (cleanValue.indexOf(",") >= 0) {
+            cleanValue = cleanValue.split(",")[0];
+        }
+        if (cleanValue.indexOf(":") >= 0) {
+            cleanValue = cleanValue.split(":")[0];
+        }
+        if (cleanValue.indexOf("#") >= 0) {
+            cleanValue = cleanValue.split("#")[1];
+        }
+        return cleanValue;
+    };
+    const headerKey = findKey(output.headers, "x-amzn-errortype");
+    if (headerKey !== undefined) {
+        return sanitizeErrorCode(output.headers[headerKey]);
+    }
+    if (data.code !== undefined) {
+        return sanitizeErrorCode(data.code);
+    }
+    if (data["__type"] !== undefined) {
+        return sanitizeErrorCode(data["__type"]);
+    }
+};