@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
package/dist-cjs/index.js CHANGED
@@ -30,9 +30,14 @@ __export(src_exports, {
30
30
  ActionGroupSignature: () => ActionGroupSignature,
31
31
  ActionInvocationType: () => ActionInvocationType,
32
32
  AgentActionGroupFilterSensitiveLog: () => AgentActionGroupFilterSensitiveLog,
33
+ AgentCollaboratorInputPayloadFilterSensitiveLog: () => AgentCollaboratorInputPayloadFilterSensitiveLog,
34
+ AgentCollaboratorInvocationInputFilterSensitiveLog: () => AgentCollaboratorInvocationInputFilterSensitiveLog,
35
+ AgentCollaboratorInvocationOutputFilterSensitiveLog: () => AgentCollaboratorInvocationOutputFilterSensitiveLog,
36
+ AgentCollaboratorOutputPayloadFilterSensitiveLog: () => AgentCollaboratorOutputPayloadFilterSensitiveLog,
33
37
  AnalyzePromptEventFilterSensitiveLog: () => AnalyzePromptEventFilterSensitiveLog,
34
38
  ApiInvocationInputFilterSensitiveLog: () => ApiInvocationInputFilterSensitiveLog,
35
39
  ApiResultFilterSensitiveLog: () => ApiResultFilterSensitiveLog,
40
+ AttributeType: () => AttributeType,
36
41
  AttributionFilterSensitiveLog: () => AttributionFilterSensitiveLog,
37
42
  BadGatewayException: () => BadGatewayException,
38
43
  BedrockAgentRuntime: () => BedrockAgentRuntime,
@@ -40,9 +45,15 @@ __export(src_exports, {
40
45
  BedrockAgentRuntimeServiceException: () => BedrockAgentRuntimeServiceException,
41
46
  ByteContentDocFilterSensitiveLog: () => ByteContentDocFilterSensitiveLog,
42
47
  ByteContentFileFilterSensitiveLog: () => ByteContentFileFilterSensitiveLog,
48
+ Caller: () => Caller,
49
+ CitationEventFilterSensitiveLog: () => CitationEventFilterSensitiveLog,
43
50
  CitationFilterSensitiveLog: () => CitationFilterSensitiveLog,
44
51
  ConfirmationState: () => ConfirmationState,
45
52
  ConflictException: () => ConflictException,
53
+ ContentBlock: () => ContentBlock,
54
+ ContentBlockFilterSensitiveLog: () => ContentBlockFilterSensitiveLog,
55
+ ConversationHistoryFilterSensitiveLog: () => ConversationHistoryFilterSensitiveLog,
56
+ ConversationRole: () => ConversationRole,
46
57
  CreationMode: () => CreationMode,
47
58
  CustomControlMethod: () => CustomControlMethod,
48
59
  CustomOrchestrationTraceEventFilterSensitiveLog: () => CustomOrchestrationTraceEventFilterSensitiveLog,
@@ -82,6 +93,7 @@ __export(src_exports, {
82
93
  FlowTraceNodeOutputEventFilterSensitiveLog: () => FlowTraceNodeOutputEventFilterSensitiveLog,
83
94
  FlowTraceNodeOutputFieldFilterSensitiveLog: () => FlowTraceNodeOutputFieldFilterSensitiveLog,
84
95
  FunctionDefinitionFilterSensitiveLog: () => FunctionDefinitionFilterSensitiveLog,
96
+ FunctionInvocationInputFilterSensitiveLog: () => FunctionInvocationInputFilterSensitiveLog,
85
97
  FunctionSchema: () => FunctionSchema,
86
98
  FunctionSchemaFilterSensitiveLog: () => FunctionSchemaFilterSensitiveLog,
87
99
  GeneratedResponsePartFilterSensitiveLog: () => GeneratedResponsePartFilterSensitiveLog,
@@ -110,6 +122,7 @@ __export(src_exports, {
110
122
  GuardrailTraceFilterSensitiveLog: () => GuardrailTraceFilterSensitiveLog,
111
123
  GuardrailWordPolicyAction: () => GuardrailWordPolicyAction,
112
124
  GuardrailWordPolicyAssessmentFilterSensitiveLog: () => GuardrailWordPolicyAssessmentFilterSensitiveLog,
125
+ ImplicitFilterConfigurationFilterSensitiveLog: () => ImplicitFilterConfigurationFilterSensitiveLog,
113
126
  InlineAgentFilePartFilterSensitiveLog: () => InlineAgentFilePartFilterSensitiveLog,
114
127
  InlineAgentPayloadPartFilterSensitiveLog: () => InlineAgentPayloadPartFilterSensitiveLog,
115
128
  InlineAgentResponseStream: () => InlineAgentResponseStream,
@@ -147,6 +160,9 @@ __export(src_exports, {
147
160
  KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog: () => KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog,
148
161
  Memory: () => Memory,
149
162
  MemoryType: () => MemoryType,
163
+ MessageFilterSensitiveLog: () => MessageFilterSensitiveLog,
164
+ MetadataAttributeSchemaFilterSensitiveLog: () => MetadataAttributeSchemaFilterSensitiveLog,
165
+ MetadataConfigurationForRerankingFilterSensitiveLog: () => MetadataConfigurationForRerankingFilterSensitiveLog,
150
166
  MetadataFilterSensitiveLog: () => MetadataFilterSensitiveLog,
151
167
  ModelInvocationInputFilterSensitiveLog: () => ModelInvocationInputFilterSensitiveLog,
152
168
  NodeType: () => NodeType,
@@ -166,6 +182,7 @@ __export(src_exports, {
166
182
  OutputFileFilterSensitiveLog: () => OutputFileFilterSensitiveLog,
167
183
  ParameterType: () => ParameterType,
168
184
  PayloadPartFilterSensitiveLog: () => PayloadPartFilterSensitiveLog,
185
+ PayloadType: () => PayloadType,
169
186
  PostProcessingModelInvocationOutputFilterSensitiveLog: () => PostProcessingModelInvocationOutputFilterSensitiveLog,
170
187
  PostProcessingParsedResponseFilterSensitiveLog: () => PostProcessingParsedResponseFilterSensitiveLog,
171
188
  PostProcessingTrace: () => PostProcessingTrace,
@@ -184,6 +201,21 @@ __export(src_exports, {
184
201
  RawResponseFilterSensitiveLog: () => RawResponseFilterSensitiveLog,
185
202
  RepromptResponseFilterSensitiveLog: () => RepromptResponseFilterSensitiveLog,
186
203
  RequireConfirmation: () => RequireConfirmation,
204
+ RerankCommand: () => RerankCommand,
205
+ RerankDocumentFilterSensitiveLog: () => RerankDocumentFilterSensitiveLog,
206
+ RerankDocumentType: () => RerankDocumentType,
207
+ RerankQueryContentType: () => RerankQueryContentType,
208
+ RerankQueryFilterSensitiveLog: () => RerankQueryFilterSensitiveLog,
209
+ RerankRequestFilterSensitiveLog: () => RerankRequestFilterSensitiveLog,
210
+ RerankResponseFilterSensitiveLog: () => RerankResponseFilterSensitiveLog,
211
+ RerankResultFilterSensitiveLog: () => RerankResultFilterSensitiveLog,
212
+ RerankSourceFilterSensitiveLog: () => RerankSourceFilterSensitiveLog,
213
+ RerankSourceType: () => RerankSourceType,
214
+ RerankTextDocumentFilterSensitiveLog: () => RerankTextDocumentFilterSensitiveLog,
215
+ RerankingConfigurationType: () => RerankingConfigurationType,
216
+ RerankingMetadataSelectionMode: () => RerankingMetadataSelectionMode,
217
+ RerankingMetadataSelectiveModeConfiguration: () => RerankingMetadataSelectiveModeConfiguration,
218
+ RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog: () => RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog,
187
219
  ResourceNotFoundException: () => ResourceNotFoundException,
188
220
  ResponseState: () => ResponseState,
189
221
  ResponseStream: () => ResponseStream,
@@ -196,15 +228,25 @@ __export(src_exports, {
196
228
  RetrieveAndGenerateCommand: () => RetrieveAndGenerateCommand,
197
229
  RetrieveAndGenerateConfigurationFilterSensitiveLog: () => RetrieveAndGenerateConfigurationFilterSensitiveLog,
198
230
  RetrieveAndGenerateInputFilterSensitiveLog: () => RetrieveAndGenerateInputFilterSensitiveLog,
231
+ RetrieveAndGenerateOutputEventFilterSensitiveLog: () => RetrieveAndGenerateOutputEventFilterSensitiveLog,
199
232
  RetrieveAndGenerateOutputFilterSensitiveLog: () => RetrieveAndGenerateOutputFilterSensitiveLog,
200
233
  RetrieveAndGenerateRequestFilterSensitiveLog: () => RetrieveAndGenerateRequestFilterSensitiveLog,
201
234
  RetrieveAndGenerateResponseFilterSensitiveLog: () => RetrieveAndGenerateResponseFilterSensitiveLog,
235
+ RetrieveAndGenerateStreamCommand: () => RetrieveAndGenerateStreamCommand,
236
+ RetrieveAndGenerateStreamRequestFilterSensitiveLog: () => RetrieveAndGenerateStreamRequestFilterSensitiveLog,
237
+ RetrieveAndGenerateStreamResponseFilterSensitiveLog: () => RetrieveAndGenerateStreamResponseFilterSensitiveLog,
238
+ RetrieveAndGenerateStreamResponseOutput: () => RetrieveAndGenerateStreamResponseOutput,
239
+ RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog: () => RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog,
202
240
  RetrieveAndGenerateType: () => RetrieveAndGenerateType,
203
241
  RetrieveCommand: () => RetrieveCommand,
204
242
  RetrieveRequestFilterSensitiveLog: () => RetrieveRequestFilterSensitiveLog,
205
243
  RetrieveResponseFilterSensitiveLog: () => RetrieveResponseFilterSensitiveLog,
206
244
  RetrievedReferenceFilterSensitiveLog: () => RetrievedReferenceFilterSensitiveLog,
207
245
  ReturnControlPayloadFilterSensitiveLog: () => ReturnControlPayloadFilterSensitiveLog,
246
+ ReturnControlResultsFilterSensitiveLog: () => ReturnControlResultsFilterSensitiveLog,
247
+ RoutingClassifierModelInvocationOutputFilterSensitiveLog: () => RoutingClassifierModelInvocationOutputFilterSensitiveLog,
248
+ RoutingClassifierTrace: () => RoutingClassifierTrace,
249
+ RoutingClassifierTraceFilterSensitiveLog: () => RoutingClassifierTraceFilterSensitiveLog,
208
250
  SearchType: () => SearchType,
209
251
  ServiceQuotaExceededException: () => ServiceQuotaExceededException,
210
252
  SessionStateFilterSensitiveLog: () => SessionStateFilterSensitiveLog,
@@ -218,8 +260,12 @@ __export(src_exports, {
218
260
  Type: () => Type,
219
261
  UsageFilterSensitiveLog: () => UsageFilterSensitiveLog,
220
262
  ValidationException: () => ValidationException,
263
+ VectorSearchBedrockRerankingConfigurationFilterSensitiveLog: () => VectorSearchBedrockRerankingConfigurationFilterSensitiveLog,
264
+ VectorSearchRerankingConfigurationFilterSensitiveLog: () => VectorSearchRerankingConfigurationFilterSensitiveLog,
265
+ VectorSearchRerankingConfigurationType: () => VectorSearchRerankingConfigurationType,
221
266
  __Client: () => import_smithy_client.Client,
222
267
  paginateGetAgentMemory: () => paginateGetAgentMemory,
268
+ paginateRerank: () => paginateRerank,
223
269
  paginateRetrieve: () => paginateRetrieve
224
270
  });
225
271
  module.exports = __toCommonJS(src_exports);
@@ -464,6 +510,38 @@ var FunctionSchema;
464
510
  return visitor._(value.$unknown[0], value.$unknown[1]);
465
511
  }, "visit");
466
512
  })(FunctionSchema || (FunctionSchema = {}));
513
+ var ConfirmationState = {
514
+ CONFIRM: "CONFIRM",
515
+ DENY: "DENY"
516
+ };
517
+ var ResponseState = {
518
+ FAILURE: "FAILURE",
519
+ REPROMPT: "REPROMPT"
520
+ };
521
+ var InvocationResultMember;
522
+ ((InvocationResultMember3) => {
523
+ InvocationResultMember3.visit = /* @__PURE__ */ __name((value, visitor) => {
524
+ if (value.apiResult !== void 0)
525
+ return visitor.apiResult(value.apiResult);
526
+ if (value.functionResult !== void 0)
527
+ return visitor.functionResult(value.functionResult);
528
+ return visitor._(value.$unknown[0], value.$unknown[1]);
529
+ }, "visit");
530
+ })(InvocationResultMember || (InvocationResultMember = {}));
531
+ var PayloadType = {
532
+ RETURN_CONTROL: "RETURN_CONTROL",
533
+ TEXT: "TEXT"
534
+ };
535
+ var InvocationInputMember;
536
+ ((InvocationInputMember2) => {
537
+ InvocationInputMember2.visit = /* @__PURE__ */ __name((value, visitor) => {
538
+ if (value.apiInvocationInput !== void 0)
539
+ return visitor.apiInvocationInput(value.apiInvocationInput);
540
+ if (value.functionInvocationInput !== void 0)
541
+ return visitor.functionInvocationInput(value.functionInvocationInput);
542
+ return visitor._(value.$unknown[0], value.$unknown[1]);
543
+ }, "visit");
544
+ })(InvocationInputMember || (InvocationInputMember = {}));
467
545
  var _BadGatewayException = class _BadGatewayException extends BedrockAgentRuntimeServiceException {
468
546
  /**
469
547
  * @internal
@@ -688,6 +766,18 @@ var FlowResponseStream;
688
766
  return visitor._(value.$unknown[0], value.$unknown[1]);
689
767
  }, "visit");
690
768
  })(FlowResponseStream || (FlowResponseStream = {}));
769
+ var ContentBlock;
770
+ ((ContentBlock3) => {
771
+ ContentBlock3.visit = /* @__PURE__ */ __name((value, visitor) => {
772
+ if (value.text !== void 0)
773
+ return visitor.text(value.text);
774
+ return visitor._(value.$unknown[0], value.$unknown[1]);
775
+ }, "visit");
776
+ })(ContentBlock || (ContentBlock = {}));
777
+ var ConversationRole = {
778
+ ASSISTANT: "assistant",
779
+ USER: "user"
780
+ };
691
781
  var FileSourceType = {
692
782
  BYTE_CONTENT: "BYTE_CONTENT",
693
783
  S3: "S3"
@@ -696,45 +786,49 @@ var FileUseCase = {
696
786
  CHAT: "CHAT",
697
787
  CODE_INTERPRETER: "CODE_INTERPRETER"
698
788
  };
789
+ var AttributeType = {
790
+ BOOLEAN: "BOOLEAN",
791
+ NUMBER: "NUMBER",
792
+ STRING: "STRING",
793
+ STRING_LIST: "STRING_LIST"
794
+ };
699
795
  var SearchType = {
700
796
  HYBRID: "HYBRID",
701
797
  SEMANTIC: "SEMANTIC"
702
798
  };
703
- var ConfirmationState = {
704
- CONFIRM: "CONFIRM",
705
- DENY: "DENY"
799
+ var RerankingMetadataSelectionMode = {
800
+ ALL: "ALL",
801
+ SELECTIVE: "SELECTIVE"
706
802
  };
707
- var ResponseState = {
708
- FAILURE: "FAILURE",
709
- REPROMPT: "REPROMPT"
710
- };
711
- var InvocationResultMember;
712
- ((InvocationResultMember3) => {
713
- InvocationResultMember3.visit = /* @__PURE__ */ __name((value, visitor) => {
714
- if (value.apiResult !== void 0)
715
- return visitor.apiResult(value.apiResult);
716
- if (value.functionResult !== void 0)
717
- return visitor.functionResult(value.functionResult);
803
+ var RerankingMetadataSelectiveModeConfiguration;
804
+ ((RerankingMetadataSelectiveModeConfiguration3) => {
805
+ RerankingMetadataSelectiveModeConfiguration3.visit = /* @__PURE__ */ __name((value, visitor) => {
806
+ if (value.fieldsToInclude !== void 0)
807
+ return visitor.fieldsToInclude(value.fieldsToInclude);
808
+ if (value.fieldsToExclude !== void 0)
809
+ return visitor.fieldsToExclude(value.fieldsToExclude);
718
810
  return visitor._(value.$unknown[0], value.$unknown[1]);
719
811
  }, "visit");
720
- })(InvocationResultMember || (InvocationResultMember = {}));
812
+ })(RerankingMetadataSelectiveModeConfiguration || (RerankingMetadataSelectiveModeConfiguration = {}));
813
+ var VectorSearchRerankingConfigurationType = {
814
+ BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL"
815
+ };
721
816
  var RetrievalResultLocationType = {
722
817
  CONFLUENCE: "CONFLUENCE",
818
+ CUSTOM: "CUSTOM",
723
819
  S3: "S3",
724
820
  SALESFORCE: "SALESFORCE",
725
821
  SHAREPOINT: "SHAREPOINT",
726
822
  WEB: "WEB"
727
823
  };
728
- var InvocationInputMember;
729
- ((InvocationInputMember2) => {
730
- InvocationInputMember2.visit = /* @__PURE__ */ __name((value, visitor) => {
731
- if (value.apiInvocationInput !== void 0)
732
- return visitor.apiInvocationInput(value.apiInvocationInput);
733
- if (value.functionInvocationInput !== void 0)
734
- return visitor.functionInvocationInput(value.functionInvocationInput);
824
+ var Caller;
825
+ ((Caller2) => {
826
+ Caller2.visit = /* @__PURE__ */ __name((value, visitor) => {
827
+ if (value.agentAliasArn !== void 0)
828
+ return visitor.agentAliasArn(value.agentAliasArn);
735
829
  return visitor._(value.$unknown[0], value.$unknown[1]);
736
830
  }, "visit");
737
- })(InvocationInputMember || (InvocationInputMember = {}));
831
+ })(Caller || (Caller = {}));
738
832
  var GuardrailAction = {
739
833
  INTERVENED: "INTERVENED",
740
834
  NONE: "NONE"
@@ -808,6 +902,7 @@ var GuardrailManagedWordType = {
808
902
  var InvocationType = {
809
903
  ACTION_GROUP: "ACTION_GROUP",
810
904
  ACTION_GROUP_CODE_INTERPRETER: "ACTION_GROUP_CODE_INTERPRETER",
905
+ AGENT_COLLABORATOR: "AGENT_COLLABORATOR",
811
906
  FINISH: "FINISH",
812
907
  KNOWLEDGE_BASE: "KNOWLEDGE_BASE"
813
908
  };
@@ -829,6 +924,7 @@ var Source = {
829
924
  };
830
925
  var Type = {
831
926
  ACTION_GROUP: "ACTION_GROUP",
927
+ AGENT_COLLABORATOR: "AGENT_COLLABORATOR",
832
928
  ASK_USER: "ASK_USER",
833
929
  FINISH: "FINISH",
834
930
  KNOWLEDGE_BASE: "KNOWLEDGE_BASE",
@@ -870,6 +966,20 @@ var PreProcessingTrace;
870
966
  return visitor._(value.$unknown[0], value.$unknown[1]);
871
967
  }, "visit");
872
968
  })(PreProcessingTrace || (PreProcessingTrace = {}));
969
+ var RoutingClassifierTrace;
970
+ ((RoutingClassifierTrace3) => {
971
+ RoutingClassifierTrace3.visit = /* @__PURE__ */ __name((value, visitor) => {
972
+ if (value.invocationInput !== void 0)
973
+ return visitor.invocationInput(value.invocationInput);
974
+ if (value.observation !== void 0)
975
+ return visitor.observation(value.observation);
976
+ if (value.modelInvocationInput !== void 0)
977
+ return visitor.modelInvocationInput(value.modelInvocationInput);
978
+ if (value.modelInvocationOutput !== void 0)
979
+ return visitor.modelInvocationOutput(value.modelInvocationOutput);
980
+ return visitor._(value.$unknown[0], value.$unknown[1]);
981
+ }, "visit");
982
+ })(RoutingClassifierTrace || (RoutingClassifierTrace = {}));
873
983
  var Trace;
874
984
  ((Trace3) => {
875
985
  Trace3.visit = /* @__PURE__ */ __name((value, visitor) => {
@@ -881,6 +991,8 @@ var Trace;
881
991
  return visitor.orchestrationTrace(value.orchestrationTrace);
882
992
  if (value.postProcessingTrace !== void 0)
883
993
  return visitor.postProcessingTrace(value.postProcessingTrace);
994
+ if (value.routingClassifierTrace !== void 0)
995
+ return visitor.routingClassifierTrace(value.routingClassifierTrace);
884
996
  if (value.failureTrace !== void 0)
885
997
  return visitor.failureTrace(value.failureTrace);
886
998
  if (value.customOrchestrationTrace !== void 0)
@@ -1005,6 +1117,19 @@ var OptimizedPromptStream;
1005
1117
  return visitor._(value.$unknown[0], value.$unknown[1]);
1006
1118
  }, "visit");
1007
1119
  })(OptimizedPromptStream || (OptimizedPromptStream = {}));
1120
+ var RerankQueryContentType = {
1121
+ TEXT: "TEXT"
1122
+ };
1123
+ var RerankingConfigurationType = {
1124
+ BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL"
1125
+ };
1126
+ var RerankDocumentType = {
1127
+ JSON: "JSON",
1128
+ TEXT: "TEXT"
1129
+ };
1130
+ var RerankSourceType = {
1131
+ INLINE: "INLINE"
1132
+ };
1008
1133
  var ExternalSourceType = {
1009
1134
  BYTE_CONTENT: "BYTE_CONTENT",
1010
1135
  S3: "S3"
@@ -1020,6 +1145,36 @@ var GuadrailAction = {
1020
1145
  INTERVENED: "INTERVENED",
1021
1146
  NONE: "NONE"
1022
1147
  };
1148
+ var RetrieveAndGenerateStreamResponseOutput;
1149
+ ((RetrieveAndGenerateStreamResponseOutput3) => {
1150
+ RetrieveAndGenerateStreamResponseOutput3.visit = /* @__PURE__ */ __name((value, visitor) => {
1151
+ if (value.output !== void 0)
1152
+ return visitor.output(value.output);
1153
+ if (value.citation !== void 0)
1154
+ return visitor.citation(value.citation);
1155
+ if (value.guardrail !== void 0)
1156
+ return visitor.guardrail(value.guardrail);
1157
+ if (value.internalServerException !== void 0)
1158
+ return visitor.internalServerException(value.internalServerException);
1159
+ if (value.validationException !== void 0)
1160
+ return visitor.validationException(value.validationException);
1161
+ if (value.resourceNotFoundException !== void 0)
1162
+ return visitor.resourceNotFoundException(value.resourceNotFoundException);
1163
+ if (value.serviceQuotaExceededException !== void 0)
1164
+ return visitor.serviceQuotaExceededException(value.serviceQuotaExceededException);
1165
+ if (value.throttlingException !== void 0)
1166
+ return visitor.throttlingException(value.throttlingException);
1167
+ if (value.accessDeniedException !== void 0)
1168
+ return visitor.accessDeniedException(value.accessDeniedException);
1169
+ if (value.conflictException !== void 0)
1170
+ return visitor.conflictException(value.conflictException);
1171
+ if (value.dependencyFailedException !== void 0)
1172
+ return visitor.dependencyFailedException(value.dependencyFailedException);
1173
+ if (value.badGatewayException !== void 0)
1174
+ return visitor.badGatewayException(value.badGatewayException);
1175
+ return visitor._(value.$unknown[0], value.$unknown[1]);
1176
+ }, "visit");
1177
+ })(RetrieveAndGenerateStreamResponseOutput || (RetrieveAndGenerateStreamResponseOutput = {}));
1023
1178
  var RetrievalFilter;
1024
1179
  ((RetrievalFilter2) => {
1025
1180
  RetrievalFilter2.visit = /* @__PURE__ */ __name((value, visitor) => {
@@ -1089,6 +1244,69 @@ var AgentActionGroupFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1089
1244
  ...obj.apiSchema && { apiSchema: APISchemaFilterSensitiveLog(obj.apiSchema) },
1090
1245
  ...obj.functionSchema && { functionSchema: FunctionSchemaFilterSensitiveLog(obj.functionSchema) }
1091
1246
  }), "AgentActionGroupFilterSensitiveLog");
1247
+ var ApiResultFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1248
+ ...obj,
1249
+ ...obj.apiPath && { apiPath: import_smithy_client.SENSITIVE_STRING }
1250
+ }), "ApiResultFilterSensitiveLog");
1251
+ var InvocationResultMemberFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1252
+ if (obj.apiResult !== void 0)
1253
+ return { apiResult: ApiResultFilterSensitiveLog(obj.apiResult) };
1254
+ if (obj.functionResult !== void 0)
1255
+ return { functionResult: obj.functionResult };
1256
+ if (obj.$unknown !== void 0)
1257
+ return { [obj.$unknown[0]]: "UNKNOWN" };
1258
+ }, "InvocationResultMemberFilterSensitiveLog");
1259
+ var ReturnControlResultsFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1260
+ ...obj,
1261
+ ...obj.returnControlInvocationResults && {
1262
+ returnControlInvocationResults: obj.returnControlInvocationResults.map(
1263
+ (item) => InvocationResultMemberFilterSensitiveLog(item)
1264
+ )
1265
+ }
1266
+ }), "ReturnControlResultsFilterSensitiveLog");
1267
+ var AgentCollaboratorInputPayloadFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1268
+ ...obj,
1269
+ ...obj.text && { text: import_smithy_client.SENSITIVE_STRING },
1270
+ ...obj.returnControlResults && {
1271
+ returnControlResults: ReturnControlResultsFilterSensitiveLog(obj.returnControlResults)
1272
+ }
1273
+ }), "AgentCollaboratorInputPayloadFilterSensitiveLog");
1274
+ var AgentCollaboratorInvocationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1275
+ ...obj,
1276
+ ...obj.input && { input: AgentCollaboratorInputPayloadFilterSensitiveLog(obj.input) }
1277
+ }), "AgentCollaboratorInvocationInputFilterSensitiveLog");
1278
+ var ApiInvocationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1279
+ ...obj,
1280
+ ...obj.apiPath && { apiPath: import_smithy_client.SENSITIVE_STRING },
1281
+ ...obj.collaboratorName && { collaboratorName: import_smithy_client.SENSITIVE_STRING }
1282
+ }), "ApiInvocationInputFilterSensitiveLog");
1283
+ var FunctionInvocationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1284
+ ...obj,
1285
+ ...obj.collaboratorName && { collaboratorName: import_smithy_client.SENSITIVE_STRING }
1286
+ }), "FunctionInvocationInputFilterSensitiveLog");
1287
+ var InvocationInputMemberFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1288
+ if (obj.apiInvocationInput !== void 0)
1289
+ return { apiInvocationInput: ApiInvocationInputFilterSensitiveLog(obj.apiInvocationInput) };
1290
+ if (obj.functionInvocationInput !== void 0)
1291
+ return { functionInvocationInput: FunctionInvocationInputFilterSensitiveLog(obj.functionInvocationInput) };
1292
+ if (obj.$unknown !== void 0)
1293
+ return { [obj.$unknown[0]]: "UNKNOWN" };
1294
+ }, "InvocationInputMemberFilterSensitiveLog");
1295
+ var ReturnControlPayloadFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1296
+ ...obj,
1297
+ ...obj.invocationInputs && {
1298
+ invocationInputs: obj.invocationInputs.map((item) => InvocationInputMemberFilterSensitiveLog(item))
1299
+ }
1300
+ }), "ReturnControlPayloadFilterSensitiveLog");
1301
+ var AgentCollaboratorOutputPayloadFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1302
+ ...obj,
1303
+ ...obj.text && { text: import_smithy_client.SENSITIVE_STRING },
1304
+ ...obj.returnControlPayload && { returnControlPayload: import_smithy_client.SENSITIVE_STRING }
1305
+ }), "AgentCollaboratorOutputPayloadFilterSensitiveLog");
1306
+ var AgentCollaboratorInvocationOutputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1307
+ ...obj,
1308
+ ...obj.output && { output: AgentCollaboratorOutputPayloadFilterSensitiveLog(obj.output) }
1309
+ }), "AgentCollaboratorInvocationOutputFilterSensitiveLog");
1092
1310
  var FlowInputContentFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1093
1311
  if (obj.document !== void 0)
1094
1312
  return { document: obj.document };
@@ -1185,6 +1403,20 @@ var InvokeFlowResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1185
1403
  ...obj,
1186
1404
  ...obj.responseStream && { responseStream: "STREAMING_CONTENT" }
1187
1405
  }), "InvokeFlowResponseFilterSensitiveLog");
1406
+ var ContentBlockFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1407
+ if (obj.text !== void 0)
1408
+ return { text: obj.text };
1409
+ if (obj.$unknown !== void 0)
1410
+ return { [obj.$unknown[0]]: "UNKNOWN" };
1411
+ }, "ContentBlockFilterSensitiveLog");
1412
+ var MessageFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1413
+ ...obj,
1414
+ ...obj.content && { content: import_smithy_client.SENSITIVE_STRING }
1415
+ }), "MessageFilterSensitiveLog");
1416
+ var ConversationHistoryFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1417
+ ...obj,
1418
+ ...obj.messages && { messages: obj.messages.map((item) => MessageFilterSensitiveLog(item)) }
1419
+ }), "ConversationHistoryFilterSensitiveLog");
1188
1420
  var ByteContentFileFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1189
1421
  ...obj,
1190
1422
  ...obj.data && { data: import_smithy_client.SENSITIVE_STRING }
@@ -1197,18 +1429,43 @@ var InputFileFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1197
1429
  ...obj,
1198
1430
  ...obj.source && { source: FileSourceFilterSensitiveLog(obj.source) }
1199
1431
  }), "InputFileFilterSensitiveLog");
1200
- var ApiResultFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1432
+ var MetadataAttributeSchemaFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1433
+ ...obj
1434
+ }), "MetadataAttributeSchemaFilterSensitiveLog");
1435
+ var ImplicitFilterConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1201
1436
  ...obj,
1202
- ...obj.apiPath && { apiPath: import_smithy_client.SENSITIVE_STRING }
1203
- }), "ApiResultFilterSensitiveLog");
1204
- var InvocationResultMemberFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1205
- if (obj.apiResult !== void 0)
1206
- return { apiResult: ApiResultFilterSensitiveLog(obj.apiResult) };
1207
- if (obj.functionResult !== void 0)
1208
- return { functionResult: obj.functionResult };
1437
+ ...obj.metadataAttributes && { metadataAttributes: import_smithy_client.SENSITIVE_STRING }
1438
+ }), "ImplicitFilterConfigurationFilterSensitiveLog");
1439
+ var RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1440
+ if (obj.fieldsToInclude !== void 0)
1441
+ return { fieldsToInclude: import_smithy_client.SENSITIVE_STRING };
1442
+ if (obj.fieldsToExclude !== void 0)
1443
+ return { fieldsToExclude: import_smithy_client.SENSITIVE_STRING };
1209
1444
  if (obj.$unknown !== void 0)
1210
1445
  return { [obj.$unknown[0]]: "UNKNOWN" };
1211
- }, "InvocationResultMemberFilterSensitiveLog");
1446
+ }, "RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog");
1447
+ var MetadataConfigurationForRerankingFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1448
+ ...obj,
1449
+ ...obj.selectiveModeConfiguration && {
1450
+ selectiveModeConfiguration: RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog(
1451
+ obj.selectiveModeConfiguration
1452
+ )
1453
+ }
1454
+ }), "MetadataConfigurationForRerankingFilterSensitiveLog");
1455
+ var VectorSearchBedrockRerankingConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1456
+ ...obj,
1457
+ ...obj.metadataConfiguration && {
1458
+ metadataConfiguration: MetadataConfigurationForRerankingFilterSensitiveLog(obj.metadataConfiguration)
1459
+ }
1460
+ }), "VectorSearchBedrockRerankingConfigurationFilterSensitiveLog");
1461
+ var VectorSearchRerankingConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1462
+ ...obj,
1463
+ ...obj.bedrockRerankingConfiguration && {
1464
+ bedrockRerankingConfiguration: VectorSearchBedrockRerankingConfigurationFilterSensitiveLog(
1465
+ obj.bedrockRerankingConfiguration
1466
+ )
1467
+ }
1468
+ }), "VectorSearchRerankingConfigurationFilterSensitiveLog");
1212
1469
  var TextResponsePartFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1213
1470
  ...obj
1214
1471
  }), "TextResponsePartFilterSensitiveLog");
@@ -1254,24 +1511,6 @@ var FilePartFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1254
1511
  ...obj,
1255
1512
  ...obj.files && { files: import_smithy_client.SENSITIVE_STRING }
1256
1513
  }), "FilePartFilterSensitiveLog");
1257
- var ApiInvocationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1258
- ...obj,
1259
- ...obj.apiPath && { apiPath: import_smithy_client.SENSITIVE_STRING }
1260
- }), "ApiInvocationInputFilterSensitiveLog");
1261
- var InvocationInputMemberFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1262
- if (obj.apiInvocationInput !== void 0)
1263
- return { apiInvocationInput: ApiInvocationInputFilterSensitiveLog(obj.apiInvocationInput) };
1264
- if (obj.functionInvocationInput !== void 0)
1265
- return { functionInvocationInput: obj.functionInvocationInput };
1266
- if (obj.$unknown !== void 0)
1267
- return { [obj.$unknown[0]]: "UNKNOWN" };
1268
- }, "InvocationInputMemberFilterSensitiveLog");
1269
- var ReturnControlPayloadFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1270
- ...obj,
1271
- ...obj.invocationInputs && {
1272
- invocationInputs: obj.invocationInputs.map((item) => InvocationInputMemberFilterSensitiveLog(item))
1273
- }
1274
- }), "ReturnControlPayloadFilterSensitiveLog");
1275
1514
  var CustomOrchestrationTraceEventFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1276
1515
  ...obj
1277
1516
  }), "CustomOrchestrationTraceEventFilterSensitiveLog");
@@ -1343,6 +1582,11 @@ var InvocationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1343
1582
  },
1344
1583
  ...obj.knowledgeBaseLookupInput && {
1345
1584
  knowledgeBaseLookupInput: KnowledgeBaseLookupInputFilterSensitiveLog(obj.knowledgeBaseLookupInput)
1585
+ },
1586
+ ...obj.agentCollaboratorInvocationInput && {
1587
+ agentCollaboratorInvocationInput: AgentCollaboratorInvocationInputFilterSensitiveLog(
1588
+ obj.agentCollaboratorInvocationInput
1589
+ )
1346
1590
  }
1347
1591
  }), "InvocationInputFilterSensitiveLog");
1348
1592
  var ModelInvocationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
@@ -1383,6 +1627,11 @@ var ObservationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1383
1627
  ...obj.actionGroupInvocationOutput && {
1384
1628
  actionGroupInvocationOutput: ActionGroupInvocationOutputFilterSensitiveLog(obj.actionGroupInvocationOutput)
1385
1629
  },
1630
+ ...obj.agentCollaboratorInvocationOutput && {
1631
+ agentCollaboratorInvocationOutput: AgentCollaboratorInvocationOutputFilterSensitiveLog(
1632
+ obj.agentCollaboratorInvocationOutput
1633
+ )
1634
+ },
1386
1635
  ...obj.knowledgeBaseLookupOutput && {
1387
1636
  knowledgeBaseLookupOutput: KnowledgeBaseLookupOutputFilterSensitiveLog(obj.knowledgeBaseLookupOutput)
1388
1637
  },
@@ -1443,6 +1692,23 @@ var PreProcessingTraceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1443
1692
  if (obj.$unknown !== void 0)
1444
1693
  return { [obj.$unknown[0]]: "UNKNOWN" };
1445
1694
  }, "PreProcessingTraceFilterSensitiveLog");
1695
+ var RoutingClassifierModelInvocationOutputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1696
+ ...obj,
1697
+ ...obj.rawResponse && { rawResponse: import_smithy_client.SENSITIVE_STRING },
1698
+ ...obj.metadata && { metadata: import_smithy_client.SENSITIVE_STRING }
1699
+ }), "RoutingClassifierModelInvocationOutputFilterSensitiveLog");
1700
+ var RoutingClassifierTraceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1701
+ if (obj.invocationInput !== void 0)
1702
+ return { invocationInput: import_smithy_client.SENSITIVE_STRING };
1703
+ if (obj.observation !== void 0)
1704
+ return { observation: import_smithy_client.SENSITIVE_STRING };
1705
+ if (obj.modelInvocationInput !== void 0)
1706
+ return { modelInvocationInput: import_smithy_client.SENSITIVE_STRING };
1707
+ if (obj.modelInvocationOutput !== void 0)
1708
+ return { modelInvocationOutput: import_smithy_client.SENSITIVE_STRING };
1709
+ if (obj.$unknown !== void 0)
1710
+ return { [obj.$unknown[0]]: "UNKNOWN" };
1711
+ }, "RoutingClassifierTraceFilterSensitiveLog");
1446
1712
  var TraceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1447
1713
  if (obj.guardrailTrace !== void 0)
1448
1714
  return { guardrailTrace: import_smithy_client.SENSITIVE_STRING };
@@ -1452,6 +1718,8 @@ var TraceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1452
1718
  return { orchestrationTrace: import_smithy_client.SENSITIVE_STRING };
1453
1719
  if (obj.postProcessingTrace !== void 0)
1454
1720
  return { postProcessingTrace: import_smithy_client.SENSITIVE_STRING };
1721
+ if (obj.routingClassifierTrace !== void 0)
1722
+ return { routingClassifierTrace: import_smithy_client.SENSITIVE_STRING };
1455
1723
  if (obj.failureTrace !== void 0)
1456
1724
  return { failureTrace: import_smithy_client.SENSITIVE_STRING };
1457
1725
  if (obj.customOrchestrationTrace !== void 0)
@@ -1461,7 +1729,9 @@ var TraceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1461
1729
  }, "TraceFilterSensitiveLog");
1462
1730
  var TracePartFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1463
1731
  ...obj,
1464
- ...obj.trace && { trace: import_smithy_client.SENSITIVE_STRING }
1732
+ ...obj.trace && { trace: import_smithy_client.SENSITIVE_STRING },
1733
+ ...obj.callerChain && { callerChain: obj.callerChain.map((item) => item) },
1734
+ ...obj.collaboratorName && { collaboratorName: import_smithy_client.SENSITIVE_STRING }
1465
1735
  }), "TracePartFilterSensitiveLog");
1466
1736
  var ResponseStreamFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1467
1737
  if (obj.chunk !== void 0)
@@ -1619,6 +1889,34 @@ var OptimizePromptResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => (
1619
1889
  ...obj,
1620
1890
  ...obj.optimizedPrompt && { optimizedPrompt: "STREAMING_CONTENT" }
1621
1891
  }), "OptimizePromptResponseFilterSensitiveLog");
1892
+ var RerankTextDocumentFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1893
+ ...obj
1894
+ }), "RerankTextDocumentFilterSensitiveLog");
1895
+ var RerankQueryFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1896
+ ...obj,
1897
+ ...obj.textQuery && { textQuery: import_smithy_client.SENSITIVE_STRING }
1898
+ }), "RerankQueryFilterSensitiveLog");
1899
+ var RerankDocumentFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1900
+ ...obj,
1901
+ ...obj.textDocument && { textDocument: import_smithy_client.SENSITIVE_STRING }
1902
+ }), "RerankDocumentFilterSensitiveLog");
1903
+ var RerankSourceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1904
+ ...obj,
1905
+ ...obj.inlineDocumentSource && { inlineDocumentSource: import_smithy_client.SENSITIVE_STRING }
1906
+ }), "RerankSourceFilterSensitiveLog");
1907
+ var RerankRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1908
+ ...obj,
1909
+ ...obj.queries && { queries: import_smithy_client.SENSITIVE_STRING },
1910
+ ...obj.sources && { sources: import_smithy_client.SENSITIVE_STRING }
1911
+ }), "RerankRequestFilterSensitiveLog");
1912
+ var RerankResultFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1913
+ ...obj,
1914
+ ...obj.document && { document: import_smithy_client.SENSITIVE_STRING }
1915
+ }), "RerankResultFilterSensitiveLog");
1916
+ var RerankResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1917
+ ...obj,
1918
+ ...obj.results && { results: obj.results.map((item) => RerankResultFilterSensitiveLog(item)) }
1919
+ }), "RerankResponseFilterSensitiveLog");
1622
1920
  var RetrieveAndGenerateInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1623
1921
  ...obj
1624
1922
  }), "RetrieveAndGenerateInputFilterSensitiveLog");
@@ -1662,6 +1960,45 @@ var RetrieveAndGenerateResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj)
1662
1960
  ...obj.output && { output: import_smithy_client.SENSITIVE_STRING },
1663
1961
  ...obj.citations && { citations: obj.citations.map((item) => CitationFilterSensitiveLog(item)) }
1664
1962
  }), "RetrieveAndGenerateResponseFilterSensitiveLog");
1963
+ var CitationEventFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1964
+ ...obj,
1965
+ ...obj.citation && { citation: CitationFilterSensitiveLog(obj.citation) }
1966
+ }), "CitationEventFilterSensitiveLog");
1967
+ var RetrieveAndGenerateOutputEventFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1968
+ ...obj
1969
+ }), "RetrieveAndGenerateOutputEventFilterSensitiveLog");
1970
+ var RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1971
+ if (obj.output !== void 0)
1972
+ return { output: import_smithy_client.SENSITIVE_STRING };
1973
+ if (obj.citation !== void 0)
1974
+ return { citation: CitationEventFilterSensitiveLog(obj.citation) };
1975
+ if (obj.guardrail !== void 0)
1976
+ return { guardrail: obj.guardrail };
1977
+ if (obj.internalServerException !== void 0)
1978
+ return { internalServerException: obj.internalServerException };
1979
+ if (obj.validationException !== void 0)
1980
+ return { validationException: obj.validationException };
1981
+ if (obj.resourceNotFoundException !== void 0)
1982
+ return { resourceNotFoundException: obj.resourceNotFoundException };
1983
+ if (obj.serviceQuotaExceededException !== void 0)
1984
+ return { serviceQuotaExceededException: obj.serviceQuotaExceededException };
1985
+ if (obj.throttlingException !== void 0)
1986
+ return { throttlingException: obj.throttlingException };
1987
+ if (obj.accessDeniedException !== void 0)
1988
+ return { accessDeniedException: obj.accessDeniedException };
1989
+ if (obj.conflictException !== void 0)
1990
+ return { conflictException: obj.conflictException };
1991
+ if (obj.dependencyFailedException !== void 0)
1992
+ return { dependencyFailedException: obj.dependencyFailedException };
1993
+ if (obj.badGatewayException !== void 0)
1994
+ return { badGatewayException: obj.badGatewayException };
1995
+ if (obj.$unknown !== void 0)
1996
+ return { [obj.$unknown[0]]: "UNKNOWN" };
1997
+ }, "RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog");
1998
+ var RetrieveAndGenerateStreamResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1999
+ ...obj,
2000
+ ...obj.stream && { stream: "STREAMING_CONTENT" }
2001
+ }), "RetrieveAndGenerateStreamResponseFilterSensitiveLog");
1665
2002
  var KnowledgeBaseQueryFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1666
2003
  ...obj
1667
2004
  }), "KnowledgeBaseQueryFilterSensitiveLog");
@@ -1707,7 +2044,13 @@ var RetrievalFilterFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
1707
2044
  }, "RetrievalFilterFilterSensitiveLog");
1708
2045
  var KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1709
2046
  ...obj,
1710
- ...obj.filter && { filter: import_smithy_client.SENSITIVE_STRING }
2047
+ ...obj.filter && { filter: import_smithy_client.SENSITIVE_STRING },
2048
+ ...obj.rerankingConfiguration && {
2049
+ rerankingConfiguration: VectorSearchRerankingConfigurationFilterSensitiveLog(obj.rerankingConfiguration)
2050
+ },
2051
+ ...obj.implicitFilterConfiguration && {
2052
+ implicitFilterConfiguration: ImplicitFilterConfigurationFilterSensitiveLog(obj.implicitFilterConfiguration)
2053
+ }
1711
2054
  }), "KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog");
1712
2055
  var KnowledgeBaseRetrievalConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1713
2056
  ...obj,
@@ -1780,6 +2123,15 @@ var RetrieveAndGenerateRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj)
1780
2123
  )
1781
2124
  }
1782
2125
  }), "RetrieveAndGenerateRequestFilterSensitiveLog");
2126
+ var RetrieveAndGenerateStreamRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
2127
+ ...obj,
2128
+ ...obj.input && { input: import_smithy_client.SENSITIVE_STRING },
2129
+ ...obj.retrieveAndGenerateConfiguration && {
2130
+ retrieveAndGenerateConfiguration: RetrieveAndGenerateConfigurationFilterSensitiveLog(
2131
+ obj.retrieveAndGenerateConfiguration
2132
+ )
2133
+ }
2134
+ }), "RetrieveAndGenerateStreamRequestFilterSensitiveLog");
1783
2135
  var SessionStateFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1784
2136
  ...obj,
1785
2137
  ...obj.returnControlInvocationResults && {
@@ -1792,6 +2144,9 @@ var SessionStateFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
1792
2144
  knowledgeBaseConfigurations: obj.knowledgeBaseConfigurations.map(
1793
2145
  (item) => KnowledgeBaseConfigurationFilterSensitiveLog(item)
1794
2146
  )
2147
+ },
2148
+ ...obj.conversationHistory && {
2149
+ conversationHistory: ConversationHistoryFilterSensitiveLog(obj.conversationHistory)
1795
2150
  }
1796
2151
  }), "SessionStateFilterSensitiveLog");
1797
2152
  var InvokeAgentRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
@@ -1832,9 +2187,10 @@ var se_GetAgentMemoryCommand = /* @__PURE__ */ __name(async (input, context) =>
1832
2187
  }, "se_GetAgentMemoryCommand");
1833
2188
  var se_InvokeAgentCommand = /* @__PURE__ */ __name(async (input, context) => {
1834
2189
  const b = (0, import_core.requestBuilder)(input, context);
1835
- const headers = {
1836
- "content-type": "application/json"
1837
- };
2190
+ const headers = (0, import_smithy_client.map)({}, import_smithy_client.isSerializableHeaderValue, {
2191
+ "content-type": "application/json",
2192
+ [_xasa]: input[_sA]
2193
+ });
1838
2194
  b.bp("/agents/{agentId}/agentAliases/{agentAliasId}/sessions/{sessionId}/text");
1839
2195
  b.p("agentId", () => input.agentId, "{agentId}", false);
1840
2196
  b.p("agentAliasId", () => input.agentAliasId, "{agentAliasId}", false);
@@ -1914,6 +2270,24 @@ var se_OptimizePromptCommand = /* @__PURE__ */ __name(async (input, context) =>
1914
2270
  b.m("POST").h(headers).b(body);
1915
2271
  return b.build();
1916
2272
  }, "se_OptimizePromptCommand");
2273
+ var se_RerankCommand = /* @__PURE__ */ __name(async (input, context) => {
2274
+ const b = (0, import_core.requestBuilder)(input, context);
2275
+ const headers = {
2276
+ "content-type": "application/json"
2277
+ };
2278
+ b.bp("/rerank");
2279
+ let body;
2280
+ body = JSON.stringify(
2281
+ (0, import_smithy_client.take)(input, {
2282
+ nextToken: [],
2283
+ queries: (_) => (0, import_smithy_client._json)(_),
2284
+ rerankingConfiguration: (_) => se_RerankingConfiguration(_, context),
2285
+ sources: (_) => se_RerankSourcesList(_, context)
2286
+ })
2287
+ );
2288
+ b.m("POST").h(headers).b(body);
2289
+ return b.build();
2290
+ }, "se_RerankCommand");
1917
2291
  var se_RetrieveCommand = /* @__PURE__ */ __name(async (input, context) => {
1918
2292
  const b = (0, import_core.requestBuilder)(input, context);
1919
2293
  const headers = {
@@ -1924,6 +2298,7 @@ var se_RetrieveCommand = /* @__PURE__ */ __name(async (input, context) => {
1924
2298
  let body;
1925
2299
  body = JSON.stringify(
1926
2300
  (0, import_smithy_client.take)(input, {
2301
+ guardrailConfiguration: (_) => (0, import_smithy_client._json)(_),
1927
2302
  nextToken: [],
1928
2303
  retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context),
1929
2304
  retrievalQuery: (_) => (0, import_smithy_client._json)(_)
@@ -1950,6 +2325,24 @@ var se_RetrieveAndGenerateCommand = /* @__PURE__ */ __name(async (input, context
1950
2325
  b.m("POST").h(headers).b(body);
1951
2326
  return b.build();
1952
2327
  }, "se_RetrieveAndGenerateCommand");
2328
+ var se_RetrieveAndGenerateStreamCommand = /* @__PURE__ */ __name(async (input, context) => {
2329
+ const b = (0, import_core.requestBuilder)(input, context);
2330
+ const headers = {
2331
+ "content-type": "application/json"
2332
+ };
2333
+ b.bp("/retrieveAndGenerateStream");
2334
+ let body;
2335
+ body = JSON.stringify(
2336
+ (0, import_smithy_client.take)(input, {
2337
+ input: (_) => (0, import_smithy_client._json)(_),
2338
+ retrieveAndGenerateConfiguration: (_) => se_RetrieveAndGenerateConfiguration(_, context),
2339
+ sessionConfiguration: (_) => (0, import_smithy_client._json)(_),
2340
+ sessionId: []
2341
+ })
2342
+ );
2343
+ b.m("POST").h(headers).b(body);
2344
+ return b.build();
2345
+ }, "se_RetrieveAndGenerateStreamCommand");
1953
2346
  var de_DeleteAgentMemoryCommand = /* @__PURE__ */ __name(async (output, context) => {
1954
2347
  if (output.statusCode !== 202 && output.statusCode >= 300) {
1955
2348
  return de_CommandError(output, context);
@@ -2024,6 +2417,21 @@ var de_OptimizePromptCommand = /* @__PURE__ */ __name(async (output, context) =>
2024
2417
  contents.optimizedPrompt = de_OptimizedPromptStream(data, context);
2025
2418
  return contents;
2026
2419
  }, "de_OptimizePromptCommand");
2420
+ var de_RerankCommand = /* @__PURE__ */ __name(async (output, context) => {
2421
+ if (output.statusCode !== 200 && output.statusCode >= 300) {
2422
+ return de_CommandError(output, context);
2423
+ }
2424
+ const contents = (0, import_smithy_client.map)({
2425
+ $metadata: deserializeMetadata(output)
2426
+ });
2427
+ const data = (0, import_smithy_client.expectNonNull)((0, import_smithy_client.expectObject)(await (0, import_core2.parseJsonBody)(output.body, context)), "body");
2428
+ const doc = (0, import_smithy_client.take)(data, {
2429
+ nextToken: import_smithy_client.expectString,
2430
+ results: (_) => de_RerankResultsList(_, context)
2431
+ });
2432
+ Object.assign(contents, doc);
2433
+ return contents;
2434
+ }, "de_RerankCommand");
2027
2435
  var de_RetrieveCommand = /* @__PURE__ */ __name(async (output, context) => {
2028
2436
  if (output.statusCode !== 200 && output.statusCode >= 300) {
2029
2437
  return de_CommandError(output, context);
@@ -2033,6 +2441,7 @@ var de_RetrieveCommand = /* @__PURE__ */ __name(async (output, context) => {
2033
2441
  });
2034
2442
  const data = (0, import_smithy_client.expectNonNull)((0, import_smithy_client.expectObject)(await (0, import_core2.parseJsonBody)(output.body, context)), "body");
2035
2443
  const doc = (0, import_smithy_client.take)(data, {
2444
+ guardrailAction: import_smithy_client.expectString,
2036
2445
  nextToken: import_smithy_client.expectString,
2037
2446
  retrievalResults: (_) => de_KnowledgeBaseRetrievalResults(_, context)
2038
2447
  });
@@ -2056,6 +2465,18 @@ var de_RetrieveAndGenerateCommand = /* @__PURE__ */ __name(async (output, contex
2056
2465
  Object.assign(contents, doc);
2057
2466
  return contents;
2058
2467
  }, "de_RetrieveAndGenerateCommand");
2468
+ var de_RetrieveAndGenerateStreamCommand = /* @__PURE__ */ __name(async (output, context) => {
2469
+ if (output.statusCode !== 200 && output.statusCode >= 300) {
2470
+ return de_CommandError(output, context);
2471
+ }
2472
+ const contents = (0, import_smithy_client.map)({
2473
+ $metadata: deserializeMetadata(output),
2474
+ [_sI]: [, output.headers[_xabkbsi]]
2475
+ });
2476
+ const data = output.body;
2477
+ contents.stream = de_RetrieveAndGenerateStreamResponseOutput(data, context);
2478
+ return contents;
2479
+ }, "de_RetrieveAndGenerateStreamCommand");
2059
2480
  var de_CommandError = /* @__PURE__ */ __name(async (output, context) => {
2060
2481
  const parsedOutput = {
2061
2482
  ...output,
@@ -2499,6 +2920,80 @@ var de_ResponseStream = /* @__PURE__ */ __name((output, context) => {
2499
2920
  return { $unknown: output };
2500
2921
  });
2501
2922
  }, "de_ResponseStream");
2923
+ var de_RetrieveAndGenerateStreamResponseOutput = /* @__PURE__ */ __name((output, context) => {
2924
+ return context.eventStreamMarshaller.deserialize(output, async (event) => {
2925
+ if (event["output"] != null) {
2926
+ return {
2927
+ output: await de_RetrieveAndGenerateOutputEvent_event(event["output"], context)
2928
+ };
2929
+ }
2930
+ if (event["citation"] != null) {
2931
+ return {
2932
+ citation: await de_CitationEvent_event(event["citation"], context)
2933
+ };
2934
+ }
2935
+ if (event["guardrail"] != null) {
2936
+ return {
2937
+ guardrail: await de_GuardrailEvent_event(event["guardrail"], context)
2938
+ };
2939
+ }
2940
+ if (event["internalServerException"] != null) {
2941
+ return {
2942
+ internalServerException: await de_InternalServerException_event(event["internalServerException"], context)
2943
+ };
2944
+ }
2945
+ if (event["validationException"] != null) {
2946
+ return {
2947
+ validationException: await de_ValidationException_event(event["validationException"], context)
2948
+ };
2949
+ }
2950
+ if (event["resourceNotFoundException"] != null) {
2951
+ return {
2952
+ resourceNotFoundException: await de_ResourceNotFoundException_event(
2953
+ event["resourceNotFoundException"],
2954
+ context
2955
+ )
2956
+ };
2957
+ }
2958
+ if (event["serviceQuotaExceededException"] != null) {
2959
+ return {
2960
+ serviceQuotaExceededException: await de_ServiceQuotaExceededException_event(
2961
+ event["serviceQuotaExceededException"],
2962
+ context
2963
+ )
2964
+ };
2965
+ }
2966
+ if (event["throttlingException"] != null) {
2967
+ return {
2968
+ throttlingException: await de_ThrottlingException_event(event["throttlingException"], context)
2969
+ };
2970
+ }
2971
+ if (event["accessDeniedException"] != null) {
2972
+ return {
2973
+ accessDeniedException: await de_AccessDeniedException_event(event["accessDeniedException"], context)
2974
+ };
2975
+ }
2976
+ if (event["conflictException"] != null) {
2977
+ return {
2978
+ conflictException: await de_ConflictException_event(event["conflictException"], context)
2979
+ };
2980
+ }
2981
+ if (event["dependencyFailedException"] != null) {
2982
+ return {
2983
+ dependencyFailedException: await de_DependencyFailedException_event(
2984
+ event["dependencyFailedException"],
2985
+ context
2986
+ )
2987
+ };
2988
+ }
2989
+ if (event["badGatewayException"] != null) {
2990
+ return {
2991
+ badGatewayException: await de_BadGatewayException_event(event["badGatewayException"], context)
2992
+ };
2993
+ }
2994
+ return { $unknown: output };
2995
+ });
2996
+ }, "de_RetrieveAndGenerateStreamResponseOutput");
2502
2997
  var de_AccessDeniedException_event = /* @__PURE__ */ __name(async (output, context) => {
2503
2998
  const parsedOutput = {
2504
2999
  ...output,
@@ -2519,6 +3014,12 @@ var de_BadGatewayException_event = /* @__PURE__ */ __name(async (output, context
2519
3014
  };
2520
3015
  return de_BadGatewayExceptionRes(parsedOutput, context);
2521
3016
  }, "de_BadGatewayException_event");
3017
+ var de_CitationEvent_event = /* @__PURE__ */ __name(async (output, context) => {
3018
+ const contents = {};
3019
+ const data = await (0, import_core2.parseJsonBody)(output.body, context);
3020
+ Object.assign(contents, de_CitationEvent(data, context));
3021
+ return contents;
3022
+ }, "de_CitationEvent_event");
2522
3023
  var de_ConflictException_event = /* @__PURE__ */ __name(async (output, context) => {
2523
3024
  const parsedOutput = {
2524
3025
  ...output,
@@ -2557,6 +3058,12 @@ var de_FlowTraceEvent_event = /* @__PURE__ */ __name(async (output, context) =>
2557
3058
  Object.assign(contents, de_FlowTraceEvent(data, context));
2558
3059
  return contents;
2559
3060
  }, "de_FlowTraceEvent_event");
3061
+ var de_GuardrailEvent_event = /* @__PURE__ */ __name(async (output, context) => {
3062
+ const contents = {};
3063
+ const data = await (0, import_core2.parseJsonBody)(output.body, context);
3064
+ Object.assign(contents, (0, import_smithy_client._json)(data));
3065
+ return contents;
3066
+ }, "de_GuardrailEvent_event");
2560
3067
  var de_InlineAgentFilePart_event = /* @__PURE__ */ __name(async (output, context) => {
2561
3068
  const contents = {};
2562
3069
  const data = await (0, import_core2.parseJsonBody)(output.body, context);
@@ -2607,6 +3114,12 @@ var de_ResourceNotFoundException_event = /* @__PURE__ */ __name(async (output, c
2607
3114
  };
2608
3115
  return de_ResourceNotFoundExceptionRes(parsedOutput, context);
2609
3116
  }, "de_ResourceNotFoundException_event");
3117
+ var de_RetrieveAndGenerateOutputEvent_event = /* @__PURE__ */ __name(async (output, context) => {
3118
+ const contents = {};
3119
+ const data = await (0, import_core2.parseJsonBody)(output.body, context);
3120
+ Object.assign(contents, (0, import_smithy_client._json)(data));
3121
+ return contents;
3122
+ }, "de_RetrieveAndGenerateOutputEvent_event");
2610
3123
  var de_ReturnControlPayload_event = /* @__PURE__ */ __name(async (output, context) => {
2611
3124
  const contents = {};
2612
3125
  const data = await (0, import_core2.parseJsonBody)(output.body, context);
@@ -2652,6 +3165,18 @@ var se_AdditionalModelRequestFields = /* @__PURE__ */ __name((input, context) =>
2652
3165
  var se_AdditionalModelRequestFieldsValue = /* @__PURE__ */ __name((input, context) => {
2653
3166
  return input;
2654
3167
  }, "se_AdditionalModelRequestFieldsValue");
3168
+ var se_BedrockRerankingConfiguration = /* @__PURE__ */ __name((input, context) => {
3169
+ return (0, import_smithy_client.take)(input, {
3170
+ modelConfiguration: (_) => se_BedrockRerankingModelConfiguration(_, context),
3171
+ numberOfResults: []
3172
+ });
3173
+ }, "se_BedrockRerankingConfiguration");
3174
+ var se_BedrockRerankingModelConfiguration = /* @__PURE__ */ __name((input, context) => {
3175
+ return (0, import_smithy_client.take)(input, {
3176
+ additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
3177
+ modelArn: []
3178
+ });
3179
+ }, "se_BedrockRerankingModelConfiguration");
2655
3180
  var se_ByteContentDoc = /* @__PURE__ */ __name((input, context) => {
2656
3181
  return (0, import_smithy_client.take)(input, {
2657
3182
  contentType: [],
@@ -2809,8 +3334,10 @@ var se_KnowledgeBases = /* @__PURE__ */ __name((input, context) => {
2809
3334
  var se_KnowledgeBaseVectorSearchConfiguration = /* @__PURE__ */ __name((input, context) => {
2810
3335
  return (0, import_smithy_client.take)(input, {
2811
3336
  filter: (_) => se_RetrievalFilter(_, context),
3337
+ implicitFilterConfiguration: import_smithy_client._json,
2812
3338
  numberOfResults: [],
2813
- overrideSearchType: []
3339
+ overrideSearchType: [],
3340
+ rerankingConfiguration: (_) => se_VectorSearchRerankingConfiguration(_, context)
2814
3341
  });
2815
3342
  }, "se_KnowledgeBaseVectorSearchConfiguration");
2816
3343
  var se_OrchestrationConfiguration = /* @__PURE__ */ __name((input, context) => {
@@ -2842,6 +3369,30 @@ var se_PromptOverrideConfiguration = /* @__PURE__ */ __name((input, context) =>
2842
3369
  promptConfigurations: (_) => se_PromptConfigurations(_, context)
2843
3370
  });
2844
3371
  }, "se_PromptOverrideConfiguration");
3372
+ var se_RerankDocument = /* @__PURE__ */ __name((input, context) => {
3373
+ return (0, import_smithy_client.take)(input, {
3374
+ jsonDocument: (_) => se_Document(_, context),
3375
+ textDocument: import_smithy_client._json,
3376
+ type: []
3377
+ });
3378
+ }, "se_RerankDocument");
3379
+ var se_RerankingConfiguration = /* @__PURE__ */ __name((input, context) => {
3380
+ return (0, import_smithy_client.take)(input, {
3381
+ bedrockRerankingConfiguration: (_) => se_BedrockRerankingConfiguration(_, context),
3382
+ type: []
3383
+ });
3384
+ }, "se_RerankingConfiguration");
3385
+ var se_RerankSource = /* @__PURE__ */ __name((input, context) => {
3386
+ return (0, import_smithy_client.take)(input, {
3387
+ inlineDocumentSource: (_) => se_RerankDocument(_, context),
3388
+ type: []
3389
+ });
3390
+ }, "se_RerankSource");
3391
+ var se_RerankSourcesList = /* @__PURE__ */ __name((input, context) => {
3392
+ return input.filter((e) => e != null).map((entry) => {
3393
+ return se_RerankSource(entry, context);
3394
+ });
3395
+ }, "se_RerankSourcesList");
2845
3396
  var se_RetrievalFilter = /* @__PURE__ */ __name((input, context) => {
2846
3397
  return RetrievalFilter.visit(input, {
2847
3398
  andAll: (value) => ({ andAll: se_RetrievalFilterList(value, context) }),
@@ -2874,6 +3425,7 @@ var se_RetrieveAndGenerateConfiguration = /* @__PURE__ */ __name((input, context
2874
3425
  }, "se_RetrieveAndGenerateConfiguration");
2875
3426
  var se_SessionState = /* @__PURE__ */ __name((input, context) => {
2876
3427
  return (0, import_smithy_client.take)(input, {
3428
+ conversationHistory: import_smithy_client._json,
2877
3429
  files: (_) => se_InputFiles(_, context),
2878
3430
  invocationId: [],
2879
3431
  knowledgeBaseConfigurations: (_) => se_KnowledgeBaseConfigurations(_, context),
@@ -2890,9 +3442,35 @@ var se_TextInferenceConfig = /* @__PURE__ */ __name((input, context) => {
2890
3442
  topP: import_smithy_client.serializeFloat
2891
3443
  });
2892
3444
  }, "se_TextInferenceConfig");
3445
+ var se_VectorSearchBedrockRerankingConfiguration = /* @__PURE__ */ __name((input, context) => {
3446
+ return (0, import_smithy_client.take)(input, {
3447
+ metadataConfiguration: import_smithy_client._json,
3448
+ modelConfiguration: (_) => se_VectorSearchBedrockRerankingModelConfiguration(_, context),
3449
+ numberOfRerankedResults: []
3450
+ });
3451
+ }, "se_VectorSearchBedrockRerankingConfiguration");
3452
+ var se_VectorSearchBedrockRerankingModelConfiguration = /* @__PURE__ */ __name((input, context) => {
3453
+ return (0, import_smithy_client.take)(input, {
3454
+ additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
3455
+ modelArn: []
3456
+ });
3457
+ }, "se_VectorSearchBedrockRerankingModelConfiguration");
3458
+ var se_VectorSearchRerankingConfiguration = /* @__PURE__ */ __name((input, context) => {
3459
+ return (0, import_smithy_client.take)(input, {
3460
+ bedrockRerankingConfiguration: (_) => se_VectorSearchBedrockRerankingConfiguration(_, context),
3461
+ type: []
3462
+ });
3463
+ }, "se_VectorSearchRerankingConfiguration");
2893
3464
  var se_Document = /* @__PURE__ */ __name((input, context) => {
2894
3465
  return input;
2895
3466
  }, "se_Document");
3467
+ var de_AgentCollaboratorInvocationOutput = /* @__PURE__ */ __name((output, context) => {
3468
+ return (0, import_smithy_client.take)(output, {
3469
+ agentCollaboratorAliasArn: import_smithy_client.expectString,
3470
+ agentCollaboratorName: import_smithy_client.expectString,
3471
+ output: import_smithy_client._json
3472
+ });
3473
+ }, "de_AgentCollaboratorInvocationOutput");
2896
3474
  var de_Attribution = /* @__PURE__ */ __name((output, context) => {
2897
3475
  return (0, import_smithy_client.take)(output, {
2898
3476
  citations: (_) => de_Citations(_, context)
@@ -2904,6 +3482,11 @@ var de_Citation = /* @__PURE__ */ __name((output, context) => {
2904
3482
  retrievedReferences: (_) => de_RetrievedReferences(_, context)
2905
3483
  });
2906
3484
  }, "de_Citation");
3485
+ var de_CitationEvent = /* @__PURE__ */ __name((output, context) => {
3486
+ return (0, import_smithy_client.take)(output, {
3487
+ citation: (_) => de_Citation(_, context)
3488
+ });
3489
+ }, "de_CitationEvent");
2907
3490
  var de_Citations = /* @__PURE__ */ __name((output, context) => {
2908
3491
  const retVal = (output || []).filter((e) => e != null).map((entry) => {
2909
3492
  return de_Citation(entry, context);
@@ -3084,6 +3667,7 @@ var de_MemorySessionSummary = /* @__PURE__ */ __name((output, context) => {
3084
3667
  }, "de_MemorySessionSummary");
3085
3668
  var de_ModelInvocationInput = /* @__PURE__ */ __name((output, context) => {
3086
3669
  return (0, import_smithy_client.take)(output, {
3670
+ foundationModel: import_smithy_client.expectString,
3087
3671
  inferenceConfiguration: (_) => de_InferenceConfiguration(_, context),
3088
3672
  overrideLambda: import_smithy_client.expectString,
3089
3673
  parserMode: import_smithy_client.expectString,
@@ -3096,6 +3680,7 @@ var de_ModelInvocationInput = /* @__PURE__ */ __name((output, context) => {
3096
3680
  var de_Observation = /* @__PURE__ */ __name((output, context) => {
3097
3681
  return (0, import_smithy_client.take)(output, {
3098
3682
  actionGroupInvocationOutput: import_smithy_client._json,
3683
+ agentCollaboratorInvocationOutput: (_) => de_AgentCollaboratorInvocationOutput(_, context),
3099
3684
  codeInterpreterInvocationOutput: import_smithy_client._json,
3100
3685
  finalResponse: import_smithy_client._json,
3101
3686
  knowledgeBaseLookupOutput: (_) => de_KnowledgeBaseLookupOutput(_, context),
@@ -3177,6 +3762,26 @@ var de_PreProcessingTrace = /* @__PURE__ */ __name((output, context) => {
3177
3762
  }
3178
3763
  return { $unknown: Object.entries(output)[0] };
3179
3764
  }, "de_PreProcessingTrace");
3765
+ var de_RerankDocument = /* @__PURE__ */ __name((output, context) => {
3766
+ return (0, import_smithy_client.take)(output, {
3767
+ jsonDocument: (_) => de_Document(_, context),
3768
+ textDocument: import_smithy_client._json,
3769
+ type: import_smithy_client.expectString
3770
+ });
3771
+ }, "de_RerankDocument");
3772
+ var de_RerankResult = /* @__PURE__ */ __name((output, context) => {
3773
+ return (0, import_smithy_client.take)(output, {
3774
+ document: (_) => de_RerankDocument(_, context),
3775
+ index: import_smithy_client.expectInt32,
3776
+ relevanceScore: import_smithy_client.limitedParseFloat32
3777
+ });
3778
+ }, "de_RerankResult");
3779
+ var de_RerankResultsList = /* @__PURE__ */ __name((output, context) => {
3780
+ const retVal = (output || []).filter((e) => e != null).map((entry) => {
3781
+ return de_RerankResult(entry, context);
3782
+ });
3783
+ return retVal;
3784
+ }, "de_RerankResultsList");
3180
3785
  var de_RetrievalResultMetadata = /* @__PURE__ */ __name((output, context) => {
3181
3786
  return Object.entries(output).reduce((acc, [key, value]) => {
3182
3787
  if (value === null) {
@@ -3202,6 +3807,29 @@ var de_RetrievedReferences = /* @__PURE__ */ __name((output, context) => {
3202
3807
  });
3203
3808
  return retVal;
3204
3809
  }, "de_RetrievedReferences");
3810
+ var de_RoutingClassifierTrace = /* @__PURE__ */ __name((output, context) => {
3811
+ if (output.invocationInput != null) {
3812
+ return {
3813
+ invocationInput: (0, import_smithy_client._json)(output.invocationInput)
3814
+ };
3815
+ }
3816
+ if (output.modelInvocationInput != null) {
3817
+ return {
3818
+ modelInvocationInput: de_ModelInvocationInput(output.modelInvocationInput, context)
3819
+ };
3820
+ }
3821
+ if (output.modelInvocationOutput != null) {
3822
+ return {
3823
+ modelInvocationOutput: (0, import_smithy_client._json)(output.modelInvocationOutput)
3824
+ };
3825
+ }
3826
+ if (output.observation != null) {
3827
+ return {
3828
+ observation: de_Observation(output.observation, context)
3829
+ };
3830
+ }
3831
+ return { $unknown: Object.entries(output)[0] };
3832
+ }, "de_RoutingClassifierTrace");
3205
3833
  var de_Trace = /* @__PURE__ */ __name((output, context) => {
3206
3834
  if (output.customOrchestrationTrace != null) {
3207
3835
  return {
@@ -3233,6 +3861,11 @@ var de_Trace = /* @__PURE__ */ __name((output, context) => {
3233
3861
  preProcessingTrace: de_PreProcessingTrace((0, import_core2.awsExpectUnion)(output.preProcessingTrace), context)
3234
3862
  };
3235
3863
  }
3864
+ if (output.routingClassifierTrace != null) {
3865
+ return {
3866
+ routingClassifierTrace: de_RoutingClassifierTrace((0, import_core2.awsExpectUnion)(output.routingClassifierTrace), context)
3867
+ };
3868
+ }
3236
3869
  return { $unknown: Object.entries(output)[0] };
3237
3870
  }, "de_Trace");
3238
3871
  var de_TracePart = /* @__PURE__ */ __name((output, context) => {
@@ -3240,6 +3873,8 @@ var de_TracePart = /* @__PURE__ */ __name((output, context) => {
3240
3873
  agentAliasId: import_smithy_client.expectString,
3241
3874
  agentId: import_smithy_client.expectString,
3242
3875
  agentVersion: import_smithy_client.expectString,
3876
+ callerChain: import_smithy_client._json,
3877
+ collaboratorName: import_smithy_client.expectString,
3243
3878
  sessionId: import_smithy_client.expectString,
3244
3879
  trace: (_) => de_Trace((0, import_core2.awsExpectUnion)(_), context)
3245
3880
  });
@@ -3258,10 +3893,13 @@ var _mI = "memoryId";
3258
3893
  var _mIa = "maxItems";
3259
3894
  var _mT = "memoryType";
3260
3895
  var _nT = "nextToken";
3896
+ var _sA = "sourceArn";
3261
3897
  var _sI = "sessionId";
3262
3898
  var _xabact = "x-amzn-bedrock-agent-content-type";
3263
3899
  var _xabami = "x-amz-bedrock-agent-memory-id";
3264
3900
  var _xabasi = "x-amz-bedrock-agent-session-id";
3901
+ var _xabkbsi = "x-amzn-bedrock-knowledge-base-session-id";
3902
+ var _xasa = "x-amz-source-arn";
3265
3903
 
3266
3904
  // src/commands/DeleteAgentMemoryCommand.ts
3267
3905
  var _DeleteAgentMemoryCommand = class _DeleteAgentMemoryCommand extends import_smithy_client.Command.classBuilder().ep(commonParams).m(function(Command, cs, config, o) {
@@ -3372,6 +4010,20 @@ var _OptimizePromptCommand = class _OptimizePromptCommand extends import_smithy_
3372
4010
  __name(_OptimizePromptCommand, "OptimizePromptCommand");
3373
4011
  var OptimizePromptCommand = _OptimizePromptCommand;
3374
4012
 
4013
+ // src/commands/RerankCommand.ts
4014
+
4015
+
4016
+
4017
+ var _RerankCommand = class _RerankCommand extends import_smithy_client.Command.classBuilder().ep(commonParams).m(function(Command, cs, config, o) {
4018
+ return [
4019
+ (0, import_middleware_serde.getSerdePlugin)(config, this.serialize, this.deserialize),
4020
+ (0, import_middleware_endpoint.getEndpointPlugin)(config, Command.getEndpointParameterInstructions())
4021
+ ];
4022
+ }).s("AmazonBedrockAgentRunTimeService", "Rerank", {}).n("BedrockAgentRuntimeClient", "RerankCommand").f(RerankRequestFilterSensitiveLog, RerankResponseFilterSensitiveLog).ser(se_RerankCommand).de(de_RerankCommand).build() {
4023
+ };
4024
+ __name(_RerankCommand, "RerankCommand");
4025
+ var RerankCommand = _RerankCommand;
4026
+
3375
4027
  // src/commands/RetrieveAndGenerateCommand.ts
3376
4028
 
3377
4029
 
@@ -3386,6 +4038,27 @@ var _RetrieveAndGenerateCommand = class _RetrieveAndGenerateCommand extends impo
3386
4038
  __name(_RetrieveAndGenerateCommand, "RetrieveAndGenerateCommand");
3387
4039
  var RetrieveAndGenerateCommand = _RetrieveAndGenerateCommand;
3388
4040
 
4041
+ // src/commands/RetrieveAndGenerateStreamCommand.ts
4042
+
4043
+
4044
+
4045
+ var _RetrieveAndGenerateStreamCommand = class _RetrieveAndGenerateStreamCommand extends import_smithy_client.Command.classBuilder().ep(commonParams).m(function(Command, cs, config, o) {
4046
+ return [
4047
+ (0, import_middleware_serde.getSerdePlugin)(config, this.serialize, this.deserialize),
4048
+ (0, import_middleware_endpoint.getEndpointPlugin)(config, Command.getEndpointParameterInstructions())
4049
+ ];
4050
+ }).s("AmazonBedrockAgentRunTimeService", "RetrieveAndGenerateStream", {
4051
+ /**
4052
+ * @internal
4053
+ */
4054
+ eventStream: {
4055
+ output: true
4056
+ }
4057
+ }).n("BedrockAgentRuntimeClient", "RetrieveAndGenerateStreamCommand").f(RetrieveAndGenerateStreamRequestFilterSensitiveLog, RetrieveAndGenerateStreamResponseFilterSensitiveLog).ser(se_RetrieveAndGenerateStreamCommand).de(de_RetrieveAndGenerateStreamCommand).build() {
4058
+ };
4059
+ __name(_RetrieveAndGenerateStreamCommand, "RetrieveAndGenerateStreamCommand");
4060
+ var RetrieveAndGenerateStreamCommand = _RetrieveAndGenerateStreamCommand;
4061
+
3389
4062
  // src/commands/RetrieveCommand.ts
3390
4063
 
3391
4064
 
@@ -3408,8 +4081,10 @@ var commands = {
3408
4081
  InvokeFlowCommand,
3409
4082
  InvokeInlineAgentCommand,
3410
4083
  OptimizePromptCommand,
4084
+ RerankCommand,
3411
4085
  RetrieveCommand,
3412
- RetrieveAndGenerateCommand
4086
+ RetrieveAndGenerateCommand,
4087
+ RetrieveAndGenerateStreamCommand
3413
4088
  };
3414
4089
  var _BedrockAgentRuntime = class _BedrockAgentRuntime extends BedrockAgentRuntimeClient {
3415
4090
  };
@@ -3421,6 +4096,10 @@ var BedrockAgentRuntime = _BedrockAgentRuntime;
3421
4096
 
3422
4097
  var paginateGetAgentMemory = (0, import_core.createPaginator)(BedrockAgentRuntimeClient, GetAgentMemoryCommand, "nextToken", "nextToken", "maxItems");
3423
4098
 
4099
+ // src/pagination/RerankPaginator.ts
4100
+
4101
+ var paginateRerank = (0, import_core.createPaginator)(BedrockAgentRuntimeClient, RerankCommand, "nextToken", "nextToken", "");
4102
+
3424
4103
  // src/pagination/RetrievePaginator.ts
3425
4104
 
3426
4105
  var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClient, RetrieveCommand, "nextToken", "nextToken", "");
@@ -3438,9 +4117,12 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3438
4117
  InvokeFlowCommand,
3439
4118
  InvokeInlineAgentCommand,
3440
4119
  OptimizePromptCommand,
4120
+ RerankCommand,
3441
4121
  RetrieveAndGenerateCommand,
4122
+ RetrieveAndGenerateStreamCommand,
3442
4123
  RetrieveCommand,
3443
4124
  paginateGetAgentMemory,
4125
+ paginateRerank,
3444
4126
  paginateRetrieve,
3445
4127
  AccessDeniedException,
3446
4128
  CustomControlMethod,
@@ -3452,6 +4134,11 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3452
4134
  ParameterType,
3453
4135
  RequireConfirmation,
3454
4136
  FunctionSchema,
4137
+ ConfirmationState,
4138
+ ResponseState,
4139
+ InvocationResultMember,
4140
+ PayloadType,
4141
+ InvocationInputMember,
3455
4142
  BadGatewayException,
3456
4143
  ConflictException,
3457
4144
  DependencyFailedException,
@@ -3468,14 +4155,17 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3468
4155
  ThrottlingException,
3469
4156
  ValidationException,
3470
4157
  FlowResponseStream,
4158
+ ContentBlock,
4159
+ ConversationRole,
3471
4160
  FileSourceType,
3472
4161
  FileUseCase,
4162
+ AttributeType,
3473
4163
  SearchType,
3474
- ConfirmationState,
3475
- ResponseState,
3476
- InvocationResultMember,
4164
+ RerankingMetadataSelectionMode,
4165
+ RerankingMetadataSelectiveModeConfiguration,
4166
+ VectorSearchRerankingConfigurationType,
3477
4167
  RetrievalResultLocationType,
3478
- InvocationInputMember,
4168
+ Caller,
3479
4169
  GuardrailAction,
3480
4170
  GuardrailContentPolicyAction,
3481
4171
  GuardrailContentFilterConfidence,
@@ -3494,6 +4184,7 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3494
4184
  OrchestrationTrace,
3495
4185
  PostProcessingTrace,
3496
4186
  PreProcessingTrace,
4187
+ RoutingClassifierTrace,
3497
4188
  Trace,
3498
4189
  ResponseStream,
3499
4190
  PromptState,
@@ -3503,10 +4194,15 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3503
4194
  InputPrompt,
3504
4195
  OptimizedPrompt,
3505
4196
  OptimizedPromptStream,
4197
+ RerankQueryContentType,
4198
+ RerankingConfigurationType,
4199
+ RerankDocumentType,
4200
+ RerankSourceType,
3506
4201
  ExternalSourceType,
3507
4202
  QueryTransformationType,
3508
4203
  RetrieveAndGenerateType,
3509
4204
  GuadrailAction,
4205
+ RetrieveAndGenerateStreamResponseOutput,
3510
4206
  RetrievalFilter,
3511
4207
  ActionGroupInvocationInputFilterSensitiveLog,
3512
4208
  ActionGroupInvocationOutputFilterSensitiveLog,
@@ -3514,6 +4210,17 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3514
4210
  FunctionDefinitionFilterSensitiveLog,
3515
4211
  FunctionSchemaFilterSensitiveLog,
3516
4212
  AgentActionGroupFilterSensitiveLog,
4213
+ ApiResultFilterSensitiveLog,
4214
+ InvocationResultMemberFilterSensitiveLog,
4215
+ ReturnControlResultsFilterSensitiveLog,
4216
+ AgentCollaboratorInputPayloadFilterSensitiveLog,
4217
+ AgentCollaboratorInvocationInputFilterSensitiveLog,
4218
+ ApiInvocationInputFilterSensitiveLog,
4219
+ FunctionInvocationInputFilterSensitiveLog,
4220
+ InvocationInputMemberFilterSensitiveLog,
4221
+ ReturnControlPayloadFilterSensitiveLog,
4222
+ AgentCollaboratorOutputPayloadFilterSensitiveLog,
4223
+ AgentCollaboratorInvocationOutputFilterSensitiveLog,
3517
4224
  FlowInputContentFilterSensitiveLog,
3518
4225
  FlowInputFilterSensitiveLog,
3519
4226
  InvokeFlowRequestFilterSensitiveLog,
@@ -3530,11 +4237,18 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3530
4237
  FlowTraceEventFilterSensitiveLog,
3531
4238
  FlowResponseStreamFilterSensitiveLog,
3532
4239
  InvokeFlowResponseFilterSensitiveLog,
4240
+ ContentBlockFilterSensitiveLog,
4241
+ MessageFilterSensitiveLog,
4242
+ ConversationHistoryFilterSensitiveLog,
3533
4243
  ByteContentFileFilterSensitiveLog,
3534
4244
  FileSourceFilterSensitiveLog,
3535
4245
  InputFileFilterSensitiveLog,
3536
- ApiResultFilterSensitiveLog,
3537
- InvocationResultMemberFilterSensitiveLog,
4246
+ MetadataAttributeSchemaFilterSensitiveLog,
4247
+ ImplicitFilterConfigurationFilterSensitiveLog,
4248
+ RerankingMetadataSelectiveModeConfigurationFilterSensitiveLog,
4249
+ MetadataConfigurationForRerankingFilterSensitiveLog,
4250
+ VectorSearchBedrockRerankingConfigurationFilterSensitiveLog,
4251
+ VectorSearchRerankingConfigurationFilterSensitiveLog,
3538
4252
  TextResponsePartFilterSensitiveLog,
3539
4253
  GeneratedResponsePartFilterSensitiveLog,
3540
4254
  RetrievalResultContentFilterSensitiveLog,
@@ -3545,9 +4259,6 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3545
4259
  PayloadPartFilterSensitiveLog,
3546
4260
  OutputFileFilterSensitiveLog,
3547
4261
  FilePartFilterSensitiveLog,
3548
- ApiInvocationInputFilterSensitiveLog,
3549
- InvocationInputMemberFilterSensitiveLog,
3550
- ReturnControlPayloadFilterSensitiveLog,
3551
4262
  CustomOrchestrationTraceEventFilterSensitiveLog,
3552
4263
  CustomOrchestrationTraceFilterSensitiveLog,
3553
4264
  FailureTraceFilterSensitiveLog,
@@ -3582,6 +4293,8 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3582
4293
  PreProcessingParsedResponseFilterSensitiveLog,
3583
4294
  PreProcessingModelInvocationOutputFilterSensitiveLog,
3584
4295
  PreProcessingTraceFilterSensitiveLog,
4296
+ RoutingClassifierModelInvocationOutputFilterSensitiveLog,
4297
+ RoutingClassifierTraceFilterSensitiveLog,
3585
4298
  TraceFilterSensitiveLog,
3586
4299
  TracePartFilterSensitiveLog,
3587
4300
  ResponseStreamFilterSensitiveLog,
@@ -3603,6 +4316,13 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3603
4316
  OptimizedPromptEventFilterSensitiveLog,
3604
4317
  OptimizedPromptStreamFilterSensitiveLog,
3605
4318
  OptimizePromptResponseFilterSensitiveLog,
4319
+ RerankTextDocumentFilterSensitiveLog,
4320
+ RerankQueryFilterSensitiveLog,
4321
+ RerankDocumentFilterSensitiveLog,
4322
+ RerankSourceFilterSensitiveLog,
4323
+ RerankRequestFilterSensitiveLog,
4324
+ RerankResultFilterSensitiveLog,
4325
+ RerankResponseFilterSensitiveLog,
3606
4326
  RetrieveAndGenerateInputFilterSensitiveLog,
3607
4327
  PromptTemplateFilterSensitiveLog,
3608
4328
  ExternalSourcesGenerationConfigurationFilterSensitiveLog,
@@ -3613,6 +4333,10 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3613
4333
  OrchestrationConfigurationFilterSensitiveLog,
3614
4334
  RetrieveAndGenerateOutputFilterSensitiveLog,
3615
4335
  RetrieveAndGenerateResponseFilterSensitiveLog,
4336
+ CitationEventFilterSensitiveLog,
4337
+ RetrieveAndGenerateOutputEventFilterSensitiveLog,
4338
+ RetrieveAndGenerateStreamResponseOutputFilterSensitiveLog,
4339
+ RetrieveAndGenerateStreamResponseFilterSensitiveLog,
3616
4340
  KnowledgeBaseQueryFilterSensitiveLog,
3617
4341
  KnowledgeBaseRetrievalResultFilterSensitiveLog,
3618
4342
  RetrieveResponseFilterSensitiveLog,
@@ -3626,6 +4350,7 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
3626
4350
  RetrieveAndGenerateConfigurationFilterSensitiveLog,
3627
4351
  InvokeInlineAgentRequestFilterSensitiveLog,
3628
4352
  RetrieveAndGenerateRequestFilterSensitiveLog,
4353
+ RetrieveAndGenerateStreamRequestFilterSensitiveLog,
3629
4354
  SessionStateFilterSensitiveLog,
3630
4355
  InvokeAgentRequestFilterSensitiveLog
3631
4356
  });