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

package/dist-es/BedrockAgentRuntime.js

@@ -1,6 +1,7 @@
 import { createAggregatedClient } from "@smithy/smithy-client";
 import { BedrockAgentRuntimeClient } from "./BedrockAgentRuntimeClient";
 import { DeleteAgentMemoryCommand, } from "./commands/DeleteAgentMemoryCommand";
+import { GenerateQueryCommand, } from "./commands/GenerateQueryCommand";
 import { GetAgentMemoryCommand, } from "./commands/GetAgentMemoryCommand";
 import { InvokeAgentCommand } from "./commands/InvokeAgentCommand";
 import { InvokeFlowCommand } from "./commands/InvokeFlowCommand";
@@ -12,6 +13,7 @@ import { RetrieveAndGenerateStreamCommand, } from "./commands/RetrieveAndGenerat
 import { RetrieveCommand } from "./commands/RetrieveCommand";
 const commands = {
     DeleteAgentMemoryCommand,
+    GenerateQueryCommand,
     GetAgentMemoryCommand,
     InvokeAgentCommand,
     InvokeFlowCommand,

package/dist-es/commands/GenerateQueryCommand.js

@@ -0,0 +1,23 @@
+import { getEndpointPlugin } from "@smithy/middleware-endpoint";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+import { commonParams } from "../endpoint/EndpointParameters";
+import { GenerateQueryRequestFilterSensitiveLog, GenerateQueryResponseFilterSensitiveLog, } from "../models/models_0";
+import { de_GenerateQueryCommand, se_GenerateQueryCommand } from "../protocols/Aws_restJson1";
+export { $Command };
+export class GenerateQueryCommand extends $Command
+    .classBuilder()
+    .ep(commonParams)
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+        getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
+    ];
+})
+    .s("AmazonBedrockAgentRunTimeService", "GenerateQuery", {})
+    .n("BedrockAgentRuntimeClient", "GenerateQueryCommand")
+    .f(GenerateQueryRequestFilterSensitiveLog, GenerateQueryResponseFilterSensitiveLog)
+    .ser(se_GenerateQueryCommand)
+    .de(de_GenerateQueryCommand)
+    .build() {
+}

package/dist-es/commands/index.js

@@ -1,4 +1,5 @@
 export * from "./DeleteAgentMemoryCommand";
+export * from "./GenerateQueryCommand";
 export * from "./GetAgentMemoryCommand";
 export * from "./InvokeAgentCommand";
 export * from "./InvokeFlowCommand";

package/dist-es/models/models_0.js

@@ -67,6 +67,38 @@ export var FunctionSchema;
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(FunctionSchema || (FunctionSchema = {}));
+export const ConfirmationState = {
+    CONFIRM: "CONFIRM",
+    DENY: "DENY",
+};
+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 PayloadType = {
+    RETURN_CONTROL: "RETURN_CONTROL",
+    TEXT: "TEXT",
+};
+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 BadGatewayException extends __BaseException {
     constructor(opts) {
         super({
@@ -251,6 +283,30 @@ export var FlowResponseStream;
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(FlowResponseStream || (FlowResponseStream = {}));
+export const InputQueryType = {
+    TEXT: "TEXT",
+};
+export const QueryTransformationMode = {
+    TEXT_TO_SQL: "TEXT_TO_SQL",
+};
+export const TextToSqlConfigurationType = {
+    KNOWLEDGE_BASE: "KNOWLEDGE_BASE",
+};
+export const GeneratedQueryType = {
+    REDSHIFT_SQL: "REDSHIFT_SQL",
+};
+export var ContentBlock;
+(function (ContentBlock) {
+    ContentBlock.visit = (value, visitor) => {
+        if (value.text !== undefined)
+            return visitor.text(value.text);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(ContentBlock || (ContentBlock = {}));
+export const ConversationRole = {
+    ASSISTANT: "assistant",
+    USER: "user",
+};
 export const FileSourceType = {
     BYTE_CONTENT: "BYTE_CONTENT",
     S3: "S3",
@@ -286,42 +342,37 @@ export var RerankingMetadataSelectiveModeConfiguration;
 export const VectorSearchRerankingConfigurationType = {
     BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL",
 };
-export const ConfirmationState = {
-    CONFIRM: "CONFIRM",
-    DENY: "DENY",
+export const RetrievalResultContentColumnType = {
+    BLOB: "BLOB",
+    BOOLEAN: "BOOLEAN",
+    DOUBLE: "DOUBLE",
+    LONG: "LONG",
+    NULL: "NULL",
+    STRING: "STRING",
 };
-export const ResponseState = {
-    FAILURE: "FAILURE",
-    REPROMPT: "REPROMPT",
+export const RetrievalResultContentType = {
+    IMAGE: "IMAGE",
+    ROW: "ROW",
+    TEXT: "TEXT",
 };
-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 = {
     CONFLUENCE: "CONFLUENCE",
     CUSTOM: "CUSTOM",
+    KENDRA: "KENDRA",
     S3: "S3",
     SALESFORCE: "SALESFORCE",
     SHAREPOINT: "SHAREPOINT",
+    SQL: "SQL",
     WEB: "WEB",
 };
-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);
+export var Caller;
+(function (Caller) {
+    Caller.visit = (value, visitor) => {
+        if (value.agentAliasArn !== undefined)
+            return visitor.agentAliasArn(value.agentAliasArn);
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
-})(InvocationInputMember || (InvocationInputMember = {}));
+})(Caller || (Caller = {}));
 export const GuardrailAction = {
     INTERVENED: "INTERVENED",
     NONE: "NONE",
@@ -395,6 +446,7 @@ export const GuardrailManagedWordType = {
 export const InvocationType = {
     ACTION_GROUP: "ACTION_GROUP",
     ACTION_GROUP_CODE_INTERPRETER: "ACTION_GROUP_CODE_INTERPRETER",
+    AGENT_COLLABORATOR: "AGENT_COLLABORATOR",
     FINISH: "FINISH",
     KNOWLEDGE_BASE: "KNOWLEDGE_BASE",
 };
@@ -416,6 +468,7 @@ export const Source = {
 };
 export const Type = {
     ACTION_GROUP: "ACTION_GROUP",
+    AGENT_COLLABORATOR: "AGENT_COLLABORATOR",
     ASK_USER: "ASK_USER",
     FINISH: "FINISH",
     KNOWLEDGE_BASE: "KNOWLEDGE_BASE",
@@ -457,6 +510,20 @@ export var PreProcessingTrace;
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(PreProcessingTrace || (PreProcessingTrace = {}));
+export var RoutingClassifierTrace;
+(function (RoutingClassifierTrace) {
+    RoutingClassifierTrace.visit = (value, visitor) => {
+        if (value.invocationInput !== undefined)
+            return visitor.invocationInput(value.invocationInput);
+        if (value.observation !== undefined)
+            return visitor.observation(value.observation);
+        if (value.modelInvocationInput !== undefined)
+            return visitor.modelInvocationInput(value.modelInvocationInput);
+        if (value.modelInvocationOutput !== undefined)
+            return visitor.modelInvocationOutput(value.modelInvocationOutput);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(RoutingClassifierTrace || (RoutingClassifierTrace = {}));
 export var Trace;
 (function (Trace) {
     Trace.visit = (value, visitor) => {
@@ -468,6 +535,8 @@ export var Trace;
             return visitor.orchestrationTrace(value.orchestrationTrace);
         if (value.postProcessingTrace !== undefined)
             return visitor.postProcessingTrace(value.postProcessingTrace);
+        if (value.routingClassifierTrace !== undefined)
+            return visitor.routingClassifierTrace(value.routingClassifierTrace);
         if (value.failureTrace !== undefined)
             return visitor.failureTrace(value.failureTrace);
         if (value.customOrchestrationTrace !== undefined)
@@ -719,6 +788,67 @@ export const AgentActionGroupFilterSensitiveLog = (obj) => ({
     ...(obj.apiSchema && { apiSchema: APISchemaFilterSensitiveLog(obj.apiSchema) }),
     ...(obj.functionSchema && { functionSchema: FunctionSchemaFilterSensitiveLog(obj.functionSchema) }),
 });
+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 ReturnControlResultsFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.returnControlInvocationResults && {
+        returnControlInvocationResults: obj.returnControlInvocationResults.map((item) => InvocationResultMemberFilterSensitiveLog(item)),
+    }),
+});
+export const AgentCollaboratorInputPayloadFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.text && { text: SENSITIVE_STRING }),
+    ...(obj.returnControlResults && {
+        returnControlResults: ReturnControlResultsFilterSensitiveLog(obj.returnControlResults),
+    }),
+});
+export const AgentCollaboratorInvocationInputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.input && { input: AgentCollaboratorInputPayloadFilterSensitiveLog(obj.input) }),
+});
+export const ApiInvocationInputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.apiPath && { apiPath: SENSITIVE_STRING }),
+    ...(obj.collaboratorName && { collaboratorName: SENSITIVE_STRING }),
+});
+export const FunctionInvocationInputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.collaboratorName && { collaboratorName: SENSITIVE_STRING }),
+});
+export const InvocationInputMemberFilterSensitiveLog = (obj) => {
+    if (obj.apiInvocationInput !== undefined)
+        return { apiInvocationInput: ApiInvocationInputFilterSensitiveLog(obj.apiInvocationInput) };
+    if (obj.functionInvocationInput !== undefined)
+        return { functionInvocationInput: FunctionInvocationInputFilterSensitiveLog(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 AgentCollaboratorOutputPayloadFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.text && { text: SENSITIVE_STRING }),
+    ...(obj.returnControlPayload && { returnControlPayload: SENSITIVE_STRING }),
+});
+export const AgentCollaboratorInvocationOutputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.output && { output: AgentCollaboratorOutputPayloadFilterSensitiveLog(obj.output) }),
+});
 export const FlowInputContentFilterSensitiveLog = (obj) => {
     if (obj.document !== undefined)
         return { document: obj.document };
@@ -815,6 +945,34 @@ export const InvokeFlowResponseFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.responseStream && { responseStream: "STREAMING_CONTENT" }),
 });
+export const QueryGenerationInputFilterSensitiveLog = (obj) => ({
+    ...obj,
+});
+export const GenerateQueryRequestFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.queryGenerationInput && { queryGenerationInput: SENSITIVE_STRING }),
+});
+export const GeneratedQueryFilterSensitiveLog = (obj) => ({
+    ...obj,
+});
+export const GenerateQueryResponseFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.queries && { queries: SENSITIVE_STRING }),
+});
+export const ContentBlockFilterSensitiveLog = (obj) => {
+    if (obj.text !== undefined)
+        return { text: obj.text };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const MessageFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.content && { content: SENSITIVE_STRING }),
+});
+export const ConversationHistoryFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.messages && { messages: obj.messages.map((item) => MessageFilterSensitiveLog(item)) }),
+});
 export const ByteContentFileFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.data && { data: SENSITIVE_STRING }),
@@ -860,18 +1018,6 @@ export const VectorSearchRerankingConfigurationFilterSensitiveLog = (obj) => ({
         bedrockRerankingConfiguration: VectorSearchBedrockRerankingConfigurationFilterSensitiveLog(obj.bedrockRerankingConfiguration),
     }),
 });
-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 TextResponsePartFilterSensitiveLog = (obj) => ({
     ...obj,
 });
@@ -879,8 +1025,12 @@ export const GeneratedResponsePartFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.textResponsePart && { textResponsePart: SENSITIVE_STRING }),
 });
+export const RetrievalResultContentColumnFilterSensitiveLog = (obj) => ({
+    ...obj,
+});
 export const RetrievalResultContentFilterSensitiveLog = (obj) => ({
     ...obj,
+    ...(obj.row && { row: SENSITIVE_STRING }),
 });
 export const RetrievalResultLocationFilterSensitiveLog = (obj) => ({
     ...obj,
@@ -917,24 +1067,6 @@ export const FilePartFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.files && { files: SENSITIVE_STRING }),
 });
-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 CustomOrchestrationTraceEventFilterSensitiveLog = (obj) => ({
     ...obj,
 });
@@ -1007,6 +1139,9 @@ export const InvocationInputFilterSensitiveLog = (obj) => ({
     ...(obj.knowledgeBaseLookupInput && {
         knowledgeBaseLookupInput: KnowledgeBaseLookupInputFilterSensitiveLog(obj.knowledgeBaseLookupInput),
     }),
+    ...(obj.agentCollaboratorInvocationInput && {
+        agentCollaboratorInvocationInput: AgentCollaboratorInvocationInputFilterSensitiveLog(obj.agentCollaboratorInvocationInput),
+    }),
 });
 export const ModelInvocationInputFilterSensitiveLog = (obj) => ({
     ...obj,
@@ -1046,6 +1181,9 @@ export const ObservationFilterSensitiveLog = (obj) => ({
     ...(obj.actionGroupInvocationOutput && {
         actionGroupInvocationOutput: ActionGroupInvocationOutputFilterSensitiveLog(obj.actionGroupInvocationOutput),
     }),
+    ...(obj.agentCollaboratorInvocationOutput && {
+        agentCollaboratorInvocationOutput: AgentCollaboratorInvocationOutputFilterSensitiveLog(obj.agentCollaboratorInvocationOutput),
+    }),
     ...(obj.knowledgeBaseLookupOutput && {
         knowledgeBaseLookupOutput: KnowledgeBaseLookupOutputFilterSensitiveLog(obj.knowledgeBaseLookupOutput),
     }),
@@ -1106,6 +1244,23 @@ export const PreProcessingTraceFilterSensitiveLog = (obj) => {
     if (obj.$unknown !== undefined)
         return { [obj.$unknown[0]]: "UNKNOWN" };
 };
+export const RoutingClassifierModelInvocationOutputFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.rawResponse && { rawResponse: SENSITIVE_STRING }),
+    ...(obj.metadata && { metadata: SENSITIVE_STRING }),
+});
+export const RoutingClassifierTraceFilterSensitiveLog = (obj) => {
+    if (obj.invocationInput !== undefined)
+        return { invocationInput: SENSITIVE_STRING };
+    if (obj.observation !== undefined)
+        return { observation: SENSITIVE_STRING };
+    if (obj.modelInvocationInput !== undefined)
+        return { modelInvocationInput: SENSITIVE_STRING };
+    if (obj.modelInvocationOutput !== undefined)
+        return { modelInvocationOutput: SENSITIVE_STRING };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
 export const TraceFilterSensitiveLog = (obj) => {
     if (obj.guardrailTrace !== undefined)
         return { guardrailTrace: SENSITIVE_STRING };
@@ -1115,6 +1270,8 @@ export const TraceFilterSensitiveLog = (obj) => {
         return { orchestrationTrace: SENSITIVE_STRING };
     if (obj.postProcessingTrace !== undefined)
         return { postProcessingTrace: SENSITIVE_STRING };
+    if (obj.routingClassifierTrace !== undefined)
+        return { routingClassifierTrace: SENSITIVE_STRING };
     if (obj.failureTrace !== undefined)
         return { failureTrace: SENSITIVE_STRING };
     if (obj.customOrchestrationTrace !== undefined)
@@ -1125,6 +1282,8 @@ export const TraceFilterSensitiveLog = (obj) => {
 export const TracePartFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.trace && { trace: SENSITIVE_STRING }),
+    ...(obj.callerChain && { callerChain: obj.callerChain.map((item) => item) }),
+    ...(obj.collaboratorName && { collaboratorName: SENSITIVE_STRING }),
 });
 export const ResponseStreamFilterSensitiveLog = (obj) => {
     if (obj.chunk !== undefined)
@@ -1524,6 +1683,9 @@ export const SessionStateFilterSensitiveLog = (obj) => ({
     ...(obj.knowledgeBaseConfigurations && {
         knowledgeBaseConfigurations: obj.knowledgeBaseConfigurations.map((item) => KnowledgeBaseConfigurationFilterSensitiveLog(item)),
     }),
+    ...(obj.conversationHistory && {
+        conversationHistory: ConversationHistoryFilterSensitiveLog(obj.conversationHistory),
+    }),
 });
 export const InvokeAgentRequestFilterSensitiveLog = (obj) => ({
     ...obj,

package/dist-es/protocols/Aws_restJson1.js

@@ -1,6 +1,6 @@
 import { awsExpectUnion as __expectUnion, loadRestJsonErrorCode, parseJsonBody as parseBody, parseJsonErrorBody as parseErrorBody, } from "@aws-sdk/core";
 import { requestBuilder as rb } from "@smithy/core";
-import { _json, collectBody, decorateServiceException as __decorateServiceException, expectInt32 as __expectInt32, expectNonNull as __expectNonNull, expectObject as __expectObject, expectString as __expectString, limitedParseDouble as __limitedParseDouble, limitedParseFloat32 as __limitedParseFloat32, map, parseRfc3339DateTimeWithOffset as __parseRfc3339DateTimeWithOffset, serializeFloat as __serializeFloat, take, withBaseException, } from "@smithy/smithy-client";
+import { _json, collectBody, decorateServiceException as __decorateServiceException, expectInt32 as __expectInt32, expectNonNull as __expectNonNull, expectObject as __expectObject, expectString as __expectString, isSerializableHeaderValue, limitedParseDouble as __limitedParseDouble, limitedParseFloat32 as __limitedParseFloat32, map, parseRfc3339DateTimeWithOffset as __parseRfc3339DateTimeWithOffset, serializeFloat as __serializeFloat, take, withBaseException, } from "@smithy/smithy-client";
 import { BedrockAgentRuntimeServiceException as __BaseException } from "../models/BedrockAgentRuntimeServiceException";
 import { AccessDeniedException, BadGatewayException, ConflictException, DependencyFailedException, FlowInputContent, InternalServerException, ResourceNotFoundException, RetrievalFilter, ServiceQuotaExceededException, ThrottlingException, ValidationException, } from "../models/models_0";
 export const se_DeleteAgentMemoryCommand = async (input, context) => {
@@ -16,6 +16,20 @@ export const se_DeleteAgentMemoryCommand = async (input, context) => {
     b.m("DELETE").h(headers).q(query).b(body);
     return b.build();
 };
+export const se_GenerateQueryCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/generateQuery");
+    let body;
+    body = JSON.stringify(take(input, {
+        queryGenerationInput: (_) => _json(_),
+        transformationConfiguration: (_) => _json(_),
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_GetAgentMemoryCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -34,9 +48,10 @@ export const se_GetAgentMemoryCommand = async (input, context) => {
 };
 export const se_InvokeAgentCommand = async (input, context) => {
     const b = rb(input, context);
-    const headers = {
+    const headers = map({}, isSerializableHeaderValue, {
         "content-type": "application/json",
-    };
+        [_xasa]: input[_sA],
+    });
     b.bp("/agents/{agentId}/agentAliases/{agentAliasId}/sessions/{sessionId}/text");
     b.p("agentId", () => input.agentId, "{agentId}", false);
     b.p("agentAliasId", () => input.agentAliasId, "{agentAliasId}", false);
@@ -183,6 +198,20 @@ export const de_DeleteAgentMemoryCommand = async (output, context) => {
     await collectBody(output.body, context);
     return contents;
 };
+export const de_GenerateQueryCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        queries: _json,
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_GetAgentMemoryCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1232,6 +1261,7 @@ const se_RetrieveAndGenerateConfiguration = (input, context) => {
 };
 const se_SessionState = (input, context) => {
     return take(input, {
+        conversationHistory: _json,
         files: (_) => se_InputFiles(_, context),
         invocationId: [],
         knowledgeBaseConfigurations: (_) => se_KnowledgeBaseConfigurations(_, context),
@@ -1270,6 +1300,13 @@ const se_VectorSearchRerankingConfiguration = (input, context) => {
 const se_Document = (input, context) => {
     return input;
 };
+const de_AgentCollaboratorInvocationOutput = (output, context) => {
+    return take(output, {
+        agentCollaboratorAliasArn: __expectString,
+        agentCollaboratorName: __expectString,
+        output: _json,
+    });
+};
 const de_Attribution = (output, context) => {
     return take(output, {
         citations: (_) => de_Citations(_, context),
@@ -1476,6 +1513,7 @@ const de_MemorySessionSummary = (output, context) => {
 };
 const de_ModelInvocationInput = (output, context) => {
     return take(output, {
+        foundationModel: __expectString,
         inferenceConfiguration: (_) => de_InferenceConfiguration(_, context),
         overrideLambda: __expectString,
         parserMode: __expectString,
@@ -1488,6 +1526,7 @@ const de_ModelInvocationInput = (output, context) => {
 const de_Observation = (output, context) => {
     return take(output, {
         actionGroupInvocationOutput: _json,
+        agentCollaboratorInvocationOutput: (_) => de_AgentCollaboratorInvocationOutput(_, context),
         codeInterpreterInvocationOutput: _json,
         finalResponse: _json,
         knowledgeBaseLookupOutput: (_) => de_KnowledgeBaseLookupOutput(_, context),
@@ -1620,6 +1659,29 @@ const de_RetrievedReferences = (output, context) => {
     });
     return retVal;
 };
+const de_RoutingClassifierTrace = (output, context) => {
+    if (output.invocationInput != null) {
+        return {
+            invocationInput: _json(output.invocationInput),
+        };
+    }
+    if (output.modelInvocationInput != null) {
+        return {
+            modelInvocationInput: de_ModelInvocationInput(output.modelInvocationInput, context),
+        };
+    }
+    if (output.modelInvocationOutput != null) {
+        return {
+            modelInvocationOutput: _json(output.modelInvocationOutput),
+        };
+    }
+    if (output.observation != null) {
+        return {
+            observation: de_Observation(output.observation, context),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
 const de_Trace = (output, context) => {
     if (output.customOrchestrationTrace != null) {
         return {
@@ -1651,6 +1713,11 @@ const de_Trace = (output, context) => {
             preProcessingTrace: de_PreProcessingTrace(__expectUnion(output.preProcessingTrace), context),
         };
     }
+    if (output.routingClassifierTrace != null) {
+        return {
+            routingClassifierTrace: de_RoutingClassifierTrace(__expectUnion(output.routingClassifierTrace), context),
+        };
+    }
     return { $unknown: Object.entries(output)[0] };
 };
 const de_TracePart = (output, context) => {
@@ -1658,6 +1725,8 @@ const de_TracePart = (output, context) => {
         agentAliasId: __expectString,
         agentId: __expectString,
         agentVersion: __expectString,
+        callerChain: _json,
+        collaboratorName: __expectString,
         sessionId: __expectString,
         trace: (_) => de_Trace(__expectUnion(_), context),
     });
@@ -1677,8 +1746,10 @@ const _mI = "memoryId";
 const _mIa = "maxItems";
 const _mT = "memoryType";
 const _nT = "nextToken";
+const _sA = "sourceArn";
 const _sI = "sessionId";
 const _xabact = "x-amzn-bedrock-agent-content-type";
 const _xabami = "x-amz-bedrock-agent-memory-id";
 const _xabasi = "x-amz-bedrock-agent-session-id";
 const _xabkbsi = "x-amzn-bedrock-knowledge-base-session-id";
+const _xasa = "x-amz-source-arn";

package/dist-types/BedrockAgentRuntime.d.ts

@@ -1,6 +1,7 @@
 import { HttpHandlerOptions as __HttpHandlerOptions } from "@smithy/types";
 import { BedrockAgentRuntimeClient } from "./BedrockAgentRuntimeClient";
 import { DeleteAgentMemoryCommandInput, DeleteAgentMemoryCommandOutput } from "./commands/DeleteAgentMemoryCommand";
+import { GenerateQueryCommandInput, GenerateQueryCommandOutput } from "./commands/GenerateQueryCommand";
 import { GetAgentMemoryCommandInput, GetAgentMemoryCommandOutput } from "./commands/GetAgentMemoryCommand";
 import { InvokeAgentCommandInput, InvokeAgentCommandOutput } from "./commands/InvokeAgentCommand";
 import { InvokeFlowCommandInput, InvokeFlowCommandOutput } from "./commands/InvokeFlowCommand";
@@ -17,6 +18,12 @@ export interface BedrockAgentRuntime {
     deleteAgentMemory(args: DeleteAgentMemoryCommandInput, options?: __HttpHandlerOptions): Promise<DeleteAgentMemoryCommandOutput>;
     deleteAgentMemory(args: DeleteAgentMemoryCommandInput, cb: (err: any, data?: DeleteAgentMemoryCommandOutput) => void): void;
     deleteAgentMemory(args: DeleteAgentMemoryCommandInput, options: __HttpHandlerOptions, cb: (err: any, data?: DeleteAgentMemoryCommandOutput) => void): void;
+    /**
+     * @see {@link GenerateQueryCommand}
+     */
+    generateQuery(args: GenerateQueryCommandInput, options?: __HttpHandlerOptions): Promise<GenerateQueryCommandOutput>;
+    generateQuery(args: GenerateQueryCommandInput, cb: (err: any, data?: GenerateQueryCommandOutput) => void): void;
+    generateQuery(args: GenerateQueryCommandInput, options: __HttpHandlerOptions, cb: (err: any, data?: GenerateQueryCommandOutput) => void): void;
     /**
      * @see {@link GetAgentMemoryCommand}
      */

package/dist-types/BedrockAgentRuntimeClient.d.ts

@@ -9,6 +9,7 @@ import { Client as __Client, DefaultsMode as __DefaultsMode, SmithyConfiguration
 import { AwsCredentialIdentityProvider, BodyLengthCalculator as __BodyLengthCalculator, CheckOptionalClientConfig as __CheckOptionalClientConfig, ChecksumConstructor as __ChecksumConstructor, Decoder as __Decoder, Encoder as __Encoder, EventStreamSerdeProvider as __EventStreamSerdeProvider, HashConstructor as __HashConstructor, HttpHandlerOptions as __HttpHandlerOptions, Logger as __Logger, Provider as __Provider, Provider, StreamCollector as __StreamCollector, UrlParser as __UrlParser, UserAgent as __UserAgent } from "@smithy/types";
 import { HttpAuthSchemeInputConfig, HttpAuthSchemeResolvedConfig } from "./auth/httpAuthSchemeProvider";
 import { DeleteAgentMemoryCommandInput, DeleteAgentMemoryCommandOutput } from "./commands/DeleteAgentMemoryCommand";
+import { GenerateQueryCommandInput, GenerateQueryCommandOutput } from "./commands/GenerateQueryCommand";
 import { GetAgentMemoryCommandInput, GetAgentMemoryCommandOutput } from "./commands/GetAgentMemoryCommand";
 import { InvokeAgentCommandInput, InvokeAgentCommandOutput } from "./commands/InvokeAgentCommand";
 import { InvokeFlowCommandInput, InvokeFlowCommandOutput } from "./commands/InvokeFlowCommand";
@@ -24,11 +25,11 @@ export { __Client };
 /**
  * @public
  */
-export type ServiceInputTypes = DeleteAgentMemoryCommandInput | GetAgentMemoryCommandInput | InvokeAgentCommandInput | InvokeFlowCommandInput | InvokeInlineAgentCommandInput | OptimizePromptCommandInput | RerankCommandInput | RetrieveAndGenerateCommandInput | RetrieveAndGenerateStreamCommandInput | RetrieveCommandInput;
+export type ServiceInputTypes = DeleteAgentMemoryCommandInput | GenerateQueryCommandInput | GetAgentMemoryCommandInput | InvokeAgentCommandInput | InvokeFlowCommandInput | InvokeInlineAgentCommandInput | OptimizePromptCommandInput | RerankCommandInput | RetrieveAndGenerateCommandInput | RetrieveAndGenerateStreamCommandInput | RetrieveCommandInput;
 /**
  * @public
  */
-export type ServiceOutputTypes = DeleteAgentMemoryCommandOutput | GetAgentMemoryCommandOutput | InvokeAgentCommandOutput | InvokeFlowCommandOutput | InvokeInlineAgentCommandOutput | OptimizePromptCommandOutput | RerankCommandOutput | RetrieveAndGenerateCommandOutput | RetrieveAndGenerateStreamCommandOutput | RetrieveCommandOutput;
+export type ServiceOutputTypes = DeleteAgentMemoryCommandOutput | GenerateQueryCommandOutput | GetAgentMemoryCommandOutput | InvokeAgentCommandOutput | InvokeFlowCommandOutput | InvokeInlineAgentCommandOutput | OptimizePromptCommandOutput | RerankCommandOutput | RetrieveAndGenerateCommandOutput | RetrieveAndGenerateStreamCommandOutput | RetrieveCommandOutput;
 /**
  * @public
  */