aws-sdk-bedrockagent 1.5.0 → 1.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 80b73b4178615eca1840c0377574039840c5368e66123261f10a064926a327bf
-  data.tar.gz: 73b552c48974c6e281c8b63bc21843b65a284d05203e3e2643599e82e6a634fd
+  metadata.gz: 4143ef9fc133a2aa9e2e22ab708a4cb72745bc76f1cd11abe0ac678d5650d6f0
+  data.tar.gz: cf65e144ace6118b6f5df92b22f0890281f2ae19dc8f1546a575d19e8b2b8ffa
 SHA512:
-  metadata.gz: 61e7c6fe4dafdffbbb21589841574d19daf132a5cb86996c0077e6149e644c60ab4654195a85e1006150fde581ef40cd1b5568c12b575b3b57d46fb7a5edc1cb
-  data.tar.gz: 33d9520e08888caab081d1055ff95448570813fa3688a949bbe88e2d3599f761d00aba1ed419fb9df7c6705e58278c9d82e18ca742bc72696cd054449b06d490
+  metadata.gz: 232bbcb77048085704f02652b46fcf8ff483c89c1274cfcdf5279d9911a50a077ff9427c847960517e46d5fbdd756849bd7a882898aef9df89052ef000648433
+  data.tar.gz: d26a5cb5f8b3a0601b0a5fd7f89c48c181a0dfdc4616cc496f56a3d75154de4fe175b1702e8981a57dfedea5e16950160acfd1ca59a9584049fa6433ed33e064

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 Unreleased Changes
 ------------------
 
+1.6.0 (2024-04-22)
+------------------
+
+* Feature - Releasing the support for simplified configuration and return of control
+
 1.5.0 (2024-04-16)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.5.0
+1.6.0

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

@@ -681,6 +681,10 @@ module Aws::BedrockAgent
     # @option params [String] :description
     #   A description of the action group.
     #
+    # @option params [Types::FunctionSchema] :function_schema
+    #   Contains details about the function schema for the action group or the
+    #   JSON or YAML-formatted payload defining the schema.
+    #
     # @option params [String] :parent_action_group_signature
     #   To allow your agent to request the user for additional information
     #   when trying to complete a task, set this field to `AMAZON.UserInput`.
@@ -704,6 +708,7 @@ module Aws::BedrockAgent
     #
     #   resp = client.create_agent_action_group({
     #     action_group_executor: {
+    #       custom_control: "RETURN_CONTROL", # accepts RETURN_CONTROL
     #       lambda: "LambdaArn",
     #     },
     #     action_group_name: "Name", # required
@@ -719,11 +724,27 @@ module Aws::BedrockAgent
     #     },
     #     client_token: "ClientToken",
     #     description: "Description",
+    #     function_schema: {
+    #       functions: [
+    #         {
+    #           description: "FunctionDescription",
+    #           name: "Name", # required
+    #           parameters: {
+    #             "Name" => {
+    #               description: "ParameterDescription",
+    #               required: false,
+    #               type: "string", # required, accepts string, number, integer, boolean, array
+    #             },
+    #           },
+    #         },
+    #       ],
+    #     },
     #     parent_action_group_signature: "AMAZON.UserInput", # accepts AMAZON.UserInput
     #   })
     #
     # @example Response structure
     #
+    #   resp.agent_action_group.action_group_executor.custom_control #=> String, one of "RETURN_CONTROL"
     #   resp.agent_action_group.action_group_executor.lambda #=> String
     #   resp.agent_action_group.action_group_id #=> String
     #   resp.agent_action_group.action_group_name #=> String
@@ -736,6 +757,13 @@ module Aws::BedrockAgent
     #   resp.agent_action_group.client_token #=> String
     #   resp.agent_action_group.created_at #=> Time
     #   resp.agent_action_group.description #=> String
+    #   resp.agent_action_group.function_schema.functions #=> Array
+    #   resp.agent_action_group.function_schema.functions[0].description #=> String
+    #   resp.agent_action_group.function_schema.functions[0].name #=> String
+    #   resp.agent_action_group.function_schema.functions[0].parameters #=> Hash
+    #   resp.agent_action_group.function_schema.functions[0].parameters["Name"].description #=> String
+    #   resp.agent_action_group.function_schema.functions[0].parameters["Name"].required #=> Boolean
+    #   resp.agent_action_group.function_schema.functions[0].parameters["Name"].type #=> String, one of "string", "number", "integer", "boolean", "array"
     #   resp.agent_action_group.parent_action_signature #=> String, one of "AMAZON.UserInput"
     #   resp.agent_action_group.updated_at #=> Time
     #
@@ -1459,6 +1487,7 @@ module Aws::BedrockAgent
     #
     # @example Response structure
     #
+    #   resp.agent_action_group.action_group_executor.custom_control #=> String, one of "RETURN_CONTROL"
     #   resp.agent_action_group.action_group_executor.lambda #=> String
     #   resp.agent_action_group.action_group_id #=> String
     #   resp.agent_action_group.action_group_name #=> String
@@ -1471,6 +1500,13 @@ module Aws::BedrockAgent
     #   resp.agent_action_group.client_token #=> String
     #   resp.agent_action_group.created_at #=> Time
     #   resp.agent_action_group.description #=> String
+    #   resp.agent_action_group.function_schema.functions #=> Array
+    #   resp.agent_action_group.function_schema.functions[0].description #=> String
+    #   resp.agent_action_group.function_schema.functions[0].name #=> String
+    #   resp.agent_action_group.function_schema.functions[0].parameters #=> Hash
+    #   resp.agent_action_group.function_schema.functions[0].parameters["Name"].description #=> String
+    #   resp.agent_action_group.function_schema.functions[0].parameters["Name"].required #=> Boolean
+    #   resp.agent_action_group.function_schema.functions[0].parameters["Name"].type #=> String, one of "string", "number", "integer", "boolean", "array"
     #   resp.agent_action_group.parent_action_signature #=> String, one of "AMAZON.UserInput"
     #   resp.agent_action_group.updated_at #=> Time
     #
@@ -2610,6 +2646,10 @@ module Aws::BedrockAgent
     # @option params [String] :description
     #   Specifies a new name for the action group.
     #
+    # @option params [Types::FunctionSchema] :function_schema
+    #   Contains details about the function schema for the action group or the
+    #   JSON or YAML-formatted payload defining the schema.
+    #
     # @option params [String] :parent_action_group_signature
     #   To allow your agent to request the user for additional information
     #   when trying to complete a task, set this field to `AMAZON.UserInput`.
@@ -2633,6 +2673,7 @@ module Aws::BedrockAgent
     #
     #   resp = client.update_agent_action_group({
     #     action_group_executor: {
+    #       custom_control: "RETURN_CONTROL", # accepts RETURN_CONTROL
     #       lambda: "LambdaArn",
     #     },
     #     action_group_id: "Id", # required
@@ -2648,11 +2689,27 @@ module Aws::BedrockAgent
     #       },
     #     },
     #     description: "Description",
+    #     function_schema: {
+    #       functions: [
+    #         {
+    #           description: "FunctionDescription",
+    #           name: "Name", # required
+    #           parameters: {
+    #             "Name" => {
+    #               description: "ParameterDescription",
+    #               required: false,
+    #               type: "string", # required, accepts string, number, integer, boolean, array
+    #             },
+    #           },
+    #         },
+    #       ],
+    #     },
     #     parent_action_group_signature: "AMAZON.UserInput", # accepts AMAZON.UserInput
     #   })
     #
     # @example Response structure
     #
+    #   resp.agent_action_group.action_group_executor.custom_control #=> String, one of "RETURN_CONTROL"
     #   resp.agent_action_group.action_group_executor.lambda #=> String
     #   resp.agent_action_group.action_group_id #=> String
     #   resp.agent_action_group.action_group_name #=> String
@@ -2665,6 +2722,13 @@ module Aws::BedrockAgent
     #   resp.agent_action_group.client_token #=> String
     #   resp.agent_action_group.created_at #=> Time
     #   resp.agent_action_group.description #=> String
+    #   resp.agent_action_group.function_schema.functions #=> Array
+    #   resp.agent_action_group.function_schema.functions[0].description #=> String
+    #   resp.agent_action_group.function_schema.functions[0].name #=> String
+    #   resp.agent_action_group.function_schema.functions[0].parameters #=> Hash
+    #   resp.agent_action_group.function_schema.functions[0].parameters["Name"].description #=> String
+    #   resp.agent_action_group.function_schema.functions[0].parameters["Name"].required #=> Boolean
+    #   resp.agent_action_group.function_schema.functions[0].parameters["Name"].type #=> String, one of "string", "number", "integer", "boolean", "array"
     #   resp.agent_action_group.parent_action_signature #=> String, one of "AMAZON.UserInput"
     #   resp.agent_action_group.updated_at #=> Time
     #
@@ -3054,7 +3118,7 @@ module Aws::BedrockAgent
         params: params,
         config: config)
       context[:gem_name] = 'aws-sdk-bedrockagent'
-      context[:gem_version] = '1.5.0'
+      context[:gem_version] = '1.6.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -64,6 +64,7 @@ module Aws::BedrockAgent
     CreateKnowledgeBaseRequest = Shapes::StructureShape.new(name: 'CreateKnowledgeBaseRequest')
     CreateKnowledgeBaseResponse = Shapes::StructureShape.new(name: 'CreateKnowledgeBaseResponse')
     CreationMode = Shapes::StringShape.new(name: 'CreationMode')
+    CustomControlMethod = Shapes::StringShape.new(name: 'CustomControlMethod')
     DataSource = Shapes::StructureShape.new(name: 'DataSource')
     DataSourceConfiguration = Shapes::StructureShape.new(name: 'DataSourceConfiguration')
     DataSourceStatus = Shapes::StringShape.new(name: 'DataSourceStatus')
@@ -93,6 +94,10 @@ module Aws::BedrockAgent
     FixedSizeChunkingConfiguration = Shapes::StructureShape.new(name: 'FixedSizeChunkingConfiguration')
     FixedSizeChunkingConfigurationMaxTokensInteger = Shapes::IntegerShape.new(name: 'FixedSizeChunkingConfigurationMaxTokensInteger')
     FixedSizeChunkingConfigurationOverlapPercentageInteger = Shapes::IntegerShape.new(name: 'FixedSizeChunkingConfigurationOverlapPercentageInteger')
+    Function = Shapes::StructureShape.new(name: 'Function')
+    FunctionDescription = Shapes::StringShape.new(name: 'FunctionDescription')
+    FunctionSchema = Shapes::UnionShape.new(name: 'FunctionSchema')
+    Functions = Shapes::ListShape.new(name: 'Functions')
     GetAgentActionGroupRequest = Shapes::StructureShape.new(name: 'GetAgentActionGroupRequest')
     GetAgentActionGroupResponse = Shapes::StructureShape.new(name: 'GetAgentActionGroupResponse')
     GetAgentAliasRequest = Shapes::StructureShape.new(name: 'GetAgentAliasRequest')
@@ -167,6 +172,9 @@ module Aws::BedrockAgent
     OpenSearchServerlessConfiguration = Shapes::StructureShape.new(name: 'OpenSearchServerlessConfiguration')
     OpenSearchServerlessFieldMapping = Shapes::StructureShape.new(name: 'OpenSearchServerlessFieldMapping')
     OpenSearchServerlessIndexName = Shapes::StringShape.new(name: 'OpenSearchServerlessIndexName')
+    ParameterDescription = Shapes::StringShape.new(name: 'ParameterDescription')
+    ParameterDetail = Shapes::StructureShape.new(name: 'ParameterDetail')
+    ParameterMap = Shapes::MapShape.new(name: 'ParameterMap')
     Payload = Shapes::StringShape.new(name: 'Payload')
     PineconeConfiguration = Shapes::StructureShape.new(name: 'PineconeConfiguration')
     PineconeConnectionString = Shapes::StringShape.new(name: 'PineconeConnectionString')
@@ -220,6 +228,7 @@ module Aws::BedrockAgent
     ThrottlingException = Shapes::StructureShape.new(name: 'ThrottlingException')
     TopK = Shapes::IntegerShape.new(name: 'TopK')
     TopP = Shapes::FloatShape.new(name: 'TopP')
+    Type = Shapes::StringShape.new(name: 'Type')
     UntagResourceRequest = Shapes::StructureShape.new(name: 'UntagResourceRequest')
     UntagResourceResponse = Shapes::StructureShape.new(name: 'UntagResourceResponse')
     UpdateAgentActionGroupRequest = Shapes::StructureShape.new(name: 'UpdateAgentActionGroupRequest')
@@ -252,8 +261,10 @@ module Aws::BedrockAgent
     AccessDeniedException.add_member(:message, Shapes::ShapeRef.new(shape: NonBlankString, location_name: "message"))
     AccessDeniedException.struct_class = Types::AccessDeniedException
 
+    ActionGroupExecutor.add_member(:custom_control, Shapes::ShapeRef.new(shape: CustomControlMethod, location_name: "customControl"))
     ActionGroupExecutor.add_member(:lambda, Shapes::ShapeRef.new(shape: LambdaArn, location_name: "lambda"))
     ActionGroupExecutor.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    ActionGroupExecutor.add_member_subclass(:custom_control, Types::ActionGroupExecutor::CustomControl)
     ActionGroupExecutor.add_member_subclass(:lambda, Types::ActionGroupExecutor::Lambda)
     ActionGroupExecutor.add_member_subclass(:unknown, Types::ActionGroupExecutor::Unknown)
     ActionGroupExecutor.struct_class = Types::ActionGroupExecutor
@@ -297,6 +308,7 @@ module Aws::BedrockAgent
     AgentActionGroup.add_member(:client_token, Shapes::ShapeRef.new(shape: ClientToken, location_name: "clientToken"))
     AgentActionGroup.add_member(:created_at, Shapes::ShapeRef.new(shape: DateTimestamp, required: true, location_name: "createdAt"))
     AgentActionGroup.add_member(:description, Shapes::ShapeRef.new(shape: Description, location_name: "description"))
+    AgentActionGroup.add_member(:function_schema, Shapes::ShapeRef.new(shape: FunctionSchema, location_name: "functionSchema"))
     AgentActionGroup.add_member(:parent_action_signature, Shapes::ShapeRef.new(shape: ActionGroupSignature, location_name: "parentActionSignature"))
     AgentActionGroup.add_member(:updated_at, Shapes::ShapeRef.new(shape: DateTimestamp, required: true, location_name: "updatedAt"))
     AgentActionGroup.struct_class = Types::AgentActionGroup
@@ -417,6 +429,7 @@ module Aws::BedrockAgent
     CreateAgentActionGroupRequest.add_member(:api_schema, Shapes::ShapeRef.new(shape: APISchema, location_name: "apiSchema"))
     CreateAgentActionGroupRequest.add_member(:client_token, Shapes::ShapeRef.new(shape: ClientToken, location_name: "clientToken", metadata: {"idempotencyToken"=>true}))
     CreateAgentActionGroupRequest.add_member(:description, Shapes::ShapeRef.new(shape: Description, location_name: "description"))
+    CreateAgentActionGroupRequest.add_member(:function_schema, Shapes::ShapeRef.new(shape: FunctionSchema, location_name: "functionSchema"))
     CreateAgentActionGroupRequest.add_member(:parent_action_group_signature, Shapes::ShapeRef.new(shape: ActionGroupSignature, location_name: "parentActionGroupSignature"))
     CreateAgentActionGroupRequest.struct_class = Types::CreateAgentActionGroupRequest
 
@@ -563,6 +576,19 @@ module Aws::BedrockAgent
     FixedSizeChunkingConfiguration.add_member(:overlap_percentage, Shapes::ShapeRef.new(shape: FixedSizeChunkingConfigurationOverlapPercentageInteger, required: true, location_name: "overlapPercentage"))
     FixedSizeChunkingConfiguration.struct_class = Types::FixedSizeChunkingConfiguration
 
+    Function.add_member(:description, Shapes::ShapeRef.new(shape: FunctionDescription, location_name: "description"))
+    Function.add_member(:name, Shapes::ShapeRef.new(shape: Name, required: true, location_name: "name"))
+    Function.add_member(:parameters, Shapes::ShapeRef.new(shape: ParameterMap, location_name: "parameters"))
+    Function.struct_class = Types::Function
+
+    FunctionSchema.add_member(:functions, Shapes::ShapeRef.new(shape: Functions, location_name: "functions"))
+    FunctionSchema.add_member(:unknown, Shapes::ShapeRef.new(shape: nil, location_name: 'unknown'))
+    FunctionSchema.add_member_subclass(:functions, Types::FunctionSchema::Functions)
+    FunctionSchema.add_member_subclass(:unknown, Types::FunctionSchema::Unknown)
+    FunctionSchema.struct_class = Types::FunctionSchema
+
+    Functions.member = Shapes::ShapeRef.new(shape: Function)
+
     GetAgentActionGroupRequest.add_member(:action_group_id, Shapes::ShapeRef.new(shape: Id, required: true, location: "uri", location_name: "actionGroupId"))
     GetAgentActionGroupRequest.add_member(:agent_id, Shapes::ShapeRef.new(shape: Id, required: true, location: "uri", location_name: "agentId"))
     GetAgentActionGroupRequest.add_member(:agent_version, Shapes::ShapeRef.new(shape: Version, required: true, location: "uri", location_name: "agentVersion"))
@@ -792,6 +818,14 @@ module Aws::BedrockAgent
     OpenSearchServerlessFieldMapping.add_member(:vector_field, Shapes::ShapeRef.new(shape: FieldName, required: true, location_name: "vectorField"))
     OpenSearchServerlessFieldMapping.struct_class = Types::OpenSearchServerlessFieldMapping
 
+    ParameterDetail.add_member(:description, Shapes::ShapeRef.new(shape: ParameterDescription, location_name: "description"))
+    ParameterDetail.add_member(:required, Shapes::ShapeRef.new(shape: Boolean, location_name: "required"))
+    ParameterDetail.add_member(:type, Shapes::ShapeRef.new(shape: Type, required: true, location_name: "type"))
+    ParameterDetail.struct_class = Types::ParameterDetail
+
+    ParameterMap.key = Shapes::ShapeRef.new(shape: Name)
+    ParameterMap.value = Shapes::ShapeRef.new(shape: ParameterDetail)
+
     PineconeConfiguration.add_member(:connection_string, Shapes::ShapeRef.new(shape: PineconeConnectionString, required: true, location_name: "connectionString"))
     PineconeConfiguration.add_member(:credentials_secret_arn, Shapes::ShapeRef.new(shape: SecretArn, required: true, location_name: "credentialsSecretArn"))
     PineconeConfiguration.add_member(:field_mapping, Shapes::ShapeRef.new(shape: PineconeFieldMapping, required: true, location_name: "fieldMapping"))
@@ -916,6 +950,7 @@ module Aws::BedrockAgent
     UpdateAgentActionGroupRequest.add_member(:agent_version, Shapes::ShapeRef.new(shape: DraftVersion, required: true, location: "uri", location_name: "agentVersion"))
     UpdateAgentActionGroupRequest.add_member(:api_schema, Shapes::ShapeRef.new(shape: APISchema, location_name: "apiSchema"))
     UpdateAgentActionGroupRequest.add_member(:description, Shapes::ShapeRef.new(shape: Description, location_name: "description"))
+    UpdateAgentActionGroupRequest.add_member(:function_schema, Shapes::ShapeRef.new(shape: FunctionSchema, location_name: "functionSchema"))
     UpdateAgentActionGroupRequest.add_member(:parent_action_group_signature, Shapes::ShapeRef.new(shape: ActionGroupSignature, location_name: "parentActionGroupSignature"))
     UpdateAgentActionGroupRequest.struct_class = Types::UpdateAgentActionGroupRequest
 

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

@@ -79,6 +79,11 @@ module Aws::BedrockAgent
     #
     # @note ActionGroupExecutor is a union - when returned from an API call exactly one value will be set and the returned type will be a subclass of ActionGroupExecutor corresponding to the set member.
     #
+    # @!attribute [rw] custom_control
+    #   To return the action group invocation results directly in the
+    #   `InvokeAgent` response, specify `RETURN_CONTROL`.
+    #   @return [String]
+    #
     # @!attribute [rw] lambda
     #   The Amazon Resource Name (ARN) of the Lambda function containing the
     #   business logic that is carried out upon invoking the action.
@@ -87,12 +92,14 @@ module Aws::BedrockAgent
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/ActionGroupExecutor AWS API Documentation
     #
     class ActionGroupExecutor < Struct.new(
+      :custom_control,
       :lambda,
       :unknown)
       SENSITIVE = []
       include Aws::Structure
       include Aws::Structure::Union
 
+      class CustomControl < ActionGroupExecutor; end
       class Lambda < ActionGroupExecutor; end
       class Unknown < ActionGroupExecutor; end
     end
@@ -335,6 +342,12 @@ module Aws::BedrockAgent
     #   The description of the action group.
     #   @return [String]
     #
+    # @!attribute [rw] function_schema
+    #   Defines functions that each define parameters that the agent needs
+    #   to invoke from the user. Each function represents an action in an
+    #   action group.
+    #   @return [Types::FunctionSchema]
+    #
     # @!attribute [rw] parent_action_signature
     #   If this field is set as `AMAZON.UserInput`, the agent can request
     #   the user for additional information when trying to complete a task.
@@ -369,6 +382,7 @@ module Aws::BedrockAgent
       :client_token,
       :created_at,
       :description,
+      :function_schema,
       :parent_action_signature,
       :updated_at)
       SENSITIVE = []
@@ -975,6 +989,11 @@ module Aws::BedrockAgent
     #   A description of the action group.
     #   @return [String]
     #
+    # @!attribute [rw] function_schema
+    #   Contains details about the function schema for the action group or
+    #   the JSON or YAML-formatted payload defining the schema.
+    #   @return [Types::FunctionSchema]
+    #
     # @!attribute [rw] parent_action_group_signature
     #   To allow your agent to request the user for additional information
     #   when trying to complete a task, set this field to
@@ -1003,6 +1022,7 @@ module Aws::BedrockAgent
       :api_schema,
       :client_token,
       :description,
+      :function_schema,
       :parent_action_group_signature)
       SENSITIVE = []
       include Aws::Structure
@@ -1707,6 +1727,97 @@ module Aws::BedrockAgent
       include Aws::Structure
     end
 
+    # Defines parameters that the agent needs to invoke from the user to
+    # complete the function. Corresponds to an action in an action group.
+    #
+    # This data type is used in the following API operations:
+    #
+    # * [CreateAgentActionGroup request][1]
+    #
+    # * [CreateAgentActionGroup response][2]
+    #
+    # * [UpdateAgentActionGroup request][3]
+    #
+    # * [UpdateAgentActionGroup response][4]
+    #
+    # * [GetAgentActionGroup response][5]
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_CreateAgentActionGroup.html#API_agent_CreateAgentActionGroup_RequestSyntax
+    # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_CreateAgentActionGroup.html#API_agent_CreateAgentActionGroup_ResponseSyntax
+    # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_UpdateAgentActionGroup.html#API_agent_UpdateAgentActionGroup_RequestSyntax
+    # [4]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_UpdateAgentActionGroup.html#API_agent_UpdateAgentActionGroup_ResponseSyntax
+    # [5]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_GetAgentActionGroup.html#API_agent_GetAgentActionGroup_ResponseSyntax
+    #
+    # @!attribute [rw] description
+    #   A description of the function and its purpose.
+    #   @return [String]
+    #
+    # @!attribute [rw] name
+    #   A name for the function.
+    #   @return [String]
+    #
+    # @!attribute [rw] parameters
+    #   The parameters that the agent elicits from the user to fulfill the
+    #   function.
+    #   @return [Hash<String,Types::ParameterDetail>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/Function AWS API Documentation
+    #
+    class Function < Struct.new(
+      :description,
+      :name,
+      :parameters)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # Defines functions that each define parameters that the agent needs to
+    # invoke from the user. Each function represents an action in an action
+    # group.
+    #
+    # This data type is used in the following API operations:
+    #
+    # * [CreateAgentActionGroup request][1]
+    #
+    # * [CreateAgentActionGroup response][2]
+    #
+    # * [UpdateAgentActionGroup request][3]
+    #
+    # * [UpdateAgentActionGroup response][4]
+    #
+    # * [GetAgentActionGroup response][5]
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_CreateAgentActionGroup.html#API_agent_CreateAgentActionGroup_RequestSyntax
+    # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_CreateAgentActionGroup.html#API_agent_CreateAgentActionGroup_ResponseSyntax
+    # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_UpdateAgentActionGroup.html#API_agent_UpdateAgentActionGroup_RequestSyntax
+    # [4]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_UpdateAgentActionGroup.html#API_agent_UpdateAgentActionGroup_ResponseSyntax
+    # [5]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_GetAgentActionGroup.html#API_agent_GetAgentActionGroup_ResponseSyntax
+    #
+    # @note FunctionSchema is a union - when making an API calls you must set exactly one of the members.
+    #
+    # @note FunctionSchema is a union - when returned from an API call exactly one value will be set and the returned type will be a subclass of FunctionSchema corresponding to the set member.
+    #
+    # @!attribute [rw] functions
+    #   A list of functions that each define an action in the action group.
+    #   @return [Array<Types::Function>]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/FunctionSchema AWS API Documentation
+    #
+    class FunctionSchema < Struct.new(
+      :functions,
+      :unknown)
+      SENSITIVE = []
+      include Aws::Structure
+      include Aws::Structure::Union
+
+      class Functions < FunctionSchema; end
+      class Unknown < FunctionSchema; end
+    end
+
     # @!attribute [rw] action_group_id
     #   The unique identifier of the action group for which to get
     #   information.
@@ -2022,7 +2133,7 @@ module Aws::BedrockAgent
     #
     # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_StartIngestionJob.html#API_agent_StartIngestionJob_ResponseSyntax
     # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_GetIngestionJob.html#API_agent_GetIngestionJob_ResponseSyntax
-    # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_ListIngestionJob.html#API_agent_ListIngestionJob_ResponseSyntax
+    # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_ListIngestionJobs.html#API_agent_ListIngestionJobs_ResponseSyntax
     #
     # @!attribute [rw] data_source_id
     #   The unique identifier of the ingested data source.
@@ -2864,6 +2975,52 @@ module Aws::BedrockAgent
       include Aws::Structure
     end
 
+    # Contains details about a parameter in a function for an action group.
+    #
+    # This data type is used in the following API operations:
+    #
+    # * [CreateAgentActionGroup request][1]
+    #
+    # * [CreateAgentActionGroup response][2]
+    #
+    # * [UpdateAgentActionGroup request][3]
+    #
+    # * [UpdateAgentActionGroup response][4]
+    #
+    # * [GetAgentActionGroup response][5]
+    #
+    #
+    #
+    # [1]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_CreateAgentActionGroup.html#API_agent_CreateAgentActionGroup_RequestSyntax
+    # [2]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_CreateAgentActionGroup.html#API_agent_CreateAgentActionGroup_ResponseSyntax
+    # [3]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_UpdateAgentActionGroup.html#API_agent_UpdateAgentActionGroup_RequestSyntax
+    # [4]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_UpdateAgentActionGroup.html#API_agent_UpdateAgentActionGroup_ResponseSyntax
+    # [5]: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_GetAgentActionGroup.html#API_agent_GetAgentActionGroup_ResponseSyntax
+    #
+    # @!attribute [rw] description
+    #   A description of the parameter. Helps the foundation model determine
+    #   how to elicit the parameters from the user.
+    #   @return [String]
+    #
+    # @!attribute [rw] required
+    #   Whether the parameter is required for the agent to complete the
+    #   function for action group invocation.
+    #   @return [Boolean]
+    #
+    # @!attribute [rw] type
+    #   The data type of the parameter.
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/ParameterDetail AWS API Documentation
+    #
+    class ParameterDetail < Struct.new(
+      :description,
+      :required,
+      :type)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains details about the storage configuration of the knowledge base
     # in Pinecone. For more information, see [Create a vector index in
     # Pinecone][1].
@@ -3509,6 +3666,11 @@ module Aws::BedrockAgent
     #   Specifies a new name for the action group.
     #   @return [String]
     #
+    # @!attribute [rw] function_schema
+    #   Contains details about the function schema for the action group or
+    #   the JSON or YAML-formatted payload defining the schema.
+    #   @return [Types::FunctionSchema]
+    #
     # @!attribute [rw] parent_action_group_signature
     #   To allow your agent to request the user for additional information
     #   when trying to complete a task, set this field to
@@ -3537,6 +3699,7 @@ module Aws::BedrockAgent
       :agent_version,
       :api_schema,
       :description,
+      :function_schema,
       :parent_action_group_signature)
       SENSITIVE = []
       include Aws::Structure

data/lib/aws-sdk-bedrockagent.rb

@@ -52,6 +52,6 @@ require_relative 'aws-sdk-bedrockagent/customizations'
 # @!group service
 module Aws::BedrockAgent
 
-  GEM_VERSION = '1.5.0'
+  GEM_VERSION = '1.6.0'
 
 end

data/sig/client.rbs

@@ -130,6 +130,7 @@ module Aws
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgent/Client.html#create_agent_action_group-instance_method
       def create_agent_action_group: (
                                        ?action_group_executor: {
+                                         custom_control: ("RETURN_CONTROL")?,
                                          lambda: ::String?
                                        },
                                        action_group_name: ::String,
@@ -145,6 +146,19 @@ module Aws
                                        },
                                        ?client_token: ::String,
                                        ?description: ::String,
+                                       ?function_schema: {
+                                         functions: Array[
+                                           {
+                                             description: ::String?,
+                                             name: ::String,
+                                             parameters: Hash[::String, {
+                                                 description: ::String?,
+                                                 required: bool?,
+                                                 type: ("string" | "number" | "integer" | "boolean" | "array")
+                                               }]?
+                                           },
+                                         ]?
+                                       },
                                        ?parent_action_group_signature: ("AMAZON.UserInput")
                                      ) -> _CreateAgentActionGroupResponseSuccess
                                    | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _CreateAgentActionGroupResponseSuccess
@@ -653,6 +667,7 @@ module Aws
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgent/Client.html#update_agent_action_group-instance_method
       def update_agent_action_group: (
                                        ?action_group_executor: {
+                                         custom_control: ("RETURN_CONTROL")?,
                                          lambda: ::String?
                                        },
                                        action_group_id: ::String,
@@ -668,6 +683,19 @@ module Aws
                                          }?
                                        },
                                        ?description: ::String,
+                                       ?function_schema: {
+                                         functions: Array[
+                                           {
+                                             description: ::String?,
+                                             name: ::String,
+                                             parameters: Hash[::String, {
+                                                 description: ::String?,
+                                                 required: bool?,
+                                                 type: ("string" | "number" | "integer" | "boolean" | "array")
+                                               }]?
+                                           },
+                                         ]?
+                                       },
                                        ?parent_action_group_signature: ("AMAZON.UserInput")
                                      ) -> _UpdateAgentActionGroupResponseSuccess
                                    | (Hash[Symbol, untyped] params, ?Hash[Symbol, untyped] options) -> _UpdateAgentActionGroupResponseSuccess

data/sig/types.rbs

@@ -28,10 +28,13 @@ module Aws::BedrockAgent
     end
 
     class ActionGroupExecutor
+      attr_accessor custom_control: ("RETURN_CONTROL")
       attr_accessor lambda: ::String
       attr_accessor unknown: untyped
       SENSITIVE: []
 
+      class CustomControl < ActionGroupExecutor
+      end
       class Lambda < ActionGroupExecutor
       end
       class Unknown < ActionGroupExecutor
@@ -80,6 +83,7 @@ module Aws::BedrockAgent
       attr_accessor client_token: ::String
       attr_accessor created_at: ::Time
       attr_accessor description: ::String
+      attr_accessor function_schema: Types::FunctionSchema
       attr_accessor parent_action_signature: ("AMAZON.UserInput")
       attr_accessor updated_at: ::Time
       SENSITIVE: []
@@ -216,6 +220,7 @@ module Aws::BedrockAgent
       attr_accessor api_schema: Types::APISchema
       attr_accessor client_token: ::String
       attr_accessor description: ::String
+      attr_accessor function_schema: Types::FunctionSchema
       attr_accessor parent_action_group_signature: ("AMAZON.UserInput")
       SENSITIVE: []
     end
@@ -411,6 +416,24 @@ module Aws::BedrockAgent
       SENSITIVE: []
     end
 
+    class Function
+      attr_accessor description: ::String
+      attr_accessor name: ::String
+      attr_accessor parameters: ::Hash[::String, Types::ParameterDetail]
+      SENSITIVE: []
+    end
+
+    class FunctionSchema
+      attr_accessor functions: ::Array[Types::Function]
+      attr_accessor unknown: untyped
+      SENSITIVE: []
+
+      class Functions < FunctionSchema
+      end
+      class Unknown < FunctionSchema
+      end
+    end
+
     class GetAgentActionGroupRequest
       attr_accessor action_group_id: ::String
       attr_accessor agent_id: ::String
@@ -724,6 +747,13 @@ module Aws::BedrockAgent
       SENSITIVE: []
     end
 
+    class ParameterDetail
+      attr_accessor description: ::String
+      attr_accessor required: bool
+      attr_accessor type: ("string" | "number" | "integer" | "boolean" | "array")
+      SENSITIVE: []
+    end
+
     class PineconeConfiguration
       attr_accessor connection_string: ::String
       attr_accessor credentials_secret_arn: ::String
@@ -880,6 +910,7 @@ module Aws::BedrockAgent
       attr_accessor agent_version: ::String
       attr_accessor api_schema: Types::APISchema
       attr_accessor description: ::String
+      attr_accessor function_schema: Types::FunctionSchema
       attr_accessor parent_action_group_signature: ("AMAZON.UserInput")
       SENSITIVE: []
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-bedrockagent
 version: !ruby/object:Gem::Version
-  version: 1.5.0
+  version: 1.6.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-04-16 00:00:00.000000000 Z
+date: 2024-04-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core