@aws-sdk/client-bedrock-agent-runtime 3.701.0 → 3.703.0

package/dist-cjs/index.js

@@ -33,6 +33,7 @@ __export(src_exports, {
   AnalyzePromptEventFilterSensitiveLog: () => AnalyzePromptEventFilterSensitiveLog,
   ApiInvocationInputFilterSensitiveLog: () => ApiInvocationInputFilterSensitiveLog,
   ApiResultFilterSensitiveLog: () => ApiResultFilterSensitiveLog,
+  AttributeType: () => AttributeType,
   AttributionFilterSensitiveLog: () => AttributionFilterSensitiveLog,
   BadGatewayException: () => BadGatewayException,
   BedrockAgentRuntime: () => BedrockAgentRuntime,
@@ -40,6 +41,7 @@ __export(src_exports, {
   BedrockAgentRuntimeServiceException: () => BedrockAgentRuntimeServiceException,
   ByteContentDocFilterSensitiveLog: () => ByteContentDocFilterSensitiveLog,
   ByteContentFileFilterSensitiveLog: () => ByteContentFileFilterSensitiveLog,
+  CitationEventFilterSensitiveLog: () => CitationEventFilterSensitiveLog,
   CitationFilterSensitiveLog: () => CitationFilterSensitiveLog,
   ConfirmationState: () => ConfirmationState,
   ConflictException: () => ConflictException,
@@ -110,6 +112,7 @@ __export(src_exports, {
   GuardrailTraceFilterSensitiveLog: () => GuardrailTraceFilterSensitiveLog,
   GuardrailWordPolicyAction: () => GuardrailWordPolicyAction,
   GuardrailWordPolicyAssessmentFilterSensitiveLog: () => GuardrailWordPolicyAssessmentFilterSensitiveLog,
+  ImplicitFilterConfigurationFilterSensitiveLog: () => ImplicitFilterConfigurationFilterSensitiveLog,
   InlineAgentFilePartFilterSensitiveLog: () => InlineAgentFilePartFilterSensitiveLog,
   InlineAgentPayloadPartFilterSensitiveLog: () => InlineAgentPayloadPartFilterSensitiveLog,
   InlineAgentResponseStream: () => InlineAgentResponseStream,
@@ -147,6 +150,8 @@ __export(src_exports, {
   KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog: () => KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog,
   Memory: () => Memory,
   MemoryType: () => MemoryType,
+  MetadataAttributeSchemaFilterSensitiveLog: () => MetadataAttributeSchemaFilterSensitiveLog,
+  MetadataConfigurationForRerankingFilterSensitiveLog: () => MetadataConfigurationForRerankingFilterSensitiveLog,
   MetadataFilterSensitiveLog: () => MetadataFilterSensitiveLog,
   ModelInvocationInputFilterSensitiveLog: () => ModelInvocationInputFilterSensitiveLog,
   NodeType: () => NodeType,
@@ -184,6 +189,21 @@ __export(src_exports, {
   RawResponseFilterSensitiveLog: () => RawResponseFilterSensitiveLog,
   RepromptResponseFilterSensitiveLog: () => RepromptResponseFilterSensitiveLog,
   RequireConfirmation: () => RequireConfirmation,
+  RerankCommand: () => RerankCommand,
+  RerankDocumentFilterSensitiveLog: () => RerankDocumentFilterSensitiveLog,
+  RerankDocumentType: () => RerankDocumentType,
+  RerankQueryContentType: () => RerankQueryContentType,
+  RerankQueryFilterSensitiveLog: () => RerankQueryFilterSensitiveLog,
+  RerankRequestFilterSensitiveLog: () => RerankRequestFilterSensitiveLog,
+  RerankResponseFilterSensitiveLog: () => RerankResponseFilterSensitiveLog,
+  RerankResultFilterSensitiveLog: () => RerankResultFilterSensitiveLog,
+  RerankSourceFilterSensitiveLog: () => RerankSourceFilterSensitiveLog,
+  RerankSourceType: () => RerankSourceType,
+  RerankTextDocumentFilterSensitiveLog: () => RerankTextDocumentFilterSensitiveLog,
+  RerankingConfigurationType: () => RerankingConfigurationType,
+  RerankingMetadataSelectionMode: () => RerankingMetadataSelectionMode,
+  RerankingMetadataSelectiveModeConfiguration: () => RerankingMetadataSelectiveModeConfiguration,
+  RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog: () => RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog,
   ResourceNotFoundException: () => ResourceNotFoundException,
   ResponseState: () => ResponseState,
   ResponseStream: () => ResponseStream,
@@ -196,9 +216,15 @@ __export(src_exports, {
   RetrieveAndGenerateCommand: () => RetrieveAndGenerateCommand,
   RetrieveAndGenerateConfigurationFilterSensitiveLog: () => RetrieveAndGenerateConfigurationFilterSensitiveLog,
   RetrieveAndGenerateInputFilterSensitiveLog: () => RetrieveAndGenerateInputFilterSensitiveLog,
+  RetrieveAndGenerateOutputEventFilterSensitiveLog: () => RetrieveAndGenerateOutputEventFilterSensitiveLog,
   RetrieveAndGenerateOutputFilterSensitiveLog: () => RetrieveAndGenerateOutputFilterSensitiveLog,
   RetrieveAndGenerateRequestFilterSensitiveLog: () => RetrieveAndGenerateRequestFilterSensitiveLog,
   RetrieveAndGenerateResponseFilterSensitiveLog: () => RetrieveAndGenerateResponseFilterSensitiveLog,
+  RetrieveAndGenerateStreamCommand: () => RetrieveAndGenerateStreamCommand,
+  RetrieveAndGenerateStreamRequestFilterSensitiveLog: () => RetrieveAndGenerateStreamRequestFilterSensitiveLog,
+  RetrieveAndGenerateStreamResponseFilterSensitiveLog: () => RetrieveAndGenerateStreamResponseFilterSensitiveLog,
+  RetrieveAndGenerateStreamResponseOutput: () => RetrieveAndGenerateStreamResponseOutput,
+  RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog: () => RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog,
   RetrieveAndGenerateType: () => RetrieveAndGenerateType,
   RetrieveCommand: () => RetrieveCommand,
   RetrieveRequestFilterSensitiveLog: () => RetrieveRequestFilterSensitiveLog,
@@ -218,8 +244,12 @@ __export(src_exports, {
   Type: () => Type,
   UsageFilterSensitiveLog: () => UsageFilterSensitiveLog,
   ValidationException: () => ValidationException,
+  VectorSearchBedrockRerankingConfigurationFilterSensitiveLog: () => VectorSearchBedrockRerankingConfigurationFilterSensitiveLog,
+  VectorSearchRerankingConfigurationFilterSensitiveLog: () => VectorSearchRerankingConfigurationFilterSensitiveLog,
+  VectorSearchRerankingConfigurationType: () => VectorSearchRerankingConfigurationType,
   __Client: () => import_smithy_client.Client,
   paginateGetAgentMemory: () => paginateGetAgentMemory,
+  paginateRerank: () => paginateRerank,
   paginateRetrieve: () => paginateRetrieve
 });
 module.exports = __toCommonJS(src_exports);
@@ -696,10 +726,33 @@ var FileUseCase = {
   CHAT: "CHAT",
   CODE_INTERPRETER: "CODE_INTERPRETER"
 };
+var AttributeType = {
+  BOOLEAN: "BOOLEAN",
+  NUMBER: "NUMBER",
+  STRING: "STRING",
+  STRING_LIST: "STRING_LIST"
+};
 var SearchType = {
   HYBRID: "HYBRID",
   SEMANTIC: "SEMANTIC"
 };
+var RerankingMetadataSelectionMode = {
+  ALL: "ALL",
+  SELECTIVE: "SELECTIVE"
+};
+var RerankingMetadataSelectiveModeConfiguration;
+((RerankingMetadataSelectiveModeConfiguration3) => {
+  RerankingMetadataSelectiveModeConfiguration3.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.fieldsToInclude !== void 0)
+      return visitor.fieldsToInclude(value.fieldsToInclude);
+    if (value.fieldsToExclude !== void 0)
+      return visitor.fieldsToExclude(value.fieldsToExclude);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(RerankingMetadataSelectiveModeConfiguration || (RerankingMetadataSelectiveModeConfiguration = {}));
+var VectorSearchRerankingConfigurationType = {
+  BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL"
+};
 var ConfirmationState = {
   CONFIRM: "CONFIRM",
   DENY: "DENY"
@@ -720,6 +773,7 @@ var InvocationResultMember;
 })(InvocationResultMember || (InvocationResultMember = {}));
 var RetrievalResultLocationType = {
   CONFLUENCE: "CONFLUENCE",
+  CUSTOM: "CUSTOM",
   S3: "S3",
   SALESFORCE: "SALESFORCE",
   SHAREPOINT: "SHAREPOINT",
@@ -1005,6 +1059,19 @@ var OptimizedPromptStream;
     return visitor._(value.$unknown[0], value.$unknown[1]);
   }, "visit");
 })(OptimizedPromptStream || (OptimizedPromptStream = {}));
+var RerankQueryContentType = {
+  TEXT: "TEXT"
+};
+var RerankingConfigurationType = {
+  BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL"
+};
+var RerankDocumentType = {
+  JSON: "JSON",
+  TEXT: "TEXT"
+};
+var RerankSourceType = {
+  INLINE: "INLINE"
+};
 var ExternalSourceType = {
   BYTE_CONTENT: "BYTE_CONTENT",
   S3: "S3"
@@ -1020,6 +1087,36 @@ var GuadrailAction = {
   INTERVENED: "INTERVENED",
   NONE: "NONE"
 };
+var RetrieveAndGenerateStreamResponseOutput;
+((RetrieveAndGenerateStreamResponseOutput3) => {
+  RetrieveAndGenerateStreamResponseOutput3.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.output !== void 0)
+      return visitor.output(value.output);
+    if (value.citation !== void 0)
+      return visitor.citation(value.citation);
+    if (value.guardrail !== void 0)
+      return visitor.guardrail(value.guardrail);
+    if (value.internalServerException !== void 0)
+      return visitor.internalServerException(value.internalServerException);
+    if (value.validationException !== void 0)
+      return visitor.validationException(value.validationException);
+    if (value.resourceNotFoundException !== void 0)
+      return visitor.resourceNotFoundException(value.resourceNotFoundException);
+    if (value.serviceQuotaExceededException !== void 0)
+      return visitor.serviceQuotaExceededException(value.serviceQuotaExceededException);
+    if (value.throttlingException !== void 0)
+      return visitor.throttlingException(value.throttlingException);
+    if (value.accessDeniedException !== void 0)
+      return visitor.accessDeniedException(value.accessDeniedException);
+    if (value.conflictException !== void 0)
+      return visitor.conflictException(value.conflictException);
+    if (value.dependencyFailedException !== void 0)
+      return visitor.dependencyFailedException(value.dependencyFailedException);
+    if (value.badGatewayException !== void 0)
+      return visitor.badGatewayException(value.badGatewayException);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(RetrieveAndGenerateStreamResponseOutput || (RetrieveAndGenerateStreamResponseOutput = {}));
 var RetrievalFilter;
 ((RetrievalFilter2) => {
   RetrievalFilter2.visit = /* @__PURE__ */ __name((value, visitor) => {
@@ -1197,6 +1294,43 @@ var InputFileFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.source && { source: FileSourceFilterSensitiveLog(obj.source) }
 }), "InputFileFilterSensitiveLog");
+var MetadataAttributeSchemaFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj
+}), "MetadataAttributeSchemaFilterSensitiveLog");
+var ImplicitFilterConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.metadataAttributes && { metadataAttributes: import_smithy_client.SENSITIVE_STRING }
+}), "ImplicitFilterConfigurationFilterSensitiveLog");
+var RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
+  if (obj.fieldsToInclude !== void 0)
+    return { fieldsToInclude: import_smithy_client.SENSITIVE_STRING };
+  if (obj.fieldsToExclude !== void 0)
+    return { fieldsToExclude: import_smithy_client.SENSITIVE_STRING };
+  if (obj.$unknown !== void 0)
+    return { [obj.$unknown[0]]: "UNKNOWN" };
+}, "RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog");
+var MetadataConfigurationForRerankingFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.selectiveModeConfiguration && {
+    selectiveModeConfiguration: RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog(
+      obj.selectiveModeConfiguration
+    )
+  }
+}), "MetadataConfigurationForRerankingFilterSensitiveLog");
+var VectorSearchBedrockRerankingConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.metadataConfiguration && {
+    metadataConfiguration: MetadataConfigurationForRerankingFilterSensitiveLog(obj.metadataConfiguration)
+  }
+}), "VectorSearchBedrockRerankingConfigurationFilterSensitiveLog");
+var VectorSearchRerankingConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.bedrockRerankingConfiguration && {
+    bedrockRerankingConfiguration: VectorSearchBedrockRerankingConfigurationFilterSensitiveLog(
+      obj.bedrockRerankingConfiguration
+    )
+  }
+}), "VectorSearchRerankingConfigurationFilterSensitiveLog");
 var ApiResultFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.apiPath && { apiPath: import_smithy_client.SENSITIVE_STRING }
@@ -1619,6 +1753,34 @@ var OptimizePromptResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => (
   ...obj,
   ...obj.optimizedPrompt && { optimizedPrompt: "STREAMING_CONTENT" }
 }), "OptimizePromptResponseFilterSensitiveLog");
+var RerankTextDocumentFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj
+}), "RerankTextDocumentFilterSensitiveLog");
+var RerankQueryFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.textQuery && { textQuery: import_smithy_client.SENSITIVE_STRING }
+}), "RerankQueryFilterSensitiveLog");
+var RerankDocumentFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.textDocument && { textDocument: import_smithy_client.SENSITIVE_STRING }
+}), "RerankDocumentFilterSensitiveLog");
+var RerankSourceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.inlineDocumentSource && { inlineDocumentSource: import_smithy_client.SENSITIVE_STRING }
+}), "RerankSourceFilterSensitiveLog");
+var RerankRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.queries && { queries: import_smithy_client.SENSITIVE_STRING },
+  ...obj.sources && { sources: import_smithy_client.SENSITIVE_STRING }
+}), "RerankRequestFilterSensitiveLog");
+var RerankResultFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.document && { document: import_smithy_client.SENSITIVE_STRING }
+}), "RerankResultFilterSensitiveLog");
+var RerankResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.results && { results: obj.results.map((item) => RerankResultFilterSensitiveLog(item)) }
+}), "RerankResponseFilterSensitiveLog");
 var RetrieveAndGenerateInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj
 }), "RetrieveAndGenerateInputFilterSensitiveLog");
@@ -1662,6 +1824,45 @@ var RetrieveAndGenerateResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj)
   ...obj.output && { output: import_smithy_client.SENSITIVE_STRING },
   ...obj.citations && { citations: obj.citations.map((item) => CitationFilterSensitiveLog(item)) }
 }), "RetrieveAndGenerateResponseFilterSensitiveLog");
+var CitationEventFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.citation && { citation: CitationFilterSensitiveLog(obj.citation) }
+}), "CitationEventFilterSensitiveLog");
+var RetrieveAndGenerateOutputEventFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj
+}), "RetrieveAndGenerateOutputEventFilterSensitiveLog");
+var RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
+  if (obj.output !== void 0)
+    return { output: import_smithy_client.SENSITIVE_STRING };
+  if (obj.citation !== void 0)
+    return { citation: CitationEventFilterSensitiveLog(obj.citation) };
+  if (obj.guardrail !== void 0)
+    return { guardrail: obj.guardrail };
+  if (obj.internalServerException !== void 0)
+    return { internalServerException: obj.internalServerException };
+  if (obj.validationException !== void 0)
+    return { validationException: obj.validationException };
+  if (obj.resourceNotFoundException !== void 0)
+    return { resourceNotFoundException: obj.resourceNotFoundException };
+  if (obj.serviceQuotaExceededException !== void 0)
+    return { serviceQuotaExceededException: obj.serviceQuotaExceededException };
+  if (obj.throttlingException !== void 0)
+    return { throttlingException: obj.throttlingException };
+  if (obj.accessDeniedException !== void 0)
+    return { accessDeniedException: obj.accessDeniedException };
+  if (obj.conflictException !== void 0)
+    return { conflictException: obj.conflictException };
+  if (obj.dependencyFailedException !== void 0)
+    return { dependencyFailedException: obj.dependencyFailedException };
+  if (obj.badGatewayException !== void 0)
+    return { badGatewayException: obj.badGatewayException };
+  if (obj.$unknown !== void 0)
+    return { [obj.$unknown[0]]: "UNKNOWN" };
+}, "RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog");
+var RetrieveAndGenerateStreamResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.stream && { stream: "STREAMING_CONTENT" }
+}), "RetrieveAndGenerateStreamResponseFilterSensitiveLog");
 var KnowledgeBaseQueryFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj
 }), "KnowledgeBaseQueryFilterSensitiveLog");
@@ -1707,7 +1908,13 @@ var RetrievalFilterFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
 }, "RetrievalFilterFilterSensitiveLog");
 var KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
-  ...obj.filter && { filter: import_smithy_client.SENSITIVE_STRING }
+  ...obj.filter && { filter: import_smithy_client.SENSITIVE_STRING },
+  ...obj.rerankingConfiguration && {
+    rerankingConfiguration: VectorSearchRerankingConfigurationFilterSensitiveLog(obj.rerankingConfiguration)
+  },
+  ...obj.implicitFilterConfiguration && {
+    implicitFilterConfiguration: ImplicitFilterConfigurationFilterSensitiveLog(obj.implicitFilterConfiguration)
+  }
 }), "KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog");
 var KnowledgeBaseRetrievalConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
@@ -1780,6 +1987,15 @@ var RetrieveAndGenerateRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj)
     )
   }
 }), "RetrieveAndGenerateRequestFilterSensitiveLog");
+var RetrieveAndGenerateStreamRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.input && { input: import_smithy_client.SENSITIVE_STRING },
+  ...obj.retrieveAndGenerateConfiguration && {
+    retrieveAndGenerateConfiguration: RetrieveAndGenerateConfigurationFilterSensitiveLog(
+      obj.retrieveAndGenerateConfiguration
+    )
+  }
+}), "RetrieveAndGenerateStreamRequestFilterSensitiveLog");
 var SessionStateFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.returnControlInvocationResults && {
@@ -1914,6 +2130,24 @@ var se_OptimizePromptCommand = /* @__PURE__ */ __name(async (input, context) =>
   b.m("POST").h(headers).b(body);
   return b.build();
 }, "se_OptimizePromptCommand");
+var se_RerankCommand = /* @__PURE__ */ __name(async (input, context) => {
+  const b = (0, import_core.requestBuilder)(input, context);
+  const headers = {
+    "content-type": "application/json"
+  };
+  b.bp("/rerank");
+  let body;
+  body = JSON.stringify(
+    (0, import_smithy_client.take)(input, {
+      nextToken: [],
+      queries: (_) => (0, import_smithy_client._json)(_),
+      rerankingConfiguration: (_) => se_RerankingConfiguration(_, context),
+      sources: (_) => se_RerankSourcesList(_, context)
+    })
+  );
+  b.m("POST").h(headers).b(body);
+  return b.build();
+}, "se_RerankCommand");
 var se_RetrieveCommand = /* @__PURE__ */ __name(async (input, context) => {
   const b = (0, import_core.requestBuilder)(input, context);
   const headers = {
@@ -1924,6 +2158,7 @@ var se_RetrieveCommand = /* @__PURE__ */ __name(async (input, context) => {
   let body;
   body = JSON.stringify(
     (0, import_smithy_client.take)(input, {
+      guardrailConfiguration: (_) => (0, import_smithy_client._json)(_),
       nextToken: [],
       retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context),
       retrievalQuery: (_) => (0, import_smithy_client._json)(_)
@@ -1950,6 +2185,24 @@ var se_RetrieveAndGenerateCommand = /* @__PURE__ */ __name(async (input, context
   b.m("POST").h(headers).b(body);
   return b.build();
 }, "se_RetrieveAndGenerateCommand");
+var se_RetrieveAndGenerateStreamCommand = /* @__PURE__ */ __name(async (input, context) => {
+  const b = (0, import_core.requestBuilder)(input, context);
+  const headers = {
+    "content-type": "application/json"
+  };
+  b.bp("/retrieveAndGenerateStream");
+  let body;
+  body = JSON.stringify(
+    (0, import_smithy_client.take)(input, {
+      input: (_) => (0, import_smithy_client._json)(_),
+      retrieveAndGenerateConfiguration: (_) => se_RetrieveAndGenerateConfiguration(_, context),
+      sessionConfiguration: (_) => (0, import_smithy_client._json)(_),
+      sessionId: []
+    })
+  );
+  b.m("POST").h(headers).b(body);
+  return b.build();
+}, "se_RetrieveAndGenerateStreamCommand");
 var de_DeleteAgentMemoryCommand = /* @__PURE__ */ __name(async (output, context) => {
   if (output.statusCode !== 202 && output.statusCode >= 300) {
     return de_CommandError(output, context);
@@ -2024,6 +2277,21 @@ var de_OptimizePromptCommand = /* @__PURE__ */ __name(async (output, context) =>
   contents.optimizedPrompt = de_OptimizedPromptStream(data, context);
   return contents;
 }, "de_OptimizePromptCommand");
+var de_RerankCommand = /* @__PURE__ */ __name(async (output, context) => {
+  if (output.statusCode !== 200 && output.statusCode >= 300) {
+    return de_CommandError(output, context);
+  }
+  const contents = (0, import_smithy_client.map)({
+    $metadata: deserializeMetadata(output)
+  });
+  const data = (0, import_smithy_client.expectNonNull)((0, import_smithy_client.expectObject)(await (0, import_core2.parseJsonBody)(output.body, context)), "body");
+  const doc = (0, import_smithy_client.take)(data, {
+    nextToken: import_smithy_client.expectString,
+    results: (_) => de_RerankResultsList(_, context)
+  });
+  Object.assign(contents, doc);
+  return contents;
+}, "de_RerankCommand");
 var de_RetrieveCommand = /* @__PURE__ */ __name(async (output, context) => {
   if (output.statusCode !== 200 && output.statusCode >= 300) {
     return de_CommandError(output, context);
@@ -2033,6 +2301,7 @@ var de_RetrieveCommand = /* @__PURE__ */ __name(async (output, context) => {
   });
   const data = (0, import_smithy_client.expectNonNull)((0, import_smithy_client.expectObject)(await (0, import_core2.parseJsonBody)(output.body, context)), "body");
   const doc = (0, import_smithy_client.take)(data, {
+    guardrailAction: import_smithy_client.expectString,
     nextToken: import_smithy_client.expectString,
     retrievalResults: (_) => de_KnowledgeBaseRetrievalResults(_, context)
   });
@@ -2056,6 +2325,18 @@ var de_RetrieveAndGenerateCommand = /* @__PURE__ */ __name(async (output, contex
   Object.assign(contents, doc);
   return contents;
 }, "de_RetrieveAndGenerateCommand");
+var de_RetrieveAndGenerateStreamCommand = /* @__PURE__ */ __name(async (output, context) => {
+  if (output.statusCode !== 200 && output.statusCode >= 300) {
+    return de_CommandError(output, context);
+  }
+  const contents = (0, import_smithy_client.map)({
+    $metadata: deserializeMetadata(output),
+    [_sI]: [, output.headers[_xabkbsi]]
+  });
+  const data = output.body;
+  contents.stream = de_RetrieveAndGenerateStreamResponseOutput(data, context);
+  return contents;
+}, "de_RetrieveAndGenerateStreamCommand");
 var de_CommandError = /* @__PURE__ */ __name(async (output, context) => {
   const parsedOutput = {
     ...output,
@@ -2499,6 +2780,80 @@ var de_ResponseStream = /* @__PURE__ */ __name((output, context) => {
     return { $unknown: output };
   });
 }, "de_ResponseStream");
+var de_RetrieveAndGenerateStreamResponseOutput = /* @__PURE__ */ __name((output, context) => {
+  return context.eventStreamMarshaller.deserialize(output, async (event) => {
+    if (event["output"] != null) {
+      return {
+        output: await de_RetrieveAndGenerateOutputEvent_event(event["output"], context)
+      };
+    }
+    if (event["citation"] != null) {
+      return {
+        citation: await de_CitationEvent_event(event["citation"], context)
+      };
+    }
+    if (event["guardrail"] != null) {
+      return {
+        guardrail: await de_GuardrailEvent_event(event["guardrail"], 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 };
+  });
+}, "de_RetrieveAndGenerateStreamResponseOutput");
 var de_AccessDeniedException_event = /* @__PURE__ */ __name(async (output, context) => {
   const parsedOutput = {
     ...output,
@@ -2519,6 +2874,12 @@ var de_BadGatewayException_event = /* @__PURE__ */ __name(async (output, context
   };
   return de_BadGatewayExceptionRes(parsedOutput, context);
 }, "de_BadGatewayException_event");
+var de_CitationEvent_event = /* @__PURE__ */ __name(async (output, context) => {
+  const contents = {};
+  const data = await (0, import_core2.parseJsonBody)(output.body, context);
+  Object.assign(contents, de_CitationEvent(data, context));
+  return contents;
+}, "de_CitationEvent_event");
 var de_ConflictException_event = /* @__PURE__ */ __name(async (output, context) => {
   const parsedOutput = {
     ...output,
@@ -2557,6 +2918,12 @@ var de_FlowTraceEvent_event = /* @__PURE__ */ __name(async (output, context) =>
   Object.assign(contents, de_FlowTraceEvent(data, context));
   return contents;
 }, "de_FlowTraceEvent_event");
+var de_GuardrailEvent_event = /* @__PURE__ */ __name(async (output, context) => {
+  const contents = {};
+  const data = await (0, import_core2.parseJsonBody)(output.body, context);
+  Object.assign(contents, (0, import_smithy_client._json)(data));
+  return contents;
+}, "de_GuardrailEvent_event");
 var de_InlineAgentFilePart_event = /* @__PURE__ */ __name(async (output, context) => {
   const contents = {};
   const data = await (0, import_core2.parseJsonBody)(output.body, context);
@@ -2607,6 +2974,12 @@ var de_ResourceNotFoundException_event = /* @__PURE__ */ __name(async (output, c
   };
   return de_ResourceNotFoundExceptionRes(parsedOutput, context);
 }, "de_ResourceNotFoundException_event");
+var de_RetrieveAndGenerateOutputEvent_event = /* @__PURE__ */ __name(async (output, context) => {
+  const contents = {};
+  const data = await (0, import_core2.parseJsonBody)(output.body, context);
+  Object.assign(contents, (0, import_smithy_client._json)(data));
+  return contents;
+}, "de_RetrieveAndGenerateOutputEvent_event");
 var de_ReturnControlPayload_event = /* @__PURE__ */ __name(async (output, context) => {
   const contents = {};
   const data = await (0, import_core2.parseJsonBody)(output.body, context);
@@ -2652,6 +3025,18 @@ var se_AdditionalModelRequestFields = /* @__PURE__ */ __name((input, context) =>
 var se_AdditionalModelRequestFieldsValue = /* @__PURE__ */ __name((input, context) => {
   return input;
 }, "se_AdditionalModelRequestFieldsValue");
+var se_BedrockRerankingConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    modelConfiguration: (_) => se_BedrockRerankingModelConfiguration(_, context),
+    numberOfResults: []
+  });
+}, "se_BedrockRerankingConfiguration");
+var se_BedrockRerankingModelConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+    modelArn: []
+  });
+}, "se_BedrockRerankingModelConfiguration");
 var se_ByteContentDoc = /* @__PURE__ */ __name((input, context) => {
   return (0, import_smithy_client.take)(input, {
     contentType: [],
@@ -2809,8 +3194,10 @@ var se_KnowledgeBases = /* @__PURE__ */ __name((input, context) => {
 var se_KnowledgeBaseVectorSearchConfiguration = /* @__PURE__ */ __name((input, context) => {
   return (0, import_smithy_client.take)(input, {
     filter: (_) => se_RetrievalFilter(_, context),
+    implicitFilterConfiguration: import_smithy_client._json,
     numberOfResults: [],
-    overrideSearchType: []
+    overrideSearchType: [],
+    rerankingConfiguration: (_) => se_VectorSearchRerankingConfiguration(_, context)
   });
 }, "se_KnowledgeBaseVectorSearchConfiguration");
 var se_OrchestrationConfiguration = /* @__PURE__ */ __name((input, context) => {
@@ -2842,6 +3229,30 @@ var se_PromptOverrideConfiguration = /* @__PURE__ */ __name((input, context) =>
     promptConfigurations: (_) => se_PromptConfigurations(_, context)
   });
 }, "se_PromptOverrideConfiguration");
+var se_RerankDocument = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    jsonDocument: (_) => se_Document(_, context),
+    textDocument: import_smithy_client._json,
+    type: []
+  });
+}, "se_RerankDocument");
+var se_RerankingConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    bedrockRerankingConfiguration: (_) => se_BedrockRerankingConfiguration(_, context),
+    type: []
+  });
+}, "se_RerankingConfiguration");
+var se_RerankSource = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    inlineDocumentSource: (_) => se_RerankDocument(_, context),
+    type: []
+  });
+}, "se_RerankSource");
+var se_RerankSourcesList = /* @__PURE__ */ __name((input, context) => {
+  return input.filter((e) => e != null).map((entry) => {
+    return se_RerankSource(entry, context);
+  });
+}, "se_RerankSourcesList");
 var se_RetrievalFilter = /* @__PURE__ */ __name((input, context) => {
   return RetrievalFilter.visit(input, {
     andAll: (value) => ({ andAll: se_RetrievalFilterList(value, context) }),
@@ -2890,6 +3301,25 @@ var se_TextInferenceConfig = /* @__PURE__ */ __name((input, context) => {
     topP: import_smithy_client.serializeFloat
   });
 }, "se_TextInferenceConfig");
+var se_VectorSearchBedrockRerankingConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    metadataConfiguration: import_smithy_client._json,
+    modelConfiguration: (_) => se_VectorSearchBedrockRerankingModelConfiguration(_, context),
+    numberOfRerankedResults: []
+  });
+}, "se_VectorSearchBedrockRerankingConfiguration");
+var se_VectorSearchBedrockRerankingModelConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+    modelArn: []
+  });
+}, "se_VectorSearchBedrockRerankingModelConfiguration");
+var se_VectorSearchRerankingConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    bedrockRerankingConfiguration: (_) => se_VectorSearchBedrockRerankingConfiguration(_, context),
+    type: []
+  });
+}, "se_VectorSearchRerankingConfiguration");
 var se_Document = /* @__PURE__ */ __name((input, context) => {
   return input;
 }, "se_Document");
@@ -2904,6 +3334,11 @@ var de_Citation = /* @__PURE__ */ __name((output, context) => {
     retrievedReferences: (_) => de_RetrievedReferences(_, context)
   });
 }, "de_Citation");
+var de_CitationEvent = /* @__PURE__ */ __name((output, context) => {
+  return (0, import_smithy_client.take)(output, {
+    citation: (_) => de_Citation(_, context)
+  });
+}, "de_CitationEvent");
 var de_Citations = /* @__PURE__ */ __name((output, context) => {
   const retVal = (output || []).filter((e) => e != null).map((entry) => {
     return de_Citation(entry, context);
@@ -3177,6 +3612,26 @@ var de_PreProcessingTrace = /* @__PURE__ */ __name((output, context) => {
   }
   return { $unknown: Object.entries(output)[0] };
 }, "de_PreProcessingTrace");
+var de_RerankDocument = /* @__PURE__ */ __name((output, context) => {
+  return (0, import_smithy_client.take)(output, {
+    jsonDocument: (_) => de_Document(_, context),
+    textDocument: import_smithy_client._json,
+    type: import_smithy_client.expectString
+  });
+}, "de_RerankDocument");
+var de_RerankResult = /* @__PURE__ */ __name((output, context) => {
+  return (0, import_smithy_client.take)(output, {
+    document: (_) => de_RerankDocument(_, context),
+    index: import_smithy_client.expectInt32,
+    relevanceScore: import_smithy_client.limitedParseFloat32
+  });
+}, "de_RerankResult");
+var de_RerankResultsList = /* @__PURE__ */ __name((output, context) => {
+  const retVal = (output || []).filter((e) => e != null).map((entry) => {
+    return de_RerankResult(entry, context);
+  });
+  return retVal;
+}, "de_RerankResultsList");
 var de_RetrievalResultMetadata = /* @__PURE__ */ __name((output, context) => {
   return Object.entries(output).reduce((acc, [key, value]) => {
     if (value === null) {
@@ -3262,6 +3717,7 @@ var _sI = "sessionId";
 var _xabact = "x-amzn-bedrock-agent-content-type";
 var _xabami = "x-amz-bedrock-agent-memory-id";
 var _xabasi = "x-amz-bedrock-agent-session-id";
+var _xabkbsi = "x-amzn-bedrock-knowledge-base-session-id";
 
 // src/commands/DeleteAgentMemoryCommand.ts
 var _DeleteAgentMemoryCommand = class _DeleteAgentMemoryCommand extends import_smithy_client.Command.classBuilder().ep(commonParams).m(function(Command, cs, config, o) {
@@ -3372,6 +3828,20 @@ var _OptimizePromptCommand = class _OptimizePromptCommand extends import_smithy_
 __name(_OptimizePromptCommand, "OptimizePromptCommand");
 var OptimizePromptCommand = _OptimizePromptCommand;
 
+// src/commands/RerankCommand.ts
+
+
+
+var _RerankCommand = class _RerankCommand 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())
+  ];
+}).s("AmazonBedrockAgentRunTimeService", "Rerank", {}).n("BedrockAgentRuntimeClient", "RerankCommand").f(RerankRequestFilterSensitiveLog, RerankResponseFilterSensitiveLog).ser(se_RerankCommand).de(de_RerankCommand).build() {
+};
+__name(_RerankCommand, "RerankCommand");
+var RerankCommand = _RerankCommand;
+
 // src/commands/RetrieveAndGenerateCommand.ts
 
 
@@ -3386,6 +3856,27 @@ var _RetrieveAndGenerateCommand = class _RetrieveAndGenerateCommand extends impo
 __name(_RetrieveAndGenerateCommand, "RetrieveAndGenerateCommand");
 var RetrieveAndGenerateCommand = _RetrieveAndGenerateCommand;
 
+// src/commands/RetrieveAndGenerateStreamCommand.ts
+
+
+
+var _RetrieveAndGenerateStreamCommand = class _RetrieveAndGenerateStreamCommand 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())
+  ];
+}).s("AmazonBedrockAgentRunTimeService", "RetrieveAndGenerateStream", {
+  /**
+   * @internal
+   */
+  eventStream: {
+    output: true
+  }
+}).n("BedrockAgentRuntimeClient", "RetrieveAndGenerateStreamCommand").f(RetrieveAndGenerateStreamRequestFilterSensitiveLog, RetrieveAndGenerateStreamResponseFilterSensitiveLog).ser(se_RetrieveAndGenerateStreamCommand).de(de_RetrieveAndGenerateStreamCommand).build() {
+};
+__name(_RetrieveAndGenerateStreamCommand, "RetrieveAndGenerateStreamCommand");
+var RetrieveAndGenerateStreamCommand = _RetrieveAndGenerateStreamCommand;
+
 // src/commands/RetrieveCommand.ts
 
 
@@ -3408,8 +3899,10 @@ var commands = {
   InvokeFlowCommand,
   InvokeInlineAgentCommand,
   OptimizePromptCommand,
+  RerankCommand,
   RetrieveCommand,
-  RetrieveAndGenerateCommand
+  RetrieveAndGenerateCommand,
+  RetrieveAndGenerateStreamCommand
 };
 var _BedrockAgentRuntime = class _BedrockAgentRuntime extends BedrockAgentRuntimeClient {
 };
@@ -3421,6 +3914,10 @@ var BedrockAgentRuntime = _BedrockAgentRuntime;
 
 var paginateGetAgentMemory = (0, import_core.createPaginator)(BedrockAgentRuntimeClient, GetAgentMemoryCommand, "nextToken", "nextToken", "maxItems");
 
+// src/pagination/RerankPaginator.ts
+
+var paginateRerank = (0, import_core.createPaginator)(BedrockAgentRuntimeClient, RerankCommand, "nextToken", "nextToken", "");
+
 // src/pagination/RetrievePaginator.ts
 
 var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClient, RetrieveCommand, "nextToken", "nextToken", "");
@@ -3438,9 +3935,12 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   InvokeFlowCommand,
   InvokeInlineAgentCommand,
   OptimizePromptCommand,
+  RerankCommand,
   RetrieveAndGenerateCommand,
+  RetrieveAndGenerateStreamCommand,
   RetrieveCommand,
   paginateGetAgentMemory,
+  paginateRerank,
   paginateRetrieve,
   AccessDeniedException,
   CustomControlMethod,
@@ -3470,7 +3970,11 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   FlowResponseStream,
   FileSourceType,
   FileUseCase,
+  AttributeType,
   SearchType,
+  RerankingMetadataSelectionMode,
+  RerankingMetadataSelectiveModeConfiguration,
+  VectorSearchRerankingConfigurationType,
   ConfirmationState,
   ResponseState,
   InvocationResultMember,
@@ -3503,10 +4007,15 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   InputPrompt,
   OptimizedPrompt,
   OptimizedPromptStream,
+  RerankQueryContentType,
+  RerankingConfigurationType,
+  RerankDocumentType,
+  RerankSourceType,
   ExternalSourceType,
   QueryTransformationType,
   RetrieveAndGenerateType,
   GuadrailAction,
+  RetrieveAndGenerateStreamResponseOutput,
   RetrievalFilter,
   ActionGroupInvocationInputFilterSensitiveLog,
   ActionGroupInvocationOutputFilterSensitiveLog,
@@ -3533,6 +4042,12 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   ByteContentFileFilterSensitiveLog,
   FileSourceFilterSensitiveLog,
   InputFileFilterSensitiveLog,
+  MetadataAttributeSchemaFilterSensitiveLog,
+  ImplicitFilterConfigurationFilterSensitiveLog,
+  RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog,
+  MetadataConfigurationForRerankingFilterSensitiveLog,
+  VectorSearchBedrockRerankingConfigurationFilterSensitiveLog,
+  VectorSearchRerankingConfigurationFilterSensitiveLog,
   ApiResultFilterSensitiveLog,
   InvocationResultMemberFilterSensitiveLog,
   TextResponsePartFilterSensitiveLog,
@@ -3603,6 +4118,13 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   OptimizedPromptEventFilterSensitiveLog,
   OptimizedPromptStreamFilterSensitiveLog,
   OptimizePromptResponseFilterSensitiveLog,
+  RerankTextDocumentFilterSensitiveLog,
+  RerankQueryFilterSensitiveLog,
+  RerankDocumentFilterSensitiveLog,
+  RerankSourceFilterSensitiveLog,
+  RerankRequestFilterSensitiveLog,
+  RerankResultFilterSensitiveLog,
+  RerankResponseFilterSensitiveLog,
   RetrieveAndGenerateInputFilterSensitiveLog,
   PromptTemplateFilterSensitiveLog,
   ExternalSourcesGenerationConfigurationFilterSensitiveLog,
@@ -3613,6 +4135,10 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   OrchestrationConfigurationFilterSensitiveLog,
   RetrieveAndGenerateOutputFilterSensitiveLog,
   RetrieveAndGenerateResponseFilterSensitiveLog,
+  CitationEventFilterSensitiveLog,
+  RetrieveAndGenerateOutputEventFilterSensitiveLog,
+  RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog,
+  RetrieveAndGenerateStreamResponseFilterSensitiveLog,
   KnowledgeBaseQueryFilterSensitiveLog,
   KnowledgeBaseRetrievalResultFilterSensitiveLog,
   RetrieveResponseFilterSensitiveLog,
@@ -3626,6 +4152,7 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   RetrieveAndGenerateConfigurationFilterSensitiveLog,
   InvokeInlineAgentRequestFilterSensitiveLog,
   RetrieveAndGenerateRequestFilterSensitiveLog,
+  RetrieveAndGenerateStreamRequestFilterSensitiveLog,
   SessionStateFilterSensitiveLog,
   InvokeAgentRequestFilterSensitiveLog
 });