@aws-sdk/client-bedrock-agent-runtime 3.556.0 → 3.561.0

package/dist-cjs/index.js

@@ -24,21 +24,32 @@ __export(src_exports, {
   AccessDeniedException: () => AccessDeniedException,
   ActionGroupInvocationInputFilterSensitiveLog: () => ActionGroupInvocationInputFilterSensitiveLog,
   ActionGroupInvocationOutputFilterSensitiveLog: () => ActionGroupInvocationOutputFilterSensitiveLog,
+  ApiInvocationInputFilterSensitiveLog: () => ApiInvocationInputFilterSensitiveLog,
+  ApiResultFilterSensitiveLog: () => ApiResultFilterSensitiveLog,
   AttributionFilterSensitiveLog: () => AttributionFilterSensitiveLog,
   BadGatewayException: () => BadGatewayException,
   BedrockAgentRuntime: () => BedrockAgentRuntime,
   BedrockAgentRuntimeClient: () => BedrockAgentRuntimeClient,
   BedrockAgentRuntimeServiceException: () => BedrockAgentRuntimeServiceException,
+  ByteContentDocFilterSensitiveLog: () => ByteContentDocFilterSensitiveLog,
   CitationFilterSensitiveLog: () => CitationFilterSensitiveLog,
   ConflictException: () => ConflictException,
   CreationMode: () => CreationMode,
   DependencyFailedException: () => DependencyFailedException,
+  ExternalSourceFilterSensitiveLog: () => ExternalSourceFilterSensitiveLog,
+  ExternalSourceType: () => ExternalSourceType,
+  ExternalSourcesGenerationConfigurationFilterSensitiveLog: () => ExternalSourcesGenerationConfigurationFilterSensitiveLog,
+  ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog: () => ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog,
   FailureTraceFilterSensitiveLog: () => FailureTraceFilterSensitiveLog,
   FinalResponseFilterSensitiveLog: () => FinalResponseFilterSensitiveLog,
   GeneratedResponsePartFilterSensitiveLog: () => GeneratedResponsePartFilterSensitiveLog,
   GenerationConfigurationFilterSensitiveLog: () => GenerationConfigurationFilterSensitiveLog,
   InternalServerException: () => InternalServerException,
   InvocationInputFilterSensitiveLog: () => InvocationInputFilterSensitiveLog,
+  InvocationInputMember: () => InvocationInputMember,
+  InvocationInputMemberFilterSensitiveLog: () => InvocationInputMemberFilterSensitiveLog,
+  InvocationResultMember: () => InvocationResultMember,
+  InvocationResultMemberFilterSensitiveLog: () => InvocationResultMemberFilterSensitiveLog,
   InvocationType: () => InvocationType,
   InvokeAgentCommand: () => InvokeAgentCommand,
   InvokeAgentRequestFilterSensitiveLog: () => InvokeAgentRequestFilterSensitiveLog,
@@ -68,6 +79,7 @@ __export(src_exports, {
   RationaleFilterSensitiveLog: () => RationaleFilterSensitiveLog,
   RepromptResponseFilterSensitiveLog: () => RepromptResponseFilterSensitiveLog,
   ResourceNotFoundException: () => ResourceNotFoundException,
+  ResponseState: () => ResponseState,
   ResponseStream: () => ResponseStream,
   ResponseStreamFilterSensitiveLog: () => ResponseStreamFilterSensitiveLog,
   RetrievalFilter: () => RetrievalFilter,
@@ -86,8 +98,10 @@ __export(src_exports, {
   RetrieveRequestFilterSensitiveLog: () => RetrieveRequestFilterSensitiveLog,
   RetrieveResponseFilterSensitiveLog: () => RetrieveResponseFilterSensitiveLog,
   RetrievedReferenceFilterSensitiveLog: () => RetrievedReferenceFilterSensitiveLog,
+  ReturnControlPayloadFilterSensitiveLog: () => ReturnControlPayloadFilterSensitiveLog,
   SearchType: () => SearchType,
   ServiceQuotaExceededException: () => ServiceQuotaExceededException,
+  SessionStateFilterSensitiveLog: () => SessionStateFilterSensitiveLog,
   Source: () => Source,
   TextResponsePartFilterSensitiveLog: () => TextResponsePartFilterSensitiveLog,
   ThrottlingException: () => ThrottlingException,
@@ -360,6 +374,20 @@ var _InternalServerException = class _InternalServerException extends BedrockAge
 };
 __name(_InternalServerException, "InternalServerException");
 var InternalServerException = _InternalServerException;
+var ResponseState = {
+  FAILURE: "FAILURE",
+  REPROMPT: "REPROMPT"
+};
+var InvocationResultMember;
+((InvocationResultMember3) => {
+  InvocationResultMember3.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.apiResult !== void 0)
+      return visitor.apiResult(value.apiResult);
+    if (value.functionResult !== void 0)
+      return visitor.functionResult(value.functionResult);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(InvocationResultMember || (InvocationResultMember = {}));
 var RetrievalResultLocationType = {
   S3: "S3"
 };
@@ -380,6 +408,16 @@ var _ResourceNotFoundException = class _ResourceNotFoundException extends Bedroc
 };
 __name(_ResourceNotFoundException, "ResourceNotFoundException");
 var ResourceNotFoundException = _ResourceNotFoundException;
+var InvocationInputMember;
+((InvocationInputMember2) => {
+  InvocationInputMember2.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.apiInvocationInput !== void 0)
+      return visitor.apiInvocationInput(value.apiInvocationInput);
+    if (value.functionInvocationInput !== void 0)
+      return visitor.functionInvocationInput(value.functionInvocationInput);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(InvocationInputMember || (InvocationInputMember = {}));
 var _ServiceQuotaExceededException = class _ServiceQuotaExceededException extends BedrockAgentRuntimeServiceException {
   /**
    * @internal
@@ -513,6 +551,8 @@ var ResponseStream;
       return visitor.chunk(value.chunk);
     if (value.trace !== void 0)
       return visitor.trace(value.trace);
+    if (value.returnControl !== void 0)
+      return visitor.returnControl(value.returnControl);
     if (value.internalServerException !== void 0)
       return visitor.internalServerException(value.internalServerException);
     if (value.validationException !== void 0)
@@ -534,11 +574,16 @@ var ResponseStream;
     return visitor._(value.$unknown[0], value.$unknown[1]);
   }, "visit");
 })(ResponseStream || (ResponseStream = {}));
+var ExternalSourceType = {
+  BYTE_CONTENT: "BYTE_CONTENT",
+  S3: "S3"
+};
 var SearchType = {
   HYBRID: "HYBRID",
   SEMANTIC: "SEMANTIC"
 };
 var RetrieveAndGenerateType = {
+  EXTERNAL_SOURCES: "EXTERNAL_SOURCES",
   KNOWLEDGE_BASE: "KNOWLEDGE_BASE"
 };
 var RetrievalFilter;
@@ -573,14 +618,36 @@ var ActionGroupInvocationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj)
   ...obj,
   ...obj.actionGroupName && { actionGroupName: import_smithy_client.SENSITIVE_STRING },
   ...obj.verb && { verb: import_smithy_client.SENSITIVE_STRING },
-  ...obj.apiPath && { apiPath: import_smithy_client.SENSITIVE_STRING }
+  ...obj.apiPath && { apiPath: import_smithy_client.SENSITIVE_STRING },
+  ...obj.function && { function: import_smithy_client.SENSITIVE_STRING }
 }), "ActionGroupInvocationInputFilterSensitiveLog");
 var ActionGroupInvocationOutputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.text && { text: import_smithy_client.SENSITIVE_STRING }
 }), "ActionGroupInvocationOutputFilterSensitiveLog");
+var ApiResultFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.apiPath && { apiPath: import_smithy_client.SENSITIVE_STRING }
+}), "ApiResultFilterSensitiveLog");
+var InvocationResultMemberFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
+  if (obj.apiResult !== void 0)
+    return { apiResult: ApiResultFilterSensitiveLog(obj.apiResult) };
+  if (obj.functionResult !== void 0)
+    return { functionResult: obj.functionResult };
+  if (obj.$unknown !== void 0)
+    return { [obj.$unknown[0]]: "UNKNOWN" };
+}, "InvocationResultMemberFilterSensitiveLog");
+var SessionStateFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.returnControlInvocationResults && {
+    returnControlInvocationResults: obj.returnControlInvocationResults.map(
+      (item) => InvocationResultMemberFilterSensitiveLog(item)
+    )
+  }
+}), "SessionStateFilterSensitiveLog");
 var InvokeAgentRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
+  ...obj.sessionState && { sessionState: SessionStateFilterSensitiveLog(obj.sessionState) },
   ...obj.inputText && { inputText: import_smithy_client.SENSITIVE_STRING }
 }), "InvokeAgentRequestFilterSensitiveLog");
 var TextResponsePartFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
@@ -620,6 +687,24 @@ var PayloadPartFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj.bytes && { bytes: import_smithy_client.SENSITIVE_STRING },
   ...obj.attribution && { attribution: AttributionFilterSensitiveLog(obj.attribution) }
 }), "PayloadPartFilterSensitiveLog");
+var ApiInvocationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.apiPath && { apiPath: import_smithy_client.SENSITIVE_STRING }
+}), "ApiInvocationInputFilterSensitiveLog");
+var InvocationInputMemberFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
+  if (obj.apiInvocationInput !== void 0)
+    return { apiInvocationInput: ApiInvocationInputFilterSensitiveLog(obj.apiInvocationInput) };
+  if (obj.functionInvocationInput !== void 0)
+    return { functionInvocationInput: obj.functionInvocationInput };
+  if (obj.$unknown !== void 0)
+    return { [obj.$unknown[0]]: "UNKNOWN" };
+}, "InvocationInputMemberFilterSensitiveLog");
+var ReturnControlPayloadFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.invocationInputs && {
+    invocationInputs: obj.invocationInputs.map((item) => InvocationInputMemberFilterSensitiveLog(item))
+  }
+}), "ReturnControlPayloadFilterSensitiveLog");
 var FailureTraceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.failureReason && { failureReason: import_smithy_client.SENSITIVE_STRING }
@@ -736,6 +821,8 @@ var ResponseStreamFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
     return { chunk: import_smithy_client.SENSITIVE_STRING };
   if (obj.trace !== void 0)
     return { trace: import_smithy_client.SENSITIVE_STRING };
+  if (obj.returnControl !== void 0)
+    return { returnControl: import_smithy_client.SENSITIVE_STRING };
   if (obj.internalServerException !== void 0)
     return { internalServerException: obj.internalServerException };
   if (obj.validationException !== void 0)
@@ -768,6 +855,26 @@ var PromptTemplateFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.textPromptTemplate && { textPromptTemplate: import_smithy_client.SENSITIVE_STRING }
 }), "PromptTemplateFilterSensitiveLog");
+var ExternalSourcesGenerationConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }
+}), "ExternalSourcesGenerationConfigurationFilterSensitiveLog");
+var ByteContentDocFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.identifier && { identifier: import_smithy_client.SENSITIVE_STRING },
+  ...obj.data && { data: import_smithy_client.SENSITIVE_STRING }
+}), "ByteContentDocFilterSensitiveLog");
+var ExternalSourceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.byteContent && { byteContent: ByteContentDocFilterSensitiveLog(obj.byteContent) }
+}), "ExternalSourceFilterSensitiveLog");
+var ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.sources && { sources: obj.sources.map((item) => ExternalSourceFilterSensitiveLog(item)) },
+  ...obj.generationConfiguration && {
+    generationConfiguration: ExternalSourcesGenerationConfigurationFilterSensitiveLog(obj.generationConfiguration)
+  }
+}), "ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog");
 var GenerationConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }
@@ -851,6 +958,11 @@ var RetrieveAndGenerateConfigurationFilterSensitiveLog = /* @__PURE__ */ __name(
     knowledgeBaseConfiguration: KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog(
       obj.knowledgeBaseConfiguration
     )
+  },
+  ...obj.externalSourcesConfiguration && {
+    externalSourcesConfiguration: ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog(
+      obj.externalSourcesConfiguration
+    )
   }
 }), "RetrieveAndGenerateConfigurationFilterSensitiveLog");
 var RetrieveAndGenerateRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
@@ -1143,6 +1255,11 @@ var de_ResponseStream = /* @__PURE__ */ __name((output, context) => {
         trace: await de_TracePart_event(event["trace"], context)
       };
     }
+    if (event["returnControl"] != null) {
+      return {
+        returnControl: await de_ReturnControlPayload_event(event["returnControl"], context)
+      };
+    }
     if (event["internalServerException"] != null) {
       return {
         internalServerException: await de_InternalServerException_event(event["internalServerException"], context)
@@ -1248,6 +1365,12 @@ var de_ResourceNotFoundException_event = /* @__PURE__ */ __name(async (output, c
   };
   return de_ResourceNotFoundExceptionRes(parsedOutput, context);
 }, "de_ResourceNotFoundException_event");
+var de_ReturnControlPayload_event = /* @__PURE__ */ __name(async (output, context) => {
+  const contents = {};
+  const data = await (0, import_core2.parseJsonBody)(output.body, context);
+  Object.assign(contents, (0, import_smithy_client._json)(data));
+  return contents;
+}, "de_ReturnControlPayload_event");
 var de_ServiceQuotaExceededException_event = /* @__PURE__ */ __name(async (output, context) => {
   const parsedOutput = {
     ...output,
@@ -1275,6 +1398,32 @@ var de_ValidationException_event = /* @__PURE__ */ __name(async (output, context
   };
   return de_ValidationExceptionRes(parsedOutput, context);
 }, "de_ValidationException_event");
+var se_ByteContentDoc = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    contentType: [],
+    data: context.base64Encoder,
+    identifier: []
+  });
+}, "se_ByteContentDoc");
+var se_ExternalSource = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    byteContent: (_) => se_ByteContentDoc(_, context),
+    s3Location: import_smithy_client._json,
+    sourceType: []
+  });
+}, "se_ExternalSource");
+var se_ExternalSources = /* @__PURE__ */ __name((input, context) => {
+  return input.filter((e) => e != null).map((entry) => {
+    return se_ExternalSource(entry, context);
+  });
+}, "se_ExternalSources");
+var se_ExternalSourcesRetrieveAndGenerateConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    generationConfiguration: import_smithy_client._json,
+    modelArn: [],
+    sources: (_) => se_ExternalSources(_, context)
+  });
+}, "se_ExternalSourcesRetrieveAndGenerateConfiguration");
 var se_FilterAttribute = /* @__PURE__ */ __name((input, context) => {
   return (0, import_smithy_client.take)(input, {
     key: [],
@@ -1327,6 +1476,7 @@ var se_RetrievalFilterList = /* @__PURE__ */ __name((input, context) => {
 }, "se_RetrievalFilterList");
 var se_RetrieveAndGenerateConfiguration = /* @__PURE__ */ __name((input, context) => {
   return (0, import_smithy_client.take)(input, {
+    externalSourcesConfiguration: (_) => se_ExternalSourcesRetrieveAndGenerateConfiguration(_, context),
     knowledgeBaseConfiguration: (_) => se_KnowledgeBaseRetrieveAndGenerateConfiguration(_, context),
     type: []
   });
@@ -1504,6 +1654,7 @@ var de_TracePart = /* @__PURE__ */ __name((output, context) => {
   return (0, import_smithy_client.take)(output, {
     agentAliasId: import_smithy_client.expectString,
     agentId: import_smithy_client.expectString,
+    agentVersion: import_smithy_client.expectString,
     sessionId: import_smithy_client.expectString,
     trace: (_) => de_Trace((0, import_core2.awsExpectUnion)(_), context)
   });
@@ -1605,8 +1756,11 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   ConflictException,
   DependencyFailedException,
   InternalServerException,
+  ResponseState,
+  InvocationResultMember,
   RetrievalResultLocationType,
   ResourceNotFoundException,
+  InvocationInputMember,
   ServiceQuotaExceededException,
   ThrottlingException,
   InvocationType,
@@ -1620,11 +1774,15 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   Trace,
   ValidationException,
   ResponseStream,
+  ExternalSourceType,
   SearchType,
   RetrieveAndGenerateType,
   RetrievalFilter,
   ActionGroupInvocationInputFilterSensitiveLog,
   ActionGroupInvocationOutputFilterSensitiveLog,
+  ApiResultFilterSensitiveLog,
+  InvocationResultMemberFilterSensitiveLog,
+  SessionStateFilterSensitiveLog,
   InvokeAgentRequestFilterSensitiveLog,
   TextResponsePartFilterSensitiveLog,
   GeneratedResponsePartFilterSensitiveLog,
@@ -1634,6 +1792,9 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   CitationFilterSensitiveLog,
   AttributionFilterSensitiveLog,
   PayloadPartFilterSensitiveLog,
+  ApiInvocationInputFilterSensitiveLog,
+  InvocationInputMemberFilterSensitiveLog,
+  ReturnControlPayloadFilterSensitiveLog,
   FailureTraceFilterSensitiveLog,
   KnowledgeBaseLookupInputFilterSensitiveLog,
   InvocationInputFilterSensitiveLog,
@@ -1656,6 +1817,10 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   InvokeAgentResponseFilterSensitiveLog,
   RetrieveAndGenerateInputFilterSensitiveLog,
   PromptTemplateFilterSensitiveLog,
+  ExternalSourcesGenerationConfigurationFilterSensitiveLog,
+  ByteContentDocFilterSensitiveLog,
+  ExternalSourceFilterSensitiveLog,
+  ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog,
   GenerationConfigurationFilterSensitiveLog,
   RetrieveAndGenerateOutputFilterSensitiveLog,
   RetrieveAndGenerateResponseFilterSensitiveLog,

package/dist-es/models/models_0.js

@@ -62,6 +62,20 @@ export class InternalServerException extends __BaseException {
         Object.setPrototypeOf(this, InternalServerException.prototype);
     }
 }
+export const ResponseState = {
+    FAILURE: "FAILURE",
+    REPROMPT: "REPROMPT",
+};
+export var InvocationResultMember;
+(function (InvocationResultMember) {
+    InvocationResultMember.visit = (value, visitor) => {
+        if (value.apiResult !== undefined)
+            return visitor.apiResult(value.apiResult);
+        if (value.functionResult !== undefined)
+            return visitor.functionResult(value.functionResult);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(InvocationResultMember || (InvocationResultMember = {}));
 export const RetrievalResultLocationType = {
     S3: "S3",
 };
@@ -77,6 +91,16 @@ export class ResourceNotFoundException extends __BaseException {
         Object.setPrototypeOf(this, ResourceNotFoundException.prototype);
     }
 }
+export var InvocationInputMember;
+(function (InvocationInputMember) {
+    InvocationInputMember.visit = (value, visitor) => {
+        if (value.apiInvocationInput !== undefined)
+            return visitor.apiInvocationInput(value.apiInvocationInput);
+        if (value.functionInvocationInput !== undefined)
+            return visitor.functionInvocationInput(value.functionInvocationInput);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(InvocationInputMember || (InvocationInputMember = {}));
 export class ServiceQuotaExceededException extends __BaseException {
     constructor(opts) {
         super({
@@ -195,6 +219,8 @@ export var ResponseStream;
             return visitor.chunk(value.chunk);
         if (value.trace !== undefined)
             return visitor.trace(value.trace);
+        if (value.returnControl !== undefined)
+            return visitor.returnControl(value.returnControl);
         if (value.internalServerException !== undefined)
             return visitor.internalServerException(value.internalServerException);
         if (value.validationException !== undefined)
@@ -216,11 +242,16 @@ export var ResponseStream;
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(ResponseStream || (ResponseStream = {}));
+export const ExternalSourceType = {
+    BYTE_CONTENT: "BYTE_CONTENT",
+    S3: "S3",
+};
 export const SearchType = {
     HYBRID: "HYBRID",
     SEMANTIC: "SEMANTIC",
 };
 export const RetrieveAndGenerateType = {
+    EXTERNAL_SOURCES: "EXTERNAL_SOURCES",
     KNOWLEDGE_BASE: "KNOWLEDGE_BASE",
 };
 export var RetrievalFilter;
@@ -256,13 +287,33 @@ export const ActionGroupInvocationInputFilterSensitiveLog = (obj) => ({
     ...(obj.actionGroupName && { actionGroupName: SENSITIVE_STRING }),
     ...(obj.verb && { verb: SENSITIVE_STRING }),
     ...(obj.apiPath && { apiPath: SENSITIVE_STRING }),
+    ...(obj.function && { function: SENSITIVE_STRING }),
 });
 export const ActionGroupInvocationOutputFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.text && { text: SENSITIVE_STRING }),
 });
+export const ApiResultFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.apiPath && { apiPath: SENSITIVE_STRING }),
+});
+export const InvocationResultMemberFilterSensitiveLog = (obj) => {
+    if (obj.apiResult !== undefined)
+        return { apiResult: ApiResultFilterSensitiveLog(obj.apiResult) };
+    if (obj.functionResult !== undefined)
+        return { functionResult: obj.functionResult };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const SessionStateFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.returnControlInvocationResults && {
+        returnControlInvocationResults: obj.returnControlInvocationResults.map((item) => InvocationResultMemberFilterSensitiveLog(item)),
+    }),
+});
 export const InvokeAgentRequestFilterSensitiveLog = (obj) => ({
     ...obj,
+    ...(obj.sessionState && { sessionState: SessionStateFilterSensitiveLog(obj.sessionState) }),
     ...(obj.inputText && { inputText: SENSITIVE_STRING }),
 });
 export const TextResponsePartFilterSensitiveLog = (obj) => ({
@@ -302,6 +353,24 @@ export const PayloadPartFilterSensitiveLog = (obj) => ({
     ...(obj.bytes && { bytes: SENSITIVE_STRING }),
     ...(obj.attribution && { attribution: AttributionFilterSensitiveLog(obj.attribution) }),
 });
+export const ApiInvocationInputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.apiPath && { apiPath: SENSITIVE_STRING }),
+});
+export const InvocationInputMemberFilterSensitiveLog = (obj) => {
+    if (obj.apiInvocationInput !== undefined)
+        return { apiInvocationInput: ApiInvocationInputFilterSensitiveLog(obj.apiInvocationInput) };
+    if (obj.functionInvocationInput !== undefined)
+        return { functionInvocationInput: obj.functionInvocationInput };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const ReturnControlPayloadFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.invocationInputs && {
+        invocationInputs: obj.invocationInputs.map((item) => InvocationInputMemberFilterSensitiveLog(item)),
+    }),
+});
 export const FailureTraceFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.failureReason && { failureReason: SENSITIVE_STRING }),
@@ -418,6 +487,8 @@ export const ResponseStreamFilterSensitiveLog = (obj) => {
         return { chunk: SENSITIVE_STRING };
     if (obj.trace !== undefined)
         return { trace: SENSITIVE_STRING };
+    if (obj.returnControl !== undefined)
+        return { returnControl: SENSITIVE_STRING };
     if (obj.internalServerException !== undefined)
         return { internalServerException: obj.internalServerException };
     if (obj.validationException !== undefined)
@@ -450,6 +521,26 @@ export const PromptTemplateFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.textPromptTemplate && { textPromptTemplate: SENSITIVE_STRING }),
 });
+export const ExternalSourcesGenerationConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }),
+});
+export const ByteContentDocFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.identifier && { identifier: SENSITIVE_STRING }),
+    ...(obj.data && { data: SENSITIVE_STRING }),
+});
+export const ExternalSourceFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.byteContent && { byteContent: ByteContentDocFilterSensitiveLog(obj.byteContent) }),
+});
+export const ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.sources && { sources: obj.sources.map((item) => ExternalSourceFilterSensitiveLog(item)) }),
+    ...(obj.generationConfiguration && {
+        generationConfiguration: ExternalSourcesGenerationConfigurationFilterSensitiveLog(obj.generationConfiguration),
+    }),
+});
 export const GenerationConfigurationFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }),
@@ -532,6 +623,9 @@ export const RetrieveAndGenerateConfigurationFilterSensitiveLog = (obj) => ({
     ...(obj.knowledgeBaseConfiguration && {
         knowledgeBaseConfiguration: KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog(obj.knowledgeBaseConfiguration),
     }),
+    ...(obj.externalSourcesConfiguration && {
+        externalSourcesConfiguration: ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog(obj.externalSourcesConfiguration),
+    }),
 });
 export const RetrieveAndGenerateRequestFilterSensitiveLog = (obj) => ({
     ...obj,

package/dist-es/protocols/Aws_restJson1.js

@@ -273,6 +273,11 @@ const de_ResponseStream = (output, context) => {
                 trace: await de_TracePart_event(event["trace"], context),
             };
         }
+        if (event["returnControl"] != null) {
+            return {
+                returnControl: await de_ReturnControlPayload_event(event["returnControl"], context),
+            };
+        }
         if (event["internalServerException"] != null) {
             return {
                 internalServerException: await de_InternalServerException_event(event["internalServerException"], context),
@@ -369,6 +374,12 @@ const de_ResourceNotFoundException_event = async (output, context) => {
     };
     return de_ResourceNotFoundExceptionRes(parsedOutput, context);
 };
+const de_ReturnControlPayload_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, _json(data));
+    return contents;
+};
 const de_ServiceQuotaExceededException_event = async (output, context) => {
     const parsedOutput = {
         ...output,
@@ -396,6 +407,34 @@ const de_ValidationException_event = async (output, context) => {
     };
     return de_ValidationExceptionRes(parsedOutput, context);
 };
+const se_ByteContentDoc = (input, context) => {
+    return take(input, {
+        contentType: [],
+        data: context.base64Encoder,
+        identifier: [],
+    });
+};
+const se_ExternalSource = (input, context) => {
+    return take(input, {
+        byteContent: (_) => se_ByteContentDoc(_, context),
+        s3Location: _json,
+        sourceType: [],
+    });
+};
+const se_ExternalSources = (input, context) => {
+    return input
+        .filter((e) => e != null)
+        .map((entry) => {
+        return se_ExternalSource(entry, context);
+    });
+};
+const se_ExternalSourcesRetrieveAndGenerateConfiguration = (input, context) => {
+    return take(input, {
+        generationConfiguration: _json,
+        modelArn: [],
+        sources: (_) => se_ExternalSources(_, context),
+    });
+};
 const se_FilterAttribute = (input, context) => {
     return take(input, {
         key: [],
@@ -450,6 +489,7 @@ const se_RetrievalFilterList = (input, context) => {
 };
 const se_RetrieveAndGenerateConfiguration = (input, context) => {
     return take(input, {
+        externalSourcesConfiguration: (_) => se_ExternalSourcesRetrieveAndGenerateConfiguration(_, context),
         knowledgeBaseConfiguration: (_) => se_KnowledgeBaseRetrieveAndGenerateConfiguration(_, context),
         type: [],
     });
@@ -633,6 +673,7 @@ const de_TracePart = (output, context) => {
     return take(output, {
         agentAliasId: __expectString,
         agentId: __expectString,
+        agentVersion: __expectString,
         sessionId: __expectString,
         trace: (_) => de_Trace(__expectUnion(_), context),
     });