@aws-sdk/client-bedrock-runtime 3.613.0 → 3.616.0

package/dist-es/BedrockRuntime.js

@@ -1,10 +1,12 @@
 import { createAggregatedClient } from "@smithy/smithy-client";
 import { BedrockRuntimeClient } from "./BedrockRuntimeClient";
+import { ApplyGuardrailCommand, } from "./commands/ApplyGuardrailCommand";
 import { ConverseCommand } from "./commands/ConverseCommand";
 import { ConverseStreamCommand, } from "./commands/ConverseStreamCommand";
 import { InvokeModelCommand } from "./commands/InvokeModelCommand";
 import { InvokeModelWithResponseStreamCommand, } from "./commands/InvokeModelWithResponseStreamCommand";
 const commands = {
+    ApplyGuardrailCommand,
     ConverseCommand,
     ConverseStreamCommand,
     InvokeModelCommand,

package/dist-es/BedrockRuntimeClient.js

@@ -20,20 +20,20 @@ export class BedrockRuntimeClient extends __Client {
         const _config_1 = resolveClientEndpointParameters(_config_0);
         const _config_2 = resolveRegionConfig(_config_1);
         const _config_3 = resolveEndpointConfig(_config_2);
-        const _config_4 = resolveRetryConfig(_config_3);
-        const _config_5 = resolveHostHeaderConfig(_config_4);
-        const _config_6 = resolveUserAgentConfig(_config_5);
+        const _config_4 = resolveHostHeaderConfig(_config_3);
+        const _config_5 = resolveUserAgentConfig(_config_4);
+        const _config_6 = resolveRetryConfig(_config_5);
         const _config_7 = resolveEventStreamSerdeConfig(_config_6);
         const _config_8 = resolveHttpAuthSchemeConfig(_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));
         this.middlewareStack.use(getLoggerPlugin(this.config));
         this.middlewareStack.use(getRecursionDetectionPlugin(this.config));
         this.middlewareStack.use(getUserAgentPlugin(this.config));
+        this.middlewareStack.use(getRetryPlugin(this.config));
+        this.middlewareStack.use(getContentLengthPlugin(this.config));
         this.middlewareStack.use(getHttpAuthSchemeEndpointRuleSetPlugin(this.config, {
             httpAuthSchemeParametersProvider: this.getDefaultHttpAuthSchemeParametersProvider(),
             identityProviderConfigProvider: this.getIdentityProviderConfigProvider(),

package/dist-es/commands/ApplyGuardrailCommand.js

@@ -0,0 +1,24 @@
+import { getEndpointPlugin } from "@smithy/middleware-endpoint";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+import { commonParams } from "../endpoint/EndpointParameters";
+import { de_ApplyGuardrailCommand, se_ApplyGuardrailCommand } from "../protocols/Aws_restJson1";
+export { $Command };
+export class ApplyGuardrailCommand extends $Command
+    .classBuilder()
+    .ep({
+    ...commonParams,
+})
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+        getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
+    ];
+})
+    .s("AmazonBedrockFrontendService", "ApplyGuardrail", {})
+    .n("BedrockRuntimeClient", "ApplyGuardrailCommand")
+    .f(void 0, void 0)
+    .ser(se_ApplyGuardrailCommand)
+    .de(de_ApplyGuardrailCommand)
+    .build() {
+}

package/dist-es/commands/index.js

@@ -1,3 +1,4 @@
+export * from "./ApplyGuardrailCommand";
 export * from "./ConverseCommand";
 export * from "./ConverseStreamCommand";
 export * from "./InvokeModelCommand";

package/dist-es/models/models_0.js

@@ -12,6 +12,161 @@ export class AccessDeniedException extends __BaseException {
         Object.setPrototypeOf(this, AccessDeniedException.prototype);
     }
 }
+export const GuardrailContentQualifier = {
+    GROUNDING_SOURCE: "grounding_source",
+    GUARD_CONTENT: "guard_content",
+    QUERY: "query",
+};
+export var GuardrailContentBlock;
+(function (GuardrailContentBlock) {
+    GuardrailContentBlock.visit = (value, visitor) => {
+        if (value.text !== undefined)
+            return visitor.text(value.text);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(GuardrailContentBlock || (GuardrailContentBlock = {}));
+export const GuardrailContentSource = {
+    INPUT: "INPUT",
+    OUTPUT: "OUTPUT",
+};
+export const GuardrailAction = {
+    GUARDRAIL_INTERVENED: "GUARDRAIL_INTERVENED",
+    NONE: "NONE",
+};
+export const GuardrailContentPolicyAction = {
+    BLOCKED: "BLOCKED",
+};
+export const GuardrailContentFilterConfidence = {
+    HIGH: "HIGH",
+    LOW: "LOW",
+    MEDIUM: "MEDIUM",
+    NONE: "NONE",
+};
+export const GuardrailContentFilterType = {
+    HATE: "HATE",
+    INSULTS: "INSULTS",
+    MISCONDUCT: "MISCONDUCT",
+    PROMPT_ATTACK: "PROMPT_ATTACK",
+    SEXUAL: "SEXUAL",
+    VIOLENCE: "VIOLENCE",
+};
+export const GuardrailContextualGroundingPolicyAction = {
+    BLOCKED: "BLOCKED",
+    NONE: "NONE",
+};
+export const GuardrailContextualGroundingFilterType = {
+    GROUNDING: "GROUNDING",
+    RELEVANCE: "RELEVANCE",
+};
+export const GuardrailSensitiveInformationPolicyAction = {
+    ANONYMIZED: "ANONYMIZED",
+    BLOCKED: "BLOCKED",
+};
+export const GuardrailPiiEntityType = {
+    ADDRESS: "ADDRESS",
+    AGE: "AGE",
+    AWS_ACCESS_KEY: "AWS_ACCESS_KEY",
+    AWS_SECRET_KEY: "AWS_SECRET_KEY",
+    CA_HEALTH_NUMBER: "CA_HEALTH_NUMBER",
+    CA_SOCIAL_INSURANCE_NUMBER: "CA_SOCIAL_INSURANCE_NUMBER",
+    CREDIT_DEBIT_CARD_CVV: "CREDIT_DEBIT_CARD_CVV",
+    CREDIT_DEBIT_CARD_EXPIRY: "CREDIT_DEBIT_CARD_EXPIRY",
+    CREDIT_DEBIT_CARD_NUMBER: "CREDIT_DEBIT_CARD_NUMBER",
+    DRIVER_ID: "DRIVER_ID",
+    EMAIL: "EMAIL",
+    INTERNATIONAL_BANK_ACCOUNT_NUMBER: "INTERNATIONAL_BANK_ACCOUNT_NUMBER",
+    IP_ADDRESS: "IP_ADDRESS",
+    LICENSE_PLATE: "LICENSE_PLATE",
+    MAC_ADDRESS: "MAC_ADDRESS",
+    NAME: "NAME",
+    PASSWORD: "PASSWORD",
+    PHONE: "PHONE",
+    PIN: "PIN",
+    SWIFT_CODE: "SWIFT_CODE",
+    UK_NATIONAL_HEALTH_SERVICE_NUMBER: "UK_NATIONAL_HEALTH_SERVICE_NUMBER",
+    UK_NATIONAL_INSURANCE_NUMBER: "UK_NATIONAL_INSURANCE_NUMBER",
+    UK_UNIQUE_TAXPAYER_REFERENCE_NUMBER: "UK_UNIQUE_TAXPAYER_REFERENCE_NUMBER",
+    URL: "URL",
+    USERNAME: "USERNAME",
+    US_BANK_ACCOUNT_NUMBER: "US_BANK_ACCOUNT_NUMBER",
+    US_BANK_ROUTING_NUMBER: "US_BANK_ROUTING_NUMBER",
+    US_INDIVIDUAL_TAX_IDENTIFICATION_NUMBER: "US_INDIVIDUAL_TAX_IDENTIFICATION_NUMBER",
+    US_PASSPORT_NUMBER: "US_PASSPORT_NUMBER",
+    US_SOCIAL_SECURITY_NUMBER: "US_SOCIAL_SECURITY_NUMBER",
+    VEHICLE_IDENTIFICATION_NUMBER: "VEHICLE_IDENTIFICATION_NUMBER",
+};
+export const GuardrailTopicPolicyAction = {
+    BLOCKED: "BLOCKED",
+};
+export const GuardrailTopicType = {
+    DENY: "DENY",
+};
+export const GuardrailWordPolicyAction = {
+    BLOCKED: "BLOCKED",
+};
+export const GuardrailManagedWordType = {
+    PROFANITY: "PROFANITY",
+};
+export class InternalServerException extends __BaseException {
+    constructor(opts) {
+        super({
+            name: "InternalServerException",
+            $fault: "server",
+            ...opts,
+        });
+        this.name = "InternalServerException";
+        this.$fault = "server";
+        Object.setPrototypeOf(this, InternalServerException.prototype);
+    }
+}
+export class ResourceNotFoundException extends __BaseException {
+    constructor(opts) {
+        super({
+            name: "ResourceNotFoundException",
+            $fault: "client",
+            ...opts,
+        });
+        this.name = "ResourceNotFoundException";
+        this.$fault = "client";
+        Object.setPrototypeOf(this, ResourceNotFoundException.prototype);
+    }
+}
+export class ServiceQuotaExceededException extends __BaseException {
+    constructor(opts) {
+        super({
+            name: "ServiceQuotaExceededException",
+            $fault: "client",
+            ...opts,
+        });
+        this.name = "ServiceQuotaExceededException";
+        this.$fault = "client";
+        Object.setPrototypeOf(this, ServiceQuotaExceededException.prototype);
+    }
+}
+export class ThrottlingException extends __BaseException {
+    constructor(opts) {
+        super({
+            name: "ThrottlingException",
+            $fault: "client",
+            ...opts,
+        });
+        this.name = "ThrottlingException";
+        this.$fault = "client";
+        Object.setPrototypeOf(this, ThrottlingException.prototype);
+    }
+}
+export class ValidationException extends __BaseException {
+    constructor(opts) {
+        super({
+            name: "ValidationException",
+            $fault: "client",
+            ...opts,
+        });
+        this.name = "ValidationException";
+        this.$fault = "client";
+        Object.setPrototypeOf(this, ValidationException.prototype);
+    }
+}
 export const GuardrailTrace = {
     DISABLED: "disabled",
     ENABLED: "enabled",
@@ -35,6 +190,11 @@ export var DocumentSource;
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(DocumentSource || (DocumentSource = {}));
+export const GuardrailConverseContentQualifier = {
+    GROUNDING_SOURCE: "grounding_source",
+    GUARD_CONTENT: "guard_content",
+    QUERY: "query",
+};
 export var GuardrailConverseContentBlock;
 (function (GuardrailConverseContentBlock) {
     GuardrailConverseContentBlock.visit = (value, visitor) => {
@@ -151,84 +311,6 @@ export const StopReason = {
     STOP_SEQUENCE: "stop_sequence",
     TOOL_USE: "tool_use",
 };
-export const GuardrailContentPolicyAction = {
-    BLOCKED: "BLOCKED",
-};
-export const GuardrailContentFilterConfidence = {
-    HIGH: "HIGH",
-    LOW: "LOW",
-    MEDIUM: "MEDIUM",
-    NONE: "NONE",
-};
-export const GuardrailContentFilterType = {
-    HATE: "HATE",
-    INSULTS: "INSULTS",
-    MISCONDUCT: "MISCONDUCT",
-    PROMPT_ATTACK: "PROMPT_ATTACK",
-    SEXUAL: "SEXUAL",
-    VIOLENCE: "VIOLENCE",
-};
-export const GuardrailSensitiveInformationPolicyAction = {
-    ANONYMIZED: "ANONYMIZED",
-    BLOCKED: "BLOCKED",
-};
-export const GuardrailPiiEntityType = {
-    ADDRESS: "ADDRESS",
-    AGE: "AGE",
-    AWS_ACCESS_KEY: "AWS_ACCESS_KEY",
-    AWS_SECRET_KEY: "AWS_SECRET_KEY",
-    CA_HEALTH_NUMBER: "CA_HEALTH_NUMBER",
-    CA_SOCIAL_INSURANCE_NUMBER: "CA_SOCIAL_INSURANCE_NUMBER",
-    CREDIT_DEBIT_CARD_CVV: "CREDIT_DEBIT_CARD_CVV",
-    CREDIT_DEBIT_CARD_EXPIRY: "CREDIT_DEBIT_CARD_EXPIRY",
-    CREDIT_DEBIT_CARD_NUMBER: "CREDIT_DEBIT_CARD_NUMBER",
-    DRIVER_ID: "DRIVER_ID",
-    EMAIL: "EMAIL",
-    INTERNATIONAL_BANK_ACCOUNT_NUMBER: "INTERNATIONAL_BANK_ACCOUNT_NUMBER",
-    IP_ADDRESS: "IP_ADDRESS",
-    LICENSE_PLATE: "LICENSE_PLATE",
-    MAC_ADDRESS: "MAC_ADDRESS",
-    NAME: "NAME",
-    PASSWORD: "PASSWORD",
-    PHONE: "PHONE",
-    PIN: "PIN",
-    SWIFT_CODE: "SWIFT_CODE",
-    UK_NATIONAL_HEALTH_SERVICE_NUMBER: "UK_NATIONAL_HEALTH_SERVICE_NUMBER",
-    UK_NATIONAL_INSURANCE_NUMBER: "UK_NATIONAL_INSURANCE_NUMBER",
-    UK_UNIQUE_TAXPAYER_REFERENCE_NUMBER: "UK_UNIQUE_TAXPAYER_REFERENCE_NUMBER",
-    URL: "URL",
-    USERNAME: "USERNAME",
-    US_BANK_ACCOUNT_NUMBER: "US_BANK_ACCOUNT_NUMBER",
-    US_BANK_ROUTING_NUMBER: "US_BANK_ROUTING_NUMBER",
-    US_INDIVIDUAL_TAX_IDENTIFICATION_NUMBER: "US_INDIVIDUAL_TAX_IDENTIFICATION_NUMBER",
-    US_PASSPORT_NUMBER: "US_PASSPORT_NUMBER",
-    US_SOCIAL_SECURITY_NUMBER: "US_SOCIAL_SECURITY_NUMBER",
-    VEHICLE_IDENTIFICATION_NUMBER: "VEHICLE_IDENTIFICATION_NUMBER",
-};
-export const GuardrailTopicPolicyAction = {
-    BLOCKED: "BLOCKED",
-};
-export const GuardrailTopicType = {
-    DENY: "DENY",
-};
-export const GuardrailWordPolicyAction = {
-    BLOCKED: "BLOCKED",
-};
-export const GuardrailManagedWordType = {
-    PROFANITY: "PROFANITY",
-};
-export class InternalServerException extends __BaseException {
-    constructor(opts) {
-        super({
-            name: "InternalServerException",
-            $fault: "server",
-            ...opts,
-        });
-        this.name = "InternalServerException";
-        this.$fault = "server";
-        Object.setPrototypeOf(this, InternalServerException.prototype);
-    }
-}
 export class ModelErrorException extends __BaseException {
     constructor(opts) {
         super({
@@ -267,42 +349,6 @@ export class ModelTimeoutException extends __BaseException {
         Object.setPrototypeOf(this, ModelTimeoutException.prototype);
     }
 }
-export class ResourceNotFoundException extends __BaseException {
-    constructor(opts) {
-        super({
-            name: "ResourceNotFoundException",
-            $fault: "client",
-            ...opts,
-        });
-        this.name = "ResourceNotFoundException";
-        this.$fault = "client";
-        Object.setPrototypeOf(this, ResourceNotFoundException.prototype);
-    }
-}
-export class ThrottlingException extends __BaseException {
-    constructor(opts) {
-        super({
-            name: "ThrottlingException",
-            $fault: "client",
-            ...opts,
-        });
-        this.name = "ThrottlingException";
-        this.$fault = "client";
-        Object.setPrototypeOf(this, ThrottlingException.prototype);
-    }
-}
-export class ValidationException extends __BaseException {
-    constructor(opts) {
-        super({
-            name: "ValidationException",
-            $fault: "client",
-            ...opts,
-        });
-        this.name = "ValidationException";
-        this.$fault = "client";
-        Object.setPrototypeOf(this, ValidationException.prototype);
-    }
-}
 export const GuardrailStreamProcessingMode = {
     ASYNC: "async",
     SYNC: "sync",
@@ -369,18 +415,6 @@ export const Trace = {
     DISABLED: "DISABLED",
     ENABLED: "ENABLED",
 };
-export class ServiceQuotaExceededException extends __BaseException {
-    constructor(opts) {
-        super({
-            name: "ServiceQuotaExceededException",
-            $fault: "client",
-            ...opts,
-        });
-        this.name = "ServiceQuotaExceededException";
-        this.$fault = "client";
-        Object.setPrototypeOf(this, ServiceQuotaExceededException.prototype);
-    }
-}
 export var ResponseStream;
 (function (ResponseStream) {
     ResponseStream.visit = (value, visitor) => {

package/dist-es/protocols/Aws_restJson1.js

@@ -1,8 +1,24 @@
 import { awsExpectUnion as __expectUnion, loadRestJsonErrorCode, parseJsonBody as parseBody, parseJsonErrorBody as parseErrorBody, } from "@aws-sdk/core";
 import { requestBuilder as rb } from "@smithy/core";
-import { _json, collectBody, decorateServiceException as __decorateServiceException, expectInt32 as __expectInt32, expectNonNull as __expectNonNull, expectObject as __expectObject, expectString as __expectString, map, serializeFloat as __serializeFloat, take, withBaseException, } from "@smithy/smithy-client";
+import { _json, collectBody, decorateServiceException as __decorateServiceException, expectInt32 as __expectInt32, expectNonNull as __expectNonNull, expectObject as __expectObject, expectString as __expectString, limitedParseDouble as __limitedParseDouble, map, serializeFloat as __serializeFloat, take, withBaseException, } from "@smithy/smithy-client";
 import { BedrockRuntimeServiceException as __BaseException } from "../models/BedrockRuntimeServiceException";
 import { AccessDeniedException, ContentBlock, DocumentSource, ImageSource, InternalServerException, ModelErrorException, ModelNotReadyException, ModelStreamErrorException, ModelTimeoutException, ResourceNotFoundException, ServiceQuotaExceededException, ThrottlingException, Tool, ToolInputSchema, ToolResultContentBlock, ValidationException, } from "../models/models_0";
+export const se_ApplyGuardrailCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/guardrail/{guardrailIdentifier}/version/{guardrailVersion}/apply");
+    b.p("guardrailIdentifier", () => input.guardrailIdentifier, "{guardrailIdentifier}", false);
+    b.p("guardrailVersion", () => input.guardrailVersion, "{guardrailVersion}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        content: (_) => _json(_),
+        source: [],
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_ConverseCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -79,6 +95,23 @@ export const se_InvokeModelWithResponseStreamCommand = async (input, context) =>
     b.m("POST").h(headers).b(body);
     return b.build();
 };
+export const de_ApplyGuardrailCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        action: __expectString,
+        assessments: (_) => de_GuardrailAssessmentList(_, context),
+        outputs: _json,
+        usage: _json,
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_ConverseCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -92,7 +125,7 @@ export const de_ConverseCommand = async (output, context) => {
         metrics: _json,
         output: (_) => de_ConverseOutput(__expectUnion(_), context),
         stopReason: __expectString,
-        trace: _json,
+        trace: (_) => de_ConverseTrace(_, context),
         usage: _json,
     });
     Object.assign(contents, doc);
@@ -146,27 +179,27 @@ const de_CommandError = async (output, context) => {
         case "InternalServerException":
         case "com.amazonaws.bedrockruntime#InternalServerException":
             throw await de_InternalServerExceptionRes(parsedOutput, context);
-        case "ModelErrorException":
-        case "com.amazonaws.bedrockruntime#ModelErrorException":
-            throw await de_ModelErrorExceptionRes(parsedOutput, context);
-        case "ModelNotReadyException":
-        case "com.amazonaws.bedrockruntime#ModelNotReadyException":
-            throw await de_ModelNotReadyExceptionRes(parsedOutput, context);
-        case "ModelTimeoutException":
-        case "com.amazonaws.bedrockruntime#ModelTimeoutException":
-            throw await de_ModelTimeoutExceptionRes(parsedOutput, context);
         case "ResourceNotFoundException":
         case "com.amazonaws.bedrockruntime#ResourceNotFoundException":
             throw await de_ResourceNotFoundExceptionRes(parsedOutput, context);
+        case "ServiceQuotaExceededException":
+        case "com.amazonaws.bedrockruntime#ServiceQuotaExceededException":
+            throw await de_ServiceQuotaExceededExceptionRes(parsedOutput, context);
         case "ThrottlingException":
         case "com.amazonaws.bedrockruntime#ThrottlingException":
             throw await de_ThrottlingExceptionRes(parsedOutput, context);
         case "ValidationException":
         case "com.amazonaws.bedrockruntime#ValidationException":
             throw await de_ValidationExceptionRes(parsedOutput, context);
-        case "ServiceQuotaExceededException":
-        case "com.amazonaws.bedrockruntime#ServiceQuotaExceededException":
-            throw await de_ServiceQuotaExceededExceptionRes(parsedOutput, context);
+        case "ModelErrorException":
+        case "com.amazonaws.bedrockruntime#ModelErrorException":
+            throw await de_ModelErrorExceptionRes(parsedOutput, context);
+        case "ModelNotReadyException":
+        case "com.amazonaws.bedrockruntime#ModelNotReadyException":
+            throw await de_ModelNotReadyExceptionRes(parsedOutput, context);
+        case "ModelTimeoutException":
+        case "com.amazonaws.bedrockruntime#ModelTimeoutException":
+            throw await de_ModelTimeoutExceptionRes(parsedOutput, context);
         case "ModelStreamErrorException":
         case "com.amazonaws.bedrockruntime#ModelStreamErrorException":
             throw await de_ModelStreamErrorExceptionRes(parsedOutput, context);
@@ -425,7 +458,7 @@ const de_ContentBlockStopEvent_event = async (output, context) => {
 const de_ConverseStreamMetadataEvent_event = async (output, context) => {
     const contents = {};
     const data = await parseBody(output.body, context);
-    Object.assign(contents, _json(data));
+    Object.assign(contents, de_ConverseStreamMetadataEvent(data, context));
     return contents;
 };
 const de_InternalServerException_event = async (output, context) => {
@@ -657,6 +690,23 @@ const de_ConverseOutput = (output, context) => {
     }
     return { $unknown: Object.entries(output)[0] };
 };
+const de_ConverseStreamMetadataEvent = (output, context) => {
+    return take(output, {
+        metrics: _json,
+        trace: (_) => de_ConverseStreamTrace(_, context),
+        usage: _json,
+    });
+};
+const de_ConverseStreamTrace = (output, context) => {
+    return take(output, {
+        guardrail: (_) => de_GuardrailTraceAssessment(_, context),
+    });
+};
+const de_ConverseTrace = (output, context) => {
+    return take(output, {
+        guardrail: (_) => de_GuardrailTraceAssessment(_, context),
+    });
+};
 const de_DocumentBlock = (output, context) => {
     return take(output, {
         format: __expectString,
@@ -672,6 +722,69 @@ const de_DocumentSource = (output, context) => {
     }
     return { $unknown: Object.entries(output)[0] };
 };
+const de_GuardrailAssessment = (output, context) => {
+    return take(output, {
+        contentPolicy: _json,
+        contextualGroundingPolicy: (_) => de_GuardrailContextualGroundingPolicyAssessment(_, context),
+        sensitiveInformationPolicy: _json,
+        topicPolicy: _json,
+        wordPolicy: _json,
+    });
+};
+const de_GuardrailAssessmentList = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_GuardrailAssessment(entry, context);
+    });
+    return retVal;
+};
+const de_GuardrailAssessmentListMap = (output, context) => {
+    return Object.entries(output).reduce((acc, [key, value]) => {
+        if (value === null) {
+            return acc;
+        }
+        acc[key] = de_GuardrailAssessmentList(value, context);
+        return acc;
+    }, {});
+};
+const de_GuardrailAssessmentMap = (output, context) => {
+    return Object.entries(output).reduce((acc, [key, value]) => {
+        if (value === null) {
+            return acc;
+        }
+        acc[key] = de_GuardrailAssessment(value, context);
+        return acc;
+    }, {});
+};
+const de_GuardrailContextualGroundingFilter = (output, context) => {
+    return take(output, {
+        action: __expectString,
+        score: __limitedParseDouble,
+        threshold: __limitedParseDouble,
+        type: __expectString,
+    });
+};
+const de_GuardrailContextualGroundingFilters = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_GuardrailContextualGroundingFilter(entry, context);
+    });
+    return retVal;
+};
+const de_GuardrailContextualGroundingPolicyAssessment = (output, context) => {
+    return take(output, {
+        filters: (_) => de_GuardrailContextualGroundingFilters(_, context),
+    });
+};
+const de_GuardrailTraceAssessment = (output, context) => {
+    return take(output, {
+        inputAssessment: (_) => de_GuardrailAssessmentMap(_, context),
+        modelOutput: _json,
+        outputAssessments: (_) => de_GuardrailAssessmentListMap(_, context),
+    });
+};
 const de_ImageBlock = (output, context) => {
     return take(output, {
         format: __expectString,

package/dist-types/BedrockRuntime.d.ts

@@ -1,10 +1,17 @@
 import { HttpHandlerOptions as __HttpHandlerOptions } from "@smithy/types";
 import { BedrockRuntimeClient } from "./BedrockRuntimeClient";
+import { ApplyGuardrailCommandInput, ApplyGuardrailCommandOutput } from "./commands/ApplyGuardrailCommand";
 import { ConverseCommandInput, ConverseCommandOutput } from "./commands/ConverseCommand";
 import { ConverseStreamCommandInput, ConverseStreamCommandOutput } from "./commands/ConverseStreamCommand";
 import { InvokeModelCommandInput, InvokeModelCommandOutput } from "./commands/InvokeModelCommand";
 import { InvokeModelWithResponseStreamCommandInput, InvokeModelWithResponseStreamCommandOutput } from "./commands/InvokeModelWithResponseStreamCommand";
 export interface BedrockRuntime {
+    /**
+     * @see {@link ApplyGuardrailCommand}
+     */
+    applyGuardrail(args: ApplyGuardrailCommandInput, options?: __HttpHandlerOptions): Promise<ApplyGuardrailCommandOutput>;
+    applyGuardrail(args: ApplyGuardrailCommandInput, cb: (err: any, data?: ApplyGuardrailCommandOutput) => void): void;
+    applyGuardrail(args: ApplyGuardrailCommandInput, options: __HttpHandlerOptions, cb: (err: any, data?: ApplyGuardrailCommandOutput) => void): void;
     /**
      * @see {@link ConverseCommand}
      */

package/dist-types/BedrockRuntimeClient.d.ts

@@ -8,6 +8,7 @@ import { HttpHandlerUserInput as __HttpHandlerUserInput } from "@smithy/protocol
 import { Client as __Client, DefaultsMode as __DefaultsMode, SmithyConfiguration as __SmithyConfiguration, SmithyResolvedConfiguration as __SmithyResolvedConfiguration } from "@smithy/smithy-client";
 import { AwsCredentialIdentityProvider, BodyLengthCalculator as __BodyLengthCalculator, CheckOptionalClientConfig as __CheckOptionalClientConfig, ChecksumConstructor as __ChecksumConstructor, Decoder as __Decoder, Encoder as __Encoder, EventStreamSerdeProvider as __EventStreamSerdeProvider, HashConstructor as __HashConstructor, HttpHandlerOptions as __HttpHandlerOptions, Logger as __Logger, Provider as __Provider, Provider, StreamCollector as __StreamCollector, UrlParser as __UrlParser, UserAgent as __UserAgent } from "@smithy/types";
 import { HttpAuthSchemeInputConfig, HttpAuthSchemeResolvedConfig } from "./auth/httpAuthSchemeProvider";
+import { ApplyGuardrailCommandInput, ApplyGuardrailCommandOutput } from "./commands/ApplyGuardrailCommand";
 import { ConverseCommandInput, ConverseCommandOutput } from "./commands/ConverseCommand";
 import { ConverseStreamCommandInput, ConverseStreamCommandOutput } from "./commands/ConverseStreamCommand";
 import { InvokeModelCommandInput, InvokeModelCommandOutput } from "./commands/InvokeModelCommand";
@@ -18,11 +19,11 @@ export { __Client };
 /**
  * @public
  */
-export type ServiceInputTypes = ConverseCommandInput | ConverseStreamCommandInput | InvokeModelCommandInput | InvokeModelWithResponseStreamCommandInput;
+export type ServiceInputTypes = ApplyGuardrailCommandInput | ConverseCommandInput | ConverseStreamCommandInput | InvokeModelCommandInput | InvokeModelWithResponseStreamCommandInput;
 /**
  * @public
  */
-export type ServiceOutputTypes = ConverseCommandOutput | ConverseStreamCommandOutput | InvokeModelCommandOutput | InvokeModelWithResponseStreamCommandOutput;
+export type ServiceOutputTypes = ApplyGuardrailCommandOutput | ConverseCommandOutput | ConverseStreamCommandOutput | InvokeModelCommandOutput | InvokeModelWithResponseStreamCommandOutput;
 /**
  * @public
  */
@@ -140,7 +141,7 @@ export interface ClientDefaults extends Partial<__SmithyConfiguration<__HttpHand
 /**
  * @public
  */
-export type BedrockRuntimeClientConfigType = Partial<__SmithyConfiguration<__HttpHandlerOptions>> & ClientDefaults & RegionInputConfig & EndpointInputConfig<EndpointParameters> & RetryInputConfig & HostHeaderInputConfig & UserAgentInputConfig & EventStreamSerdeInputConfig & HttpAuthSchemeInputConfig & ClientInputEndpointParameters;
+export type BedrockRuntimeClientConfigType = Partial<__SmithyConfiguration<__HttpHandlerOptions>> & ClientDefaults & RegionInputConfig & EndpointInputConfig<EndpointParameters> & HostHeaderInputConfig & UserAgentInputConfig & RetryInputConfig & EventStreamSerdeInputConfig & HttpAuthSchemeInputConfig & ClientInputEndpointParameters;
 /**
  * @public
  *
@@ -151,7 +152,7 @@ export interface BedrockRuntimeClientConfig extends BedrockRuntimeClientConfigTy
 /**
  * @public
  */
-export type BedrockRuntimeClientResolvedConfigType = __SmithyResolvedConfiguration<__HttpHandlerOptions> & Required<ClientDefaults> & RuntimeExtensionsConfig & RegionResolvedConfig & EndpointResolvedConfig<EndpointParameters> & RetryResolvedConfig & HostHeaderResolvedConfig & UserAgentResolvedConfig & EventStreamSerdeResolvedConfig & HttpAuthSchemeResolvedConfig & ClientResolvedEndpointParameters;
+export type BedrockRuntimeClientResolvedConfigType = __SmithyResolvedConfiguration<__HttpHandlerOptions> & Required<ClientDefaults> & RuntimeExtensionsConfig & RegionResolvedConfig & EndpointResolvedConfig<EndpointParameters> & HostHeaderResolvedConfig & UserAgentResolvedConfig & RetryResolvedConfig & EventStreamSerdeResolvedConfig & HttpAuthSchemeResolvedConfig & ClientResolvedEndpointParameters;
 /**
  * @public
  *