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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (33) hide show
  1. package/README.md +16 -0
  2. package/dist-cjs/index.js +791 -66
  3. package/dist-es/BedrockAgentRuntime.js +4 -0
  4. package/dist-es/commands/RerankCommand.js +23 -0
  5. package/dist-es/commands/RetrieveAndGenerateStreamCommand.js +27 -0
  6. package/dist-es/commands/index.js +2 -0
  7. package/dist-es/models/models_0.js +345 -48
  8. package/dist-es/pagination/RerankPaginator.js +4 -0
  9. package/dist-es/pagination/index.js +1 -0
  10. package/dist-es/protocols/Aws_restJson1.js +277 -3
  11. package/dist-types/BedrockAgentRuntime.d.ts +14 -0
  12. package/dist-types/BedrockAgentRuntimeClient.d.ts +4 -2
  13. package/dist-types/commands/InvokeAgentCommand.d.ts +396 -22
  14. package/dist-types/commands/InvokeInlineAgentCommand.d.ts +376 -21
  15. package/dist-types/commands/RerankCommand.d.ts +143 -0
  16. package/dist-types/commands/RetrieveAndGenerateCommand.d.ts +42 -2
  17. package/dist-types/commands/RetrieveAndGenerateStreamCommand.d.ts +371 -0
  18. package/dist-types/commands/RetrieveCommand.d.ts +46 -1
  19. package/dist-types/commands/index.d.ts +2 -0
  20. package/dist-types/models/models_0.d.ts +1713 -404
  21. package/dist-types/pagination/RerankPaginator.d.ts +7 -0
  22. package/dist-types/pagination/index.d.ts +1 -0
  23. package/dist-types/protocols/Aws_restJson1.d.ts +18 -0
  24. package/dist-types/ts3.4/BedrockAgentRuntime.d.ts +34 -0
  25. package/dist-types/ts3.4/BedrockAgentRuntimeClient.d.ts +12 -0
  26. package/dist-types/ts3.4/commands/RerankCommand.d.ts +41 -0
  27. package/dist-types/ts3.4/commands/RetrieveAndGenerateStreamCommand.d.ts +51 -0
  28. package/dist-types/ts3.4/commands/index.d.ts +2 -0
  29. package/dist-types/ts3.4/models/models_0.d.ts +764 -110
  30. package/dist-types/ts3.4/pagination/RerankPaginator.d.ts +11 -0
  31. package/dist-types/ts3.4/pagination/index.d.ts +1 -0
  32. package/dist-types/ts3.4/protocols/Aws_restJson1.d.ts +24 -0
  33. package/package.json +1 -1
@@ -6,7 +6,9 @@ import { InvokeAgentCommand } from "./commands/InvokeAgentCommand";
6
6
  import { InvokeFlowCommand } from "./commands/InvokeFlowCommand";
7
7
  import { InvokeInlineAgentCommand, } from "./commands/InvokeInlineAgentCommand";
8
8
  import { OptimizePromptCommand, } from "./commands/OptimizePromptCommand";
9
+ import { RerankCommand } from "./commands/RerankCommand";
9
10
  import { RetrieveAndGenerateCommand, } from "./commands/RetrieveAndGenerateCommand";
11
+ import { RetrieveAndGenerateStreamCommand, } from "./commands/RetrieveAndGenerateStreamCommand";
10
12
  import { RetrieveCommand } from "./commands/RetrieveCommand";
11
13
  const commands = {
12
14
  DeleteAgentMemoryCommand,
@@ -15,8 +17,10 @@ const commands = {
15
17
  InvokeFlowCommand,
16
18
  InvokeInlineAgentCommand,
17
19
  OptimizePromptCommand,
20
+ RerankCommand,
18
21
  RetrieveCommand,
19
22
  RetrieveAndGenerateCommand,
23
+ RetrieveAndGenerateStreamCommand,
20
24
  };
21
25
  export class BedrockAgentRuntime extends BedrockAgentRuntimeClient {
22
26
  }
@@ -0,0 +1,23 @@
1
+ import { getEndpointPlugin } from "@smithy/middleware-endpoint";
2
+ import { getSerdePlugin } from "@smithy/middleware-serde";
3
+ import { Command as $Command } from "@smithy/smithy-client";
4
+ import { commonParams } from "../endpoint/EndpointParameters";
5
+ import { RerankRequestFilterSensitiveLog, RerankResponseFilterSensitiveLog, } from "../models/models_0";
6
+ import { de_RerankCommand, se_RerankCommand } from "../protocols/Aws_restJson1";
7
+ export { $Command };
8
+ export class RerankCommand extends $Command
9
+ .classBuilder()
10
+ .ep(commonParams)
11
+ .m(function (Command, cs, config, o) {
12
+ return [
13
+ getSerdePlugin(config, this.serialize, this.deserialize),
14
+ getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
15
+ ];
16
+ })
17
+ .s("AmazonBedrockAgentRunTimeService", "Rerank", {})
18
+ .n("BedrockAgentRuntimeClient", "RerankCommand")
19
+ .f(RerankRequestFilterSensitiveLog, RerankResponseFilterSensitiveLog)
20
+ .ser(se_RerankCommand)
21
+ .de(de_RerankCommand)
22
+ .build() {
23
+ }
@@ -0,0 +1,27 @@
1
+ import { getEndpointPlugin } from "@smithy/middleware-endpoint";
2
+ import { getSerdePlugin } from "@smithy/middleware-serde";
3
+ import { Command as $Command } from "@smithy/smithy-client";
4
+ import { commonParams } from "../endpoint/EndpointParameters";
5
+ import { RetrieveAndGenerateStreamRequestFilterSensitiveLog, RetrieveAndGenerateStreamResponseFilterSensitiveLog, } from "../models/models_0";
6
+ import { de_RetrieveAndGenerateStreamCommand, se_RetrieveAndGenerateStreamCommand } from "../protocols/Aws_restJson1";
7
+ export { $Command };
8
+ export class RetrieveAndGenerateStreamCommand extends $Command
9
+ .classBuilder()
10
+ .ep(commonParams)
11
+ .m(function (Command, cs, config, o) {
12
+ return [
13
+ getSerdePlugin(config, this.serialize, this.deserialize),
14
+ getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
15
+ ];
16
+ })
17
+ .s("AmazonBedrockAgentRunTimeService", "RetrieveAndGenerateStream", {
18
+ eventStream: {
19
+ output: true,
20
+ },
21
+ })
22
+ .n("BedrockAgentRuntimeClient", "RetrieveAndGenerateStreamCommand")
23
+ .f(RetrieveAndGenerateStreamRequestFilterSensitiveLog, RetrieveAndGenerateStreamResponseFilterSensitiveLog)
24
+ .ser(se_RetrieveAndGenerateStreamCommand)
25
+ .de(de_RetrieveAndGenerateStreamCommand)
26
+ .build() {
27
+ }
@@ -4,5 +4,7 @@ export * from "./InvokeAgentCommand";
4
4
  export * from "./InvokeFlowCommand";
5
5
  export * from "./InvokeInlineAgentCommand";
6
6
  export * from "./OptimizePromptCommand";
7
+ export * from "./RerankCommand";
7
8
  export * from "./RetrieveAndGenerateCommand";
9
+ export * from "./RetrieveAndGenerateStreamCommand";
8
10
  export * from "./RetrieveCommand";
@@ -67,6 +67,38 @@ export var FunctionSchema;
67
67
  return visitor._(value.$unknown[0], value.$unknown[1]);
68
68
  };
69
69
  })(FunctionSchema || (FunctionSchema = {}));
70
+ export const ConfirmationState = {
71
+ CONFIRM: "CONFIRM",
72
+ DENY: "DENY",
73
+ };
74
+ export const ResponseState = {
75
+ FAILURE: "FAILURE",
76
+ REPROMPT: "REPROMPT",
77
+ };
78
+ export var InvocationResultMember;
79
+ (function (InvocationResultMember) {
80
+ InvocationResultMember.visit = (value, visitor) => {
81
+ if (value.apiResult !== undefined)
82
+ return visitor.apiResult(value.apiResult);
83
+ if (value.functionResult !== undefined)
84
+ return visitor.functionResult(value.functionResult);
85
+ return visitor._(value.$unknown[0], value.$unknown[1]);
86
+ };
87
+ })(InvocationResultMember || (InvocationResultMember = {}));
88
+ export const PayloadType = {
89
+ RETURN_CONTROL: "RETURN_CONTROL",
90
+ TEXT: "TEXT",
91
+ };
92
+ export var InvocationInputMember;
93
+ (function (InvocationInputMember) {
94
+ InvocationInputMember.visit = (value, visitor) => {
95
+ if (value.apiInvocationInput !== undefined)
96
+ return visitor.apiInvocationInput(value.apiInvocationInput);
97
+ if (value.functionInvocationInput !== undefined)
98
+ return visitor.functionInvocationInput(value.functionInvocationInput);
99
+ return visitor._(value.$unknown[0], value.$unknown[1]);
100
+ };
101
+ })(InvocationInputMember || (InvocationInputMember = {}));
70
102
  export class BadGatewayException extends __BaseException {
71
103
  constructor(opts) {
72
104
  super({
@@ -251,6 +283,18 @@ export var FlowResponseStream;
251
283
  return visitor._(value.$unknown[0], value.$unknown[1]);
252
284
  };
253
285
  })(FlowResponseStream || (FlowResponseStream = {}));
286
+ export var ContentBlock;
287
+ (function (ContentBlock) {
288
+ ContentBlock.visit = (value, visitor) => {
289
+ if (value.text !== undefined)
290
+ return visitor.text(value.text);
291
+ return visitor._(value.$unknown[0], value.$unknown[1]);
292
+ };
293
+ })(ContentBlock || (ContentBlock = {}));
294
+ export const ConversationRole = {
295
+ ASSISTANT: "assistant",
296
+ USER: "user",
297
+ };
254
298
  export const FileSourceType = {
255
299
  BYTE_CONTENT: "BYTE_CONTENT",
256
300
  S3: "S3",
@@ -259,45 +303,49 @@ export const FileUseCase = {
259
303
  CHAT: "CHAT",
260
304
  CODE_INTERPRETER: "CODE_INTERPRETER",
261
305
  };
306
+ export const AttributeType = {
307
+ BOOLEAN: "BOOLEAN",
308
+ NUMBER: "NUMBER",
309
+ STRING: "STRING",
310
+ STRING_LIST: "STRING_LIST",
311
+ };
262
312
  export const SearchType = {
263
313
  HYBRID: "HYBRID",
264
314
  SEMANTIC: "SEMANTIC",
265
315
  };
266
- export const ConfirmationState = {
267
- CONFIRM: "CONFIRM",
268
- DENY: "DENY",
269
- };
270
- export const ResponseState = {
271
- FAILURE: "FAILURE",
272
- REPROMPT: "REPROMPT",
316
+ export const RerankingMetadataSelectionMode = {
317
+ ALL: "ALL",
318
+ SELECTIVE: "SELECTIVE",
273
319
  };
274
- export var InvocationResultMember;
275
- (function (InvocationResultMember) {
276
- InvocationResultMember.visit = (value, visitor) => {
277
- if (value.apiResult !== undefined)
278
- return visitor.apiResult(value.apiResult);
279
- if (value.functionResult !== undefined)
280
- return visitor.functionResult(value.functionResult);
320
+ export var RerankingMetadataSelectiveModeConfiguration;
321
+ (function (RerankingMetadataSelectiveModeConfiguration) {
322
+ RerankingMetadataSelectiveModeConfiguration.visit = (value, visitor) => {
323
+ if (value.fieldsToInclude !== undefined)
324
+ return visitor.fieldsToInclude(value.fieldsToInclude);
325
+ if (value.fieldsToExclude !== undefined)
326
+ return visitor.fieldsToExclude(value.fieldsToExclude);
281
327
  return visitor._(value.$unknown[0], value.$unknown[1]);
282
328
  };
283
- })(InvocationResultMember || (InvocationResultMember = {}));
329
+ })(RerankingMetadataSelectiveModeConfiguration || (RerankingMetadataSelectiveModeConfiguration = {}));
330
+ export const VectorSearchRerankingConfigurationType = {
331
+ BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL",
332
+ };
284
333
  export const RetrievalResultLocationType = {
285
334
  CONFLUENCE: "CONFLUENCE",
335
+ CUSTOM: "CUSTOM",
286
336
  S3: "S3",
287
337
  SALESFORCE: "SALESFORCE",
288
338
  SHAREPOINT: "SHAREPOINT",
289
339
  WEB: "WEB",
290
340
  };
291
- export var InvocationInputMember;
292
- (function (InvocationInputMember) {
293
- InvocationInputMember.visit = (value, visitor) => {
294
- if (value.apiInvocationInput !== undefined)
295
- return visitor.apiInvocationInput(value.apiInvocationInput);
296
- if (value.functionInvocationInput !== undefined)
297
- return visitor.functionInvocationInput(value.functionInvocationInput);
341
+ export var Caller;
342
+ (function (Caller) {
343
+ Caller.visit = (value, visitor) => {
344
+ if (value.agentAliasArn !== undefined)
345
+ return visitor.agentAliasArn(value.agentAliasArn);
298
346
  return visitor._(value.$unknown[0], value.$unknown[1]);
299
347
  };
300
- })(InvocationInputMember || (InvocationInputMember = {}));
348
+ })(Caller || (Caller = {}));
301
349
  export const GuardrailAction = {
302
350
  INTERVENED: "INTERVENED",
303
351
  NONE: "NONE",
@@ -371,6 +419,7 @@ export const GuardrailManagedWordType = {
371
419
  export const InvocationType = {
372
420
  ACTION_GROUP: "ACTION_GROUP",
373
421
  ACTION_GROUP_CODE_INTERPRETER: "ACTION_GROUP_CODE_INTERPRETER",
422
+ AGENT_COLLABORATOR: "AGENT_COLLABORATOR",
374
423
  FINISH: "FINISH",
375
424
  KNOWLEDGE_BASE: "KNOWLEDGE_BASE",
376
425
  };
@@ -392,6 +441,7 @@ export const Source = {
392
441
  };
393
442
  export const Type = {
394
443
  ACTION_GROUP: "ACTION_GROUP",
444
+ AGENT_COLLABORATOR: "AGENT_COLLABORATOR",
395
445
  ASK_USER: "ASK_USER",
396
446
  FINISH: "FINISH",
397
447
  KNOWLEDGE_BASE: "KNOWLEDGE_BASE",
@@ -433,6 +483,20 @@ export var PreProcessingTrace;
433
483
  return visitor._(value.$unknown[0], value.$unknown[1]);
434
484
  };
435
485
  })(PreProcessingTrace || (PreProcessingTrace = {}));
486
+ export var RoutingClassifierTrace;
487
+ (function (RoutingClassifierTrace) {
488
+ RoutingClassifierTrace.visit = (value, visitor) => {
489
+ if (value.invocationInput !== undefined)
490
+ return visitor.invocationInput(value.invocationInput);
491
+ if (value.observation !== undefined)
492
+ return visitor.observation(value.observation);
493
+ if (value.modelInvocationInput !== undefined)
494
+ return visitor.modelInvocationInput(value.modelInvocationInput);
495
+ if (value.modelInvocationOutput !== undefined)
496
+ return visitor.modelInvocationOutput(value.modelInvocationOutput);
497
+ return visitor._(value.$unknown[0], value.$unknown[1]);
498
+ };
499
+ })(RoutingClassifierTrace || (RoutingClassifierTrace = {}));
436
500
  export var Trace;
437
501
  (function (Trace) {
438
502
  Trace.visit = (value, visitor) => {
@@ -444,6 +508,8 @@ export var Trace;
444
508
  return visitor.orchestrationTrace(value.orchestrationTrace);
445
509
  if (value.postProcessingTrace !== undefined)
446
510
  return visitor.postProcessingTrace(value.postProcessingTrace);
511
+ if (value.routingClassifierTrace !== undefined)
512
+ return visitor.routingClassifierTrace(value.routingClassifierTrace);
447
513
  if (value.failureTrace !== undefined)
448
514
  return visitor.failureTrace(value.failureTrace);
449
515
  if (value.customOrchestrationTrace !== undefined)
@@ -568,6 +634,19 @@ export var OptimizedPromptStream;
568
634
  return visitor._(value.$unknown[0], value.$unknown[1]);
569
635
  };
570
636
  })(OptimizedPromptStream || (OptimizedPromptStream = {}));
637
+ export const RerankQueryContentType = {
638
+ TEXT: "TEXT",
639
+ };
640
+ export const RerankingConfigurationType = {
641
+ BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL",
642
+ };
643
+ export const RerankDocumentType = {
644
+ JSON: "JSON",
645
+ TEXT: "TEXT",
646
+ };
647
+ export const RerankSourceType = {
648
+ INLINE: "INLINE",
649
+ };
571
650
  export const ExternalSourceType = {
572
651
  BYTE_CONTENT: "BYTE_CONTENT",
573
652
  S3: "S3",
@@ -583,6 +662,36 @@ export const GuadrailAction = {
583
662
  INTERVENED: "INTERVENED",
584
663
  NONE: "NONE",
585
664
  };
665
+ export var RetrieveAndGenerateStreamResponseOutput;
666
+ (function (RetrieveAndGenerateStreamResponseOutput) {
667
+ RetrieveAndGenerateStreamResponseOutput.visit = (value, visitor) => {
668
+ if (value.output !== undefined)
669
+ return visitor.output(value.output);
670
+ if (value.citation !== undefined)
671
+ return visitor.citation(value.citation);
672
+ if (value.guardrail !== undefined)
673
+ return visitor.guardrail(value.guardrail);
674
+ if (value.internalServerException !== undefined)
675
+ return visitor.internalServerException(value.internalServerException);
676
+ if (value.validationException !== undefined)
677
+ return visitor.validationException(value.validationException);
678
+ if (value.resourceNotFoundException !== undefined)
679
+ return visitor.resourceNotFoundException(value.resourceNotFoundException);
680
+ if (value.serviceQuotaExceededException !== undefined)
681
+ return visitor.serviceQuotaExceededException(value.serviceQuotaExceededException);
682
+ if (value.throttlingException !== undefined)
683
+ return visitor.throttlingException(value.throttlingException);
684
+ if (value.accessDeniedException !== undefined)
685
+ return visitor.accessDeniedException(value.accessDeniedException);
686
+ if (value.conflictException !== undefined)
687
+ return visitor.conflictException(value.conflictException);
688
+ if (value.dependencyFailedException !== undefined)
689
+ return visitor.dependencyFailedException(value.dependencyFailedException);
690
+ if (value.badGatewayException !== undefined)
691
+ return visitor.badGatewayException(value.badGatewayException);
692
+ return visitor._(value.$unknown[0], value.$unknown[1]);
693
+ };
694
+ })(RetrieveAndGenerateStreamResponseOutput || (RetrieveAndGenerateStreamResponseOutput = {}));
586
695
  export var RetrievalFilter;
587
696
  (function (RetrievalFilter) {
588
697
  RetrievalFilter.visit = (value, visitor) => {
@@ -652,6 +761,67 @@ export const AgentActionGroupFilterSensitiveLog = (obj) => ({
652
761
  ...(obj.apiSchema && { apiSchema: APISchemaFilterSensitiveLog(obj.apiSchema) }),
653
762
  ...(obj.functionSchema && { functionSchema: FunctionSchemaFilterSensitiveLog(obj.functionSchema) }),
654
763
  });
764
+ export const ApiResultFilterSensitiveLog = (obj) => ({
765
+ ...obj,
766
+ ...(obj.apiPath && { apiPath: SENSITIVE_STRING }),
767
+ });
768
+ export const InvocationResultMemberFilterSensitiveLog = (obj) => {
769
+ if (obj.apiResult !== undefined)
770
+ return { apiResult: ApiResultFilterSensitiveLog(obj.apiResult) };
771
+ if (obj.functionResult !== undefined)
772
+ return { functionResult: obj.functionResult };
773
+ if (obj.$unknown !== undefined)
774
+ return { [obj.$unknown[0]]: "UNKNOWN" };
775
+ };
776
+ export const ReturnControlResultsFilterSensitiveLog = (obj) => ({
777
+ ...obj,
778
+ ...(obj.returnControlInvocationResults && {
779
+ returnControlInvocationResults: obj.returnControlInvocationResults.map((item) => InvocationResultMemberFilterSensitiveLog(item)),
780
+ }),
781
+ });
782
+ export const AgentCollaboratorInputPayloadFilterSensitiveLog = (obj) => ({
783
+ ...obj,
784
+ ...(obj.text && { text: SENSITIVE_STRING }),
785
+ ...(obj.returnControlResults && {
786
+ returnControlResults: ReturnControlResultsFilterSensitiveLog(obj.returnControlResults),
787
+ }),
788
+ });
789
+ export const AgentCollaboratorInvocationInputFilterSensitiveLog = (obj) => ({
790
+ ...obj,
791
+ ...(obj.input && { input: AgentCollaboratorInputPayloadFilterSensitiveLog(obj.input) }),
792
+ });
793
+ export const ApiInvocationInputFilterSensitiveLog = (obj) => ({
794
+ ...obj,
795
+ ...(obj.apiPath && { apiPath: SENSITIVE_STRING }),
796
+ ...(obj.collaboratorName && { collaboratorName: SENSITIVE_STRING }),
797
+ });
798
+ export const FunctionInvocationInputFilterSensitiveLog = (obj) => ({
799
+ ...obj,
800
+ ...(obj.collaboratorName && { collaboratorName: SENSITIVE_STRING }),
801
+ });
802
+ export const InvocationInputMemberFilterSensitiveLog = (obj) => {
803
+ if (obj.apiInvocationInput !== undefined)
804
+ return { apiInvocationInput: ApiInvocationInputFilterSensitiveLog(obj.apiInvocationInput) };
805
+ if (obj.functionInvocationInput !== undefined)
806
+ return { functionInvocationInput: FunctionInvocationInputFilterSensitiveLog(obj.functionInvocationInput) };
807
+ if (obj.$unknown !== undefined)
808
+ return { [obj.$unknown[0]]: "UNKNOWN" };
809
+ };
810
+ export const ReturnControlPayloadFilterSensitiveLog = (obj) => ({
811
+ ...obj,
812
+ ...(obj.invocationInputs && {
813
+ invocationInputs: obj.invocationInputs.map((item) => InvocationInputMemberFilterSensitiveLog(item)),
814
+ }),
815
+ });
816
+ export const AgentCollaboratorOutputPayloadFilterSensitiveLog = (obj) => ({
817
+ ...obj,
818
+ ...(obj.text && { text: SENSITIVE_STRING }),
819
+ ...(obj.returnControlPayload && { returnControlPayload: SENSITIVE_STRING }),
820
+ });
821
+ export const AgentCollaboratorInvocationOutputFilterSensitiveLog = (obj) => ({
822
+ ...obj,
823
+ ...(obj.output && { output: AgentCollaboratorOutputPayloadFilterSensitiveLog(obj.output) }),
824
+ });
655
825
  export const FlowInputContentFilterSensitiveLog = (obj) => {
656
826
  if (obj.document !== undefined)
657
827
  return { document: obj.document };
@@ -748,6 +918,20 @@ export const InvokeFlowResponseFilterSensitiveLog = (obj) => ({
748
918
  ...obj,
749
919
  ...(obj.responseStream && { responseStream: "STREAMING_CONTENT" }),
750
920
  });
921
+ export const ContentBlockFilterSensitiveLog = (obj) => {
922
+ if (obj.text !== undefined)
923
+ return { text: obj.text };
924
+ if (obj.$unknown !== undefined)
925
+ return { [obj.$unknown[0]]: "UNKNOWN" };
926
+ };
927
+ export const MessageFilterSensitiveLog = (obj) => ({
928
+ ...obj,
929
+ ...(obj.content && { content: SENSITIVE_STRING }),
930
+ });
931
+ export const ConversationHistoryFilterSensitiveLog = (obj) => ({
932
+ ...obj,
933
+ ...(obj.messages && { messages: obj.messages.map((item) => MessageFilterSensitiveLog(item)) }),
934
+ });
751
935
  export const ByteContentFileFilterSensitiveLog = (obj) => ({
752
936
  ...obj,
753
937
  ...(obj.data && { data: SENSITIVE_STRING }),
@@ -760,18 +944,39 @@ export const InputFileFilterSensitiveLog = (obj) => ({
760
944
  ...obj,
761
945
  ...(obj.source && { source: FileSourceFilterSensitiveLog(obj.source) }),
762
946
  });
763
- export const ApiResultFilterSensitiveLog = (obj) => ({
947
+ export const MetadataAttributeSchemaFilterSensitiveLog = (obj) => ({
764
948
  ...obj,
765
- ...(obj.apiPath && { apiPath: SENSITIVE_STRING }),
766
949
  });
767
- export const InvocationResultMemberFilterSensitiveLog = (obj) => {
768
- if (obj.apiResult !== undefined)
769
- return { apiResult: ApiResultFilterSensitiveLog(obj.apiResult) };
770
- if (obj.functionResult !== undefined)
771
- return { functionResult: obj.functionResult };
950
+ export const ImplicitFilterConfigurationFilterSensitiveLog = (obj) => ({
951
+ ...obj,
952
+ ...(obj.metadataAttributes && { metadataAttributes: SENSITIVE_STRING }),
953
+ });
954
+ export const RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog = (obj) => {
955
+ if (obj.fieldsToInclude !== undefined)
956
+ return { fieldsToInclude: SENSITIVE_STRING };
957
+ if (obj.fieldsToExclude !== undefined)
958
+ return { fieldsToExclude: SENSITIVE_STRING };
772
959
  if (obj.$unknown !== undefined)
773
960
  return { [obj.$unknown[0]]: "UNKNOWN" };
774
961
  };
962
+ export const MetadataConfigurationForRerankingFilterSensitiveLog = (obj) => ({
963
+ ...obj,
964
+ ...(obj.selectiveModeConfiguration && {
965
+ selectiveModeConfiguration: RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog(obj.selectiveModeConfiguration),
966
+ }),
967
+ });
968
+ export const VectorSearchBedrockRerankingConfigurationFilterSensitiveLog = (obj) => ({
969
+ ...obj,
970
+ ...(obj.metadataConfiguration && {
971
+ metadataConfiguration: MetadataConfigurationForRerankingFilterSensitiveLog(obj.metadataConfiguration),
972
+ }),
973
+ });
974
+ export const VectorSearchRerankingConfigurationFilterSensitiveLog = (obj) => ({
975
+ ...obj,
976
+ ...(obj.bedrockRerankingConfiguration && {
977
+ bedrockRerankingConfiguration: VectorSearchBedrockRerankingConfigurationFilterSensitiveLog(obj.bedrockRerankingConfiguration),
978
+ }),
979
+ });
775
980
  export const TextResponsePartFilterSensitiveLog = (obj) => ({
776
981
  ...obj,
777
982
  });
@@ -817,24 +1022,6 @@ export const FilePartFilterSensitiveLog = (obj) => ({
817
1022
  ...obj,
818
1023
  ...(obj.files && { files: SENSITIVE_STRING }),
819
1024
  });
820
- export const ApiInvocationInputFilterSensitiveLog = (obj) => ({
821
- ...obj,
822
- ...(obj.apiPath && { apiPath: SENSITIVE_STRING }),
823
- });
824
- export const InvocationInputMemberFilterSensitiveLog = (obj) => {
825
- if (obj.apiInvocationInput !== undefined)
826
- return { apiInvocationInput: ApiInvocationInputFilterSensitiveLog(obj.apiInvocationInput) };
827
- if (obj.functionInvocationInput !== undefined)
828
- return { functionInvocationInput: obj.functionInvocationInput };
829
- if (obj.$unknown !== undefined)
830
- return { [obj.$unknown[0]]: "UNKNOWN" };
831
- };
832
- export const ReturnControlPayloadFilterSensitiveLog = (obj) => ({
833
- ...obj,
834
- ...(obj.invocationInputs && {
835
- invocationInputs: obj.invocationInputs.map((item) => InvocationInputMemberFilterSensitiveLog(item)),
836
- }),
837
- });
838
1025
  export const CustomOrchestrationTraceEventFilterSensitiveLog = (obj) => ({
839
1026
  ...obj,
840
1027
  });
@@ -907,6 +1094,9 @@ export const InvocationInputFilterSensitiveLog = (obj) => ({
907
1094
  ...(obj.knowledgeBaseLookupInput && {
908
1095
  knowledgeBaseLookupInput: KnowledgeBaseLookupInputFilterSensitiveLog(obj.knowledgeBaseLookupInput),
909
1096
  }),
1097
+ ...(obj.agentCollaboratorInvocationInput && {
1098
+ agentCollaboratorInvocationInput: AgentCollaboratorInvocationInputFilterSensitiveLog(obj.agentCollaboratorInvocationInput),
1099
+ }),
910
1100
  });
911
1101
  export const ModelInvocationInputFilterSensitiveLog = (obj) => ({
912
1102
  ...obj,
@@ -946,6 +1136,9 @@ export const ObservationFilterSensitiveLog = (obj) => ({
946
1136
  ...(obj.actionGroupInvocationOutput && {
947
1137
  actionGroupInvocationOutput: ActionGroupInvocationOutputFilterSensitiveLog(obj.actionGroupInvocationOutput),
948
1138
  }),
1139
+ ...(obj.agentCollaboratorInvocationOutput && {
1140
+ agentCollaboratorInvocationOutput: AgentCollaboratorInvocationOutputFilterSensitiveLog(obj.agentCollaboratorInvocationOutput),
1141
+ }),
949
1142
  ...(obj.knowledgeBaseLookupOutput && {
950
1143
  knowledgeBaseLookupOutput: KnowledgeBaseLookupOutputFilterSensitiveLog(obj.knowledgeBaseLookupOutput),
951
1144
  }),
@@ -1006,6 +1199,23 @@ export const PreProcessingTraceFilterSensitiveLog = (obj) => {
1006
1199
  if (obj.$unknown !== undefined)
1007
1200
  return { [obj.$unknown[0]]: "UNKNOWN" };
1008
1201
  };
1202
+ export const RoutingClassifierModelInvocationOutputFilterSensitiveLog = (obj) => ({
1203
+ ...obj,
1204
+ ...(obj.rawResponse && { rawResponse: SENSITIVE_STRING }),
1205
+ ...(obj.metadata && { metadata: SENSITIVE_STRING }),
1206
+ });
1207
+ export const RoutingClassifierTraceFilterSensitiveLog = (obj) => {
1208
+ if (obj.invocationInput !== undefined)
1209
+ return { invocationInput: SENSITIVE_STRING };
1210
+ if (obj.observation !== undefined)
1211
+ return { observation: SENSITIVE_STRING };
1212
+ if (obj.modelInvocationInput !== undefined)
1213
+ return { modelInvocationInput: SENSITIVE_STRING };
1214
+ if (obj.modelInvocationOutput !== undefined)
1215
+ return { modelInvocationOutput: SENSITIVE_STRING };
1216
+ if (obj.$unknown !== undefined)
1217
+ return { [obj.$unknown[0]]: "UNKNOWN" };
1218
+ };
1009
1219
  export const TraceFilterSensitiveLog = (obj) => {
1010
1220
  if (obj.guardrailTrace !== undefined)
1011
1221
  return { guardrailTrace: SENSITIVE_STRING };
@@ -1015,6 +1225,8 @@ export const TraceFilterSensitiveLog = (obj) => {
1015
1225
  return { orchestrationTrace: SENSITIVE_STRING };
1016
1226
  if (obj.postProcessingTrace !== undefined)
1017
1227
  return { postProcessingTrace: SENSITIVE_STRING };
1228
+ if (obj.routingClassifierTrace !== undefined)
1229
+ return { routingClassifierTrace: SENSITIVE_STRING };
1018
1230
  if (obj.failureTrace !== undefined)
1019
1231
  return { failureTrace: SENSITIVE_STRING };
1020
1232
  if (obj.customOrchestrationTrace !== undefined)
@@ -1025,6 +1237,8 @@ export const TraceFilterSensitiveLog = (obj) => {
1025
1237
  export const TracePartFilterSensitiveLog = (obj) => ({
1026
1238
  ...obj,
1027
1239
  ...(obj.trace && { trace: SENSITIVE_STRING }),
1240
+ ...(obj.callerChain && { callerChain: obj.callerChain.map((item) => item) }),
1241
+ ...(obj.collaboratorName && { collaboratorName: SENSITIVE_STRING }),
1028
1242
  });
1029
1243
  export const ResponseStreamFilterSensitiveLog = (obj) => {
1030
1244
  if (obj.chunk !== undefined)
@@ -1180,6 +1394,34 @@ export const OptimizePromptResponseFilterSensitiveLog = (obj) => ({
1180
1394
  ...obj,
1181
1395
  ...(obj.optimizedPrompt && { optimizedPrompt: "STREAMING_CONTENT" }),
1182
1396
  });
1397
+ export const RerankTextDocumentFilterSensitiveLog = (obj) => ({
1398
+ ...obj,
1399
+ });
1400
+ export const RerankQueryFilterSensitiveLog = (obj) => ({
1401
+ ...obj,
1402
+ ...(obj.textQuery && { textQuery: SENSITIVE_STRING }),
1403
+ });
1404
+ export const RerankDocumentFilterSensitiveLog = (obj) => ({
1405
+ ...obj,
1406
+ ...(obj.textDocument && { textDocument: SENSITIVE_STRING }),
1407
+ });
1408
+ export const RerankSourceFilterSensitiveLog = (obj) => ({
1409
+ ...obj,
1410
+ ...(obj.inlineDocumentSource && { inlineDocumentSource: SENSITIVE_STRING }),
1411
+ });
1412
+ export const RerankRequestFilterSensitiveLog = (obj) => ({
1413
+ ...obj,
1414
+ ...(obj.queries && { queries: SENSITIVE_STRING }),
1415
+ ...(obj.sources && { sources: SENSITIVE_STRING }),
1416
+ });
1417
+ export const RerankResultFilterSensitiveLog = (obj) => ({
1418
+ ...obj,
1419
+ ...(obj.document && { document: SENSITIVE_STRING }),
1420
+ });
1421
+ export const RerankResponseFilterSensitiveLog = (obj) => ({
1422
+ ...obj,
1423
+ ...(obj.results && { results: obj.results.map((item) => RerankResultFilterSensitiveLog(item)) }),
1424
+ });
1183
1425
  export const RetrieveAndGenerateInputFilterSensitiveLog = (obj) => ({
1184
1426
  ...obj,
1185
1427
  });
@@ -1223,6 +1465,45 @@ export const RetrieveAndGenerateResponseFilterSensitiveLog = (obj) => ({
1223
1465
  ...(obj.output && { output: SENSITIVE_STRING }),
1224
1466
  ...(obj.citations && { citations: obj.citations.map((item) => CitationFilterSensitiveLog(item)) }),
1225
1467
  });
1468
+ export const CitationEventFilterSensitiveLog = (obj) => ({
1469
+ ...obj,
1470
+ ...(obj.citation && { citation: CitationFilterSensitiveLog(obj.citation) }),
1471
+ });
1472
+ export const RetrieveAndGenerateOutputEventFilterSensitiveLog = (obj) => ({
1473
+ ...obj,
1474
+ });
1475
+ export const RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog = (obj) => {
1476
+ if (obj.output !== undefined)
1477
+ return { output: SENSITIVE_STRING };
1478
+ if (obj.citation !== undefined)
1479
+ return { citation: CitationEventFilterSensitiveLog(obj.citation) };
1480
+ if (obj.guardrail !== undefined)
1481
+ return { guardrail: obj.guardrail };
1482
+ if (obj.internalServerException !== undefined)
1483
+ return { internalServerException: obj.internalServerException };
1484
+ if (obj.validationException !== undefined)
1485
+ return { validationException: obj.validationException };
1486
+ if (obj.resourceNotFoundException !== undefined)
1487
+ return { resourceNotFoundException: obj.resourceNotFoundException };
1488
+ if (obj.serviceQuotaExceededException !== undefined)
1489
+ return { serviceQuotaExceededException: obj.serviceQuotaExceededException };
1490
+ if (obj.throttlingException !== undefined)
1491
+ return { throttlingException: obj.throttlingException };
1492
+ if (obj.accessDeniedException !== undefined)
1493
+ return { accessDeniedException: obj.accessDeniedException };
1494
+ if (obj.conflictException !== undefined)
1495
+ return { conflictException: obj.conflictException };
1496
+ if (obj.dependencyFailedException !== undefined)
1497
+ return { dependencyFailedException: obj.dependencyFailedException };
1498
+ if (obj.badGatewayException !== undefined)
1499
+ return { badGatewayException: obj.badGatewayException };
1500
+ if (obj.$unknown !== undefined)
1501
+ return { [obj.$unknown[0]]: "UNKNOWN" };
1502
+ };
1503
+ export const RetrieveAndGenerateStreamResponseFilterSensitiveLog = (obj) => ({
1504
+ ...obj,
1505
+ ...(obj.stream && { stream: "STREAMING_CONTENT" }),
1506
+ });
1226
1507
  export const KnowledgeBaseQueryFilterSensitiveLog = (obj) => ({
1227
1508
  ...obj,
1228
1509
  });
@@ -1269,6 +1550,12 @@ export const RetrievalFilterFilterSensitiveLog = (obj) => {
1269
1550
  export const KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog = (obj) => ({
1270
1551
  ...obj,
1271
1552
  ...(obj.filter && { filter: SENSITIVE_STRING }),
1553
+ ...(obj.rerankingConfiguration && {
1554
+ rerankingConfiguration: VectorSearchRerankingConfigurationFilterSensitiveLog(obj.rerankingConfiguration),
1555
+ }),
1556
+ ...(obj.implicitFilterConfiguration && {
1557
+ implicitFilterConfiguration: ImplicitFilterConfigurationFilterSensitiveLog(obj.implicitFilterConfiguration),
1558
+ }),
1272
1559
  });
1273
1560
  export const KnowledgeBaseRetrievalConfigurationFilterSensitiveLog = (obj) => ({
1274
1561
  ...obj,
@@ -1335,6 +1622,13 @@ export const RetrieveAndGenerateRequestFilterSensitiveLog = (obj) => ({
1335
1622
  retrieveAndGenerateConfiguration: RetrieveAndGenerateConfigurationFilterSensitiveLog(obj.retrieveAndGenerateConfiguration),
1336
1623
  }),
1337
1624
  });
1625
+ export const RetrieveAndGenerateStreamRequestFilterSensitiveLog = (obj) => ({
1626
+ ...obj,
1627
+ ...(obj.input && { input: SENSITIVE_STRING }),
1628
+ ...(obj.retrieveAndGenerateConfiguration && {
1629
+ retrieveAndGenerateConfiguration: RetrieveAndGenerateConfigurationFilterSensitiveLog(obj.retrieveAndGenerateConfiguration),
1630
+ }),
1631
+ });
1338
1632
  export const SessionStateFilterSensitiveLog = (obj) => ({
1339
1633
  ...obj,
1340
1634
  ...(obj.returnControlInvocationResults && {
@@ -1344,6 +1638,9 @@ export const SessionStateFilterSensitiveLog = (obj) => ({
1344
1638
  ...(obj.knowledgeBaseConfigurations && {
1345
1639
  knowledgeBaseConfigurations: obj.knowledgeBaseConfigurations.map((item) => KnowledgeBaseConfigurationFilterSensitiveLog(item)),
1346
1640
  }),
1641
+ ...(obj.conversationHistory && {
1642
+ conversationHistory: ConversationHistoryFilterSensitiveLog(obj.conversationHistory),
1643
+ }),
1347
1644
  });
1348
1645
  export const InvokeAgentRequestFilterSensitiveLog = (obj) => ({
1349
1646
  ...obj,
@@ -0,0 +1,4 @@
1
+ import { createPaginator } from "@smithy/core";
2
+ import { BedrockAgentRuntimeClient } from "../BedrockAgentRuntimeClient";
3
+ import { RerankCommand } from "../commands/RerankCommand";
4
+ export const paginateRerank = createPaginator(BedrockAgentRuntimeClient, RerankCommand, "nextToken", "nextToken", "");
@@ -1,3 +1,4 @@
1
1
  export * from "./GetAgentMemoryPaginator";
2
2
  export * from "./Interfaces";
3
+ export * from "./RerankPaginator";
3
4
  export * from "./RetrievePaginator";