@aws-sdk/client-bedrock-runtime 3.839.0 → 3.842.0

package/dist-cjs/auth/httpAuthSchemeProvider.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.resolveHttpAuthSchemeConfig = exports.defaultBedrockRuntimeHttpAuthSchemeProvider = exports.defaultBedrockRuntimeHttpAuthSchemeParametersProvider = void 0;
 const core_1 = require("@aws-sdk/core");
+const core_2 = require("@smithy/core");
 const util_middleware_1 = require("@smithy/util-middleware");
 const defaultBedrockRuntimeHttpAuthSchemeParametersProvider = async (config, context, input) => {
     return {
@@ -28,20 +29,36 @@ function createAwsAuthSigv4HttpAuthOption(authParameters) {
         }),
     };
 }
+function createSmithyApiHttpBearerAuthHttpAuthOption(authParameters) {
+    return {
+        schemeId: "smithy.api#httpBearerAuth",
+        propertiesExtractor: ({ profile, filepath, configFilepath, ignoreCache }, context) => ({
+            identityProperties: {
+                profile,
+                filepath,
+                configFilepath,
+                ignoreCache,
+            },
+        }),
+    };
+}
 const defaultBedrockRuntimeHttpAuthSchemeProvider = (authParameters) => {
     const options = [];
     switch (authParameters.operation) {
         default: {
             options.push(createAwsAuthSigv4HttpAuthOption(authParameters));
+            options.push(createSmithyApiHttpBearerAuthHttpAuthOption(authParameters));
         }
     }
     return options;
 };
 exports.defaultBedrockRuntimeHttpAuthSchemeProvider = defaultBedrockRuntimeHttpAuthSchemeProvider;
 const resolveHttpAuthSchemeConfig = (config) => {
+    const token = (0, core_2.memoizeIdentityProvider)(config.token, core_2.isIdentityExpired, core_2.doesIdentityRequireRefresh);
     const config_0 = (0, core_1.resolveAwsSdkSigV4Config)(config);
     return Object.assign(config_0, {
         authSchemePreference: (0, util_middleware_1.normalizeProvider)(config.authSchemePreference ?? []),
+        token,
     });
 };
 exports.resolveHttpAuthSchemeConfig = resolveHttpAuthSchemeConfig;

package/dist-cjs/index.js

@@ -33,6 +33,9 @@ __export(index_exports, {
   BidirectionalInputPayloadPartFilterSensitiveLog: () => BidirectionalInputPayloadPartFilterSensitiveLog,
   BidirectionalOutputPayloadPartFilterSensitiveLog: () => BidirectionalOutputPayloadPartFilterSensitiveLog,
   CachePointType: () => CachePointType,
+  CitationGeneratedContent: () => CitationGeneratedContent,
+  CitationLocation: () => CitationLocation,
+  CitationSourceContent: () => CitationSourceContent,
   ConflictException: () => ConflictException,
   ContentBlock: () => ContentBlock,
   ContentBlockDelta: () => ContentBlockDelta,
@@ -51,6 +54,7 @@ __export(index_exports, {
   ConverseStreamOutputFilterSensitiveLog: () => ConverseStreamOutputFilterSensitiveLog,
   ConverseStreamRequestFilterSensitiveLog: () => ConverseStreamRequestFilterSensitiveLog,
   ConverseStreamResponseFilterSensitiveLog: () => ConverseStreamResponseFilterSensitiveLog,
+  DocumentContentBlock: () => DocumentContentBlock,
   DocumentFormat: () => DocumentFormat,
   DocumentSource: () => DocumentSource,
   GetAsyncInvokeCommand: () => GetAsyncInvokeCommand,
@@ -150,6 +154,7 @@ var import_middleware_host_header = require("@aws-sdk/middleware-host-header");
 var import_middleware_logger = require("@aws-sdk/middleware-logger");
 var import_middleware_recursion_detection = require("@aws-sdk/middleware-recursion-detection");
 var import_middleware_user_agent = require("@aws-sdk/middleware-user-agent");
+var import_middleware_websocket = require("@aws-sdk/middleware-websocket");
 var import_config_resolver = require("@smithy/config-resolver");
 var import_core = require("@smithy/core");
 var import_eventstream_serde_config_resolver = require("@smithy/eventstream-serde-config-resolver");
@@ -187,6 +192,7 @@ var getHttpAuthExtensionConfiguration = /* @__PURE__ */ __name((runtimeConfig) =
   const _httpAuthSchemes = runtimeConfig.httpAuthSchemes;
   let _httpAuthSchemeProvider = runtimeConfig.httpAuthSchemeProvider;
   let _credentials = runtimeConfig.credentials;
+  let _token = runtimeConfig.token;
   return {
     setHttpAuthScheme(httpAuthScheme) {
       const index = _httpAuthSchemes.findIndex((scheme) => scheme.schemeId === httpAuthScheme.schemeId);
@@ -210,6 +216,12 @@ var getHttpAuthExtensionConfiguration = /* @__PURE__ */ __name((runtimeConfig) =
     },
     credentials() {
       return _credentials;
+    },
+    setToken(token) {
+      _token = token;
+    },
+    token() {
+      return _token;
     }
   };
 }, "getHttpAuthExtensionConfiguration");
@@ -217,7 +229,8 @@ var resolveHttpAuthRuntimeConfig = /* @__PURE__ */ __name((config) => {
   return {
     httpAuthSchemes: config.httpAuthSchemes(),
     httpAuthSchemeProvider: config.httpAuthSchemeProvider(),
-    credentials: config.credentials()
+    credentials: config.credentials(),
+    token: config.token()
   };
 }, "resolveHttpAuthRuntimeConfig");
 
@@ -261,8 +274,9 @@ var BedrockRuntimeClient = class extends import_smithy_client.Client {
     const _config_7 = (0, import_eventstream_serde_config_resolver.resolveEventStreamSerdeConfig)(_config_6);
     const _config_8 = (0, import_httpAuthSchemeProvider.resolveHttpAuthSchemeConfig)(_config_7);
     const _config_9 = (0, import_middleware_eventstream.resolveEventStreamConfig)(_config_8);
-    const _config_10 = resolveRuntimeExtensions(_config_9, configuration?.extensions || []);
-    this.config = _config_10;
+    const _config_10 = (0, import_middleware_websocket.resolveWebSocketConfig)(_config_9);
+    const _config_11 = resolveRuntimeExtensions(_config_10, configuration?.extensions || []);
+    this.config = _config_11;
     this.middlewareStack.use((0, import_middleware_user_agent.getUserAgentPlugin)(this.config));
     this.middlewareStack.use((0, import_middleware_retry.getRetryPlugin)(this.config));
     this.middlewareStack.use((0, import_middleware_content_length.getContentLengthPlugin)(this.config));
@@ -273,7 +287,8 @@ var BedrockRuntimeClient = class extends import_smithy_client.Client {
       (0, import_core.getHttpAuthSchemeEndpointRuleSetPlugin)(this.config, {
         httpAuthSchemeParametersProvider: import_httpAuthSchemeProvider.defaultBedrockRuntimeHttpAuthSchemeParametersProvider,
         identityProviderConfigProvider: /* @__PURE__ */ __name(async (config) => new import_core.DefaultIdentityProviderConfig({
-          "aws.auth#sigv4": config.credentials
+          "aws.auth#sigv4": config.credentials,
+          "smithy.api#httpBearerAuth": config.token
         }), "identityProviderConfigProvider")
       })
     );
@@ -607,6 +622,29 @@ var GuardrailTrace = {
 var CachePointType = {
   DEFAULT: "default"
 };
+var CitationLocation;
+((CitationLocation3) => {
+  CitationLocation3.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.documentChar !== void 0) return visitor.documentChar(value.documentChar);
+    if (value.documentPage !== void 0) return visitor.documentPage(value.documentPage);
+    if (value.documentChunk !== void 0) return visitor.documentChunk(value.documentChunk);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(CitationLocation || (CitationLocation = {}));
+var CitationSourceContent;
+((CitationSourceContent3) => {
+  CitationSourceContent3.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.text !== void 0) return visitor.text(value.text);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(CitationSourceContent || (CitationSourceContent = {}));
+var CitationGeneratedContent;
+((CitationGeneratedContent3) => {
+  CitationGeneratedContent3.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.text !== void 0) return visitor.text(value.text);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(CitationGeneratedContent || (CitationGeneratedContent = {}));
 var DocumentFormat = {
   CSV: "csv",
   DOC: "doc",
@@ -618,11 +656,20 @@ var DocumentFormat = {
   XLS: "xls",
   XLSX: "xlsx"
 };
+var DocumentContentBlock;
+((DocumentContentBlock3) => {
+  DocumentContentBlock3.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.text !== void 0) return visitor.text(value.text);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(DocumentContentBlock || (DocumentContentBlock = {}));
 var DocumentSource;
 ((DocumentSource2) => {
   DocumentSource2.visit = /* @__PURE__ */ __name((value, visitor) => {
     if (value.bytes !== void 0) return visitor.bytes(value.bytes);
     if (value.s3Location !== void 0) return visitor.s3Location(value.s3Location);
+    if (value.text !== void 0) return visitor.text(value.text);
+    if (value.content !== void 0) return visitor.content(value.content);
     return visitor._(value.$unknown[0], value.$unknown[1]);
   }, "visit");
 })(DocumentSource || (DocumentSource = {}));
@@ -718,6 +765,7 @@ var ContentBlock;
     if (value.guardContent !== void 0) return visitor.guardContent(value.guardContent);
     if (value.cachePoint !== void 0) return visitor.cachePoint(value.cachePoint);
     if (value.reasoningContent !== void 0) return visitor.reasoningContent(value.reasoningContent);
+    if (value.citationsContent !== void 0) return visitor.citationsContent(value.citationsContent);
     return visitor._(value.$unknown[0], value.$unknown[1]);
   }, "visit");
 })(ContentBlock || (ContentBlock = {}));
@@ -870,6 +918,7 @@ var ContentBlockDelta;
     if (value.text !== void 0) return visitor.text(value.text);
     if (value.toolUse !== void 0) return visitor.toolUse(value.toolUse);
     if (value.reasoningContent !== void 0) return visitor.reasoningContent(value.reasoningContent);
+    if (value.citation !== void 0) return visitor.citation(value.citation);
     return visitor._(value.$unknown[0], value.$unknown[1]);
   }, "visit");
 })(ContentBlockDelta || (ContentBlockDelta = {}));
@@ -1044,6 +1093,7 @@ var ContentBlockFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
     return { guardContent: GuardrailConverseContentBlockFilterSensitiveLog(obj.guardContent) };
   if (obj.cachePoint !== void 0) return { cachePoint: obj.cachePoint };
   if (obj.reasoningContent !== void 0) return { reasoningContent: import_smithy_client.SENSITIVE_STRING };
+  if (obj.citationsContent !== void 0) return { citationsContent: obj.citationsContent };
   if (obj.$unknown !== void 0) return { [obj.$unknown[0]]: "UNKNOWN" };
 }, "ContentBlockFilterSensitiveLog");
 var MessageFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
@@ -1091,6 +1141,7 @@ var ContentBlockDeltaFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
   if (obj.text !== void 0) return { text: obj.text };
   if (obj.toolUse !== void 0) return { toolUse: obj.toolUse };
   if (obj.reasoningContent !== void 0) return { reasoningContent: import_smithy_client.SENSITIVE_STRING };
+  if (obj.citation !== void 0) return { citation: obj.citation };
   if (obj.$unknown !== void 0) return { [obj.$unknown[0]]: "UNKNOWN" };
 }, "ContentBlockDeltaFilterSensitiveLog");
 var ContentBlockDeltaEventFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
@@ -1976,6 +2027,7 @@ var se_BidirectionalInputPayloadPart = /* @__PURE__ */ __name((input, context) =
 var se_ContentBlock = /* @__PURE__ */ __name((input, context) => {
   return ContentBlock.visit(input, {
     cachePoint: /* @__PURE__ */ __name((value) => ({ cachePoint: (0, import_smithy_client._json)(value) }), "cachePoint"),
+    citationsContent: /* @__PURE__ */ __name((value) => ({ citationsContent: (0, import_smithy_client._json)(value) }), "citationsContent"),
     document: /* @__PURE__ */ __name((value) => ({ document: se_DocumentBlock(value, context) }), "document"),
     guardContent: /* @__PURE__ */ __name((value) => ({ guardContent: se_GuardrailConverseContentBlock(value, context) }), "guardContent"),
     image: /* @__PURE__ */ __name((value) => ({ image: se_ImageBlock(value, context) }), "image"),
@@ -1994,6 +2046,8 @@ var se_ContentBlocks = /* @__PURE__ */ __name((input, context) => {
 }, "se_ContentBlocks");
 var se_DocumentBlock = /* @__PURE__ */ __name((input, context) => {
   return (0, import_smithy_client.take)(input, {
+    citations: import_smithy_client._json,
+    context: [],
     format: [],
     name: [],
     source: /* @__PURE__ */ __name((_) => se_DocumentSource(_, context), "source")
@@ -2002,7 +2056,9 @@ var se_DocumentBlock = /* @__PURE__ */ __name((input, context) => {
 var se_DocumentSource = /* @__PURE__ */ __name((input, context) => {
   return DocumentSource.visit(input, {
     bytes: /* @__PURE__ */ __name((value) => ({ bytes: context.base64Encoder(value) }), "bytes"),
+    content: /* @__PURE__ */ __name((value) => ({ content: (0, import_smithy_client._json)(value) }), "content"),
     s3Location: /* @__PURE__ */ __name((value) => ({ s3Location: (0, import_smithy_client._json)(value) }), "s3Location"),
+    text: /* @__PURE__ */ __name((value) => ({ text: value }), "text"),
     _: /* @__PURE__ */ __name((name, value) => ({ [name]: value }), "_")
   });
 }, "se_DocumentSource");
@@ -2210,6 +2266,11 @@ var de_ContentBlock = /* @__PURE__ */ __name((output, context) => {
       cachePoint: (0, import_smithy_client._json)(output.cachePoint)
     };
   }
+  if (output.citationsContent != null) {
+    return {
+      citationsContent: (0, import_smithy_client._json)(output.citationsContent)
+    };
+  }
   if (output.document != null) {
     return {
       document: de_DocumentBlock(output.document, context)
@@ -2251,6 +2312,11 @@ var de_ContentBlock = /* @__PURE__ */ __name((output, context) => {
   return { $unknown: Object.entries(output)[0] };
 }, "de_ContentBlock");
 var de_ContentBlockDelta = /* @__PURE__ */ __name((output, context) => {
+  if (output.citation != null) {
+    return {
+      citation: (0, import_smithy_client._json)(output.citation)
+    };
+  }
   if (output.reasoningContent != null) {
     return {
       reasoningContent: de_ReasoningContentBlockDelta((0, import_core2.awsExpectUnion)(output.reasoningContent), context)
@@ -2308,6 +2374,8 @@ var de_ConverseTrace = /* @__PURE__ */ __name((output, context) => {
 }, "de_ConverseTrace");
 var de_DocumentBlock = /* @__PURE__ */ __name((output, context) => {
   return (0, import_smithy_client.take)(output, {
+    citations: import_smithy_client._json,
+    context: import_smithy_client.expectString,
     format: import_smithy_client.expectString,
     name: import_smithy_client.expectString,
     source: /* @__PURE__ */ __name((_) => de_DocumentSource((0, import_core2.awsExpectUnion)(_), context), "source")
@@ -2319,11 +2387,19 @@ var de_DocumentSource = /* @__PURE__ */ __name((output, context) => {
       bytes: context.base64Decoder(output.bytes)
     };
   }
+  if (output.content != null) {
+    return {
+      content: (0, import_smithy_client._json)(output.content)
+    };
+  }
   if (output.s3Location != null) {
     return {
       s3Location: (0, import_smithy_client._json)(output.s3Location)
     };
   }
+  if ((0, import_smithy_client.expectString)(output.text) !== void 0) {
+    return { text: (0, import_smithy_client.expectString)(output.text) };
+  }
   return { $unknown: Object.entries(output)[0] };
 }, "de_DocumentSource");
 var de_GuardrailAssessment = /* @__PURE__ */ __name((output, context) => {
@@ -2657,11 +2733,15 @@ var InvokeModelCommand = class extends import_smithy_client.Command.classBuilder
 
 
 
+
 var InvokeModelWithBidirectionalStreamCommand = class extends import_smithy_client.Command.classBuilder().ep(commonParams).m(function(Command, cs, config, o) {
   return [
     (0, import_middleware_serde.getSerdePlugin)(config, this.serialize, this.deserialize),
     (0, import_middleware_endpoint.getEndpointPlugin)(config, Command.getEndpointParameterInstructions()),
-    (0, import_middleware_eventstream.getEventStreamPlugin)(config)
+    (0, import_middleware_eventstream.getEventStreamPlugin)(config),
+    (0, import_middleware_websocket.getWebSocketPlugin)(config, {
+      headerPrefix: "x-amz-bedrock-"
+    })
   ];
 }).s("AmazonBedrockFrontendService", "InvokeModelWithBidirectionalStream", {
   /**
@@ -2805,7 +2885,11 @@ var paginateListAsyncInvokes = (0, import_core.createPaginator)(BedrockRuntimeCl
   GuardrailManagedWordType,
   GuardrailTrace,
   CachePointType,
+  CitationLocation,
+  CitationSourceContent,
+  CitationGeneratedContent,
   DocumentFormat,
+  DocumentContentBlock,
   DocumentSource,
   GuardrailConverseImageFormat,
   GuardrailConverseImageSource,

package/dist-cjs/runtimeConfig.browser.js

@@ -4,6 +4,7 @@ exports.getRuntimeConfig = void 0;
 const tslib_1 = require("tslib");
 const package_json_1 = tslib_1.__importDefault(require("../package.json"));
 const sha256_browser_1 = require("@aws-crypto/sha256-browser");
+const middleware_websocket_1 = require("@aws-sdk/middleware-websocket");
 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");
@@ -27,12 +28,11 @@ const getRuntimeConfig = (config) => {
         credentialDefaultProvider: config?.credentialDefaultProvider ?? ((_) => () => Promise.reject(new Error("Credential is missing"))),
         defaultUserAgentProvider: config?.defaultUserAgentProvider ??
             (0, util_user_agent_browser_1.createDefaultUserAgentProvider)({ serviceId: clientSharedValues.serviceId, clientVersion: package_json_1.default.version }),
-        eventStreamPayloadHandlerProvider: config?.eventStreamPayloadHandlerProvider ??
-            (() => ({ handle: (0, invalid_dependency_1.invalidFunction)("event stream request is not supported in browser.") })),
+        eventStreamPayloadHandlerProvider: config?.eventStreamPayloadHandlerProvider ?? middleware_websocket_1.eventStreamPayloadHandlerProvider,
         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: fetch_http_handler_1.FetchHttpHandler.create(config?.requestHandler ?? defaultConfigProvider),
+        requestHandler: middleware_websocket_1.WebSocketFetchHandler.create(config?.requestHandler ?? defaultConfigProvider, fetch_http_handler_1.FetchHttpHandler.create(defaultConfigProvider)),
         retryMode: config?.retryMode ?? (async () => (await defaultConfigProvider()).retryMode || util_retry_1.DEFAULT_RETRY_MODE),
         sha256: config?.sha256 ?? sha256_browser_1.Sha256,
         streamCollector: config?.streamCollector ?? fetch_http_handler_1.streamCollector,

package/dist-cjs/runtimeConfig.js

@@ -6,8 +6,10 @@ const package_json_1 = tslib_1.__importDefault(require("../package.json"));
 const core_1 = require("@aws-sdk/core");
 const credential_provider_node_1 = require("@aws-sdk/credential-provider-node");
 const eventstream_handler_node_1 = require("@aws-sdk/eventstream-handler-node");
+const token_providers_1 = require("@aws-sdk/token-providers");
 const util_user_agent_node_1 = require("@aws-sdk/util-user-agent-node");
 const config_resolver_1 = require("@smithy/config-resolver");
+const core_2 = require("@smithy/core");
 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");
@@ -28,6 +30,7 @@ const getRuntimeConfig = (config) => {
     const loaderConfig = {
         profile: config?.profile,
         logger: clientSharedValues.logger,
+        signingName: "bedrock",
     };
     return {
         ...clientSharedValues,
@@ -41,6 +44,26 @@ const getRuntimeConfig = (config) => {
             (0, util_user_agent_node_1.createDefaultUserAgentProvider)({ serviceId: clientSharedValues.serviceId, clientVersion: package_json_1.default.version }),
         eventStreamPayloadHandlerProvider: config?.eventStreamPayloadHandlerProvider ?? eventstream_handler_node_1.eventStreamPayloadHandlerProvider,
         eventStreamSerdeProvider: config?.eventStreamSerdeProvider ?? eventstream_serde_node_1.eventStreamSerdeProvider,
+        httpAuthSchemes: config?.httpAuthSchemes ?? [
+            {
+                schemeId: "aws.auth#sigv4",
+                identityProvider: (ipc) => ipc.getIdentityProvider("aws.auth#sigv4"),
+                signer: new core_1.AwsSdkSigV4Signer(),
+            },
+            {
+                schemeId: "smithy.api#httpBearerAuth",
+                identityProvider: (ipc) => ipc.getIdentityProvider("smithy.api#httpBearerAuth") ||
+                    (async (idProps) => {
+                        try {
+                            return await (0, token_providers_1.fromEnvSigningName)({ signingName: "bedrock" })();
+                        }
+                        catch (error) {
+                            return await (0, token_providers_1.nodeProvider)(idProps)(idProps);
+                        }
+                    }),
+                signer: new core_2.HttpBearerAuthSigner(),
+            },
+        ],
         maxAttempts: config?.maxAttempts ?? (0, node_config_provider_1.loadConfig)(middleware_retry_1.NODE_MAX_ATTEMPT_CONFIG_OPTIONS, config),
         region: config?.region ??
             (0, node_config_provider_1.loadConfig)(config_resolver_1.NODE_REGION_CONFIG_OPTIONS, { ...config_resolver_1.NODE_REGION_CONFIG_FILE_OPTIONS, ...loaderConfig }),

package/dist-cjs/runtimeConfig.shared.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getRuntimeConfig = void 0;
 const core_1 = require("@aws-sdk/core");
+const core_2 = require("@smithy/core");
 const smithy_client_1 = require("@smithy/smithy-client");
 const url_parser_1 = require("@smithy/url-parser");
 const util_base64_1 = require("@smithy/util-base64");
@@ -23,6 +24,11 @@ const getRuntimeConfig = (config) => {
                 identityProvider: (ipc) => ipc.getIdentityProvider("aws.auth#sigv4"),
                 signer: new core_1.AwsSdkSigV4Signer(),
             },
+            {
+                schemeId: "smithy.api#httpBearerAuth",
+                identityProvider: (ipc) => ipc.getIdentityProvider("smithy.api#httpBearerAuth"),
+                signer: new core_2.HttpBearerAuthSigner(),
+            },
         ],
         logger: config?.logger ?? new smithy_client_1.NoOpLogger(),
         serviceId: config?.serviceId ?? "Bedrock Runtime",

package/dist-es/BedrockRuntimeClient.js

@@ -3,6 +3,7 @@ import { getHostHeaderPlugin, resolveHostHeaderConfig, } from "@aws-sdk/middlewa
 import { getLoggerPlugin } from "@aws-sdk/middleware-logger";
 import { getRecursionDetectionPlugin } from "@aws-sdk/middleware-recursion-detection";
 import { getUserAgentPlugin, resolveUserAgentConfig, } from "@aws-sdk/middleware-user-agent";
+import { resolveWebSocketConfig } from "@aws-sdk/middleware-websocket";
 import { resolveRegionConfig } from "@smithy/config-resolver";
 import { DefaultIdentityProviderConfig, getHttpAuthSchemeEndpointRuleSetPlugin, getHttpSigningPlugin, } from "@smithy/core";
 import { resolveEventStreamSerdeConfig, } from "@smithy/eventstream-serde-config-resolver";
@@ -30,8 +31,9 @@ export class BedrockRuntimeClient extends __Client {
         const _config_7 = resolveEventStreamSerdeConfig(_config_6);
         const _config_8 = resolveHttpAuthSchemeConfig(_config_7);
         const _config_9 = resolveEventStreamConfig(_config_8);
-        const _config_10 = resolveRuntimeExtensions(_config_9, configuration?.extensions || []);
-        this.config = _config_10;
+        const _config_10 = resolveWebSocketConfig(_config_9);
+        const _config_11 = resolveRuntimeExtensions(_config_10, configuration?.extensions || []);
+        this.config = _config_11;
         this.middlewareStack.use(getUserAgentPlugin(this.config));
         this.middlewareStack.use(getRetryPlugin(this.config));
         this.middlewareStack.use(getContentLengthPlugin(this.config));
@@ -42,6 +44,7 @@ export class BedrockRuntimeClient extends __Client {
             httpAuthSchemeParametersProvider: defaultBedrockRuntimeHttpAuthSchemeParametersProvider,
             identityProviderConfigProvider: async (config) => new DefaultIdentityProviderConfig({
                 "aws.auth#sigv4": config.credentials,
+                "smithy.api#httpBearerAuth": config.token,
             }),
         }));
         this.middlewareStack.use(getHttpSigningPlugin(this.config));

package/dist-es/auth/httpAuthExtensionConfiguration.js

@@ -2,6 +2,7 @@ export const getHttpAuthExtensionConfiguration = (runtimeConfig) => {
     const _httpAuthSchemes = runtimeConfig.httpAuthSchemes;
     let _httpAuthSchemeProvider = runtimeConfig.httpAuthSchemeProvider;
     let _credentials = runtimeConfig.credentials;
+    let _token = runtimeConfig.token;
     return {
         setHttpAuthScheme(httpAuthScheme) {
             const index = _httpAuthSchemes.findIndex((scheme) => scheme.schemeId === httpAuthScheme.schemeId);
@@ -27,6 +28,12 @@ export const getHttpAuthExtensionConfiguration = (runtimeConfig) => {
         credentials() {
             return _credentials;
         },
+        setToken(token) {
+            _token = token;
+        },
+        token() {
+            return _token;
+        },
     };
 };
 export const resolveHttpAuthRuntimeConfig = (config) => {
@@ -34,5 +41,6 @@ export const resolveHttpAuthRuntimeConfig = (config) => {
         httpAuthSchemes: config.httpAuthSchemes(),
         httpAuthSchemeProvider: config.httpAuthSchemeProvider(),
         credentials: config.credentials(),
+        token: config.token(),
     };
 };

package/dist-es/auth/httpAuthSchemeProvider.js

@@ -1,4 +1,5 @@
 import { resolveAwsSdkSigV4Config, } from "@aws-sdk/core";
+import { doesIdentityRequireRefresh, isIdentityExpired, memoizeIdentityProvider } from "@smithy/core";
 import { getSmithyContext, normalizeProvider } from "@smithy/util-middleware";
 export const defaultBedrockRuntimeHttpAuthSchemeParametersProvider = async (config, context, input) => {
     return {
@@ -24,18 +25,34 @@ function createAwsAuthSigv4HttpAuthOption(authParameters) {
         }),
     };
 }
+function createSmithyApiHttpBearerAuthHttpAuthOption(authParameters) {
+    return {
+        schemeId: "smithy.api#httpBearerAuth",
+        propertiesExtractor: ({ profile, filepath, configFilepath, ignoreCache }, context) => ({
+            identityProperties: {
+                profile,
+                filepath,
+                configFilepath,
+                ignoreCache,
+            },
+        }),
+    };
+}
 export const defaultBedrockRuntimeHttpAuthSchemeProvider = (authParameters) => {
     const options = [];
     switch (authParameters.operation) {
         default: {
             options.push(createAwsAuthSigv4HttpAuthOption(authParameters));
+            options.push(createSmithyApiHttpBearerAuthHttpAuthOption(authParameters));
         }
     }
     return options;
 };
 export const resolveHttpAuthSchemeConfig = (config) => {
+    const token = memoizeIdentityProvider(config.token, isIdentityExpired, doesIdentityRequireRefresh);
     const config_0 = resolveAwsSdkSigV4Config(config);
     return Object.assign(config_0, {
         authSchemePreference: normalizeProvider(config.authSchemePreference ?? []),
+        token,
     });
 };

package/dist-es/commands/InvokeModelWithBidirectionalStreamCommand.js

@@ -1,4 +1,5 @@
 import { getEventStreamPlugin } from "@aws-sdk/middleware-eventstream";
+import { getWebSocketPlugin } from "@aws-sdk/middleware-websocket";
 import { getEndpointPlugin } from "@smithy/middleware-endpoint";
 import { getSerdePlugin } from "@smithy/middleware-serde";
 import { Command as $Command } from "@smithy/smithy-client";
@@ -14,6 +15,9 @@ export class InvokeModelWithBidirectionalStreamCommand extends $Command
         getSerdePlugin(config, this.serialize, this.deserialize),
         getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
         getEventStreamPlugin(config),
+        getWebSocketPlugin(config, {
+            headerPrefix: "x-amz-bedrock-",
+        }),
     ];
 })
     .s("AmazonBedrockFrontendService", "InvokeModelWithBidirectionalStream", {

package/dist-es/models/models_0.js

@@ -247,6 +247,34 @@ export const GuardrailTrace = {
 export const CachePointType = {
     DEFAULT: "default",
 };
+export var CitationLocation;
+(function (CitationLocation) {
+    CitationLocation.visit = (value, visitor) => {
+        if (value.documentChar !== undefined)
+            return visitor.documentChar(value.documentChar);
+        if (value.documentPage !== undefined)
+            return visitor.documentPage(value.documentPage);
+        if (value.documentChunk !== undefined)
+            return visitor.documentChunk(value.documentChunk);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(CitationLocation || (CitationLocation = {}));
+export var CitationSourceContent;
+(function (CitationSourceContent) {
+    CitationSourceContent.visit = (value, visitor) => {
+        if (value.text !== undefined)
+            return visitor.text(value.text);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(CitationSourceContent || (CitationSourceContent = {}));
+export var CitationGeneratedContent;
+(function (CitationGeneratedContent) {
+    CitationGeneratedContent.visit = (value, visitor) => {
+        if (value.text !== undefined)
+            return visitor.text(value.text);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(CitationGeneratedContent || (CitationGeneratedContent = {}));
 export const DocumentFormat = {
     CSV: "csv",
     DOC: "doc",
@@ -258,6 +286,14 @@ export const DocumentFormat = {
     XLS: "xls",
     XLSX: "xlsx",
 };
+export var DocumentContentBlock;
+(function (DocumentContentBlock) {
+    DocumentContentBlock.visit = (value, visitor) => {
+        if (value.text !== undefined)
+            return visitor.text(value.text);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(DocumentContentBlock || (DocumentContentBlock = {}));
 export var DocumentSource;
 (function (DocumentSource) {
     DocumentSource.visit = (value, visitor) => {
@@ -265,6 +301,10 @@ export var DocumentSource;
             return visitor.bytes(value.bytes);
         if (value.s3Location !== undefined)
             return visitor.s3Location(value.s3Location);
+        if (value.text !== undefined)
+            return visitor.text(value.text);
+        if (value.content !== undefined)
+            return visitor.content(value.content);
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(DocumentSource || (DocumentSource = {}));
@@ -383,6 +423,8 @@ export var ContentBlock;
             return visitor.cachePoint(value.cachePoint);
         if (value.reasoningContent !== undefined)
             return visitor.reasoningContent(value.reasoningContent);
+        if (value.citationsContent !== undefined)
+            return visitor.citationsContent(value.citationsContent);
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(ContentBlock || (ContentBlock = {}));
@@ -526,6 +568,8 @@ export var ContentBlockDelta;
             return visitor.toolUse(value.toolUse);
         if (value.reasoningContent !== undefined)
             return visitor.reasoningContent(value.reasoningContent);
+        if (value.citation !== undefined)
+            return visitor.citation(value.citation);
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(ContentBlockDelta || (ContentBlockDelta = {}));
@@ -725,6 +769,8 @@ export const ContentBlockFilterSensitiveLog = (obj) => {
         return { cachePoint: obj.cachePoint };
     if (obj.reasoningContent !== undefined)
         return { reasoningContent: SENSITIVE_STRING };
+    if (obj.citationsContent !== undefined)
+        return { citationsContent: obj.citationsContent };
     if (obj.$unknown !== undefined)
         return { [obj.$unknown[0]]: "UNKNOWN" };
 };
@@ -785,6 +831,8 @@ export const ContentBlockDeltaFilterSensitiveLog = (obj) => {
         return { toolUse: obj.toolUse };
     if (obj.reasoningContent !== undefined)
         return { reasoningContent: SENSITIVE_STRING };
+    if (obj.citation !== undefined)
+        return { citation: obj.citation };
     if (obj.$unknown !== undefined)
         return { [obj.$unknown[0]]: "UNKNOWN" };
 };