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

package/README.md

@@ -210,6 +210,14 @@ DeleteAgentMemory
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent-runtime/command/DeleteAgentMemoryCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/DeleteAgentMemoryCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/DeleteAgentMemoryCommandOutput/)
 
+</details>
+<details>
+<summary>
+GenerateQuery
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent-runtime/command/GenerateQueryCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/GenerateQueryCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/GenerateQueryCommandOutput/)
+
 </details>
 <details>
 <summary>

package/dist-cjs/index.js

@@ -96,6 +96,11 @@ __export(src_exports, {
   FunctionInvocationInputFilterSensitiveLog: () => FunctionInvocationInputFilterSensitiveLog,
   FunctionSchema: () => FunctionSchema,
   FunctionSchemaFilterSensitiveLog: () => FunctionSchemaFilterSensitiveLog,
+  GenerateQueryCommand: () => GenerateQueryCommand,
+  GenerateQueryRequestFilterSensitiveLog: () => GenerateQueryRequestFilterSensitiveLog,
+  GenerateQueryResponseFilterSensitiveLog: () => GenerateQueryResponseFilterSensitiveLog,
+  GeneratedQueryFilterSensitiveLog: () => GeneratedQueryFilterSensitiveLog,
+  GeneratedQueryType: () => GeneratedQueryType,
   GeneratedResponsePartFilterSensitiveLog: () => GeneratedResponsePartFilterSensitiveLog,
   GenerationConfigurationFilterSensitiveLog: () => GenerationConfigurationFilterSensitiveLog,
   GetAgentMemoryCommand: () => GetAgentMemoryCommand,
@@ -133,6 +138,7 @@ __export(src_exports, {
   InputFileFilterSensitiveLog: () => InputFileFilterSensitiveLog,
   InputPrompt: () => InputPrompt,
   InputPromptFilterSensitiveLog: () => InputPromptFilterSensitiveLog,
+  InputQueryType: () => InputQueryType,
   InternalServerException: () => InternalServerException,
   InvocationInputFilterSensitiveLog: () => InvocationInputFilterSensitiveLog,
   InvocationInputMember: () => InvocationInputMember,
@@ -196,6 +202,8 @@ __export(src_exports, {
   PromptState: () => PromptState,
   PromptTemplateFilterSensitiveLog: () => PromptTemplateFilterSensitiveLog,
   PromptType: () => PromptType,
+  QueryGenerationInputFilterSensitiveLog: () => QueryGenerationInputFilterSensitiveLog,
+  QueryTransformationMode: () => QueryTransformationMode,
   QueryTransformationType: () => QueryTransformationType,
   RationaleFilterSensitiveLog: () => RationaleFilterSensitiveLog,
   RawResponseFilterSensitiveLog: () => RawResponseFilterSensitiveLog,
@@ -222,7 +230,10 @@ __export(src_exports, {
   ResponseStreamFilterSensitiveLog: () => ResponseStreamFilterSensitiveLog,
   RetrievalFilter: () => RetrievalFilter,
   RetrievalFilterFilterSensitiveLog: () => RetrievalFilterFilterSensitiveLog,
+  RetrievalResultContentColumnFilterSensitiveLog: () => RetrievalResultContentColumnFilterSensitiveLog,
+  RetrievalResultContentColumnType: () => RetrievalResultContentColumnType,
   RetrievalResultContentFilterSensitiveLog: () => RetrievalResultContentFilterSensitiveLog,
+  RetrievalResultContentType: () => RetrievalResultContentType,
   RetrievalResultLocationFilterSensitiveLog: () => RetrievalResultLocationFilterSensitiveLog,
   RetrievalResultLocationType: () => RetrievalResultLocationType,
   RetrieveAndGenerateCommand: () => RetrieveAndGenerateCommand,
@@ -253,6 +264,7 @@ __export(src_exports, {
   Source: () => Source,
   TextPromptFilterSensitiveLog: () => TextPromptFilterSensitiveLog,
   TextResponsePartFilterSensitiveLog: () => TextResponsePartFilterSensitiveLog,
+  TextToSqlConfigurationType: () => TextToSqlConfigurationType,
   ThrottlingException: () => ThrottlingException,
   Trace: () => Trace,
   TraceFilterSensitiveLog: () => TraceFilterSensitiveLog,
@@ -766,6 +778,18 @@ var FlowResponseStream;
     return visitor._(value.$unknown[0], value.$unknown[1]);
   }, "visit");
 })(FlowResponseStream || (FlowResponseStream = {}));
+var InputQueryType = {
+  TEXT: "TEXT"
+};
+var QueryTransformationMode = {
+  TEXT_TO_SQL: "TEXT_TO_SQL"
+};
+var TextToSqlConfigurationType = {
+  KNOWLEDGE_BASE: "KNOWLEDGE_BASE"
+};
+var GeneratedQueryType = {
+  REDSHIFT_SQL: "REDSHIFT_SQL"
+};
 var ContentBlock;
 ((ContentBlock3) => {
   ContentBlock3.visit = /* @__PURE__ */ __name((value, visitor) => {
@@ -813,12 +837,27 @@ var RerankingMetadataSelectiveModeConfiguration;
 var VectorSearchRerankingConfigurationType = {
   BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL"
 };
+var RetrievalResultContentColumnType = {
+  BLOB: "BLOB",
+  BOOLEAN: "BOOLEAN",
+  DOUBLE: "DOUBLE",
+  LONG: "LONG",
+  NULL: "NULL",
+  STRING: "STRING"
+};
+var RetrievalResultContentType = {
+  IMAGE: "IMAGE",
+  ROW: "ROW",
+  TEXT: "TEXT"
+};
 var RetrievalResultLocationType = {
   CONFLUENCE: "CONFLUENCE",
   CUSTOM: "CUSTOM",
+  KENDRA: "KENDRA",
   S3: "S3",
   SALESFORCE: "SALESFORCE",
   SHAREPOINT: "SHAREPOINT",
+  SQL: "SQL",
   WEB: "WEB"
 };
 var Caller;
@@ -1403,6 +1442,20 @@ var InvokeFlowResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.responseStream && { responseStream: "STREAMING_CONTENT" }
 }), "InvokeFlowResponseFilterSensitiveLog");
+var QueryGenerationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj
+}), "QueryGenerationInputFilterSensitiveLog");
+var GenerateQueryRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.queryGenerationInput && { queryGenerationInput: import_smithy_client.SENSITIVE_STRING }
+}), "GenerateQueryRequestFilterSensitiveLog");
+var GeneratedQueryFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj
+}), "GeneratedQueryFilterSensitiveLog");
+var GenerateQueryResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.queries && { queries: import_smithy_client.SENSITIVE_STRING }
+}), "GenerateQueryResponseFilterSensitiveLog");
 var ContentBlockFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
   if (obj.text !== void 0)
     return { text: obj.text };
@@ -1473,8 +1526,12 @@ var GeneratedResponsePartFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.textResponsePart && { textResponsePart: import_smithy_client.SENSITIVE_STRING }
 }), "GeneratedResponsePartFilterSensitiveLog");
-var RetrievalResultContentFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+var RetrievalResultContentColumnFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj
+}), "RetrievalResultContentColumnFilterSensitiveLog");
+var RetrievalResultContentFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.row && { row: import_smithy_client.SENSITIVE_STRING }
 }), "RetrievalResultContentFilterSensitiveLog");
 var RetrievalResultLocationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj
@@ -2169,6 +2226,22 @@ var se_DeleteAgentMemoryCommand = /* @__PURE__ */ __name(async (input, context)
   b.m("DELETE").h(headers).q(query).b(body);
   return b.build();
 }, "se_DeleteAgentMemoryCommand");
+var se_GenerateQueryCommand = /* @__PURE__ */ __name(async (input, context) => {
+  const b = (0, import_core.requestBuilder)(input, context);
+  const headers = {
+    "content-type": "application/json"
+  };
+  b.bp("/generateQuery");
+  let body;
+  body = JSON.stringify(
+    (0, import_smithy_client.take)(input, {
+      queryGenerationInput: (_) => (0, import_smithy_client._json)(_),
+      transformationConfiguration: (_) => (0, import_smithy_client._json)(_)
+    })
+  );
+  b.m("POST").h(headers).b(body);
+  return b.build();
+}, "se_GenerateQueryCommand");
 var se_GetAgentMemoryCommand = /* @__PURE__ */ __name(async (input, context) => {
   const b = (0, import_core.requestBuilder)(input, context);
   const headers = {};
@@ -2353,6 +2426,20 @@ var de_DeleteAgentMemoryCommand = /* @__PURE__ */ __name(async (output, context)
   await (0, import_smithy_client.collectBody)(output.body, context);
   return contents;
 }, "de_DeleteAgentMemoryCommand");
+var de_GenerateQueryCommand = /* @__PURE__ */ __name(async (output, context) => {
+  if (output.statusCode !== 200 && output.statusCode >= 300) {
+    return de_CommandError(output, context);
+  }
+  const contents = (0, import_smithy_client.map)({
+    $metadata: deserializeMetadata(output)
+  });
+  const data = (0, import_smithy_client.expectNonNull)((0, import_smithy_client.expectObject)(await (0, import_core2.parseJsonBody)(output.body, context)), "body");
+  const doc = (0, import_smithy_client.take)(data, {
+    queries: import_smithy_client._json
+  });
+  Object.assign(contents, doc);
+  return contents;
+}, "de_GenerateQueryCommand");
 var de_GetAgentMemoryCommand = /* @__PURE__ */ __name(async (output, context) => {
   if (output.statusCode !== 200 && output.statusCode >= 300) {
     return de_CommandError(output, context);
@@ -3912,6 +3999,20 @@ var _DeleteAgentMemoryCommand = class _DeleteAgentMemoryCommand extends import_s
 __name(_DeleteAgentMemoryCommand, "DeleteAgentMemoryCommand");
 var DeleteAgentMemoryCommand = _DeleteAgentMemoryCommand;
 
+// src/commands/GenerateQueryCommand.ts
+
+
+
+var _GenerateQueryCommand = class _GenerateQueryCommand extends import_smithy_client.Command.classBuilder().ep(commonParams).m(function(Command, cs, config, o) {
+  return [
+    (0, import_middleware_serde.getSerdePlugin)(config, this.serialize, this.deserialize),
+    (0, import_middleware_endpoint.getEndpointPlugin)(config, Command.getEndpointParameterInstructions())
+  ];
+}).s("AmazonBedrockAgentRunTimeService", "GenerateQuery", {}).n("BedrockAgentRuntimeClient", "GenerateQueryCommand").f(GenerateQueryRequestFilterSensitiveLog, GenerateQueryResponseFilterSensitiveLog).ser(se_GenerateQueryCommand).de(de_GenerateQueryCommand).build() {
+};
+__name(_GenerateQueryCommand, "GenerateQueryCommand");
+var GenerateQueryCommand = _GenerateQueryCommand;
+
 // src/commands/GetAgentMemoryCommand.ts
 
 
@@ -4076,6 +4177,7 @@ var RetrieveCommand = _RetrieveCommand;
 // src/BedrockAgentRuntime.ts
 var commands = {
   DeleteAgentMemoryCommand,
+  GenerateQueryCommand,
   GetAgentMemoryCommand,
   InvokeAgentCommand,
   InvokeFlowCommand,
@@ -4112,6 +4214,7 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   BedrockAgentRuntime,
   $Command,
   DeleteAgentMemoryCommand,
+  GenerateQueryCommand,
   GetAgentMemoryCommand,
   InvokeAgentCommand,
   InvokeFlowCommand,
@@ -4155,6 +4258,10 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   ThrottlingException,
   ValidationException,
   FlowResponseStream,
+  InputQueryType,
+  QueryTransformationMode,
+  TextToSqlConfigurationType,
+  GeneratedQueryType,
   ContentBlock,
   ConversationRole,
   FileSourceType,
@@ -4164,6 +4271,8 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   RerankingMetadataSelectionMode,
   RerankingMetadataSelectiveModeConfiguration,
   VectorSearchRerankingConfigurationType,
+  RetrievalResultContentColumnType,
+  RetrievalResultContentType,
   RetrievalResultLocationType,
   Caller,
   GuardrailAction,
@@ -4237,6 +4346,10 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   FlowTraceEventFilterSensitiveLog,
   FlowResponseStreamFilterSensitiveLog,
   InvokeFlowResponseFilterSensitiveLog,
+  QueryGenerationInputFilterSensitiveLog,
+  GenerateQueryRequestFilterSensitiveLog,
+  GeneratedQueryFilterSensitiveLog,
+  GenerateQueryResponseFilterSensitiveLog,
   ContentBlockFilterSensitiveLog,
   MessageFilterSensitiveLog,
   ConversationHistoryFilterSensitiveLog,
@@ -4251,6 +4364,7 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   VectorSearchRerankingConfigurationFilterSensitiveLog,
   TextResponsePartFilterSensitiveLog,
   GeneratedResponsePartFilterSensitiveLog,
+  RetrievalResultContentColumnFilterSensitiveLog,
   RetrievalResultContentFilterSensitiveLog,
   RetrievalResultLocationFilterSensitiveLog,
   RetrievedReferenceFilterSensitiveLog,

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

@@ -283,6 +283,18 @@ 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) => {
@@ -330,12 +342,27 @@ export var RerankingMetadataSelectiveModeConfiguration;
 export const VectorSearchRerankingConfigurationType = {
     BEDROCK_RERANKING_MODEL: "BEDROCK_RERANKING_MODEL",
 };
+export const RetrievalResultContentColumnType = {
+    BLOB: "BLOB",
+    BOOLEAN: "BOOLEAN",
+    DOUBLE: "DOUBLE",
+    LONG: "LONG",
+    NULL: "NULL",
+    STRING: "STRING",
+};
+export const RetrievalResultContentType = {
+    IMAGE: "IMAGE",
+    ROW: "ROW",
+    TEXT: "TEXT",
+};
 export const RetrievalResultLocationType = {
     CONFLUENCE: "CONFLUENCE",
     CUSTOM: "CUSTOM",
+    KENDRA: "KENDRA",
     S3: "S3",
     SALESFORCE: "SALESFORCE",
     SHAREPOINT: "SHAREPOINT",
+    SQL: "SQL",
     WEB: "WEB",
 };
 export var Caller;
@@ -918,6 +945,20 @@ 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 };
@@ -984,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,

package/dist-es/protocols/Aws_restJson1.js

@@ -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 = {};
@@ -184,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);

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
  */

package/dist-types/commands/GenerateQueryCommand.d.ts

@@ -0,0 +1,115 @@
+import { Command as $Command } from "@smithy/smithy-client";
+import { MetadataBearer as __MetadataBearer } from "@smithy/types";
+import { BedrockAgentRuntimeClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../BedrockAgentRuntimeClient";
+import { GenerateQueryRequest, GenerateQueryResponse } from "../models/models_0";
+/**
+ * @public
+ */
+export type { __MetadataBearer };
+export { $Command };
+/**
+ * @public
+ *
+ * The input for {@link GenerateQueryCommand}.
+ */
+export interface GenerateQueryCommandInput extends GenerateQueryRequest {
+}
+/**
+ * @public
+ *
+ * The output of {@link GenerateQueryCommand}.
+ */
+export interface GenerateQueryCommandOutput extends GenerateQueryResponse, __MetadataBearer {
+}
+declare const GenerateQueryCommand_base: {
+    new (input: GenerateQueryCommandInput): import("@smithy/smithy-client").CommandImpl<GenerateQueryCommandInput, GenerateQueryCommandOutput, BedrockAgentRuntimeClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes>;
+    new (__0_0: GenerateQueryCommandInput): import("@smithy/smithy-client").CommandImpl<GenerateQueryCommandInput, GenerateQueryCommandOutput, BedrockAgentRuntimeClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes>;
+    getEndpointParameterInstructions(): import("@smithy/middleware-endpoint").EndpointParameterInstructions;
+};
+/**
+ * <p>Generates an SQL query from a natural language query. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/knowledge-base-generate-query.html">Generate a query for structured data</a> in the Amazon Bedrock User Guide.</p>
+ * @example
+ * Use a bare-bones client and the command you need to make an API call.
+ * ```javascript
+ * import { BedrockAgentRuntimeClient, GenerateQueryCommand } from "@aws-sdk/client-bedrock-agent-runtime"; // ES Modules import
+ * // const { BedrockAgentRuntimeClient, GenerateQueryCommand } = require("@aws-sdk/client-bedrock-agent-runtime"); // CommonJS import
+ * const client = new BedrockAgentRuntimeClient(config);
+ * const input = { // GenerateQueryRequest
+ *   queryGenerationInput: { // QueryGenerationInput
+ *     type: "TEXT", // required
+ *     text: "STRING_VALUE", // required
+ *   },
+ *   transformationConfiguration: { // TransformationConfiguration
+ *     mode: "TEXT_TO_SQL", // required
+ *     textToSqlConfiguration: { // TextToSqlConfiguration
+ *       type: "KNOWLEDGE_BASE", // required
+ *       knowledgeBaseConfiguration: { // TextToSqlKnowledgeBaseConfiguration
+ *         knowledgeBaseArn: "STRING_VALUE", // required
+ *       },
+ *     },
+ *   },
+ * };
+ * const command = new GenerateQueryCommand(input);
+ * const response = await client.send(command);
+ * // { // GenerateQueryResponse
+ * //   queries: [ // GeneratedQueries
+ * //     { // GeneratedQuery
+ * //       type: "REDSHIFT_SQL",
+ * //       sql: "STRING_VALUE",
+ * //     },
+ * //   ],
+ * // };
+ *
+ * ```
+ *
+ * @param GenerateQueryCommandInput - {@link GenerateQueryCommandInput}
+ * @returns {@link GenerateQueryCommandOutput}
+ * @see {@link GenerateQueryCommandInput} for command's `input` shape.
+ * @see {@link GenerateQueryCommandOutput} for command's `response` shape.
+ * @see {@link BedrockAgentRuntimeClientResolvedConfig | config} for BedrockAgentRuntimeClient's `config` shape.
+ *
+ * @throws {@link AccessDeniedException} (client fault)
+ *  <p>The request is denied because of missing access permissions. Check your permissions and retry your request.</p>
+ *
+ * @throws {@link BadGatewayException} (server fault)
+ *  <p>There was an issue with a dependency due to a server issue. Retry your request.</p>
+ *
+ * @throws {@link ConflictException} (client fault)
+ *  <p>There was a conflict performing an operation. Resolve the conflict and retry your request.</p>
+ *
+ * @throws {@link DependencyFailedException} (client fault)
+ *  <p>There was an issue with a dependency. Check the resource configurations and retry the request.</p>
+ *
+ * @throws {@link InternalServerException} (server fault)
+ *  <p>An internal server error occurred. Retry your request.</p>
+ *
+ * @throws {@link ResourceNotFoundException} (client fault)
+ *  <p>The specified resource Amazon Resource Name (ARN) was not found. Check the Amazon Resource Name (ARN) and try your request again.</p>
+ *
+ * @throws {@link ServiceQuotaExceededException} (client fault)
+ *  <p>The number of requests exceeds the service quota. Resubmit your request later.</p>
+ *
+ * @throws {@link ThrottlingException} (client fault)
+ *  <p>The number of requests exceeds the limit. Resubmit your request later.</p>
+ *
+ * @throws {@link ValidationException} (client fault)
+ *  <p>Input validation failed. Check your request parameters and retry the request.</p>
+ *
+ * @throws {@link BedrockAgentRuntimeServiceException}
+ * <p>Base exception class for all service exceptions from BedrockAgentRuntime service.</p>
+ *
+ * @public
+ */
+export declare class GenerateQueryCommand extends GenerateQueryCommand_base {
+    /** @internal type navigation helper, not in runtime. */
+    protected static __types: {
+        api: {
+            input: GenerateQueryRequest;
+            output: GenerateQueryResponse;
+        };
+        sdk: {
+            input: GenerateQueryCommandInput;
+            output: GenerateQueryCommandOutput;
+        };
+    };
+}