aws-sdk-bedrockagentruntime 1.7.0 → 1.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3a7807ae877e929d1ebf772ede849e026038aef63d4cab849617d61b17ca4ab2
-  data.tar.gz: 65456ef078ed1cf91f3f6bb9771417e2a8754a9112f73f40a967c8e2d931a04c
+  metadata.gz: d0b33ecfc30b93d8a65929e3cd819ade75f55b90acf67de7f7bb0f665a912561
+  data.tar.gz: 3f3ec35ccc289e5a22ccc82b5e4c3910b32e8534437b413f924b6b4e44bf657c
 SHA512:
-  metadata.gz: c87fb48cd5dca38d6bc19cab8ffd706e48d38b824eb5fab4bcada125e8c03de0b2bf8e9fa2952390d2b3a52b5f9691a9c4ef1dbfd41750f7841d9ae1af99b0b7
-  data.tar.gz: 5c541812632089ec05e6323e2297a14795fd4964742bd7fab30c5ca2d365828c62c9babe1460cbcd9d0aa866364692cf21fafdc191e81fcad22b5362b8672258
+  metadata.gz: 7f2063b06a4d1430646b9970907d489ccd2845dab6105f682fdff15942930260e915afcd6845eb031795b7fed763db2eeb5b0206535933f908a171d5fbb98f2a
+  data.tar.gz: 7560d744e4abfe1c081844c095cebcd69e970ac17e049f40a9a8bc377c7a2ec8f2e4021b1a3cbd8576e18c923da76c988117d3330859e0006d6587ee037fc231

data/CHANGELOG.md

@@ -1,6 +1,16 @@
 Unreleased Changes
 ------------------
 
+1.9.0 (2024-05-13)
+------------------
+
+* Feature - Code Generated Changes, see `./build_tools` or `aws-sdk-core`'s CHANGELOG.md for details.
+
+1.8.0 (2024-05-09)
+------------------
+
+* Feature - This release adds support to provide guardrail configuration and modify inference parameters that are then used in RetrieveAndGenerate API in Agents for Amazon Bedrock.
+
 1.7.0 (2024-04-25)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.7.0
+1.9.0

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

@@ -312,8 +312,9 @@ module Aws::BedrockAgentRuntime
     #
     #   @option options [String] :sdk_ua_app_id
     #     A unique and opaque application ID that is appended to the
-    #     User-Agent header as app/<sdk_ua_app_id>. It should have a
-    #     maximum length of 50.
+    #     User-Agent header as app/sdk_ua_app_id. It should have a
+    #     maximum length of 50. This variable is sourced from environment
+    #     variable AWS_SDK_UA_APP_ID or the shared config profile attribute sdk_ua_app_id.
     #
     #   @option options [String] :secret_access_key
     #
@@ -423,13 +424,13 @@ module Aws::BedrockAgentRuntime
 
     # @!group API Operations
 
-    # Sends a prompt for the agent to process and respond to. Use return
-    # control event type for function calling.
-    #
     # <note markdown="1"> The CLI doesn't support `InvokeAgent`.
     #
     #  </note>
     #
+    # Sends a prompt for the agent to process and respond to. Note the
+    # following fields for the request:
+    #
     # * To continue the same conversation with an agent, use the same
     #   `sessionId` value in the request.
     #
@@ -442,9 +443,8 @@ module Aws::BedrockAgentRuntime
     # * End a conversation by setting `endSession` to `true`.
     #
     # * In the `sessionState` object, you can include attributes for the
-    #   session or prompt or parameters returned from the action group.
-    #
-    # * Use return control event type for function calling.
+    #   session or prompt or, if you configured an action group to return
+    #   control, results from invocation of the action group.
     #
     # The response is returned in the `bytes` field of the `chunk` object.
     #
@@ -454,6 +454,10 @@ module Aws::BedrockAgentRuntime
     # * If you set `enableTrace` to `true` in the request, you can trace the
     #   agent's steps and reasoning process that led it to the response.
     #
+    # * If the action predicted was configured to return control, the
+    #   response returns parameters for the action, elicited from the user,
+    #   in the `returnControl` field.
+    #
     # * Errors are also surfaced in the response.
     #
     #
@@ -480,6 +484,11 @@ module Aws::BedrockAgentRuntime
     # @option params [String] :input_text
     #   The prompt text to send the agent.
     #
+    #   <note markdown="1"> If you include `returnControlInvocationResults` in the `sessionState`
+    #   field, the `inputText` field will be ignored.
+    #
+    #    </note>
+    #
     # @option params [required, String] :session_id
     #   The unique identifier of the session. Use the same value across
     #   requests to continue the same conversation.
@@ -488,6 +497,11 @@ module Aws::BedrockAgentRuntime
     #   Contains parameters that specify various attributes of the session.
     #   For more information, see [Control session context][1].
     #
+    #   <note markdown="1"> If you include `returnControlInvocationResults` in the `sessionState`
+    #   field, the `inputText` field will be ignored.
+    #
+    #    </note>
+    #
     #
     #
     #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
@@ -1049,6 +1063,7 @@ module Aws::BedrockAgentRuntime
     # @return [Types::RetrieveAndGenerateResponse] Returns a {Seahorse::Client::Response response} object which responds to the following methods:
     #
     #   * {Types::RetrieveAndGenerateResponse#citations #citations} => Array&lt;Types::Citation&gt;
+    #   * {Types::RetrieveAndGenerateResponse#guardrail_action #guardrail_action} => String
     #   * {Types::RetrieveAndGenerateResponse#output #output} => Types::RetrieveAndGenerateOutput
     #   * {Types::RetrieveAndGenerateResponse#session_id #session_id} => String
     #
@@ -1061,6 +1076,22 @@ module Aws::BedrockAgentRuntime
     #     retrieve_and_generate_configuration: {
     #       external_sources_configuration: {
     #         generation_configuration: {
+    #           additional_model_request_fields: {
+    #             "AdditionalModelRequestFieldsKey" => {
+    #             },
+    #           },
+    #           guardrail_configuration: {
+    #             guardrail_id: "GuardrailConfigurationGuardrailIdString", # required
+    #             guardrail_version: "GuardrailConfigurationGuardrailVersionString", # required
+    #           },
+    #           inference_config: {
+    #             text_inference_config: {
+    #               max_tokens: 1,
+    #               stop_sequences: ["RAGStopSequencesMemberString"],
+    #               temperature: 1.0,
+    #               top_p: 1.0,
+    #             },
+    #           },
     #           prompt_template: {
     #             text_prompt_template: "TextPromptTemplate",
     #           },
@@ -1082,6 +1113,22 @@ module Aws::BedrockAgentRuntime
     #       },
     #       knowledge_base_configuration: {
     #         generation_configuration: {
+    #           additional_model_request_fields: {
+    #             "AdditionalModelRequestFieldsKey" => {
+    #             },
+    #           },
+    #           guardrail_configuration: {
+    #             guardrail_id: "GuardrailConfigurationGuardrailIdString", # required
+    #             guardrail_version: "GuardrailConfigurationGuardrailVersionString", # required
+    #           },
+    #           inference_config: {
+    #             text_inference_config: {
+    #               max_tokens: 1,
+    #               stop_sequences: ["RAGStopSequencesMemberString"],
+    #               temperature: 1.0,
+    #               top_p: 1.0,
+    #             },
+    #           },
     #           prompt_template: {
     #             text_prompt_template: "TextPromptTemplate",
     #           },
@@ -1171,6 +1218,7 @@ module Aws::BedrockAgentRuntime
     #   resp.citations[0].retrieved_references[0].location.s3_location.uri #=> String
     #   resp.citations[0].retrieved_references[0].location.type #=> String, one of "S3"
     #   resp.citations[0].retrieved_references[0].metadata #=> Hash
+    #   resp.guardrail_action #=> String, one of "INTERVENED", "NONE"
     #   resp.output.text #=> String
     #   resp.session_id #=> String
     #
@@ -1196,7 +1244,7 @@ module Aws::BedrockAgentRuntime
         params: params,
         config: config)
       context[:gem_name] = 'aws-sdk-bedrockagentruntime'
-      context[:gem_version] = '1.7.0'
+      context[:gem_version] = '1.9.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -18,6 +18,9 @@ module Aws::BedrockAgentRuntime
     ActionGroupInvocationOutput = Shapes::StructureShape.new(name: 'ActionGroupInvocationOutput')
     ActionGroupName = Shapes::StringShape.new(name: 'ActionGroupName')
     ActionGroupOutputString = Shapes::StringShape.new(name: 'ActionGroupOutputString')
+    AdditionalModelRequestFields = Shapes::MapShape.new(name: 'AdditionalModelRequestFields')
+    AdditionalModelRequestFieldsKey = Shapes::StringShape.new(name: 'AdditionalModelRequestFieldsKey')
+    AdditionalModelRequestFieldsValue = Shapes::DocumentShape.new(name: 'AdditionalModelRequestFieldsValue', document: true)
     AgentAliasId = Shapes::StringShape.new(name: 'AgentAliasId')
     AgentId = Shapes::StringShape.new(name: 'AgentId')
     AgentVersion = Shapes::StringShape.new(name: 'AgentVersion')
@@ -62,7 +65,12 @@ module Aws::BedrockAgentRuntime
     FunctionResult = Shapes::StructureShape.new(name: 'FunctionResult')
     GeneratedResponsePart = Shapes::StructureShape.new(name: 'GeneratedResponsePart')
     GenerationConfiguration = Shapes::StructureShape.new(name: 'GenerationConfiguration')
+    GuadrailAction = Shapes::StringShape.new(name: 'GuadrailAction')
+    GuardrailConfiguration = Shapes::StructureShape.new(name: 'GuardrailConfiguration')
+    GuardrailConfigurationGuardrailIdString = Shapes::StringShape.new(name: 'GuardrailConfigurationGuardrailIdString')
+    GuardrailConfigurationGuardrailVersionString = Shapes::StringShape.new(name: 'GuardrailConfigurationGuardrailVersionString')
     Identifier = Shapes::StringShape.new(name: 'Identifier')
+    InferenceConfig = Shapes::StructureShape.new(name: 'InferenceConfig')
     InferenceConfiguration = Shapes::StructureShape.new(name: 'InferenceConfiguration')
     InputText = Shapes::StringShape.new(name: 'InputText')
     Integer = Shapes::IntegerShape.new(name: 'Integer')
@@ -88,6 +96,7 @@ module Aws::BedrockAgentRuntime
     KnowledgeBaseVectorSearchConfiguration = Shapes::StructureShape.new(name: 'KnowledgeBaseVectorSearchConfiguration')
     KnowledgeBaseVectorSearchConfigurationNumberOfResultsInteger = Shapes::IntegerShape.new(name: 'KnowledgeBaseVectorSearchConfigurationNumberOfResultsInteger')
     LambdaArn = Shapes::StringShape.new(name: 'LambdaArn')
+    MaxTokens = Shapes::IntegerShape.new(name: 'MaxTokens')
     MaximumLength = Shapes::IntegerShape.new(name: 'MaximumLength')
     MimeType = Shapes::StringShape.new(name: 'MimeType')
     ModelInvocationInput = Shapes::StructureShape.new(name: 'ModelInvocationInput')
@@ -112,6 +121,8 @@ module Aws::BedrockAgentRuntime
     PromptText = Shapes::StringShape.new(name: 'PromptText')
     PromptType = Shapes::StringShape.new(name: 'PromptType')
     PropertyParameters = Shapes::StructureShape.new(name: 'PropertyParameters')
+    RAGStopSequences = Shapes::ListShape.new(name: 'RAGStopSequences')
+    RAGStopSequencesMemberString = Shapes::StringShape.new(name: 'RAGStopSequencesMemberString')
     Rationale = Shapes::StructureShape.new(name: 'Rationale')
     RationaleString = Shapes::StringShape.new(name: 'RationaleString')
     RepromptResponse = Shapes::StructureShape.new(name: 'RepromptResponse')
@@ -157,6 +168,7 @@ module Aws::BedrockAgentRuntime
     StopSequences = Shapes::ListShape.new(name: 'StopSequences')
     String = Shapes::StringShape.new(name: 'String')
     Temperature = Shapes::FloatShape.new(name: 'Temperature')
+    TextInferenceConfig = Shapes::StructureShape.new(name: 'TextInferenceConfig')
     TextPromptTemplate = Shapes::StringShape.new(name: 'TextPromptTemplate')
     TextResponsePart = Shapes::StructureShape.new(name: 'TextResponsePart')
     ThrottlingException = Shapes::StructureShape.new(name: 'ThrottlingException')
@@ -184,6 +196,9 @@ module Aws::BedrockAgentRuntime
     ActionGroupInvocationOutput.add_member(:text, Shapes::ShapeRef.new(shape: ActionGroupOutputString, location_name: "text"))
     ActionGroupInvocationOutput.struct_class = Types::ActionGroupInvocationOutput
 
+    AdditionalModelRequestFields.key = Shapes::ShapeRef.new(shape: AdditionalModelRequestFieldsKey)
+    AdditionalModelRequestFields.value = Shapes::ShapeRef.new(shape: AdditionalModelRequestFieldsValue)
+
     ApiContentMap.key = Shapes::ShapeRef.new(shape: String)
     ApiContentMap.value = Shapes::ShapeRef.new(shape: PropertyParameters)
 
@@ -250,6 +265,9 @@ module Aws::BedrockAgentRuntime
 
     ExternalSources.member = Shapes::ShapeRef.new(shape: ExternalSource)
 
+    ExternalSourcesGenerationConfiguration.add_member(:additional_model_request_fields, Shapes::ShapeRef.new(shape: AdditionalModelRequestFields, location_name: "additionalModelRequestFields"))
+    ExternalSourcesGenerationConfiguration.add_member(:guardrail_configuration, Shapes::ShapeRef.new(shape: GuardrailConfiguration, location_name: "guardrailConfiguration"))
+    ExternalSourcesGenerationConfiguration.add_member(:inference_config, Shapes::ShapeRef.new(shape: InferenceConfig, location_name: "inferenceConfig"))
     ExternalSourcesGenerationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
     ExternalSourcesGenerationConfiguration.struct_class = Types::ExternalSourcesGenerationConfiguration
 
@@ -290,9 +308,19 @@ module Aws::BedrockAgentRuntime
     GeneratedResponsePart.add_member(:text_response_part, Shapes::ShapeRef.new(shape: TextResponsePart, location_name: "textResponsePart"))
     GeneratedResponsePart.struct_class = Types::GeneratedResponsePart
 
+    GenerationConfiguration.add_member(:additional_model_request_fields, Shapes::ShapeRef.new(shape: AdditionalModelRequestFields, location_name: "additionalModelRequestFields"))
+    GenerationConfiguration.add_member(:guardrail_configuration, Shapes::ShapeRef.new(shape: GuardrailConfiguration, location_name: "guardrailConfiguration"))
+    GenerationConfiguration.add_member(:inference_config, Shapes::ShapeRef.new(shape: InferenceConfig, location_name: "inferenceConfig"))
     GenerationConfiguration.add_member(:prompt_template, Shapes::ShapeRef.new(shape: PromptTemplate, location_name: "promptTemplate"))
     GenerationConfiguration.struct_class = Types::GenerationConfiguration
 
+    GuardrailConfiguration.add_member(:guardrail_id, Shapes::ShapeRef.new(shape: GuardrailConfigurationGuardrailIdString, required: true, location_name: "guardrailId"))
+    GuardrailConfiguration.add_member(:guardrail_version, Shapes::ShapeRef.new(shape: GuardrailConfigurationGuardrailVersionString, required: true, location_name: "guardrailVersion"))
+    GuardrailConfiguration.struct_class = Types::GuardrailConfiguration
+
+    InferenceConfig.add_member(:text_inference_config, Shapes::ShapeRef.new(shape: TextInferenceConfig, location_name: "textInferenceConfig"))
+    InferenceConfig.struct_class = Types::InferenceConfig
+
     InferenceConfiguration.add_member(:maximum_length, Shapes::ShapeRef.new(shape: MaximumLength, location_name: "maximumLength"))
     InferenceConfiguration.add_member(:stop_sequences, Shapes::ShapeRef.new(shape: StopSequences, location_name: "stopSequences"))
     InferenceConfiguration.add_member(:temperature, Shapes::ShapeRef.new(shape: Temperature, location_name: "temperature"))
@@ -457,6 +485,8 @@ module Aws::BedrockAgentRuntime
     PropertyParameters.add_member(:properties, Shapes::ShapeRef.new(shape: ParameterList, location_name: "properties"))
     PropertyParameters.struct_class = Types::PropertyParameters
 
+    RAGStopSequences.member = Shapes::ShapeRef.new(shape: RAGStopSequencesMemberString)
+
     Rationale.add_member(:text, Shapes::ShapeRef.new(shape: RationaleString, location_name: "text"))
     Rationale.add_member(:trace_id, Shapes::ShapeRef.new(shape: TraceId, location_name: "traceId"))
     Rationale.struct_class = Types::Rationale
@@ -547,6 +577,7 @@ module Aws::BedrockAgentRuntime
     RetrieveAndGenerateRequest.struct_class = Types::RetrieveAndGenerateRequest
 
     RetrieveAndGenerateResponse.add_member(:citations, Shapes::ShapeRef.new(shape: Citations, location_name: "citations"))
+    RetrieveAndGenerateResponse.add_member(:guardrail_action, Shapes::ShapeRef.new(shape: GuadrailAction, location_name: "guardrailAction"))
     RetrieveAndGenerateResponse.add_member(:output, Shapes::ShapeRef.new(shape: RetrieveAndGenerateOutput, required: true, location_name: "output"))
     RetrieveAndGenerateResponse.add_member(:session_id, Shapes::ShapeRef.new(shape: SessionId, required: true, location_name: "sessionId"))
     RetrieveAndGenerateResponse.struct_class = Types::RetrieveAndGenerateResponse
@@ -598,6 +629,12 @@ module Aws::BedrockAgentRuntime
 
     StopSequences.member = Shapes::ShapeRef.new(shape: String)
 
+    TextInferenceConfig.add_member(:max_tokens, Shapes::ShapeRef.new(shape: MaxTokens, location_name: "maxTokens"))
+    TextInferenceConfig.add_member(:stop_sequences, Shapes::ShapeRef.new(shape: RAGStopSequences, location_name: "stopSequences"))
+    TextInferenceConfig.add_member(:temperature, Shapes::ShapeRef.new(shape: Temperature, location_name: "temperature"))
+    TextInferenceConfig.add_member(:top_p, Shapes::ShapeRef.new(shape: TopP, location_name: "topP"))
+    TextInferenceConfig.struct_class = Types::TextInferenceConfig
+
     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

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

@@ -91,13 +91,13 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * In the `returnControl` field of the [Retrieve response][1]
+    # * In the `returnControl` field of the [InvokeAgent response][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] action_group
     #   The action group that the API operation belongs to.
@@ -137,13 +137,13 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * [Retrieve response][1]
+    # * [InvokeAgent response][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] name
     #   The name of the parameter.
@@ -172,13 +172,13 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * [Retrieve response][1]
+    # * [InvokeAgent response][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] content
     #   The content of the request body. The key of the object in this field
@@ -198,13 +198,14 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * In the `returnControlInvocationResults` of the [Retrieve request][1]
+    # * In the `returnControlInvocationResults` of the [InvokeAgent
+    #   request][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_RequestSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_RequestSyntax
     #
     # @!attribute [rw] action_group
     #   The action group that the API operation belongs to.
@@ -225,8 +226,8 @@ module Aws::BedrockAgentRuntime
     #
     # @!attribute [rw] response_body
     #   The response body from the API operation. The key of the object is
-    #   the content type. The response may be returned directly or from the
-    #   Lambda function.
+    #   the content type (currently, only `TEXT` is supported). The response
+    #   may be returned directly or from the Lambda function.
     #   @return [Hash<String,Types::ContentBody>]
     #
     # @!attribute [rw] response_state
@@ -317,13 +318,13 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * [Retrieve response][1] – in the `citations` field
+    # * [InvokeAgent response][1] – in the `citations` field
     #
     # * [RetrieveAndGenerate response][2] – in the `citations` field
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_ResponseSyntax
     #
     # @!attribute [rw] generated_response_part
@@ -363,14 +364,14 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * In the `returnControlInvocationResults` field of the [Retrieve
+    # * In the `returnControlInvocationResults` field of the [InvokeAgent
     #   request][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_RequestSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_RequestSyntax
     #
     # @!attribute [rw] body
     #   The body of the API response.
@@ -434,6 +435,22 @@ module Aws::BedrockAgentRuntime
     # Contains the generation configuration of the external source wrapper
     # object.
     #
+    # @!attribute [rw] additional_model_request_fields
+    #   Additional model parameters and their corresponding values not
+    #   included in the textInferenceConfig structure for an external
+    #   source. Takes in custom model parameters specific to the language
+    #   model being used.
+    #   @return [Hash<String,Hash,Array,String,Numeric,Boolean>]
+    #
+    # @!attribute [rw] guardrail_configuration
+    #   The configuration details for the guardrail.
+    #   @return [Types::GuardrailConfiguration]
+    #
+    # @!attribute [rw] inference_config
+    #   Configuration settings for inference when using RetrieveAndGenerate
+    #   to generate responses while using an external source.
+    #   @return [Types::InferenceConfig]
+    #
     # @!attribute [rw] prompt_template
     #   Contain the textPromptTemplate string for the external source
     #   wrapper object.
@@ -442,6 +459,9 @@ module Aws::BedrockAgentRuntime
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/ExternalSourcesGenerationConfiguration AWS API Documentation
     #
     class ExternalSourcesGenerationConfiguration < Struct.new(
+      :additional_model_request_fields,
+      :guardrail_configuration,
+      :inference_config,
       :prompt_template)
       SENSITIVE = []
       include Aws::Structure
@@ -545,13 +565,13 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * In the `returnControl` field of the [Retrieve response][1]
+    # * In the `returnControl` field of the [InvokeAgent response][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] action_group
     #   The action group that the function belongs to.
@@ -579,13 +599,13 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * In the `returnControl` field of the [Retrieve response][1]
+    # * In the `returnControl` field of the [InvokeAgent response][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] name
     #   The name of the parameter.
@@ -614,13 +634,14 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * In the `returnControlInvocationResults` of the [Retrieve request][1]
+    # * In the `returnControlInvocationResults` of the [InvokeAgent
+    #   request][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_RequestSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_RequestSyntax
     #
     # @!attribute [rw] action_group
     #   The action group that the function belongs to.
@@ -631,8 +652,10 @@ module Aws::BedrockAgentRuntime
     #   @return [String]
     #
     # @!attribute [rw] response_body
-    #   The response from the function call using the parameters. The
-    #   response may be returned directly or from the Lambda function.
+    #   The response from the function call using the parameters. The key of
+    #   the object is the content type (currently, only `TEXT` is
+    #   supported). The response may be returned directly or from the Lambda
+    #   function.
     #   @return [Hash<String,Types::ContentBody>]
     #
     # @!attribute [rw] response_state
@@ -659,14 +682,14 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * [Retrieve response][1] – in the `generatedResponsePart` field
+    # * [InvokeAgent response][1] – in the `generatedResponsePart` field
     #
     # * [RetrieveAndGenerate response][2] – in the `generatedResponsePart`
     #   field
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_ResponseSyntax
     #
     # @!attribute [rw] text_response_part
@@ -695,6 +718,22 @@ module Aws::BedrockAgentRuntime
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_RequestSyntax
     #
+    # @!attribute [rw] additional_model_request_fields
+    #   Additional model parameters and corresponding values not included in
+    #   the textInferenceConfig structure for a knowledge base. This allows
+    #   users to provide custom model parameters specific to the language
+    #   model being used.
+    #   @return [Hash<String,Hash,Array,String,Numeric,Boolean>]
+    #
+    # @!attribute [rw] guardrail_configuration
+    #   The configuration details for the guardrail.
+    #   @return [Types::GuardrailConfiguration]
+    #
+    # @!attribute [rw] inference_config
+    #   Configuration settings for inference when using RetrieveAndGenerate
+    #   to generate responses while using a knowledge base as a source.
+    #   @return [Types::InferenceConfig]
+    #
     # @!attribute [rw] prompt_template
     #   Contains the template for the prompt that's sent to the model for
     #   response generation.
@@ -703,11 +742,49 @@ module Aws::BedrockAgentRuntime
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/GenerationConfiguration AWS API Documentation
     #
     class GenerationConfiguration < Struct.new(
+      :additional_model_request_fields,
+      :guardrail_configuration,
+      :inference_config,
       :prompt_template)
       SENSITIVE = []
       include Aws::Structure
     end
 
+    # The configuration details for the guardrail.
+    #
+    # @!attribute [rw] guardrail_id
+    #   The unique identifier for the guardrail.
+    #   @return [String]
+    #
+    # @!attribute [rw] guardrail_version
+    #   The version of the guardrail.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/GuardrailConfiguration AWS API Documentation
+    #
+    class GuardrailConfiguration < Struct.new(
+      :guardrail_id,
+      :guardrail_version)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # The configuration for inference settings when generating responses
+    # using RetrieveAndGenerate.
+    #
+    # @!attribute [rw] text_inference_config
+    #   Configuration settings specific to text generation while generating
+    #   responses using RetrieveAndGenerate.
+    #   @return [Types::TextInferenceConfig]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/InferenceConfig AWS API Documentation
+    #
+    class InferenceConfig < Struct.new(
+      :text_inference_config)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Specifications about the inference parameters that were provided
     # alongside the prompt. These are specified in the
     # [PromptOverrideConfiguration][1] object that was set when the agent
@@ -816,13 +893,13 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * In the `returnControl` field of the [Retrieve response][1]
+    # * In the `returnControl` field of the [InvokeAgent response][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @note InvocationInputMember is a union - when returned from an API call exactly one value will be set and the returned type will be a subclass of InvocationInputMember corresponding to the set member.
     #
@@ -851,17 +928,21 @@ module Aws::BedrockAgentRuntime
       class Unknown < InvocationInputMember; end
     end
 
-    # A result from the action group invocation.
+    # A result from the invocation of an action. For more information, see
+    # [Return control to the agent developer][1] and [Control session
+    # context][2].
     #
     # This data type is used in the following API operations:
     #
-    # * [Retrieve request][1]
+    # * [InvokeAgent request][3]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_RequestSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-returncontrol.html
+    # [2]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
+    # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_RequestSyntax
     #
     # @note InvocationResultMember is a union - when making an API calls you must set exactly one of the members.
     #
@@ -911,6 +992,11 @@ module Aws::BedrockAgentRuntime
     #
     # @!attribute [rw] input_text
     #   The prompt text to send the agent.
+    #
+    #   <note markdown="1"> If you include `returnControlInvocationResults` in the
+    #   `sessionState` field, the `inputText` field will be ignored.
+    #
+    #    </note>
     #   @return [String]
     #
     # @!attribute [rw] session_id
@@ -922,6 +1008,11 @@ module Aws::BedrockAgentRuntime
     #   Contains parameters that specify various attributes of the session.
     #   For more information, see [Control session context][1].
     #
+    #   <note markdown="1"> If you include `returnControlInvocationResults` in the
+    #   `sessionState` field, the `inputText` field will be ignored.
+    #
+    #    </note>
+    #
     #
     #
     #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
@@ -1838,12 +1929,13 @@ module Aws::BedrockAgentRuntime
     #
     # * [RetrieveAndGenerate response][2] – in the `content` field
     #
-    # * [Retrieve response][1] – in the `content` field
+    # * [InvokeAgent response][3] – in the `content` field
     #
     #
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
     # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_ResponseSyntax
+    # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] text
     #   The cited text from the data source.
@@ -1865,12 +1957,13 @@ module Aws::BedrockAgentRuntime
     #
     # * [RetrieveAndGenerate response][2] – in the `location` field
     #
-    # * [Retrieve response][1] – in the `locatino` field
+    # * [InvokeAgent response][3] – in the `locatino` field
     #
     #
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
     # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_ResponseSyntax
+    # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] s3_location
     #   Contains the S3 location of the data source.
@@ -1897,12 +1990,13 @@ module Aws::BedrockAgentRuntime
     #
     # * [RetrieveAndGenerate response][2] – in the `s3Location` field
     #
-    # * [Retrieve response][1] – in the `s3Location` field
+    # * [InvokeAgent response][3] – in the `s3Location` field
     #
     #
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
     # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_ResponseSyntax
+    # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] uri
     #   The S3 URI of the data source.
@@ -2039,6 +2133,10 @@ module Aws::BedrockAgentRuntime
     #   sources.
     #   @return [Array<Types::Citation>]
     #
+    # @!attribute [rw] guardrail_action
+    #   Specifies if there is a guardrail intervention in the response.
+    #   @return [String]
+    #
     # @!attribute [rw] output
     #   Contains the response generated from querying the knowledge base.
     #   @return [Types::RetrieveAndGenerateOutput]
@@ -2052,6 +2150,7 @@ module Aws::BedrockAgentRuntime
     #
     class RetrieveAndGenerateResponse < Struct.new(
       :citations,
+      :guardrail_action,
       :output,
       :session_id)
       SENSITIVE = [:output]
@@ -2143,12 +2242,12 @@ module Aws::BedrockAgentRuntime
     # * [RetrieveAndGenerate response][1] – in the `retrievedReferences`
     #   field
     #
-    # * [Retrieve response][2] – in the `retrievedReferences` field
+    # * [InvokeAgent response][2] – in the `retrievedReferences` field
     #
     #
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_ResponseSyntax
-    # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] content
     #   Contains the cited text from the data source.
@@ -2182,13 +2281,13 @@ module Aws::BedrockAgentRuntime
     #
     # This data type is used in the following API operations:
     #
-    # * [Retrieve response][1]
+    # * [InvokeAgent response][1]
     #
     # ^
     #
     #
     #
-    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] invocation_id
     #   The identifier of the action group invocation.
@@ -2254,7 +2353,17 @@ module Aws::BedrockAgentRuntime
     # [3]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
     #
     # @!attribute [rw] invocation_id
-    #   The identifier of the invocation.
+    #   The identifier of the invocation of an action. This value must match
+    #   the `invocationId` returned in the `InvokeAgent` response for the
+    #   action whose results are provided in the
+    #   `returnControlInvocationResults` field. For more information, see
+    #   [Return control to the agent developer][1] and [Control session
+    #   context][2].
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-returncontrol.html
+    #   [2]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
     #   @return [String]
     #
     # @!attribute [rw] prompt_session_attributes
@@ -2271,7 +2380,17 @@ module Aws::BedrockAgentRuntime
     #
     # @!attribute [rw] return_control_invocation_results
     #   Contains information about the results from the action group
-    #   invocation.
+    #   invocation. For more information, see [Return control to the agent
+    #   developer][1] and [Control session context][2].
+    #
+    #   <note markdown="1"> If you include this field, the `inputText` field will be ignored.
+    #
+    #    </note>
+    #
+    #
+    #
+    #   [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-returncontrol.html
+    #   [2]: https://docs.aws.amazon.com/bedrock/latest/userguide/agents-session-state.html
     #   @return [Array<Types::InvocationResultMember>]
     #
     # @!attribute [rw] session_attributes
@@ -2297,12 +2416,12 @@ module Aws::BedrockAgentRuntime
     #
     # * [RetrieveAndGenerate response][1] – in the `span` field
     #
-    # * [Retrieve response][2] – in the `span` field
+    # * [InvokeAgent response][2] – in the `span` field
     #
     #
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_ResponseSyntax
-    # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] end
     #   Where the text with a citation ends in the generated output.
@@ -2321,6 +2440,62 @@ module Aws::BedrockAgentRuntime
       include Aws::Structure
     end
 
+    # Configuration settings for text generation using a language model via
+    # the RetrieveAndGenerate operation. Includes parameters like
+    # temperature, top-p, maximum token count, and stop sequences.
+    #
+    # <note markdown="1"> The valid range of `maxTokens` depends on the accepted values for your
+    # chosen model's inference parameters. To see the inference parameters
+    # for your model, see [Inference parameters for foundation models.][1]
+    #
+    #  </note>
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters.html
+    #
+    # @!attribute [rw] max_tokens
+    #   The maximum number of tokens to generate in the output text. Do not
+    #   use the minimum of 0 or the maximum of 65536. The limit values
+    #   described here are arbitary values, for actual values consult the
+    #   limits defined by your specific model.
+    #   @return [Integer]
+    #
+    # @!attribute [rw] stop_sequences
+    #   A list of sequences of characters that, if generated, will cause the
+    #   model to stop generating further tokens. Do not use a minimum length
+    #   of 1 or a maximum length of 1000. The limit values described here
+    #   are arbitary values, for actual values consult the limits defined by
+    #   your specific model.
+    #   @return [Array<String>]
+    #
+    # @!attribute [rw] temperature
+    #   Controls the random-ness of text generated by the language model,
+    #   influencing how much the model sticks to the most predictable next
+    #   words versus exploring more surprising options. A lower temperature
+    #   value (e.g. 0.2 or 0.3) makes model outputs more deterministic or
+    #   predictable, while a higher temperature (e.g. 0.8 or 0.9) makes the
+    #   outputs more creative or unpredictable.
+    #   @return [Float]
+    #
+    # @!attribute [rw] top_p
+    #   A probability distribution threshold which controls what the model
+    #   considers for the set of possible next tokens. The model will only
+    #   consider the top p% of the probability distribution when generating
+    #   the next token.
+    #   @return [Float]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-runtime-2023-07-26/TextInferenceConfig AWS API Documentation
+    #
+    class TextInferenceConfig < Struct.new(
+      :max_tokens,
+      :stop_sequences,
+      :temperature,
+      :top_p)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains the part of the generated text that contains a citation,
     # alongside where it begins and ends.
     #
@@ -2328,12 +2503,12 @@ module Aws::BedrockAgentRuntime
     #
     # * [RetrieveAndGenerate response][1] – in the `textResponsePart` field
     #
-    # * [Retrieve response][2] – in the `textResponsePart` field
+    # * [InvokeAgent response][2] – in the `textResponsePart` field
     #
     #
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_RetrieveAndGenerate.html#API_agent-runtime_RetrieveAndGenerate_ResponseSyntax
-    # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_Retrieve.html#API_agent-runtime_Retrieve_ResponseSyntax
+    # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent-runtime_InvokeAgent.html#API_agent-runtime_InvokeAgent_ResponseSyntax
     #
     # @!attribute [rw] span
     #   Contains information about where the text with a citation begins and

data/lib/aws-sdk-bedrockagentruntime.rb

@@ -53,6 +53,6 @@ require_relative 'aws-sdk-bedrockagentruntime/event_streams'
 # @!group service
 module Aws::BedrockAgentRuntime
 
-  GEM_VERSION = '1.7.0'
+  GEM_VERSION = '1.9.0'
 
 end

data/sig/client.rbs

@@ -196,6 +196,7 @@ module Aws
       interface _RetrieveAndGenerateResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::RetrieveAndGenerateResponse]
         def citations: () -> ::Array[Types::Citation]
+        def guardrail_action: () -> ("INTERVENED" | "NONE")
         def output: () -> Types::RetrieveAndGenerateOutput
         def session_id: () -> ::String
       end
@@ -207,6 +208,20 @@ module Aws
                                    ?retrieve_and_generate_configuration: {
                                      external_sources_configuration: {
                                        generation_configuration: {
+                                         additional_model_request_fields: Hash[::String, {
+                                           }]?,
+                                         guardrail_configuration: {
+                                           guardrail_id: ::String,
+                                           guardrail_version: ::String
+                                         }?,
+                                         inference_config: {
+                                           text_inference_config: {
+                                             max_tokens: ::Integer?,
+                                             stop_sequences: Array[::String]?,
+                                             temperature: ::Float?,
+                                             top_p: ::Float?
+                                           }?
+                                         }?,
                                          prompt_template: {
                                            text_prompt_template: ::String?
                                          }?
@@ -228,6 +243,20 @@ module Aws
                                      }?,
                                      knowledge_base_configuration: {
                                        generation_configuration: {
+                                         additional_model_request_fields: Hash[::String, {
+                                           }]?,
+                                         guardrail_configuration: {
+                                           guardrail_id: ::String,
+                                           guardrail_version: ::String
+                                         }?,
+                                         inference_config: {
+                                           text_inference_config: {
+                                             max_tokens: ::Integer?,
+                                             stop_sequences: Array[::String]?,
+                                             temperature: ::Float?,
+                                             top_p: ::Float?
+                                           }?
+                                         }?,
                                          prompt_template: {
                                            text_prompt_template: ::String?
                                          }?

data/sig/types.rbs

@@ -111,6 +111,9 @@ module Aws::BedrockAgentRuntime
     end
 
     class ExternalSourcesGenerationConfiguration
+      attr_accessor additional_model_request_fields: ::Hash[::String, untyped]
+      attr_accessor guardrail_configuration: Types::GuardrailConfiguration
+      attr_accessor inference_config: Types::InferenceConfig
       attr_accessor prompt_template: Types::PromptTemplate
       SENSITIVE: []
     end
@@ -167,10 +170,24 @@ module Aws::BedrockAgentRuntime
     end
 
     class GenerationConfiguration
+      attr_accessor additional_model_request_fields: ::Hash[::String, untyped]
+      attr_accessor guardrail_configuration: Types::GuardrailConfiguration
+      attr_accessor inference_config: Types::InferenceConfig
       attr_accessor prompt_template: Types::PromptTemplate
       SENSITIVE: []
     end
 
+    class GuardrailConfiguration
+      attr_accessor guardrail_id: ::String
+      attr_accessor guardrail_version: ::String
+      SENSITIVE: []
+    end
+
+    class InferenceConfig
+      attr_accessor text_inference_config: Types::TextInferenceConfig
+      SENSITIVE: []
+    end
+
     class InferenceConfiguration
       attr_accessor maximum_length: ::Integer
       attr_accessor stop_sequences: ::Array[::String]
@@ -507,6 +524,7 @@ module Aws::BedrockAgentRuntime
 
     class RetrieveAndGenerateResponse
       attr_accessor citations: ::Array[Types::Citation]
+      attr_accessor guardrail_action: ("INTERVENED" | "NONE")
       attr_accessor output: Types::RetrieveAndGenerateOutput
       attr_accessor session_id: ::String
       SENSITIVE: [:output]
@@ -570,6 +588,14 @@ module Aws::BedrockAgentRuntime
       SENSITIVE: []
     end
 
+    class TextInferenceConfig
+      attr_accessor max_tokens: ::Integer
+      attr_accessor stop_sequences: ::Array[::String]
+      attr_accessor temperature: ::Float
+      attr_accessor top_p: ::Float
+      SENSITIVE: []
+    end
+
     class TextResponsePart
       attr_accessor span: Types::Span
       attr_accessor text: ::String

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-bedrockagentruntime
 version: !ruby/object:Gem::Version
-  version: 1.7.0
+  version: 1.9.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-04-25 00:00:00.000000000 Z
+date: 2024-05-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core