@aws-sdk/client-bedrock-agent-runtime 3.696.0 → 3.697.0

package/README.md

@@ -234,6 +234,14 @@ InvokeFlow
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent-runtime/command/InvokeFlowCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/InvokeFlowCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/InvokeFlowCommandOutput/)
 
+</details>
+<details>
+<summary>
+OptimizePrompt
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent-runtime/command/OptimizePromptCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/OptimizePromptCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/OptimizePromptCommandOutput/)
+
 </details>
 <details>
 <summary>

package/dist-cjs/index.js

@@ -25,6 +25,7 @@ __export(src_exports, {
   ActionGroupInvocationInputFilterSensitiveLog: () => ActionGroupInvocationInputFilterSensitiveLog,
   ActionGroupInvocationOutputFilterSensitiveLog: () => ActionGroupInvocationOutputFilterSensitiveLog,
   ActionInvocationType: () => ActionInvocationType,
+  AnalyzePromptEventFilterSensitiveLog: () => AnalyzePromptEventFilterSensitiveLog,
   ApiInvocationInputFilterSensitiveLog: () => ApiInvocationInputFilterSensitiveLog,
   ApiResultFilterSensitiveLog: () => ApiResultFilterSensitiveLog,
   AttributionFilterSensitiveLog: () => AttributionFilterSensitiveLog,
@@ -99,6 +100,8 @@ __export(src_exports, {
   GuardrailWordPolicyAction: () => GuardrailWordPolicyAction,
   GuardrailWordPolicyAssessmentFilterSensitiveLog: () => GuardrailWordPolicyAssessmentFilterSensitiveLog,
   InputFileFilterSensitiveLog: () => InputFileFilterSensitiveLog,
+  InputPrompt: () => InputPrompt,
+  InputPromptFilterSensitiveLog: () => InputPromptFilterSensitiveLog,
   InternalServerException: () => InternalServerException,
   InvocationInputFilterSensitiveLog: () => InvocationInputFilterSensitiveLog,
   InvocationInputMember: () => InvocationInputMember,
@@ -126,6 +129,14 @@ __export(src_exports, {
   ModelInvocationInputFilterSensitiveLog: () => ModelInvocationInputFilterSensitiveLog,
   NodeType: () => NodeType,
   ObservationFilterSensitiveLog: () => ObservationFilterSensitiveLog,
+  OptimizePromptCommand: () => OptimizePromptCommand,
+  OptimizePromptRequestFilterSensitiveLog: () => OptimizePromptRequestFilterSensitiveLog,
+  OptimizePromptResponseFilterSensitiveLog: () => OptimizePromptResponseFilterSensitiveLog,
+  OptimizedPrompt: () => OptimizedPrompt,
+  OptimizedPromptEventFilterSensitiveLog: () => OptimizedPromptEventFilterSensitiveLog,
+  OptimizedPromptFilterSensitiveLog: () => OptimizedPromptFilterSensitiveLog,
+  OptimizedPromptStream: () => OptimizedPromptStream,
+  OptimizedPromptStreamFilterSensitiveLog: () => OptimizedPromptStreamFilterSensitiveLog,
   OrchestrationConfigurationFilterSensitiveLog: () => OrchestrationConfigurationFilterSensitiveLog,
   OrchestrationModelInvocationOutputFilterSensitiveLog: () => OrchestrationModelInvocationOutputFilterSensitiveLog,
   OrchestrationTrace: () => OrchestrationTrace,
@@ -171,6 +182,7 @@ __export(src_exports, {
   ServiceQuotaExceededException: () => ServiceQuotaExceededException,
   SessionStateFilterSensitiveLog: () => SessionStateFilterSensitiveLog,
   Source: () => Source,
+  TextPromptFilterSensitiveLog: () => TextPromptFilterSensitiveLog,
   TextResponsePartFilterSensitiveLog: () => TextResponsePartFilterSensitiveLog,
   ThrottlingException: () => ThrottlingException,
   Trace: () => Trace,
@@ -843,6 +855,44 @@ var Memory;
     return visitor._(value.$unknown[0], value.$unknown[1]);
   }, "visit");
 })(Memory || (Memory = {}));
+var InputPrompt;
+((InputPrompt3) => {
+  InputPrompt3.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.textPrompt !== void 0)
+      return visitor.textPrompt(value.textPrompt);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(InputPrompt || (InputPrompt = {}));
+var OptimizedPrompt;
+((OptimizedPrompt2) => {
+  OptimizedPrompt2.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.textPrompt !== void 0)
+      return visitor.textPrompt(value.textPrompt);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(OptimizedPrompt || (OptimizedPrompt = {}));
+var OptimizedPromptStream;
+((OptimizedPromptStream3) => {
+  OptimizedPromptStream3.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.optimizedPromptEvent !== void 0)
+      return visitor.optimizedPromptEvent(value.optimizedPromptEvent);
+    if (value.analyzePromptEvent !== void 0)
+      return visitor.analyzePromptEvent(value.analyzePromptEvent);
+    if (value.internalServerException !== void 0)
+      return visitor.internalServerException(value.internalServerException);
+    if (value.throttlingException !== void 0)
+      return visitor.throttlingException(value.throttlingException);
+    if (value.validationException !== void 0)
+      return visitor.validationException(value.validationException);
+    if (value.dependencyFailedException !== void 0)
+      return visitor.dependencyFailedException(value.dependencyFailedException);
+    if (value.accessDeniedException !== void 0)
+      return visitor.accessDeniedException(value.accessDeniedException);
+    if (value.badGatewayException !== void 0)
+      return visitor.badGatewayException(value.badGatewayException);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(OptimizedPromptStream || (OptimizedPromptStream = {}));
 var ExternalSourceType = {
   BYTE_CONTENT: "BYTE_CONTENT",
   S3: "S3"
@@ -1300,6 +1350,56 @@ var InvokeAgentResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.completion && { completion: "STREAMING_CONTENT" }
 }), "InvokeAgentResponseFilterSensitiveLog");
+var TextPromptFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj
+}), "TextPromptFilterSensitiveLog");
+var InputPromptFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
+  if (obj.textPrompt !== void 0)
+    return { textPrompt: import_smithy_client.SENSITIVE_STRING };
+  if (obj.$unknown !== void 0)
+    return { [obj.$unknown[0]]: "UNKNOWN" };
+}, "InputPromptFilterSensitiveLog");
+var OptimizePromptRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.input && { input: InputPromptFilterSensitiveLog(obj.input) }
+}), "OptimizePromptRequestFilterSensitiveLog");
+var AnalyzePromptEventFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj
+}), "AnalyzePromptEventFilterSensitiveLog");
+var OptimizedPromptFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
+  if (obj.textPrompt !== void 0)
+    return { textPrompt: import_smithy_client.SENSITIVE_STRING };
+  if (obj.$unknown !== void 0)
+    return { [obj.$unknown[0]]: "UNKNOWN" };
+}, "OptimizedPromptFilterSensitiveLog");
+var OptimizedPromptEventFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.optimizedPrompt && { optimizedPrompt: OptimizedPromptFilterSensitiveLog(obj.optimizedPrompt) }
+}), "OptimizedPromptEventFilterSensitiveLog");
+var OptimizedPromptStreamFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
+  if (obj.optimizedPromptEvent !== void 0)
+    return { optimizedPromptEvent: import_smithy_client.SENSITIVE_STRING };
+  if (obj.analyzePromptEvent !== void 0)
+    return { analyzePromptEvent: import_smithy_client.SENSITIVE_STRING };
+  if (obj.internalServerException !== void 0)
+    return { internalServerException: obj.internalServerException };
+  if (obj.throttlingException !== void 0)
+    return { throttlingException: obj.throttlingException };
+  if (obj.validationException !== void 0)
+    return { validationException: obj.validationException };
+  if (obj.dependencyFailedException !== void 0)
+    return { dependencyFailedException: obj.dependencyFailedException };
+  if (obj.accessDeniedException !== void 0)
+    return { accessDeniedException: obj.accessDeniedException };
+  if (obj.badGatewayException !== void 0)
+    return { badGatewayException: obj.badGatewayException };
+  if (obj.$unknown !== void 0)
+    return { [obj.$unknown[0]]: "UNKNOWN" };
+}, "OptimizedPromptStreamFilterSensitiveLog");
+var OptimizePromptResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.optimizedPrompt && { optimizedPrompt: "STREAMING_CONTENT" }
+}), "OptimizePromptResponseFilterSensitiveLog");
 var RetrieveAndGenerateInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj
 }), "RetrieveAndGenerateInputFilterSensitiveLog");
@@ -1533,6 +1633,22 @@ var se_InvokeFlowCommand = /* @__PURE__ */ __name(async (input, context) => {
   b.m("POST").h(headers).b(body);
   return b.build();
 }, "se_InvokeFlowCommand");
+var se_OptimizePromptCommand = /* @__PURE__ */ __name(async (input, context) => {
+  const b = (0, import_core.requestBuilder)(input, context);
+  const headers = {
+    "content-type": "application/json"
+  };
+  b.bp("/optimize-prompt");
+  let body;
+  body = JSON.stringify(
+    (0, import_smithy_client.take)(input, {
+      input: (_) => (0, import_smithy_client._json)(_),
+      targetModelId: []
+    })
+  );
+  b.m("POST").h(headers).b(body);
+  return b.build();
+}, "se_OptimizePromptCommand");
 var se_RetrieveCommand = /* @__PURE__ */ __name(async (input, context) => {
   const b = (0, import_core.requestBuilder)(input, context);
   const headers = {
@@ -1619,6 +1735,17 @@ var de_InvokeFlowCommand = /* @__PURE__ */ __name(async (output, context) => {
   contents.responseStream = de_FlowResponseStream(data, context);
   return contents;
 }, "de_InvokeFlowCommand");
+var de_OptimizePromptCommand = /* @__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 = output.body;
+  contents.optimizedPrompt = de_OptimizedPromptStream(data, context);
+  return contents;
+}, "de_OptimizePromptCommand");
 var de_RetrieveCommand = /* @__PURE__ */ __name(async (output, context) => {
   if (output.statusCode !== 200 && output.statusCode >= 300) {
     return de_CommandError(output, context);
@@ -1888,6 +2015,54 @@ var de_FlowResponseStream = /* @__PURE__ */ __name((output, context) => {
     return { $unknown: output };
   });
 }, "de_FlowResponseStream");
+var de_OptimizedPromptStream = /* @__PURE__ */ __name((output, context) => {
+  return context.eventStreamMarshaller.deserialize(output, async (event) => {
+    if (event["optimizedPromptEvent"] != null) {
+      return {
+        optimizedPromptEvent: await de_OptimizedPromptEvent_event(event["optimizedPromptEvent"], context)
+      };
+    }
+    if (event["analyzePromptEvent"] != null) {
+      return {
+        analyzePromptEvent: await de_AnalyzePromptEvent_event(event["analyzePromptEvent"], context)
+      };
+    }
+    if (event["internalServerException"] != null) {
+      return {
+        internalServerException: await de_InternalServerException_event(event["internalServerException"], context)
+      };
+    }
+    if (event["throttlingException"] != null) {
+      return {
+        throttlingException: await de_ThrottlingException_event(event["throttlingException"], context)
+      };
+    }
+    if (event["validationException"] != null) {
+      return {
+        validationException: await de_ValidationException_event(event["validationException"], context)
+      };
+    }
+    if (event["dependencyFailedException"] != null) {
+      return {
+        dependencyFailedException: await de_DependencyFailedException_event(
+          event["dependencyFailedException"],
+          context
+        )
+      };
+    }
+    if (event["accessDeniedException"] != null) {
+      return {
+        accessDeniedException: await de_AccessDeniedException_event(event["accessDeniedException"], context)
+      };
+    }
+    if (event["badGatewayException"] != null) {
+      return {
+        badGatewayException: await de_BadGatewayException_event(event["badGatewayException"], context)
+      };
+    }
+    return { $unknown: output };
+  });
+}, "de_OptimizedPromptStream");
 var de_ResponseStream = /* @__PURE__ */ __name((output, context) => {
   return context.eventStreamMarshaller.deserialize(output, async (event) => {
     if (event["chunk"] != null) {
@@ -1974,6 +2149,12 @@ var de_AccessDeniedException_event = /* @__PURE__ */ __name(async (output, conte
   };
   return de_AccessDeniedExceptionRes(parsedOutput, context);
 }, "de_AccessDeniedException_event");
+var de_AnalyzePromptEvent_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_AnalyzePromptEvent_event");
 var de_BadGatewayException_event = /* @__PURE__ */ __name(async (output, context) => {
   const parsedOutput = {
     ...output,
@@ -2026,6 +2207,12 @@ var de_InternalServerException_event = /* @__PURE__ */ __name(async (output, con
   };
   return de_InternalServerExceptionRes(parsedOutput, context);
 }, "de_InternalServerException_event");
+var de_OptimizedPromptEvent_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_OptimizedPromptEvent_event");
 var de_PayloadPart_event = /* @__PURE__ */ __name(async (output, context) => {
   const contents = {};
   const data = await (0, import_core2.parseJsonBody)(output.body, context);
@@ -2689,6 +2876,27 @@ var _InvokeFlowCommand = class _InvokeFlowCommand extends import_smithy_client.C
 __name(_InvokeFlowCommand, "InvokeFlowCommand");
 var InvokeFlowCommand = _InvokeFlowCommand;
 
+// src/commands/OptimizePromptCommand.ts
+
+
+
+var _OptimizePromptCommand = class _OptimizePromptCommand 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", "OptimizePrompt", {
+  /**
+   * @internal
+   */
+  eventStream: {
+    output: true
+  }
+}).n("BedrockAgentRuntimeClient", "OptimizePromptCommand").f(OptimizePromptRequestFilterSensitiveLog, OptimizePromptResponseFilterSensitiveLog).ser(se_OptimizePromptCommand).de(de_OptimizePromptCommand).build() {
+};
+__name(_OptimizePromptCommand, "OptimizePromptCommand");
+var OptimizePromptCommand = _OptimizePromptCommand;
+
 // src/commands/RetrieveAndGenerateCommand.ts
 
 
@@ -2723,6 +2931,7 @@ var commands = {
   GetAgentMemoryCommand,
   InvokeAgentCommand,
   InvokeFlowCommand,
+  OptimizePromptCommand,
   RetrieveCommand,
   RetrieveAndGenerateCommand
 };
@@ -2751,6 +2960,7 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   GetAgentMemoryCommand,
   InvokeAgentCommand,
   InvokeFlowCommand,
+  OptimizePromptCommand,
   RetrieveAndGenerateCommand,
   RetrieveCommand,
   paginateGetAgentMemory,
@@ -2804,6 +3014,9 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   ResponseStream,
   MemoryType,
   Memory,
+  InputPrompt,
+  OptimizedPrompt,
+  OptimizedPromptStream,
   ExternalSourceType,
   QueryTransformationType,
   RetrieveAndGenerateType,
@@ -2881,6 +3094,14 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   TracePartFilterSensitiveLog,
   ResponseStreamFilterSensitiveLog,
   InvokeAgentResponseFilterSensitiveLog,
+  TextPromptFilterSensitiveLog,
+  InputPromptFilterSensitiveLog,
+  OptimizePromptRequestFilterSensitiveLog,
+  AnalyzePromptEventFilterSensitiveLog,
+  OptimizedPromptFilterSensitiveLog,
+  OptimizedPromptEventFilterSensitiveLog,
+  OptimizedPromptStreamFilterSensitiveLog,
+  OptimizePromptResponseFilterSensitiveLog,
   RetrieveAndGenerateInputFilterSensitiveLog,
   PromptTemplateFilterSensitiveLog,
   ExternalSourcesGenerationConfigurationFilterSensitiveLog,

package/dist-es/BedrockAgentRuntime.js

@@ -4,6 +4,7 @@ import { DeleteAgentMemoryCommand, } from "./commands/DeleteAgentMemoryCommand";
 import { GetAgentMemoryCommand, } from "./commands/GetAgentMemoryCommand";
 import { InvokeAgentCommand } from "./commands/InvokeAgentCommand";
 import { InvokeFlowCommand } from "./commands/InvokeFlowCommand";
+import { OptimizePromptCommand, } from "./commands/OptimizePromptCommand";
 import { RetrieveAndGenerateCommand, } from "./commands/RetrieveAndGenerateCommand";
 import { RetrieveCommand } from "./commands/RetrieveCommand";
 const commands = {
@@ -11,6 +12,7 @@ const commands = {
     GetAgentMemoryCommand,
     InvokeAgentCommand,
     InvokeFlowCommand,
+    OptimizePromptCommand,
     RetrieveCommand,
     RetrieveAndGenerateCommand,
 };

package/dist-es/commands/OptimizePromptCommand.js

@@ -0,0 +1,27 @@
+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 { OptimizePromptRequestFilterSensitiveLog, OptimizePromptResponseFilterSensitiveLog, } from "../models/models_0";
+import { de_OptimizePromptCommand, se_OptimizePromptCommand } from "../protocols/Aws_restJson1";
+export { $Command };
+export class OptimizePromptCommand extends $Command
+    .classBuilder()
+    .ep(commonParams)
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+        getEndpointPlugin(config, Command.getEndpointParameterInstructions()),
+    ];
+})
+    .s("AmazonBedrockAgentRunTimeService", "OptimizePrompt", {
+    eventStream: {
+        output: true,
+    },
+})
+    .n("BedrockAgentRuntimeClient", "OptimizePromptCommand")
+    .f(OptimizePromptRequestFilterSensitiveLog, OptimizePromptResponseFilterSensitiveLog)
+    .ser(se_OptimizePromptCommand)
+    .de(de_OptimizePromptCommand)
+    .build() {
+}

package/dist-es/commands/index.js

@@ -2,5 +2,6 @@ export * from "./DeleteAgentMemoryCommand";
 export * from "./GetAgentMemoryCommand";
 export * from "./InvokeAgentCommand";
 export * from "./InvokeFlowCommand";
+export * from "./OptimizePromptCommand";
 export * from "./RetrieveAndGenerateCommand";
 export * from "./RetrieveCommand";

package/dist-es/models/models_0.js

@@ -445,6 +445,44 @@ export var Memory;
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(Memory || (Memory = {}));
+export var InputPrompt;
+(function (InputPrompt) {
+    InputPrompt.visit = (value, visitor) => {
+        if (value.textPrompt !== undefined)
+            return visitor.textPrompt(value.textPrompt);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(InputPrompt || (InputPrompt = {}));
+export var OptimizedPrompt;
+(function (OptimizedPrompt) {
+    OptimizedPrompt.visit = (value, visitor) => {
+        if (value.textPrompt !== undefined)
+            return visitor.textPrompt(value.textPrompt);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(OptimizedPrompt || (OptimizedPrompt = {}));
+export var OptimizedPromptStream;
+(function (OptimizedPromptStream) {
+    OptimizedPromptStream.visit = (value, visitor) => {
+        if (value.optimizedPromptEvent !== undefined)
+            return visitor.optimizedPromptEvent(value.optimizedPromptEvent);
+        if (value.analyzePromptEvent !== undefined)
+            return visitor.analyzePromptEvent(value.analyzePromptEvent);
+        if (value.internalServerException !== undefined)
+            return visitor.internalServerException(value.internalServerException);
+        if (value.throttlingException !== undefined)
+            return visitor.throttlingException(value.throttlingException);
+        if (value.validationException !== undefined)
+            return visitor.validationException(value.validationException);
+        if (value.dependencyFailedException !== undefined)
+            return visitor.dependencyFailedException(value.dependencyFailedException);
+        if (value.accessDeniedException !== undefined)
+            return visitor.accessDeniedException(value.accessDeniedException);
+        if (value.badGatewayException !== undefined)
+            return visitor.badGatewayException(value.badGatewayException);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(OptimizedPromptStream || (OptimizedPromptStream = {}));
 export const ExternalSourceType = {
     BYTE_CONTENT: "BYTE_CONTENT",
     S3: "S3",
@@ -902,6 +940,56 @@ export const InvokeAgentResponseFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.completion && { completion: "STREAMING_CONTENT" }),
 });
+export const TextPromptFilterSensitiveLog = (obj) => ({
+    ...obj,
+});
+export const InputPromptFilterSensitiveLog = (obj) => {
+    if (obj.textPrompt !== undefined)
+        return { textPrompt: SENSITIVE_STRING };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const OptimizePromptRequestFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.input && { input: InputPromptFilterSensitiveLog(obj.input) }),
+});
+export const AnalyzePromptEventFilterSensitiveLog = (obj) => ({
+    ...obj,
+});
+export const OptimizedPromptFilterSensitiveLog = (obj) => {
+    if (obj.textPrompt !== undefined)
+        return { textPrompt: SENSITIVE_STRING };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const OptimizedPromptEventFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.optimizedPrompt && { optimizedPrompt: OptimizedPromptFilterSensitiveLog(obj.optimizedPrompt) }),
+});
+export const OptimizedPromptStreamFilterSensitiveLog = (obj) => {
+    if (obj.optimizedPromptEvent !== undefined)
+        return { optimizedPromptEvent: SENSITIVE_STRING };
+    if (obj.analyzePromptEvent !== undefined)
+        return { analyzePromptEvent: SENSITIVE_STRING };
+    if (obj.internalServerException !== undefined)
+        return { internalServerException: obj.internalServerException };
+    if (obj.throttlingException !== undefined)
+        return { throttlingException: obj.throttlingException };
+    if (obj.validationException !== undefined)
+        return { validationException: obj.validationException };
+    if (obj.dependencyFailedException !== undefined)
+        return { dependencyFailedException: obj.dependencyFailedException };
+    if (obj.accessDeniedException !== undefined)
+        return { accessDeniedException: obj.accessDeniedException };
+    if (obj.badGatewayException !== undefined)
+        return { badGatewayException: obj.badGatewayException };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const OptimizePromptResponseFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.optimizedPrompt && { optimizedPrompt: "STREAMING_CONTENT" }),
+});
 export const RetrieveAndGenerateInputFilterSensitiveLog = (obj) => ({
     ...obj,
 });

package/dist-es/protocols/Aws_restJson1.js

@@ -68,6 +68,20 @@ export const se_InvokeFlowCommand = async (input, context) => {
     b.m("POST").h(headers).b(body);
     return b.build();
 };
+export const se_OptimizePromptCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/optimize-prompt");
+    let body;
+    body = JSON.stringify(take(input, {
+        input: (_) => _json(_),
+        targetModelId: [],
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_RetrieveCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -150,6 +164,17 @@ export const de_InvokeFlowCommand = async (output, context) => {
     contents.responseStream = de_FlowResponseStream(data, context);
     return contents;
 };
+export const de_OptimizePromptCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = output.body;
+    contents.optimizedPrompt = de_OptimizedPromptStream(data, context);
+    return contents;
+};
 export const de_RetrieveCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -410,6 +435,51 @@ const de_FlowResponseStream = (output, context) => {
         return { $unknown: output };
     });
 };
+const de_OptimizedPromptStream = (output, context) => {
+    return context.eventStreamMarshaller.deserialize(output, async (event) => {
+        if (event["optimizedPromptEvent"] != null) {
+            return {
+                optimizedPromptEvent: await de_OptimizedPromptEvent_event(event["optimizedPromptEvent"], context),
+            };
+        }
+        if (event["analyzePromptEvent"] != null) {
+            return {
+                analyzePromptEvent: await de_AnalyzePromptEvent_event(event["analyzePromptEvent"], context),
+            };
+        }
+        if (event["internalServerException"] != null) {
+            return {
+                internalServerException: await de_InternalServerException_event(event["internalServerException"], context),
+            };
+        }
+        if (event["throttlingException"] != null) {
+            return {
+                throttlingException: await de_ThrottlingException_event(event["throttlingException"], context),
+            };
+        }
+        if (event["validationException"] != null) {
+            return {
+                validationException: await de_ValidationException_event(event["validationException"], context),
+            };
+        }
+        if (event["dependencyFailedException"] != null) {
+            return {
+                dependencyFailedException: await de_DependencyFailedException_event(event["dependencyFailedException"], context),
+            };
+        }
+        if (event["accessDeniedException"] != null) {
+            return {
+                accessDeniedException: await de_AccessDeniedException_event(event["accessDeniedException"], context),
+            };
+        }
+        if (event["badGatewayException"] != null) {
+            return {
+                badGatewayException: await de_BadGatewayException_event(event["badGatewayException"], context),
+            };
+        }
+        return { $unknown: output };
+    });
+};
 const de_ResponseStream = (output, context) => {
     return context.eventStreamMarshaller.deserialize(output, async (event) => {
         if (event["chunk"] != null) {
@@ -487,6 +557,12 @@ const de_AccessDeniedException_event = async (output, context) => {
     };
     return de_AccessDeniedExceptionRes(parsedOutput, context);
 };
+const de_AnalyzePromptEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, _json(data));
+    return contents;
+};
 const de_BadGatewayException_event = async (output, context) => {
     const parsedOutput = {
         ...output,
@@ -539,6 +615,12 @@ const de_InternalServerException_event = async (output, context) => {
     };
     return de_InternalServerExceptionRes(parsedOutput, context);
 };
+const de_OptimizedPromptEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, _json(data));
+    return contents;
+};
 const de_PayloadPart_event = async (output, context) => {
     const contents = {};
     const data = await parseBody(output.body, context);

package/dist-types/BedrockAgentRuntime.d.ts

@@ -4,6 +4,7 @@ import { DeleteAgentMemoryCommandInput, DeleteAgentMemoryCommandOutput } from ".
 import { GetAgentMemoryCommandInput, GetAgentMemoryCommandOutput } from "./commands/GetAgentMemoryCommand";
 import { InvokeAgentCommandInput, InvokeAgentCommandOutput } from "./commands/InvokeAgentCommand";
 import { InvokeFlowCommandInput, InvokeFlowCommandOutput } from "./commands/InvokeFlowCommand";
+import { OptimizePromptCommandInput, OptimizePromptCommandOutput } from "./commands/OptimizePromptCommand";
 import { RetrieveAndGenerateCommandInput, RetrieveAndGenerateCommandOutput } from "./commands/RetrieveAndGenerateCommand";
 import { RetrieveCommandInput, RetrieveCommandOutput } from "./commands/RetrieveCommand";
 export interface BedrockAgentRuntime {
@@ -31,6 +32,12 @@ export interface BedrockAgentRuntime {
     invokeFlow(args: InvokeFlowCommandInput, options?: __HttpHandlerOptions): Promise<InvokeFlowCommandOutput>;
     invokeFlow(args: InvokeFlowCommandInput, cb: (err: any, data?: InvokeFlowCommandOutput) => void): void;
     invokeFlow(args: InvokeFlowCommandInput, options: __HttpHandlerOptions, cb: (err: any, data?: InvokeFlowCommandOutput) => void): void;
+    /**
+     * @see {@link OptimizePromptCommand}
+     */
+    optimizePrompt(args: OptimizePromptCommandInput, options?: __HttpHandlerOptions): Promise<OptimizePromptCommandOutput>;
+    optimizePrompt(args: OptimizePromptCommandInput, cb: (err: any, data?: OptimizePromptCommandOutput) => void): void;
+    optimizePrompt(args: OptimizePromptCommandInput, options: __HttpHandlerOptions, cb: (err: any, data?: OptimizePromptCommandOutput) => void): void;
     /**
      * @see {@link RetrieveCommand}
      */

package/dist-types/BedrockAgentRuntimeClient.d.ts

@@ -12,6 +12,7 @@ import { DeleteAgentMemoryCommandInput, DeleteAgentMemoryCommandOutput } from ".
 import { GetAgentMemoryCommandInput, GetAgentMemoryCommandOutput } from "./commands/GetAgentMemoryCommand";
 import { InvokeAgentCommandInput, InvokeAgentCommandOutput } from "./commands/InvokeAgentCommand";
 import { InvokeFlowCommandInput, InvokeFlowCommandOutput } from "./commands/InvokeFlowCommand";
+import { OptimizePromptCommandInput, OptimizePromptCommandOutput } from "./commands/OptimizePromptCommand";
 import { RetrieveAndGenerateCommandInput, RetrieveAndGenerateCommandOutput } from "./commands/RetrieveAndGenerateCommand";
 import { RetrieveCommandInput, RetrieveCommandOutput } from "./commands/RetrieveCommand";
 import { ClientInputEndpointParameters, ClientResolvedEndpointParameters, EndpointParameters } from "./endpoint/EndpointParameters";
@@ -20,11 +21,11 @@ export { __Client };
 /**
  * @public
  */
-export type ServiceInputTypes = DeleteAgentMemoryCommandInput | GetAgentMemoryCommandInput | InvokeAgentCommandInput | InvokeFlowCommandInput | RetrieveAndGenerateCommandInput | RetrieveCommandInput;
+export type ServiceInputTypes = DeleteAgentMemoryCommandInput | GetAgentMemoryCommandInput | InvokeAgentCommandInput | InvokeFlowCommandInput | OptimizePromptCommandInput | RetrieveAndGenerateCommandInput | RetrieveCommandInput;
 /**
  * @public
  */
-export type ServiceOutputTypes = DeleteAgentMemoryCommandOutput | GetAgentMemoryCommandOutput | InvokeAgentCommandOutput | InvokeFlowCommandOutput | RetrieveAndGenerateCommandOutput | RetrieveCommandOutput;
+export type ServiceOutputTypes = DeleteAgentMemoryCommandOutput | GetAgentMemoryCommandOutput | InvokeAgentCommandOutput | InvokeFlowCommandOutput | OptimizePromptCommandOutput | RetrieveAndGenerateCommandOutput | RetrieveCommandOutput;
 /**
  * @public
  */

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

@@ -27,7 +27,7 @@ declare const InvokeFlowCommand_base: {
     getEndpointParameterInstructions(): import("@smithy/middleware-endpoint").EndpointParameterInstructions;
 };
 /**
- * <p>Invokes an alias of a flow to run the inputs that you specify and return the output of each node as a stream. If there's an error, the error is returned. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/flows-test.html">Test a flow in Amazon Bedrock</a> in the Amazon Bedrock User Guide.</p>
+ * <p>Invokes an alias of a flow to run the inputs that you specify and return the output of each node as a stream. If there's an error, the error is returned. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/flows-test.html">Test a flow in Amazon Bedrock</a> in the <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/what-is-service.html">Amazon Bedrock User Guide</a>.</p>
  *          <note>
  *             <p>The CLI doesn't support streaming operations in Amazon Bedrock, including <code>InvokeFlow</code>.</p>
  *          </note>