aws-sdk-bedrockagentruntime 1.30.0 → 1.32.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ddd19c39eedacc41d2bef21c75ff2669e9badf69d4b784b9b828ef5c9482d225
-  data.tar.gz: b3b78d61228b5a8db3fbe3efa336e94f11b1f83b453c7aa1f726d7ffab940ef5
+  metadata.gz: 9a5c8f9ada774c7b6b0f076b035f69819f5b0d0c04912808fda03f994b242e96
+  data.tar.gz: 9142d0bc53700130ddda7545064b22ae00b04e60057c0643add3f5cfa1cb59c5
 SHA512:
-  metadata.gz: 3ce22f6888e86e1c8303524cbd67791e688f760182d326fb88f856c98aabefe428310b52b954d243f0d7a8e8a46e79c2e3eae70e8f0acde4d47e24d9b7ef1543
-  data.tar.gz: 9fae515be407940cd0821112ef9248f5f714e139d332da4f3d6a222ca589332c59cf161729493ae69a309ac894fd614664d7fba3e9bb967f7630aa4ae35402d2
+  metadata.gz: 5e5d6f501e8d9cc30c2310099696aebf003fbc3e9d2d228276643178ea15f29c819537d5942a6f63e66bfc5e7656b62a81f2fca55e114422476a91af7d22ad96
+  data.tar.gz: 54e0b35d92200a5b96f66ab97b14ebe698f5ff7aa4f0ed0334522b28dad7451b0270f632a4e2f75bf97c4849b8da4a3ae0ca43dbb6ca5b4d4b52ac3cb97c3461

data/CHANGELOG.md

@@ -1,6 +1,16 @@
 Unreleased Changes
 ------------------
 
+1.32.0 (2024-11-20)
+------------------
+
+* Feature - Releasing new Prompt Optimization to enhance your prompts for improved performance
+
+1.31.0 (2024-11-08)
+------------------
+
+* Feature - This release adds trace functionality to Bedrock Prompt Flows
+
 1.30.0 (2024-11-06)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.30.0
+1.32.0

data/lib/aws-sdk-bedrockagentruntime/client.rb

@@ -1248,7 +1248,7 @@ module Aws::BedrockAgentRuntime
     # 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 [Test a flow in Amazon
-    # Bedrock][1] in the Amazon Bedrock User Guide.
+    # Bedrock][1] in the [Amazon Bedrock User Guide][2].
     #
     # <note markdown="1"> The CLI doesn't support streaming operations in Amazon Bedrock,
     # including `InvokeFlow`.
@@ -1258,6 +1258,17 @@ module Aws::BedrockAgentRuntime
     #
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/flows-test.html
+    # [2]: https://docs.aws.amazon.com/bedrock/latest/userguide/what-is-service.html
+    #
+    # @option params [Boolean] :enable_trace
+    #   Specifies whether to return the trace for the flow or not. Traces
+    #   track inputs and outputs for nodes in the flow. For more information,
+    #   see [Track each step in your prompt flow by viewing its trace in
+    #   Amazon Bedrock][1].
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/flows-trace.html
     #
     # @option params [required, String] :flow_alias_identifier
     #   The unique identifier of the flow alias.
@@ -1334,6 +1345,9 @@ module Aws::BedrockAgentRuntime
     #       handler.on_flow_output_event_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::flowOutputEvent
     #       end
+    #       handler.on_flow_trace_event_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::flowTraceEvent
+    #       end
     #       handler.on_internal_server_exception_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::internalServerException
     #       end
@@ -1374,6 +1388,9 @@ module Aws::BedrockAgentRuntime
     #       stream.on_flow_output_event_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::flowOutputEvent
     #       end
+    #       stream.on_flow_trace_event_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::flowTraceEvent
+    #       end
     #       stream.on_internal_server_exception_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::internalServerException
     #       end
@@ -1414,6 +1431,9 @@ module Aws::BedrockAgentRuntime
     #       handler.on_flow_output_event_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::flowOutputEvent
     #       end
+    #       handler.on_flow_trace_event_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::flowTraceEvent
+    #       end
     #       handler.on_internal_server_exception_event do |event|
     #         event # => Aws::BedrockAgentRuntime::Types::internalServerException
     #       end
@@ -1449,6 +1469,7 @@ module Aws::BedrockAgentRuntime
     # @example Request syntax with placeholder values
     #
     #   resp = client.invoke_flow({
+    #     enable_trace: false,
     #     flow_alias_identifier: "FlowAliasIdentifier", # required
     #     flow_identifier: "FlowIdentifier", # required
     #     inputs: [ # required
@@ -1467,7 +1488,7 @@ module Aws::BedrockAgentRuntime
     #
     #   All events are available at resp.response_stream:
     #   resp.response_stream #=> Enumerator
-    #   resp.response_stream.event_types #=> [:access_denied_exception, :bad_gateway_exception, :conflict_exception, :dependency_failed_exception, :flow_completion_event, :flow_output_event, :internal_server_exception, :resource_not_found_exception, :service_quota_exceeded_exception, :throttling_exception, :validation_exception]
+    #   resp.response_stream.event_types #=> [:access_denied_exception, :bad_gateway_exception, :conflict_exception, :dependency_failed_exception, :flow_completion_event, :flow_output_event, :flow_trace_event, :internal_server_exception, :resource_not_found_exception, :service_quota_exceeded_exception, :throttling_exception, :validation_exception]
     #
     #   For :access_denied_exception event available at #on_access_denied_exception_event callback and response eventstream enumerator:
     #   event.message #=> String
@@ -1490,6 +1511,20 @@ module Aws::BedrockAgentRuntime
     #   event.node_name #=> String
     #   event.node_type #=> String, one of "FlowInputNode", "FlowOutputNode", "LambdaFunctionNode", "KnowledgeBaseNode", "PromptNode", "ConditionNode", "LexNode"
     #
+    #   For :flow_trace_event event available at #on_flow_trace_event_event callback and response eventstream enumerator:
+    #   event.trace.condition_node_result_trace.node_name #=> String
+    #   event.trace.condition_node_result_trace.satisfied_conditions #=> Array
+    #   event.trace.condition_node_result_trace.satisfied_conditions[0].condition_name #=> String
+    #   event.trace.condition_node_result_trace.timestamp #=> Time
+    #   event.trace.node_input_trace.fields #=> Array
+    #   event.trace.node_input_trace.fields[0].node_input_name #=> String
+    #   event.trace.node_input_trace.node_name #=> String
+    #   event.trace.node_input_trace.timestamp #=> Time
+    #   event.trace.node_output_trace.fields #=> Array
+    #   event.trace.node_output_trace.fields[0].node_output_name #=> String
+    #   event.trace.node_output_trace.node_name #=> String
+    #   event.trace.node_output_trace.timestamp #=> Time
+    #
     #   For :internal_server_exception event available at #on_internal_server_exception_event callback and response eventstream enumerator:
     #   event.message #=> String
     #
@@ -1532,6 +1567,242 @@ module Aws::BedrockAgentRuntime
       req.send_request(options, &block)
     end
 
+    # Optimizes a prompt for the task that you specify. For more
+    # information, see [Optimize a prompt][1] in the [Amazon Bedrock User
+    # Guide][2].
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/prompt-management-optimize.html
+    # [2]: https://docs.aws.amazon.com/bedrock/latest/userguide/what-is-service.html
+    #
+    # @option params [required, Types::InputPrompt] :input
+    #   Contains the prompt to optimize.
+    #
+    # @option params [required, String] :target_model_id
+    #   The unique identifier of the model that you want to optimize the
+    #   prompt for.
+    #
+    # @return [Types::OptimizePromptResponse] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
+    #
+    #   * {Types::OptimizePromptResponse#optimized_prompt #optimized_prompt} => Types::OptimizedPromptStream
+    #
+    # @example EventStream Operation Example
+    #
+    #   You can process the event once it arrives immediately, or wait until the
+    #   full response is complete and iterate through the eventstream enumerator.
+    #
+    #   To interact with event immediately, you need to register #optimize_prompt
+    #   with callbacks. Callbacks can be registered for specific events or for all
+    #   events, including error events.
+    #
+    #   Callbacks can be passed into the `:event_stream_handler` option or within a
+    #   block statement attached to the #optimize_prompt call directly. Hybrid
+    #   pattern of both is also supported.
+    #
+    #   `:event_stream_handler` option takes in either a Proc object or
+    #   Aws::BedrockAgentRuntime::EventStreams::OptimizedPromptStream object.
+    #
+    #   Usage pattern a): Callbacks with a block attached to #optimize_prompt
+    #     Example for registering callbacks for all event types and an error event
+    #
+    #     client.optimize_prompt( # params input# ) do |stream|
+    #       stream.on_error_event do |event|
+    #         # catch unmodeled error event in the stream
+    #         raise event
+    #         # => Aws::Errors::EventError
+    #         # event.event_type => :error
+    #         # event.error_code => String
+    #         # event.error_message => String
+    #       end
+    #
+    #       stream.on_event do |event|
+    #         # process all events arrive
+    #         puts event.event_type
+    #         ...
+    #       end
+    #
+    #     end
+    #
+    #   Usage pattern b): Pass in `:event_stream_handler` for #optimize_prompt
+    #
+    #     1) Create a Aws::BedrockAgentRuntime::EventStreams::OptimizedPromptStream object
+    #     Example for registering callbacks with specific events
+    #
+    #       handler = Aws::BedrockAgentRuntime::EventStreams::OptimizedPromptStream.new
+    #       handler.on_access_denied_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::accessDeniedException
+    #       end
+    #       handler.on_analyze_prompt_event_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::analyzePromptEvent
+    #       end
+    #       handler.on_bad_gateway_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::badGatewayException
+    #       end
+    #       handler.on_dependency_failed_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::dependencyFailedException
+    #       end
+    #       handler.on_internal_server_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::internalServerException
+    #       end
+    #       handler.on_optimized_prompt_event_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::optimizedPromptEvent
+    #       end
+    #       handler.on_throttling_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::throttlingException
+    #       end
+    #       handler.on_validation_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::validationException
+    #       end
+    #
+    #     client.optimize_prompt( # params input #, event_stream_handler: handler)
+    #
+    #     2) Use a Ruby Proc object
+    #     Example for registering callbacks with specific events
+    #
+    #     handler = Proc.new do |stream|
+    #       stream.on_access_denied_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::accessDeniedException
+    #       end
+    #       stream.on_analyze_prompt_event_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::analyzePromptEvent
+    #       end
+    #       stream.on_bad_gateway_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::badGatewayException
+    #       end
+    #       stream.on_dependency_failed_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::dependencyFailedException
+    #       end
+    #       stream.on_internal_server_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::internalServerException
+    #       end
+    #       stream.on_optimized_prompt_event_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::optimizedPromptEvent
+    #       end
+    #       stream.on_throttling_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::throttlingException
+    #       end
+    #       stream.on_validation_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::validationException
+    #       end
+    #     end
+    #
+    #     client.optimize_prompt( # params input #, event_stream_handler: handler)
+    #
+    #   Usage pattern c): Hybrid pattern of a) and b)
+    #
+    #       handler = Aws::BedrockAgentRuntime::EventStreams::OptimizedPromptStream.new
+    #       handler.on_access_denied_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::accessDeniedException
+    #       end
+    #       handler.on_analyze_prompt_event_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::analyzePromptEvent
+    #       end
+    #       handler.on_bad_gateway_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::badGatewayException
+    #       end
+    #       handler.on_dependency_failed_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::dependencyFailedException
+    #       end
+    #       handler.on_internal_server_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::internalServerException
+    #       end
+    #       handler.on_optimized_prompt_event_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::optimizedPromptEvent
+    #       end
+    #       handler.on_throttling_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::throttlingException
+    #       end
+    #       handler.on_validation_exception_event do |event|
+    #         event # => Aws::BedrockAgentRuntime::Types::validationException
+    #       end
+    #
+    #     client.optimize_prompt( # params input #, event_stream_handler: handler) do |stream|
+    #       stream.on_error_event do |event|
+    #         # catch unmodeled error event in the stream
+    #         raise event
+    #         # => Aws::Errors::EventError
+    #         # event.event_type => :error
+    #         # event.error_code => String
+    #         # event.error_message => String
+    #       end
+    #     end
+    #
+    #   You can also iterate through events after the response complete.
+    #
+    #   Events are available at resp.optimized_prompt # => Enumerator
+    #   For parameter input example, please refer to following request syntax
+    #
+    # @example Request syntax with placeholder values
+    #
+    #   resp = client.optimize_prompt({
+    #     input: { # required
+    #       text_prompt: {
+    #         text: "TextPromptTextString", # required
+    #       },
+    #     },
+    #     target_model_id: "OptimizePromptRequestTargetModelIdString", # required
+    #   })
+    #
+    # @example Response structure
+    #
+    #   All events are available at resp.optimized_prompt:
+    #   resp.optimized_prompt #=> Enumerator
+    #   resp.optimized_prompt.event_types #=> [:access_denied_exception, :analyze_prompt_event, :bad_gateway_exception, :dependency_failed_exception, :internal_server_exception, :optimized_prompt_event, :throttling_exception, :validation_exception]
+    #
+    #   For :access_denied_exception event available at #on_access_denied_exception_event callback and response eventstream enumerator:
+    #   event.message #=> String
+    #
+    #   For :analyze_prompt_event event available at #on_analyze_prompt_event_event callback and response eventstream enumerator:
+    #   event.message #=> String
+    #
+    #   For :bad_gateway_exception event available at #on_bad_gateway_exception_event callback and response eventstream enumerator:
+    #   event.message #=> String
+    #   event.resource_name #=> String
+    #
+    #   For :dependency_failed_exception event available at #on_dependency_failed_exception_event callback and response eventstream enumerator:
+    #   event.message #=> String
+    #   event.resource_name #=> String
+    #
+    #   For :internal_server_exception event available at #on_internal_server_exception_event callback and response eventstream enumerator:
+    #   event.message #=> String
+    #
+    #   For :optimized_prompt_event event available at #on_optimized_prompt_event_event callback and response eventstream enumerator:
+    #   event.optimized_prompt.text_prompt.text #=> String
+    #
+    #   For :throttling_exception event available at #on_throttling_exception_event callback and response eventstream enumerator:
+    #   event.message #=> String
+    #
+    #   For :validation_exception event available at #on_validation_exception_event callback and response eventstream enumerator:
+    #   event.message #=> String
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/OptimizePrompt AWS API Documentation
+    #
+    # @overload optimize_prompt(params = {})
+    # @param [Hash] params ({})
+    def optimize_prompt(params = {}, options = {}, &block)
+      params = params.dup
+      event_stream_handler = case handler = params.delete(:event_stream_handler)
+        when EventStreams::OptimizedPromptStream then handler
+        when Proc then EventStreams::OptimizedPromptStream.new.tap(&handler)
+        when nil then EventStreams::OptimizedPromptStream.new
+        else
+          msg = "expected :event_stream_handler to be a block or "\
+                "instance of Aws::BedrockAgentRuntime::EventStreams::OptimizedPromptStream"\
+                ", got `#{handler.inspect}` instead"
+          raise ArgumentError, msg
+        end
+
+      yield(event_stream_handler) if block_given?
+
+      req = build_request(:optimize_prompt, params)
+
+      req.context[:event_stream_handler] = event_stream_handler
+      req.handlers.add(Aws::Binary::DecodeHandler, priority: 95)
+
+      req.send_request(options, &block)
+    end
+
     # Queries a knowledge base and retrieves information from it.
     #
     # @option params [required, String] :knowledge_base_id
@@ -1920,7 +2191,7 @@ module Aws::BedrockAgentRuntime
         tracer: tracer
       )
       context[:gem_name] = 'aws-sdk-bedrockagentruntime'
-      context[:gem_version] = '1.30.0'
+      context[:gem_version] = '1.32.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

data/lib/aws-sdk-bedrockagentruntime/client_api.rb

@@ -26,6 +26,7 @@ module Aws::BedrockAgentRuntime
     AgentAliasId = Shapes::StringShape.new(name: 'AgentAliasId')
     AgentId = Shapes::StringShape.new(name: 'AgentId')
     AgentVersion = Shapes::StringShape.new(name: 'AgentVersion')
+    AnalyzePromptEvent = Shapes::StructureShape.new(name: 'AnalyzePromptEvent')
     ApiContentMap = Shapes::MapShape.new(name: 'ApiContentMap')
     ApiInvocationInput = Shapes::StructureShape.new(name: 'ApiInvocationInput')
     ApiParameter = Shapes::StructureShape.new(name: 'ApiParameter')
@@ -85,6 +86,19 @@ module Aws::BedrockAgentRuntime
     FlowOutputContent = Shapes::UnionShape.new(name: 'FlowOutputContent')
     FlowOutputEvent = Shapes::StructureShape.new(name: 'FlowOutputEvent')
     FlowResponseStream = Shapes::StructureShape.new(name: 'FlowResponseStream')
+    FlowTrace = Shapes::UnionShape.new(name: 'FlowTrace')
+    FlowTraceCondition = Shapes::StructureShape.new(name: 'FlowTraceCondition')
+    FlowTraceConditionNodeResultEvent = Shapes::StructureShape.new(name: 'FlowTraceConditionNodeResultEvent')
+    FlowTraceConditions = Shapes::ListShape.new(name: 'FlowTraceConditions')
+    FlowTraceEvent = Shapes::StructureShape.new(name: 'FlowTraceEvent')
+    FlowTraceNodeInputContent = Shapes::UnionShape.new(name: 'FlowTraceNodeInputContent')
+    FlowTraceNodeInputEvent = Shapes::StructureShape.new(name: 'FlowTraceNodeInputEvent')
+    FlowTraceNodeInputField = Shapes::StructureShape.new(name: 'FlowTraceNodeInputField')
+    FlowTraceNodeInputFields = Shapes::ListShape.new(name: 'FlowTraceNodeInputFields')
+    FlowTraceNodeOutputContent = Shapes::UnionShape.new(name: 'FlowTraceNodeOutputContent')
+    FlowTraceNodeOutputEvent = Shapes::StructureShape.new(name: 'FlowTraceNodeOutputEvent')
+    FlowTraceNodeOutputField = Shapes::StructureShape.new(name: 'FlowTraceNodeOutputField')
+    FlowTraceNodeOutputFields = Shapes::ListShape.new(name: 'FlowTraceNodeOutputFields')
     Function = Shapes::StringShape.new(name: 'Function')
     FunctionInvocationInput = Shapes::StructureShape.new(name: 'FunctionInvocationInput')
     FunctionParameter = Shapes::StructureShape.new(name: 'FunctionParameter')
@@ -132,6 +146,7 @@ module Aws::BedrockAgentRuntime
     InferenceConfiguration = Shapes::StructureShape.new(name: 'InferenceConfiguration')
     InputFile = Shapes::StructureShape.new(name: 'InputFile')
     InputFiles = Shapes::ListShape.new(name: 'InputFiles')
+    InputPrompt = Shapes::UnionShape.new(name: 'InputPrompt')
     InputText = Shapes::StringShape.new(name: 'InputText')
     Integer = Shapes::IntegerShape.new(name: 'Integer')
     InternalServerException = Shapes::StructureShape.new(name: 'InternalServerException')
@@ -172,11 +187,18 @@ module Aws::BedrockAgentRuntime
     MimeType = Shapes::StringShape.new(name: 'MimeType')
     ModelInvocationInput = Shapes::StructureShape.new(name: 'ModelInvocationInput')
     NextToken = Shapes::StringShape.new(name: 'NextToken')
+    NodeInputName = Shapes::StringShape.new(name: 'NodeInputName')
     NodeName = Shapes::StringShape.new(name: 'NodeName')
     NodeOutputName = Shapes::StringShape.new(name: 'NodeOutputName')
     NodeType = Shapes::StringShape.new(name: 'NodeType')
     NonBlankString = Shapes::StringShape.new(name: 'NonBlankString')
     Observation = Shapes::StructureShape.new(name: 'Observation')
+    OptimizePromptRequest = Shapes::StructureShape.new(name: 'OptimizePromptRequest')
+    OptimizePromptRequestTargetModelIdString = Shapes::StringShape.new(name: 'OptimizePromptRequestTargetModelIdString')
+    OptimizePromptResponse = Shapes::StructureShape.new(name: 'OptimizePromptResponse')
+    OptimizedPrompt = Shapes::UnionShape.new(name: 'OptimizedPrompt')
+    OptimizedPromptEvent = Shapes::StructureShape.new(name: 'OptimizedPromptEvent')
+    OptimizedPromptStream = Shapes::StructureShape.new(name: 'OptimizedPromptStream')
     OrchestrationConfiguration = Shapes::StructureShape.new(name: 'OrchestrationConfiguration')
     OrchestrationModelInvocationOutput = Shapes::StructureShape.new(name: 'OrchestrationModelInvocationOutput')
     OrchestrationTrace = Shapes::UnionShape.new(name: 'OrchestrationTrace')
@@ -256,7 +278,9 @@ module Aws::BedrockAgentRuntime
     SummaryText = Shapes::StringShape.new(name: 'SummaryText')
     Temperature = Shapes::FloatShape.new(name: 'Temperature')
     TextInferenceConfig = Shapes::StructureShape.new(name: 'TextInferenceConfig')
+    TextPrompt = Shapes::StructureShape.new(name: 'TextPrompt')
     TextPromptTemplate = Shapes::StringShape.new(name: 'TextPromptTemplate')
+    TextPromptTextString = Shapes::StringShape.new(name: 'TextPromptTextString')
     TextResponsePart = Shapes::StructureShape.new(name: 'TextResponsePart')
     ThrottlingException = Shapes::StructureShape.new(name: 'ThrottlingException')
     TopK = Shapes::IntegerShape.new(name: 'TopK')
@@ -289,6 +313,9 @@ module Aws::BedrockAgentRuntime
     AdditionalModelRequestFields.key = Shapes::ShapeRef.new(shape: AdditionalModelRequestFieldsKey)
     AdditionalModelRequestFields.value = Shapes::ShapeRef.new(shape: AdditionalModelRequestFieldsValue)
 
+    AnalyzePromptEvent.add_member(:message, Shapes::ShapeRef.new(shape: String, location_name: "message"))
+    AnalyzePromptEvent.struct_class = Types::AnalyzePromptEvent
+
     ApiContentMap.key = Shapes::ShapeRef.new(shape: String)
     ApiContentMap.value = Shapes::ShapeRef.new(shape: PropertyParameters)
 
@@ -443,6 +470,7 @@ module Aws::BedrockAgentRuntime
     FlowResponseStream.add_member(:dependency_failed_exception, Shapes::ShapeRef.new(shape: DependencyFailedException, location_name: "dependencyFailedException"))
     FlowResponseStream.add_member(:flow_completion_event, Shapes::ShapeRef.new(shape: FlowCompletionEvent, event: true, location_name: "flowCompletionEvent"))
     FlowResponseStream.add_member(:flow_output_event, Shapes::ShapeRef.new(shape: FlowOutputEvent, event: true, location_name: "flowOutputEvent"))
+    FlowResponseStream.add_member(:flow_trace_event, Shapes::ShapeRef.new(shape: FlowTraceEvent, event: true, location_name: "flowTraceEvent"))
     FlowResponseStream.add_member(:internal_server_exception, Shapes::ShapeRef.new(shape: InternalServerException, location_name: "internalServerException"))
     FlowResponseStream.add_member(:resource_not_found_exception, Shapes::ShapeRef.new(shape: ResourceNotFoundException, location_name: "resourceNotFoundException"))
     FlowResponseStream.add_member(:service_quota_exceeded_exception, Shapes::ShapeRef.new(shape: ServiceQuotaExceededException, location_name: "serviceQuotaExceededException"))
@@ -450,6 +478,63 @@ module Aws::BedrockAgentRuntime
     FlowResponseStream.add_member(:validation_exception, Shapes::ShapeRef.new(shape: ValidationException, location_name: "validationException"))
     FlowResponseStream.struct_class = Types::FlowResponseStream
 
+    FlowTrace.add_member(:condition_node_result_trace, Shapes::ShapeRef.new(shape: FlowTraceConditionNodeResultEvent, location_name: "conditionNodeResultTrace"))
+    FlowTrace.add_member(:node_input_trace, Shapes::ShapeRef.new(shape: FlowTraceNodeInputEvent, location_name: "nodeInputTrace"))
+    FlowTrace.add_member(:node_output_trace, Shapes::ShapeRef.new(shape: FlowTraceNodeOutputEvent, location_name: "nodeOutputTrace"))
+    FlowTrace.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    FlowTrace.add_member_subclass(:condition_node_result_trace, Types::FlowTrace::ConditionNodeResultTrace)
+    FlowTrace.add_member_subclass(:node_input_trace, Types::FlowTrace::NodeInputTrace)
+    FlowTrace.add_member_subclass(:node_output_trace, Types::FlowTrace::NodeOutputTrace)
+    FlowTrace.add_member_subclass(:unknown, Types::FlowTrace::Unknown)
+    FlowTrace.struct_class = Types::FlowTrace
+
+    FlowTraceCondition.add_member(:condition_name, Shapes::ShapeRef.new(shape: String, required: true, location_name: "conditionName"))
+    FlowTraceCondition.struct_class = Types::FlowTraceCondition
+
+    FlowTraceConditionNodeResultEvent.add_member(:node_name, Shapes::ShapeRef.new(shape: NodeName, required: true, location_name: "nodeName"))
+    FlowTraceConditionNodeResultEvent.add_member(:satisfied_conditions, Shapes::ShapeRef.new(shape: FlowTraceConditions, required: true, location_name: "satisfiedConditions"))
+    FlowTraceConditionNodeResultEvent.add_member(:timestamp, Shapes::ShapeRef.new(shape: DateTimestamp, required: true, location_name: "timestamp"))
+    FlowTraceConditionNodeResultEvent.struct_class = Types::FlowTraceConditionNodeResultEvent
+
+    FlowTraceConditions.member = Shapes::ShapeRef.new(shape: FlowTraceCondition)
+
+    FlowTraceEvent.add_member(:trace, Shapes::ShapeRef.new(shape: FlowTrace, required: true, location_name: "trace"))
+    FlowTraceEvent.struct_class = Types::FlowTraceEvent
+
+    FlowTraceNodeInputContent.add_member(:document, Shapes::ShapeRef.new(shape: Document, location_name: "document"))
+    FlowTraceNodeInputContent.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    FlowTraceNodeInputContent.add_member_subclass(:document, Types::FlowTraceNodeInputContent::Document)
+    FlowTraceNodeInputContent.add_member_subclass(:unknown, Types::FlowTraceNodeInputContent::Unknown)
+    FlowTraceNodeInputContent.struct_class = Types::FlowTraceNodeInputContent
+
+    FlowTraceNodeInputEvent.add_member(:fields, Shapes::ShapeRef.new(shape: FlowTraceNodeInputFields, required: true, location_name: "fields"))
+    FlowTraceNodeInputEvent.add_member(:node_name, Shapes::ShapeRef.new(shape: NodeName, required: true, location_name: "nodeName"))
+    FlowTraceNodeInputEvent.add_member(:timestamp, Shapes::ShapeRef.new(shape: DateTimestamp, required: true, location_name: "timestamp"))
+    FlowTraceNodeInputEvent.struct_class = Types::FlowTraceNodeInputEvent
+
+    FlowTraceNodeInputField.add_member(:content, Shapes::ShapeRef.new(shape: FlowTraceNodeInputContent, required: true, location_name: "content"))
+    FlowTraceNodeInputField.add_member(:node_input_name, Shapes::ShapeRef.new(shape: NodeInputName, required: true, location_name: "nodeInputName"))
+    FlowTraceNodeInputField.struct_class = Types::FlowTraceNodeInputField
+
+    FlowTraceNodeInputFields.member = Shapes::ShapeRef.new(shape: FlowTraceNodeInputField)
+
+    FlowTraceNodeOutputContent.add_member(:document, Shapes::ShapeRef.new(shape: Document, location_name: "document"))
+    FlowTraceNodeOutputContent.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    FlowTraceNodeOutputContent.add_member_subclass(:document, Types::FlowTraceNodeOutputContent::Document)
+    FlowTraceNodeOutputContent.add_member_subclass(:unknown, Types::FlowTraceNodeOutputContent::Unknown)
+    FlowTraceNodeOutputContent.struct_class = Types::FlowTraceNodeOutputContent
+
+    FlowTraceNodeOutputEvent.add_member(:fields, Shapes::ShapeRef.new(shape: FlowTraceNodeOutputFields, required: true, location_name: "fields"))
+    FlowTraceNodeOutputEvent.add_member(:node_name, Shapes::ShapeRef.new(shape: NodeName, required: true, location_name: "nodeName"))
+    FlowTraceNodeOutputEvent.add_member(:timestamp, Shapes::ShapeRef.new(shape: DateTimestamp, required: true, location_name: "timestamp"))
+    FlowTraceNodeOutputEvent.struct_class = Types::FlowTraceNodeOutputEvent
+
+    FlowTraceNodeOutputField.add_member(:content, Shapes::ShapeRef.new(shape: FlowTraceNodeOutputContent, required: true, location_name: "content"))
+    FlowTraceNodeOutputField.add_member(:node_output_name, Shapes::ShapeRef.new(shape: NodeOutputName, required: true, location_name: "nodeOutputName"))
+    FlowTraceNodeOutputField.struct_class = Types::FlowTraceNodeOutputField
+
+    FlowTraceNodeOutputFields.member = Shapes::ShapeRef.new(shape: FlowTraceNodeOutputField)
+
     FunctionInvocationInput.add_member(:action_group, Shapes::ShapeRef.new(shape: String, required: true, location_name: "actionGroup"))
     FunctionInvocationInput.add_member(:action_invocation_type, Shapes::ShapeRef.new(shape: ActionInvocationType, location_name: "actionInvocationType"))
     FunctionInvocationInput.add_member(:function, Shapes::ShapeRef.new(shape: String, location_name: "function"))
@@ -582,6 +667,12 @@ module Aws::BedrockAgentRuntime
 
     InputFiles.member = Shapes::ShapeRef.new(shape: InputFile)
 
+    InputPrompt.add_member(:text_prompt, Shapes::ShapeRef.new(shape: TextPrompt, location_name: "textPrompt"))
+    InputPrompt.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    InputPrompt.add_member_subclass(:text_prompt, Types::InputPrompt::TextPrompt)
+    InputPrompt.add_member_subclass(:unknown, Types::InputPrompt::Unknown)
+    InputPrompt.struct_class = Types::InputPrompt
+
     InternalServerException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
     InternalServerException.struct_class = Types::InternalServerException
 
@@ -628,6 +719,7 @@ module Aws::BedrockAgentRuntime
     InvokeAgentResponse[:payload] = :completion
     InvokeAgentResponse[:payload_member] = InvokeAgentResponse.member(:completion)
 
+    InvokeFlowRequest.add_member(:enable_trace, Shapes::ShapeRef.new(shape: Boolean, location_name: "enableTrace"))
     InvokeFlowRequest.add_member(:flow_alias_identifier, Shapes::ShapeRef.new(shape: FlowAliasIdentifier, required: true, location: "uri", location_name: "flowAliasIdentifier"))
     InvokeFlowRequest.add_member(:flow_identifier, Shapes::ShapeRef.new(shape: FlowIdentifier, required: true, location: "uri", location_name: "flowIdentifier"))
     InvokeFlowRequest.add_member(:inputs, Shapes::ShapeRef.new(shape: FlowInputs, required: true, location_name: "inputs"))
@@ -713,6 +805,34 @@ module Aws::BedrockAgentRuntime
     Observation.add_member(:type, Shapes::ShapeRef.new(shape: Type, location_name: "type"))
     Observation.struct_class = Types::Observation
 
+    OptimizePromptRequest.add_member(:input, Shapes::ShapeRef.new(shape: InputPrompt, required: true, location_name: "input"))
+    OptimizePromptRequest.add_member(:target_model_id, Shapes::ShapeRef.new(shape: OptimizePromptRequestTargetModelIdString, required: true, location_name: "targetModelId"))
+    OptimizePromptRequest.struct_class = Types::OptimizePromptRequest
+
+    OptimizePromptResponse.add_member(:optimized_prompt, Shapes::ShapeRef.new(shape: OptimizedPromptStream, required: true, eventstream: true, location_name: "optimizedPrompt"))
+    OptimizePromptResponse.struct_class = Types::OptimizePromptResponse
+    OptimizePromptResponse[:payload] = :optimized_prompt
+    OptimizePromptResponse[:payload_member] = OptimizePromptResponse.member(:optimized_prompt)
+
+    OptimizedPrompt.add_member(:text_prompt, Shapes::ShapeRef.new(shape: TextPrompt, location_name: "textPrompt"))
+    OptimizedPrompt.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    OptimizedPrompt.add_member_subclass(:text_prompt, Types::OptimizedPrompt::TextPrompt)
+    OptimizedPrompt.add_member_subclass(:unknown, Types::OptimizedPrompt::Unknown)
+    OptimizedPrompt.struct_class = Types::OptimizedPrompt
+
+    OptimizedPromptEvent.add_member(:optimized_prompt, Shapes::ShapeRef.new(shape: OptimizedPrompt, location_name: "optimizedPrompt"))
+    OptimizedPromptEvent.struct_class = Types::OptimizedPromptEvent
+
+    OptimizedPromptStream.add_member(:access_denied_exception, Shapes::ShapeRef.new(shape: AccessDeniedException, location_name: "accessDeniedException"))
+    OptimizedPromptStream.add_member(:analyze_prompt_event, Shapes::ShapeRef.new(shape: AnalyzePromptEvent, event: true, location_name: "analyzePromptEvent"))
+    OptimizedPromptStream.add_member(:bad_gateway_exception, Shapes::ShapeRef.new(shape: BadGatewayException, location_name: "badGatewayException"))
+    OptimizedPromptStream.add_member(:dependency_failed_exception, Shapes::ShapeRef.new(shape: DependencyFailedException, location_name: "dependencyFailedException"))
+    OptimizedPromptStream.add_member(:internal_server_exception, Shapes::ShapeRef.new(shape: InternalServerException, location_name: "internalServerException"))
+    OptimizedPromptStream.add_member(:optimized_prompt_event, Shapes::ShapeRef.new(shape: OptimizedPromptEvent, event: true, location_name: "optimizedPromptEvent"))
+    OptimizedPromptStream.add_member(:throttling_exception, Shapes::ShapeRef.new(shape: ThrottlingException, location_name: "throttlingException"))
+    OptimizedPromptStream.add_member(:validation_exception, Shapes::ShapeRef.new(shape: ValidationException, location_name: "validationException"))
+    OptimizedPromptStream.struct_class = Types::OptimizedPromptStream
+
     OrchestrationConfiguration.add_member(:additional_model_request_fields, Shapes::ShapeRef.new(shape: AdditionalModelRequestFields, location_name: "additionalModelRequestFields"))
     OrchestrationConfiguration.add_member(:inference_config, Shapes::ShapeRef.new(shape: InferenceConfig, location_name: "inferenceConfig"))
     OrchestrationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
@@ -984,6 +1104,9 @@ module Aws::BedrockAgentRuntime
     TextInferenceConfig.add_member(:top_p, Shapes::ShapeRef.new(shape: TopP, location_name: "topP"))
     TextInferenceConfig.struct_class = Types::TextInferenceConfig
 
+    TextPrompt.add_member(:text, Shapes::ShapeRef.new(shape: TextPromptTextString, required: true, location_name: "text"))
+    TextPrompt.struct_class = Types::TextPrompt
+
     TextResponsePart.add_member(:span, Shapes::ShapeRef.new(shape: Span, location_name: "span"))
     TextResponsePart.add_member(:text, Shapes::ShapeRef.new(shape: String, location_name: "text"))
     TextResponsePart.struct_class = Types::TextResponsePart
@@ -1113,6 +1236,20 @@ module Aws::BedrockAgentRuntime
         o.errors << Shapes::ShapeRef.new(shape: ServiceQuotaExceededException)
       end)
 
+      api.add_operation(:optimize_prompt, Seahorse::Model::Operation.new.tap do |o|
+        o.name = "OptimizePrompt"
+        o.http_method = "POST"
+        o.http_request_uri = "/optimize-prompt"
+        o.input = Shapes::ShapeRef.new(shape: OptimizePromptRequest)
+        o.output = Shapes::ShapeRef.new(shape: OptimizePromptResponse)
+        o.errors << Shapes::ShapeRef.new(shape: ValidationException)
+        o.errors << Shapes::ShapeRef.new(shape: InternalServerException)
+        o.errors << Shapes::ShapeRef.new(shape: DependencyFailedException)
+        o.errors << Shapes::ShapeRef.new(shape: BadGatewayException)
+        o.errors << Shapes::ShapeRef.new(shape: ThrottlingException)
+        o.errors << Shapes::ShapeRef.new(shape: AccessDeniedException)
+      end)
+
       api.add_operation(:retrieve, Seahorse::Model::Operation.new.tap do |o|
         o.name = "Retrieve"
         o.http_method = "POST"