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

package/dist-es/BedrockAgentRuntime.js

@@ -6,7 +6,9 @@ import { InvokeAgentCommand } from "./commands/InvokeAgentCommand";
 import { InvokeFlowCommand } from "./commands/InvokeFlowCommand";
 import { InvokeInlineAgentCommand, } from "./commands/InvokeInlineAgentCommand";
 import { OptimizePromptCommand, } from "./commands/OptimizePromptCommand";
+import { RerankCommand } from "./commands/RerankCommand";
 import { RetrieveAndGenerateCommand, } from "./commands/RetrieveAndGenerateCommand";
+import { RetrieveAndGenerateStreamCommand, } from "./commands/RetrieveAndGenerateStreamCommand";
 import { RetrieveCommand } from "./commands/RetrieveCommand";
 const commands = {
     DeleteAgentMemoryCommand,
@@ -15,8 +17,10 @@ const commands = {
     InvokeFlowCommand,
     InvokeInlineAgentCommand,
     OptimizePromptCommand,
+    RerankCommand,
     RetrieveCommand,
     RetrieveAndGenerateCommand,
+    RetrieveAndGenerateStreamCommand,
 };
 export class BedrockAgentRuntime extends BedrockAgentRuntimeClient {
 }

package/dist-es/commands/RerankCommand.js

@@ -0,0 +1,23 @@
+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 { RerankRequestFilterSensitiveLog, RerankResponseFilterSensitiveLog, } from "../models/models_0";
+import { de_RerankCommand, se_RerankCommand } from "../protocols/Aws_restJson1";
+export { $Command };
+export class RerankCommand extends $Command
+    .classBuilder()
+    .ep(commonParams)
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+        getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
+    ];
+})
+    .s("AmazonBedrockAgentRunTimeService", "Rerank", {})
+    .n("BedrockAgentRuntimeClient", "RerankCommand")
+    .f(RerankRequestFilterSensitiveLog, RerankResponseFilterSensitiveLog)
+    .ser(se_RerankCommand)
+    .de(de_RerankCommand)
+    .build() {
+}

package/dist-es/commands/RetrieveAndGenerateStreamCommand.js

@@ -0,0 +1,27 @@
+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 { RetrieveAndGenerateStreamRequestFilterSensitiveLog, RetrieveAndGenerateStreamResponseFilterSensitiveLog, } from "../models/models_0";
+import { de_RetrieveAndGenerateStreamCommand, se_RetrieveAndGenerateStreamCommand } from "../protocols/Aws_restJson1";
+export { $Command };
+export class RetrieveAndGenerateStreamCommand extends $Command
+    .classBuilder()
+    .ep(commonParams)
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+        getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
+    ];
+})
+    .s("AmazonBedrockAgentRunTimeService", "RetrieveAndGenerateStream", {
+    eventStream: {
+        output: true,
+    },
+})
+    .n("BedrockAgentRuntimeClient", "RetrieveAndGenerateStreamCommand")
+    .f(RetrieveAndGenerateStreamRequestFilterSensitiveLog, RetrieveAndGenerateStreamResponseFilterSensitiveLog)
+    .ser(se_RetrieveAndGenerateStreamCommand)
+    .de(de_RetrieveAndGenerateStreamCommand)
+    .build() {
+}

package/dist-es/commands/index.js

@@ -4,5 +4,7 @@ export * from "./InvokeAgentCommand";
 export * from "./InvokeFlowCommand";
 export * from "./InvokeInlineAgentCommand";
 export * from "./OptimizePromptCommand";
+export * from "./RerankCommand";
 export * from "./RetrieveAndGenerateCommand";
+export * from "./RetrieveAndGenerateStreamCommand";
 export * from "./RetrieveCommand";

package/dist-es/models/models_0.js

@@ -259,10 +259,33 @@ export const FileUseCase = {
     CHAT: "CHAT",
     CODE_INTERPRETER: "CODE_INTERPRETER",
 };
+export const AttributeType = {
+    BOOLEAN: "BOOLEAN",
+    NUMBER: "NUMBER",
+    STRING: "STRING",
+    STRING_LIST: "STRING_LIST",
+};
 export const SearchType = {
     HYBRID: "HYBRID",
     SEMANTIC: "SEMANTIC",
 };
+export const RerankingMetadataSelectionMode = {
+    ALL: "ALL",
+    SELECTIVE: "SELECTIVE",
+};
+export var RerankingMetadataSelectiveModeConfiguration;
+(function (RerankingMetadataSelectiveModeConfiguration) {
+    RerankingMetadataSelectiveModeConfiguration.visit = (value, visitor) => {
+        if (value.fieldsToInclude !== undefined)
+            return visitor.fieldsToInclude(value.fieldsToInclude);
+        if (value.fieldsToExclude !== undefined)
+            return visitor.fieldsToExclude(value.fieldsToExclude);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(RerankingMetadataSelectiveModeConfiguration || (RerankingMetadataSelectiveModeConfiguration = {}));
+export const VectorSearchRerankingConfigurationType = {
+    BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL",
+};
 export const ConfirmationState = {
     CONFIRM: "CONFIRM",
     DENY: "DENY",
@@ -283,6 +306,7 @@ export var InvocationResultMember;
 })(InvocationResultMember || (InvocationResultMember = {}));
 export const RetrievalResultLocationType = {
     CONFLUENCE: "CONFLUENCE",
+    CUSTOM: "CUSTOM",
     S3: "S3",
     SALESFORCE: "SALESFORCE",
     SHAREPOINT: "SHAREPOINT",
@@ -446,6 +470,8 @@ export var Trace;
             return visitor.postProcessingTrace(value.postProcessingTrace);
         if (value.failureTrace !== undefined)
             return visitor.failureTrace(value.failureTrace);
+        if (value.customOrchestrationTrace !== undefined)
+            return visitor.customOrchestrationTrace(value.customOrchestrationTrace);
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(Trace || (Trace = {}));
@@ -566,6 +592,19 @@ export var OptimizedPromptStream;
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(OptimizedPromptStream || (OptimizedPromptStream = {}));
+export const RerankQueryContentType = {
+    TEXT: "TEXT",
+};
+export const RerankingConfigurationType = {
+    BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL",
+};
+export const RerankDocumentType = {
+    JSON: "JSON",
+    TEXT: "TEXT",
+};
+export const RerankSourceType = {
+    INLINE: "INLINE",
+};
 export const ExternalSourceType = {
     BYTE_CONTENT: "BYTE_CONTENT",
     S3: "S3",
@@ -581,6 +620,36 @@ export const GuadrailAction = {
     INTERVENED: "INTERVENED",
     NONE: "NONE",
 };
+export var RetrieveAndGenerateStreamResponseOutput;
+(function (RetrieveAndGenerateStreamResponseOutput) {
+    RetrieveAndGenerateStreamResponseOutput.visit = (value, visitor) => {
+        if (value.output !== undefined)
+            return visitor.output(value.output);
+        if (value.citation !== undefined)
+            return visitor.citation(value.citation);
+        if (value.guardrail !== undefined)
+            return visitor.guardrail(value.guardrail);
+        if (value.internalServerException !== undefined)
+            return visitor.internalServerException(value.internalServerException);
+        if (value.validationException !== undefined)
+            return visitor.validationException(value.validationException);
+        if (value.resourceNotFoundException !== undefined)
+            return visitor.resourceNotFoundException(value.resourceNotFoundException);
+        if (value.serviceQuotaExceededException !== undefined)
+            return visitor.serviceQuotaExceededException(value.serviceQuotaExceededException);
+        if (value.throttlingException !== undefined)
+            return visitor.throttlingException(value.throttlingException);
+        if (value.accessDeniedException !== undefined)
+            return visitor.accessDeniedException(value.accessDeniedException);
+        if (value.conflictException !== undefined)
+            return visitor.conflictException(value.conflictException);
+        if (value.dependencyFailedException !== undefined)
+            return visitor.dependencyFailedException(value.dependencyFailedException);
+        if (value.badGatewayException !== undefined)
+            return visitor.badGatewayException(value.badGatewayException);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(RetrieveAndGenerateStreamResponseOutput || (RetrieveAndGenerateStreamResponseOutput = {}));
 export var RetrievalFilter;
 (function (RetrievalFilter) {
     RetrievalFilter.visit = (value, visitor) => {
@@ -758,6 +827,39 @@ export const InputFileFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.source && { source: FileSourceFilterSensitiveLog(obj.source) }),
 });
+export const MetadataAttributeSchemaFilterSensitiveLog = (obj) => ({
+    ...obj,
+});
+export const ImplicitFilterConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.metadataAttributes && { metadataAttributes: SENSITIVE_STRING }),
+});
+export const RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog = (obj) => {
+    if (obj.fieldsToInclude !== undefined)
+        return { fieldsToInclude: SENSITIVE_STRING };
+    if (obj.fieldsToExclude !== undefined)
+        return { fieldsToExclude: SENSITIVE_STRING };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const MetadataConfigurationForRerankingFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.selectiveModeConfiguration && {
+        selectiveModeConfiguration: RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog(obj.selectiveModeConfiguration),
+    }),
+});
+export const VectorSearchBedrockRerankingConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.metadataConfiguration && {
+        metadataConfiguration: MetadataConfigurationForRerankingFilterSensitiveLog(obj.metadataConfiguration),
+    }),
+});
+export const VectorSearchRerankingConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.bedrockRerankingConfiguration && {
+        bedrockRerankingConfiguration: VectorSearchBedrockRerankingConfigurationFilterSensitiveLog(obj.bedrockRerankingConfiguration),
+    }),
+});
 export const ApiResultFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.apiPath && { apiPath: SENSITIVE_STRING }),
@@ -833,6 +935,13 @@ export const ReturnControlPayloadFilterSensitiveLog = (obj) => ({
         invocationInputs: obj.invocationInputs.map((item) => InvocationInputMemberFilterSensitiveLog(item)),
     }),
 });
+export const CustomOrchestrationTraceEventFilterSensitiveLog = (obj) => ({
+    ...obj,
+});
+export const CustomOrchestrationTraceFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.event && { event: SENSITIVE_STRING }),
+});
 export const FailureTraceFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.failureReason && { failureReason: SENSITIVE_STRING }),
@@ -1008,6 +1117,8 @@ export const TraceFilterSensitiveLog = (obj) => {
         return { postProcessingTrace: SENSITIVE_STRING };
     if (obj.failureTrace !== undefined)
         return { failureTrace: SENSITIVE_STRING };
+    if (obj.customOrchestrationTrace !== undefined)
+        return { customOrchestrationTrace: SENSITIVE_STRING };
     if (obj.$unknown !== undefined)
         return { [obj.$unknown[0]]: "UNKNOWN" };
 };
@@ -1169,6 +1280,34 @@ export const OptimizePromptResponseFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.optimizedPrompt && { optimizedPrompt: "STREAMING_CONTENT" }),
 });
+export const RerankTextDocumentFilterSensitiveLog = (obj) => ({
+    ...obj,
+});
+export const RerankQueryFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.textQuery && { textQuery: SENSITIVE_STRING }),
+});
+export const RerankDocumentFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.textDocument && { textDocument: SENSITIVE_STRING }),
+});
+export const RerankSourceFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.inlineDocumentSource && { inlineDocumentSource: SENSITIVE_STRING }),
+});
+export const RerankRequestFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.queries && { queries: SENSITIVE_STRING }),
+    ...(obj.sources && { sources: SENSITIVE_STRING }),
+});
+export const RerankResultFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.document && { document: SENSITIVE_STRING }),
+});
+export const RerankResponseFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.results && { results: obj.results.map((item) => RerankResultFilterSensitiveLog(item)) }),
+});
 export const RetrieveAndGenerateInputFilterSensitiveLog = (obj) => ({
     ...obj,
 });
@@ -1212,6 +1351,45 @@ export const RetrieveAndGenerateResponseFilterSensitiveLog = (obj) => ({
     ...(obj.output && { output: SENSITIVE_STRING }),
     ...(obj.citations && { citations: obj.citations.map((item) => CitationFilterSensitiveLog(item)) }),
 });
+export const CitationEventFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.citation && { citation: CitationFilterSensitiveLog(obj.citation) }),
+});
+export const RetrieveAndGenerateOutputEventFilterSensitiveLog = (obj) => ({
+    ...obj,
+});
+export const RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog = (obj) => {
+    if (obj.output !== undefined)
+        return { output: SENSITIVE_STRING };
+    if (obj.citation !== undefined)
+        return { citation: CitationEventFilterSensitiveLog(obj.citation) };
+    if (obj.guardrail !== undefined)
+        return { guardrail: obj.guardrail };
+    if (obj.internalServerException !== undefined)
+        return { internalServerException: obj.internalServerException };
+    if (obj.validationException !== undefined)
+        return { validationException: obj.validationException };
+    if (obj.resourceNotFoundException !== undefined)
+        return { resourceNotFoundException: obj.resourceNotFoundException };
+    if (obj.serviceQuotaExceededException !== undefined)
+        return { serviceQuotaExceededException: obj.serviceQuotaExceededException };
+    if (obj.throttlingException !== undefined)
+        return { throttlingException: obj.throttlingException };
+    if (obj.accessDeniedException !== undefined)
+        return { accessDeniedException: obj.accessDeniedException };
+    if (obj.conflictException !== undefined)
+        return { conflictException: obj.conflictException };
+    if (obj.dependencyFailedException !== undefined)
+        return { dependencyFailedException: obj.dependencyFailedException };
+    if (obj.badGatewayException !== undefined)
+        return { badGatewayException: obj.badGatewayException };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const RetrieveAndGenerateStreamResponseFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.stream && { stream: "STREAMING_CONTENT" }),
+});
 export const KnowledgeBaseQueryFilterSensitiveLog = (obj) => ({
     ...obj,
 });
@@ -1258,6 +1436,12 @@ export const RetrievalFilterFilterSensitiveLog = (obj) => {
 export const KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.filter && { filter: SENSITIVE_STRING }),
+    ...(obj.rerankingConfiguration && {
+        rerankingConfiguration: VectorSearchRerankingConfigurationFilterSensitiveLog(obj.rerankingConfiguration),
+    }),
+    ...(obj.implicitFilterConfiguration && {
+        implicitFilterConfiguration: ImplicitFilterConfigurationFilterSensitiveLog(obj.implicitFilterConfiguration),
+    }),
 });
 export const KnowledgeBaseRetrievalConfigurationFilterSensitiveLog = (obj) => ({
     ...obj,
@@ -1324,6 +1508,13 @@ export const RetrieveAndGenerateRequestFilterSensitiveLog = (obj) => ({
         retrieveAndGenerateConfiguration: RetrieveAndGenerateConfigurationFilterSensitiveLog(obj.retrieveAndGenerateConfiguration),
     }),
 });
+export const RetrieveAndGenerateStreamRequestFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.input && { input: SENSITIVE_STRING }),
+    ...(obj.retrieveAndGenerateConfiguration && {
+        retrieveAndGenerateConfiguration: RetrieveAndGenerateConfigurationFilterSensitiveLog(obj.retrieveAndGenerateConfiguration),
+    }),
+});
 export const SessionStateFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.returnControlInvocationResults && {

package/dist-es/pagination/RerankPaginator.js

@@ -0,0 +1,4 @@
+import { createPaginator } from "@smithy/core";
+import { BedrockAgentRuntimeClient } from "../BedrockAgentRuntimeClient";
+import { RerankCommand } from "../commands/RerankCommand";
+export const paginateRerank = createPaginator(BedrockAgentRuntimeClient, RerankCommand, "nextToken", "nextToken", "");

package/dist-es/pagination/index.js

@@ -1,3 +1,4 @@
 export * from "./GetAgentMemoryPaginator";
 export * from "./Interfaces";
+export * from "./RerankPaginator";
 export * from "./RetrievePaginator";