aws-sdk-bedrockagent 1.5.0 → 1.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 80b73b4178615eca1840c0377574039840c5368e66123261f10a064926a327bf
-  data.tar.gz: 73b552c48974c6e281c8b63bc21843b65a284d05203e3e2643599e82e6a634fd
+  metadata.gz: bcb2814f409a0dd580ca9fac184a721497956f9dc4f2dc3d42e2a527501f348f
+  data.tar.gz: 1c8674b5d0d383286f891e7499d3f8149c8281b90a4c853ae163351112e9c1e0
 SHA512:
-  metadata.gz: 61e7c6fe4dafdffbbb21589841574d19daf132a5cb86996c0077e6149e644c60ab4654195a85e1006150fde581ef40cd1b5568c12b575b3b57d46fb7a5edc1cb
-  data.tar.gz: 33d9520e08888caab081d1055ff95448570813fa3688a949bbe88e2d3599f761d00aba1ed419fb9df7c6705e58278c9d82e18ca742bc72696cd054449b06d490
+  metadata.gz: 8668ed7e1e8d33ed7be55f8df2090e4d0bc450216d5862cb46b4f1f3fbb901316c123e2eddccc9235e5175c3d4ccc0465c147bda1082cc3e49306533f0a15fdd
+  data.tar.gz: d280ca0c93898e4c5fa82482f65f62d4c3af1820aebc4fd9e3460da3a92dfb65d865fbbc05a3c8a7e129489aee30aeeb663531394b8ee37b7b7c2c835a694192

data/CHANGELOG.md

@@ -1,6 +1,16 @@
 Unreleased Changes
 ------------------
 
+1.7.0 (2024-04-23)
+------------------
+
+* Feature - Introducing the ability to create multiple data sources per knowledge base, specify S3 buckets as data sources from external accounts, and exposing levers to define the deletion behavior of the underlying vector store data.
+
+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.7.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
     #
@@ -845,6 +873,9 @@ module Aws::BedrockAgent
     #
     #   [1]: https://docs.aws.amazon.com/AWSEC2/latest/APIReference/Run_Instance_Idempotency.html
     #
+    # @option params [String] :data_deletion_policy
+    #   The deletion policy for the requested data source
+    #
     # @option params [required, Types::DataSourceConfiguration] :data_source_configuration
     #   Contains metadata about where the data source is stored.
     #
@@ -872,9 +903,11 @@ module Aws::BedrockAgent
     #
     #   resp = client.create_data_source({
     #     client_token: "ClientToken",
+    #     data_deletion_policy: "RETAIN", # accepts RETAIN, DELETE
     #     data_source_configuration: { # required
     #       s3_configuration: {
     #         bucket_arn: "S3BucketArn", # required
+    #         bucket_owner_account_id: "BucketOwnerAccountId",
     #         inclusion_prefixes: ["S3Prefix"],
     #       },
     #       type: "S3", # required, accepts S3
@@ -899,16 +932,20 @@ module Aws::BedrockAgent
     # @example Response structure
     #
     #   resp.data_source.created_at #=> Time
+    #   resp.data_source.data_deletion_policy #=> String, one of "RETAIN", "DELETE"
     #   resp.data_source.data_source_configuration.s3_configuration.bucket_arn #=> String
+    #   resp.data_source.data_source_configuration.s3_configuration.bucket_owner_account_id #=> String
     #   resp.data_source.data_source_configuration.s3_configuration.inclusion_prefixes #=> Array
     #   resp.data_source.data_source_configuration.s3_configuration.inclusion_prefixes[0] #=> String
     #   resp.data_source.data_source_configuration.type #=> String, one of "S3"
     #   resp.data_source.data_source_id #=> String
     #   resp.data_source.description #=> String
+    #   resp.data_source.failure_reasons #=> Array
+    #   resp.data_source.failure_reasons[0] #=> String
     #   resp.data_source.knowledge_base_id #=> String
     #   resp.data_source.name #=> String
     #   resp.data_source.server_side_encryption_configuration.kms_key_arn #=> String
-    #   resp.data_source.status #=> String, one of "AVAILABLE", "DELETING"
+    #   resp.data_source.status #=> String, one of "AVAILABLE", "DELETING", "DELETE_UNSUCCESSFUL"
     #   resp.data_source.updated_at #=> Time
     #   resp.data_source.vector_ingestion_configuration.chunking_configuration.chunking_strategy #=> String, one of "FIXED_SIZE", "NONE"
     #   resp.data_source.vector_ingestion_configuration.chunking_configuration.fixed_size_chunking_configuration.max_tokens #=> Integer
@@ -1082,7 +1119,7 @@ module Aws::BedrockAgent
     #   resp.knowledge_base.knowledge_base_id #=> String
     #   resp.knowledge_base.name #=> String
     #   resp.knowledge_base.role_arn #=> String
-    #   resp.knowledge_base.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED"
+    #   resp.knowledge_base.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED", "DELETE_UNSUCCESSFUL"
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.collection_arn #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.field_mapping.metadata_field #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.field_mapping.text_field #=> String
@@ -1295,7 +1332,7 @@ module Aws::BedrockAgent
     #
     #   resp.data_source_id #=> String
     #   resp.knowledge_base_id #=> String
-    #   resp.status #=> String, one of "AVAILABLE", "DELETING"
+    #   resp.status #=> String, one of "AVAILABLE", "DELETING", "DELETE_UNSUCCESSFUL"
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/DeleteDataSource AWS API Documentation
     #
@@ -1331,7 +1368,7 @@ module Aws::BedrockAgent
     # @example Response structure
     #
     #   resp.knowledge_base_id #=> String
-    #   resp.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED"
+    #   resp.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED", "DELETE_UNSUCCESSFUL"
     #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/DeleteKnowledgeBase AWS API Documentation
     #
@@ -1459,6 +1496,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 +1509,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
     #
@@ -1658,16 +1703,20 @@ module Aws::BedrockAgent
     # @example Response structure
     #
     #   resp.data_source.created_at #=> Time
+    #   resp.data_source.data_deletion_policy #=> String, one of "RETAIN", "DELETE"
     #   resp.data_source.data_source_configuration.s3_configuration.bucket_arn #=> String
+    #   resp.data_source.data_source_configuration.s3_configuration.bucket_owner_account_id #=> String
     #   resp.data_source.data_source_configuration.s3_configuration.inclusion_prefixes #=> Array
     #   resp.data_source.data_source_configuration.s3_configuration.inclusion_prefixes[0] #=> String
     #   resp.data_source.data_source_configuration.type #=> String, one of "S3"
     #   resp.data_source.data_source_id #=> String
     #   resp.data_source.description #=> String
+    #   resp.data_source.failure_reasons #=> Array
+    #   resp.data_source.failure_reasons[0] #=> String
     #   resp.data_source.knowledge_base_id #=> String
     #   resp.data_source.name #=> String
     #   resp.data_source.server_side_encryption_configuration.kms_key_arn #=> String
-    #   resp.data_source.status #=> String, one of "AVAILABLE", "DELETING"
+    #   resp.data_source.status #=> String, one of "AVAILABLE", "DELETING", "DELETE_UNSUCCESSFUL"
     #   resp.data_source.updated_at #=> Time
     #   resp.data_source.vector_ingestion_configuration.chunking_configuration.chunking_strategy #=> String, one of "FIXED_SIZE", "NONE"
     #   resp.data_source.vector_ingestion_configuration.chunking_configuration.fixed_size_chunking_configuration.max_tokens #=> Integer
@@ -1763,7 +1812,7 @@ module Aws::BedrockAgent
     #   resp.knowledge_base.knowledge_base_id #=> String
     #   resp.knowledge_base.name #=> String
     #   resp.knowledge_base.role_arn #=> String
-    #   resp.knowledge_base.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED"
+    #   resp.knowledge_base.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED", "DELETE_UNSUCCESSFUL"
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.collection_arn #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.field_mapping.metadata_field #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.field_mapping.text_field #=> String
@@ -2108,7 +2157,7 @@ module Aws::BedrockAgent
     #   resp.data_source_summaries[0].description #=> String
     #   resp.data_source_summaries[0].knowledge_base_id #=> String
     #   resp.data_source_summaries[0].name #=> String
-    #   resp.data_source_summaries[0].status #=> String, one of "AVAILABLE", "DELETING"
+    #   resp.data_source_summaries[0].status #=> String, one of "AVAILABLE", "DELETING", "DELETE_UNSUCCESSFUL"
     #   resp.data_source_summaries[0].updated_at #=> Time
     #   resp.next_token #=> String
     #
@@ -2240,7 +2289,7 @@ module Aws::BedrockAgent
     #   resp.knowledge_base_summaries[0].description #=> String
     #   resp.knowledge_base_summaries[0].knowledge_base_id #=> String
     #   resp.knowledge_base_summaries[0].name #=> String
-    #   resp.knowledge_base_summaries[0].status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED"
+    #   resp.knowledge_base_summaries[0].status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED", "DELETE_UNSUCCESSFUL"
     #   resp.knowledge_base_summaries[0].updated_at #=> Time
     #   resp.next_token #=> String
     #
@@ -2610,6 +2659,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 +2686,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 +2702,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 +2735,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
     #
@@ -2805,6 +2882,9 @@ module Aws::BedrockAgent
     # You can't change the `chunkingConfiguration` after you create the
     # data source. Specify the existing `chunkingConfiguration`.
     #
+    # @option params [String] :data_deletion_policy
+    #   The data deletion policy of the updated data source.
+    #
     # @option params [required, Types::DataSourceConfiguration] :data_source_configuration
     #   Contains details about the storage configuration of the data source.
     #
@@ -2834,9 +2914,11 @@ module Aws::BedrockAgent
     # @example Request syntax with placeholder values
     #
     #   resp = client.update_data_source({
+    #     data_deletion_policy: "RETAIN", # accepts RETAIN, DELETE
     #     data_source_configuration: { # required
     #       s3_configuration: {
     #         bucket_arn: "S3BucketArn", # required
+    #         bucket_owner_account_id: "BucketOwnerAccountId",
     #         inclusion_prefixes: ["S3Prefix"],
     #       },
     #       type: "S3", # required, accepts S3
@@ -2862,16 +2944,20 @@ module Aws::BedrockAgent
     # @example Response structure
     #
     #   resp.data_source.created_at #=> Time
+    #   resp.data_source.data_deletion_policy #=> String, one of "RETAIN", "DELETE"
     #   resp.data_source.data_source_configuration.s3_configuration.bucket_arn #=> String
+    #   resp.data_source.data_source_configuration.s3_configuration.bucket_owner_account_id #=> String
     #   resp.data_source.data_source_configuration.s3_configuration.inclusion_prefixes #=> Array
     #   resp.data_source.data_source_configuration.s3_configuration.inclusion_prefixes[0] #=> String
     #   resp.data_source.data_source_configuration.type #=> String, one of "S3"
     #   resp.data_source.data_source_id #=> String
     #   resp.data_source.description #=> String
+    #   resp.data_source.failure_reasons #=> Array
+    #   resp.data_source.failure_reasons[0] #=> String
     #   resp.data_source.knowledge_base_id #=> String
     #   resp.data_source.name #=> String
     #   resp.data_source.server_side_encryption_configuration.kms_key_arn #=> String
-    #   resp.data_source.status #=> String, one of "AVAILABLE", "DELETING"
+    #   resp.data_source.status #=> String, one of "AVAILABLE", "DELETING", "DELETE_UNSUCCESSFUL"
     #   resp.data_source.updated_at #=> Time
     #   resp.data_source.vector_ingestion_configuration.chunking_configuration.chunking_strategy #=> String, one of "FIXED_SIZE", "NONE"
     #   resp.data_source.vector_ingestion_configuration.chunking_configuration.fixed_size_chunking_configuration.max_tokens #=> Integer
@@ -3004,7 +3090,7 @@ module Aws::BedrockAgent
     #   resp.knowledge_base.knowledge_base_id #=> String
     #   resp.knowledge_base.name #=> String
     #   resp.knowledge_base.role_arn #=> String
-    #   resp.knowledge_base.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED"
+    #   resp.knowledge_base.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED", "DELETE_UNSUCCESSFUL"
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.collection_arn #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.field_mapping.metadata_field #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.field_mapping.text_field #=> String
@@ -3054,7 +3140,7 @@ module Aws::BedrockAgent
         params: params,
         config: config)
       context[:gem_name] = 'aws-sdk-bedrockagent'
-      context[:gem_version] = '1.5.0'
+      context[:gem_version] = '1.7.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -48,6 +48,7 @@ module Aws::BedrockAgent
     BasePromptTemplate = Shapes::StringShape.new(name: 'BasePromptTemplate')
     BedrockEmbeddingModelArn = Shapes::StringShape.new(name: 'BedrockEmbeddingModelArn')
     Boolean = Shapes::BooleanShape.new(name: 'Boolean')
+    BucketOwnerAccountId = Shapes::StringShape.new(name: 'BucketOwnerAccountId')
     ChunkingConfiguration = Shapes::StructureShape.new(name: 'ChunkingConfiguration')
     ChunkingStrategy = Shapes::StringShape.new(name: 'ChunkingStrategy')
     ClientToken = Shapes::StringShape.new(name: 'ClientToken')
@@ -64,6 +65,8 @@ 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')
+    DataDeletionPolicy = Shapes::StringShape.new(name: 'DataDeletionPolicy')
     DataSource = Shapes::StructureShape.new(name: 'DataSource')
     DataSourceConfiguration = Shapes::StructureShape.new(name: 'DataSourceConfiguration')
     DataSourceStatus = Shapes::StringShape.new(name: 'DataSourceStatus')
@@ -93,6 +96,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 +174,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 +230,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 +263,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 +310,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 +431,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
 
@@ -450,6 +465,7 @@ module Aws::BedrockAgent
     CreateAgentResponse.struct_class = Types::CreateAgentResponse
 
     CreateDataSourceRequest.add_member(:client_token, Shapes::ShapeRef.new(shape: ClientToken, location_name: "clientToken", metadata: {"idempotencyToken"=>true}))
+    CreateDataSourceRequest.add_member(:data_deletion_policy, Shapes::ShapeRef.new(shape: DataDeletionPolicy, location_name: "dataDeletionPolicy"))
     CreateDataSourceRequest.add_member(:data_source_configuration, Shapes::ShapeRef.new(shape: DataSourceConfiguration, required: true, location_name: "dataSourceConfiguration"))
     CreateDataSourceRequest.add_member(:description, Shapes::ShapeRef.new(shape: Description, location_name: "description"))
     CreateDataSourceRequest.add_member(:knowledge_base_id, Shapes::ShapeRef.new(shape: Id, required: true, location: "uri", location_name: "knowledgeBaseId"))
@@ -474,9 +490,11 @@ module Aws::BedrockAgent
     CreateKnowledgeBaseResponse.struct_class = Types::CreateKnowledgeBaseResponse
 
     DataSource.add_member(:created_at, Shapes::ShapeRef.new(shape: DateTimestamp, required: true, location_name: "createdAt"))
+    DataSource.add_member(:data_deletion_policy, Shapes::ShapeRef.new(shape: DataDeletionPolicy, location_name: "dataDeletionPolicy"))
     DataSource.add_member(:data_source_configuration, Shapes::ShapeRef.new(shape: DataSourceConfiguration, required: true, location_name: "dataSourceConfiguration"))
     DataSource.add_member(:data_source_id, Shapes::ShapeRef.new(shape: Id, required: true, location_name: "dataSourceId"))
     DataSource.add_member(:description, Shapes::ShapeRef.new(shape: Description, location_name: "description"))
+    DataSource.add_member(:failure_reasons, Shapes::ShapeRef.new(shape: FailureReasons, location_name: "failureReasons"))
     DataSource.add_member(:knowledge_base_id, Shapes::ShapeRef.new(shape: Id, required: true, location_name: "knowledgeBaseId"))
     DataSource.add_member(:name, Shapes::ShapeRef.new(shape: Name, required: true, location_name: "name"))
     DataSource.add_member(:server_side_encryption_configuration, Shapes::ShapeRef.new(shape: ServerSideEncryptionConfiguration, location_name: "serverSideEncryptionConfiguration"))
@@ -563,6 +581,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 +823,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"))
@@ -855,6 +894,7 @@ module Aws::BedrockAgent
     ResourceNotFoundException.struct_class = Types::ResourceNotFoundException
 
     S3DataSourceConfiguration.add_member(:bucket_arn, Shapes::ShapeRef.new(shape: S3BucketArn, required: true, location_name: "bucketArn"))
+    S3DataSourceConfiguration.add_member(:bucket_owner_account_id, Shapes::ShapeRef.new(shape: BucketOwnerAccountId, location_name: "bucketOwnerAccountId"))
     S3DataSourceConfiguration.add_member(:inclusion_prefixes, Shapes::ShapeRef.new(shape: S3Prefixes, location_name: "inclusionPrefixes"))
     S3DataSourceConfiguration.struct_class = Types::S3DataSourceConfiguration
 
@@ -916,6 +956,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
 
@@ -956,6 +997,7 @@ module Aws::BedrockAgent
     UpdateAgentResponse.add_member(:agent, Shapes::ShapeRef.new(shape: Agent, required: true, location_name: "agent"))
     UpdateAgentResponse.struct_class = Types::UpdateAgentResponse
 
+    UpdateDataSourceRequest.add_member(:data_deletion_policy, Shapes::ShapeRef.new(shape: DataDeletionPolicy, location_name: "dataDeletionPolicy"))
     UpdateDataSourceRequest.add_member(:data_source_configuration, Shapes::ShapeRef.new(shape: DataSourceConfiguration, required: true, location_name: "dataSourceConfiguration"))
     UpdateDataSourceRequest.add_member(:data_source_id, Shapes::ShapeRef.new(shape: Id, required: true, location: "uri", location_name: "dataSourceId"))
     UpdateDataSourceRequest.add_member(:description, Shapes::ShapeRef.new(shape: Description, location_name: "description"))

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
@@ -1186,6 +1206,10 @@ module Aws::BedrockAgent
     #   [1]: https://docs.aws.amazon.com/AWSEC2/latest/APIReference/Run_Instance_Idempotency.html
     #   @return [String]
     #
+    # @!attribute [rw] data_deletion_policy
+    #   The deletion policy for the requested data source
+    #   @return [String]
+    #
     # @!attribute [rw] data_source_configuration
     #   Contains metadata about where the data source is stored.
     #   @return [Types::DataSourceConfiguration]
@@ -1217,6 +1241,7 @@ module Aws::BedrockAgent
     #
     class CreateDataSourceRequest < Struct.new(
       :client_token,
+      :data_deletion_policy,
       :data_source_configuration,
       :description,
       :knowledge_base_id,
@@ -1313,6 +1338,10 @@ module Aws::BedrockAgent
     #   The time at which the data source was created.
     #   @return [Time]
     #
+    # @!attribute [rw] data_deletion_policy
+    #   The deletion policy for the data source.
+    #   @return [String]
+    #
     # @!attribute [rw] data_source_configuration
     #   Contains details about how the data source is stored.
     #   @return [Types::DataSourceConfiguration]
@@ -1325,6 +1354,10 @@ module Aws::BedrockAgent
     #   The description of the data source.
     #   @return [String]
     #
+    # @!attribute [rw] failure_reasons
+    #   The details of the failure reasons related to the data source.
+    #   @return [Array<String>]
+    #
     # @!attribute [rw] knowledge_base_id
     #   The unique identifier of the knowledge base to which the data source
     #   belongs.
@@ -1361,9 +1394,11 @@ module Aws::BedrockAgent
     #
     class DataSource < Struct.new(
       :created_at,
+      :data_deletion_policy,
       :data_source_configuration,
       :data_source_id,
       :description,
+      :failure_reasons,
       :knowledge_base_id,
       :name,
       :server_side_encryption_configuration,
@@ -1707,6 +1742,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 +2148,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 +2990,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].
@@ -3245,6 +3417,10 @@ module Aws::BedrockAgent
     #   source.
     #   @return [String]
     #
+    # @!attribute [rw] bucket_owner_account_id
+    #   The account ID for the owner of the S3 bucket.
+    #   @return [String]
+    #
     # @!attribute [rw] inclusion_prefixes
     #   A list of S3 prefixes that define the object containing the data
     #   sources. For more information, see [Organizing objects using
@@ -3259,6 +3435,7 @@ module Aws::BedrockAgent
     #
     class S3DataSourceConfiguration < Struct.new(
       :bucket_arn,
+      :bucket_owner_account_id,
       :inclusion_prefixes)
       SENSITIVE = []
       include Aws::Structure
@@ -3509,6 +3686,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 +3719,7 @@ module Aws::BedrockAgent
       :agent_version,
       :api_schema,
       :description,
+      :function_schema,
       :parent_action_group_signature)
       SENSITIVE = []
       include Aws::Structure
@@ -3730,6 +3913,10 @@ module Aws::BedrockAgent
       include Aws::Structure
     end
 
+    # @!attribute [rw] data_deletion_policy
+    #   The data deletion policy of the updated data source.
+    #   @return [String]
+    #
     # @!attribute [rw] data_source_configuration
     #   Contains details about the storage configuration of the data source.
     #   @return [Types::DataSourceConfiguration]
@@ -3763,6 +3950,7 @@ module Aws::BedrockAgent
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/UpdateDataSourceRequest AWS API Documentation
     #
     class UpdateDataSourceRequest < Struct.new(
+      :data_deletion_policy,
       :data_source_configuration,
       :data_source_id,
       :description,

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.7.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
@@ -175,9 +189,11 @@ module Aws
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgent/Client.html#create_data_source-instance_method
       def create_data_source: (
                                 ?client_token: ::String,
+                                ?data_deletion_policy: ("RETAIN" | "DELETE"),
                                 data_source_configuration: {
                                   s3_configuration: {
                                     bucket_arn: ::String,
+                                    bucket_owner_account_id: ::String?,
                                     inclusion_prefixes: Array[::String]?
                                   }?,
                                   type: ("S3")
@@ -318,7 +334,7 @@ module Aws
         include ::Seahorse::Client::_ResponseSuccess[Types::DeleteDataSourceResponse]
         def data_source_id: () -> ::String
         def knowledge_base_id: () -> ::String
-        def status: () -> ("AVAILABLE" | "DELETING")
+        def status: () -> ("AVAILABLE" | "DELETING" | "DELETE_UNSUCCESSFUL")
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgent/Client.html#delete_data_source-instance_method
       def delete_data_source: (
@@ -330,7 +346,7 @@ module Aws
       interface _DeleteKnowledgeBaseResponseSuccess
         include ::Seahorse::Client::_ResponseSuccess[Types::DeleteKnowledgeBaseResponse]
         def knowledge_base_id: () -> ::String
-        def status: () -> ("CREATING" | "ACTIVE" | "DELETING" | "UPDATING" | "FAILED")
+        def status: () -> ("CREATING" | "ACTIVE" | "DELETING" | "UPDATING" | "FAILED" | "DELETE_UNSUCCESSFUL")
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgent/Client.html#delete_knowledge_base-instance_method
       def delete_knowledge_base: (
@@ -653,6 +669,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 +685,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
@@ -710,9 +740,11 @@ module Aws
       end
       # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/BedrockAgent/Client.html#update_data_source-instance_method
       def update_data_source: (
+                                ?data_deletion_policy: ("RETAIN" | "DELETE"),
                                 data_source_configuration: {
                                   s3_configuration: {
                                     bucket_arn: ::String,
+                                    bucket_owner_account_id: ::String?,
                                     inclusion_prefixes: Array[::String]?
                                   }?,
                                   type: ("S3")

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
@@ -261,6 +266,7 @@ module Aws::BedrockAgent
 
     class CreateDataSourceRequest
       attr_accessor client_token: ::String
+      attr_accessor data_deletion_policy: ("RETAIN" | "DELETE")
       attr_accessor data_source_configuration: Types::DataSourceConfiguration
       attr_accessor description: ::String
       attr_accessor knowledge_base_id: ::String
@@ -293,13 +299,15 @@ module Aws::BedrockAgent
 
     class DataSource
       attr_accessor created_at: ::Time
+      attr_accessor data_deletion_policy: ("RETAIN" | "DELETE")
       attr_accessor data_source_configuration: Types::DataSourceConfiguration
       attr_accessor data_source_id: ::String
       attr_accessor description: ::String
+      attr_accessor failure_reasons: ::Array[::String]
       attr_accessor knowledge_base_id: ::String
       attr_accessor name: ::String
       attr_accessor server_side_encryption_configuration: Types::ServerSideEncryptionConfiguration
-      attr_accessor status: ("AVAILABLE" | "DELETING")
+      attr_accessor status: ("AVAILABLE" | "DELETING" | "DELETE_UNSUCCESSFUL")
       attr_accessor updated_at: ::Time
       attr_accessor vector_ingestion_configuration: Types::VectorIngestionConfiguration
       SENSITIVE: []
@@ -316,7 +324,7 @@ module Aws::BedrockAgent
       attr_accessor description: ::String
       attr_accessor knowledge_base_id: ::String
       attr_accessor name: ::String
-      attr_accessor status: ("AVAILABLE" | "DELETING")
+      attr_accessor status: ("AVAILABLE" | "DELETING" | "DELETE_UNSUCCESSFUL")
       attr_accessor updated_at: ::Time
       SENSITIVE: []
     end
@@ -380,7 +388,7 @@ module Aws::BedrockAgent
     class DeleteDataSourceResponse
       attr_accessor data_source_id: ::String
       attr_accessor knowledge_base_id: ::String
-      attr_accessor status: ("AVAILABLE" | "DELETING")
+      attr_accessor status: ("AVAILABLE" | "DELETING" | "DELETE_UNSUCCESSFUL")
       SENSITIVE: []
     end
 
@@ -391,7 +399,7 @@ module Aws::BedrockAgent
 
     class DeleteKnowledgeBaseResponse
       attr_accessor knowledge_base_id: ::String
-      attr_accessor status: ("CREATING" | "ACTIVE" | "DELETING" | "UPDATING" | "FAILED")
+      attr_accessor status: ("CREATING" | "ACTIVE" | "DELETING" | "UPDATING" | "FAILED" | "DELETE_UNSUCCESSFUL")
       SENSITIVE: []
     end
 
@@ -411,6 +419,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
@@ -572,7 +598,7 @@ module Aws::BedrockAgent
       attr_accessor knowledge_base_id: ::String
       attr_accessor name: ::String
       attr_accessor role_arn: ::String
-      attr_accessor status: ("CREATING" | "ACTIVE" | "DELETING" | "UPDATING" | "FAILED")
+      attr_accessor status: ("CREATING" | "ACTIVE" | "DELETING" | "UPDATING" | "FAILED" | "DELETE_UNSUCCESSFUL")
       attr_accessor storage_configuration: Types::StorageConfiguration
       attr_accessor updated_at: ::Time
       SENSITIVE: []
@@ -588,7 +614,7 @@ module Aws::BedrockAgent
       attr_accessor description: ::String
       attr_accessor knowledge_base_id: ::String
       attr_accessor name: ::String
-      attr_accessor status: ("CREATING" | "ACTIVE" | "DELETING" | "UPDATING" | "FAILED")
+      attr_accessor status: ("CREATING" | "ACTIVE" | "DELETING" | "UPDATING" | "FAILED" | "DELETE_UNSUCCESSFUL")
       attr_accessor updated_at: ::Time
       SENSITIVE: []
     end
@@ -724,6 +750,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
@@ -806,6 +839,7 @@ module Aws::BedrockAgent
 
     class S3DataSourceConfiguration
       attr_accessor bucket_arn: ::String
+      attr_accessor bucket_owner_account_id: ::String
       attr_accessor inclusion_prefixes: ::Array[::String]
       SENSITIVE: []
     end
@@ -880,6 +914,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
@@ -936,6 +971,7 @@ module Aws::BedrockAgent
     end
 
     class UpdateDataSourceRequest
+      attr_accessor data_deletion_policy: ("RETAIN" | "DELETE")
       attr_accessor data_source_configuration: Types::DataSourceConfiguration
       attr_accessor data_source_id: ::String
       attr_accessor description: ::String

metadata

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