aws-sdk-bedrockagent 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 421c4af321deb6eaef8ae07288ba866340052236ce4d263891a00368f201d9ac
-  data.tar.gz: e03c63fdfd1758ed667fc153c7042ab9a6f4456255b4902bb6cde4e8340ebbee
+  metadata.gz: 18afad3033d8046f6e84ffa350ac115d5c8c9b2ddf76529ec91f1afee13a3976
+  data.tar.gz: 47ec503df4d8dd33dd6bf67feee1f7ac3ad3c54e56c29a64510154918c2fa16b
 SHA512:
-  metadata.gz: 68434e938593c334e373517a48ef7e7df476fbd60b52131e5d4077776146463dc39a784d8c598d6da203732f7794d8ff5297899d90277d222abb96354d19b3d9
-  data.tar.gz: 870c3c8d028666f2dbe576b5186ca76be07ea350d3b84d79d87ea1ccefb979c320b1434e95c3adbb76dc6d37aa866b7eb139603c83e31b1a8efa6ca9285142b9
+  metadata.gz: 6b6b152b3440c7e6db66582c207503c53b01c24bd45b36c5bad6869a61b17403141292c19514432ae7f27a1130a142c01496fb6109f0dece9c8adfd507c3d22d
+  data.tar.gz: 01da5ec955929b371554485f08110885432571bc1fc18e4daadfa189e20ada70b2a68f48c1489df87d58f3cad8832b0a1d6c184e8736e5d23add6cfddb1cfcc8

data/CHANGELOG.md

@@ -1,6 +1,11 @@
 Unreleased Changes
 ------------------
 
+1.1.0 (2023-12-21)
+------------------
+
+* Feature - This release introduces Amazon Aurora as a vector store on Knowledge Bases for Amazon Bedrock
+
 1.0.0 (2023-11-28)
 ------------------
 

data/VERSION

@@ -1 +1 @@
-1.0.0
+1.1.0

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

@@ -841,7 +841,7 @@ module Aws::BedrockAgent
     #       },
     #     },
     #     storage_configuration: { # required
-    #       type: "OPENSEARCH_SERVERLESS", # required, accepts OPENSEARCH_SERVERLESS, PINECONE, REDIS_ENTERPRISE_CLOUD
+    #       type: "OPENSEARCH_SERVERLESS", # required, accepts OPENSEARCH_SERVERLESS, PINECONE, REDIS_ENTERPRISE_CLOUD, RDS
     #       opensearch_serverless_configuration: {
     #         collection_arn: "OpenSearchServerlessCollectionArn", # required
     #         vector_index_name: "OpenSearchServerlessIndexName", # required
@@ -870,6 +870,18 @@ module Aws::BedrockAgent
     #           metadata_field: "FieldName", # required
     #         },
     #       },
+    #       rds_configuration: {
+    #         resource_arn: "RdsArn", # required
+    #         credentials_secret_arn: "SecretArn", # required
+    #         database_name: "RdsDatabaseName", # required
+    #         table_name: "RdsTableName", # required
+    #         field_mapping: { # required
+    #           primary_key_field: "ColumnName", # required
+    #           vector_field: "ColumnName", # required
+    #           text_field: "ColumnName", # required
+    #           metadata_field: "ColumnName", # required
+    #         },
+    #       },
     #     },
     #     tags: {
     #       "TagKey" => "TagValue",
@@ -885,7 +897,7 @@ module Aws::BedrockAgent
     #   resp.knowledge_base.role_arn #=> String
     #   resp.knowledge_base.knowledge_base_configuration.type #=> String, one of "VECTOR"
     #   resp.knowledge_base.knowledge_base_configuration.vector_knowledge_base_configuration.embedding_model_arn #=> String
-    #   resp.knowledge_base.storage_configuration.type #=> String, one of "OPENSEARCH_SERVERLESS", "PINECONE", "REDIS_ENTERPRISE_CLOUD"
+    #   resp.knowledge_base.storage_configuration.type #=> String, one of "OPENSEARCH_SERVERLESS", "PINECONE", "REDIS_ENTERPRISE_CLOUD", "RDS"
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.collection_arn #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.vector_index_name #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.field_mapping.vector_field #=> String
@@ -902,6 +914,14 @@ module Aws::BedrockAgent
     #   resp.knowledge_base.storage_configuration.redis_enterprise_cloud_configuration.field_mapping.vector_field #=> String
     #   resp.knowledge_base.storage_configuration.redis_enterprise_cloud_configuration.field_mapping.text_field #=> String
     #   resp.knowledge_base.storage_configuration.redis_enterprise_cloud_configuration.field_mapping.metadata_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.resource_arn #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.credentials_secret_arn #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.database_name #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.table_name #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.primary_key_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.vector_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.text_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.metadata_field #=> String
     #   resp.knowledge_base.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED"
     #   resp.knowledge_base.created_at #=> Time
     #   resp.knowledge_base.updated_at #=> Time
@@ -1539,7 +1559,7 @@ module Aws::BedrockAgent
     #   resp.knowledge_base.role_arn #=> String
     #   resp.knowledge_base.knowledge_base_configuration.type #=> String, one of "VECTOR"
     #   resp.knowledge_base.knowledge_base_configuration.vector_knowledge_base_configuration.embedding_model_arn #=> String
-    #   resp.knowledge_base.storage_configuration.type #=> String, one of "OPENSEARCH_SERVERLESS", "PINECONE", "REDIS_ENTERPRISE_CLOUD"
+    #   resp.knowledge_base.storage_configuration.type #=> String, one of "OPENSEARCH_SERVERLESS", "PINECONE", "REDIS_ENTERPRISE_CLOUD", "RDS"
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.collection_arn #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.vector_index_name #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.field_mapping.vector_field #=> String
@@ -1556,6 +1576,14 @@ module Aws::BedrockAgent
     #   resp.knowledge_base.storage_configuration.redis_enterprise_cloud_configuration.field_mapping.vector_field #=> String
     #   resp.knowledge_base.storage_configuration.redis_enterprise_cloud_configuration.field_mapping.text_field #=> String
     #   resp.knowledge_base.storage_configuration.redis_enterprise_cloud_configuration.field_mapping.metadata_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.resource_arn #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.credentials_secret_arn #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.database_name #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.table_name #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.primary_key_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.vector_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.text_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.metadata_field #=> String
     #   resp.knowledge_base.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED"
     #   resp.knowledge_base.created_at #=> Time
     #   resp.knowledge_base.updated_at #=> Time
@@ -2564,7 +2592,7 @@ module Aws::BedrockAgent
     #       },
     #     },
     #     storage_configuration: { # required
-    #       type: "OPENSEARCH_SERVERLESS", # required, accepts OPENSEARCH_SERVERLESS, PINECONE, REDIS_ENTERPRISE_CLOUD
+    #       type: "OPENSEARCH_SERVERLESS", # required, accepts OPENSEARCH_SERVERLESS, PINECONE, REDIS_ENTERPRISE_CLOUD, RDS
     #       opensearch_serverless_configuration: {
     #         collection_arn: "OpenSearchServerlessCollectionArn", # required
     #         vector_index_name: "OpenSearchServerlessIndexName", # required
@@ -2593,6 +2621,18 @@ module Aws::BedrockAgent
     #           metadata_field: "FieldName", # required
     #         },
     #       },
+    #       rds_configuration: {
+    #         resource_arn: "RdsArn", # required
+    #         credentials_secret_arn: "SecretArn", # required
+    #         database_name: "RdsDatabaseName", # required
+    #         table_name: "RdsTableName", # required
+    #         field_mapping: { # required
+    #           primary_key_field: "ColumnName", # required
+    #           vector_field: "ColumnName", # required
+    #           text_field: "ColumnName", # required
+    #           metadata_field: "ColumnName", # required
+    #         },
+    #       },
     #     },
     #   })
     #
@@ -2605,7 +2645,7 @@ module Aws::BedrockAgent
     #   resp.knowledge_base.role_arn #=> String
     #   resp.knowledge_base.knowledge_base_configuration.type #=> String, one of "VECTOR"
     #   resp.knowledge_base.knowledge_base_configuration.vector_knowledge_base_configuration.embedding_model_arn #=> String
-    #   resp.knowledge_base.storage_configuration.type #=> String, one of "OPENSEARCH_SERVERLESS", "PINECONE", "REDIS_ENTERPRISE_CLOUD"
+    #   resp.knowledge_base.storage_configuration.type #=> String, one of "OPENSEARCH_SERVERLESS", "PINECONE", "REDIS_ENTERPRISE_CLOUD", "RDS"
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.collection_arn #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.vector_index_name #=> String
     #   resp.knowledge_base.storage_configuration.opensearch_serverless_configuration.field_mapping.vector_field #=> String
@@ -2622,6 +2662,14 @@ module Aws::BedrockAgent
     #   resp.knowledge_base.storage_configuration.redis_enterprise_cloud_configuration.field_mapping.vector_field #=> String
     #   resp.knowledge_base.storage_configuration.redis_enterprise_cloud_configuration.field_mapping.text_field #=> String
     #   resp.knowledge_base.storage_configuration.redis_enterprise_cloud_configuration.field_mapping.metadata_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.resource_arn #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.credentials_secret_arn #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.database_name #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.table_name #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.primary_key_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.vector_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.text_field #=> String
+    #   resp.knowledge_base.storage_configuration.rds_configuration.field_mapping.metadata_field #=> String
     #   resp.knowledge_base.status #=> String, one of "CREATING", "ACTIVE", "DELETING", "UPDATING", "FAILED"
     #   resp.knowledge_base.created_at #=> Time
     #   resp.knowledge_base.updated_at #=> Time
@@ -2650,7 +2698,7 @@ module Aws::BedrockAgent
         params: params,
         config: config)
       context[:gem_name] = 'aws-sdk-bedrockagent'
-      context[:gem_version] = '1.0.0'
+      context[:gem_version] = '1.1.0'
       Seahorse::Client::Request.new(handlers, context)
     end
 

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

@@ -51,6 +51,7 @@ module Aws::BedrockAgent
     ChunkingConfiguration = Shapes::StructureShape.new(name: 'ChunkingConfiguration')
     ChunkingStrategy = Shapes::StringShape.new(name: 'ChunkingStrategy')
     ClientToken = Shapes::StringShape.new(name: 'ClientToken')
+    ColumnName = Shapes::StringShape.new(name: 'ColumnName')
     ConflictException = Shapes::StructureShape.new(name: 'ConflictException')
     CreateAgentActionGroupRequest = Shapes::StructureShape.new(name: 'CreateAgentActionGroupRequest')
     CreateAgentActionGroupResponse = Shapes::StructureShape.new(name: 'CreateAgentActionGroupResponse')
@@ -179,6 +180,11 @@ module Aws::BedrockAgent
     PromptOverrideConfiguration = Shapes::StructureShape.new(name: 'PromptOverrideConfiguration')
     PromptState = Shapes::StringShape.new(name: 'PromptState')
     PromptType = Shapes::StringShape.new(name: 'PromptType')
+    RdsArn = Shapes::StringShape.new(name: 'RdsArn')
+    RdsConfiguration = Shapes::StructureShape.new(name: 'RdsConfiguration')
+    RdsDatabaseName = Shapes::StringShape.new(name: 'RdsDatabaseName')
+    RdsFieldMapping = Shapes::StructureShape.new(name: 'RdsFieldMapping')
+    RdsTableName = Shapes::StringShape.new(name: 'RdsTableName')
     RecommendedAction = Shapes::StringShape.new(name: 'RecommendedAction')
     RecommendedActions = Shapes::ListShape.new(name: 'RecommendedActions')
     RedisEnterpriseCloudConfiguration = Shapes::StructureShape.new(name: 'RedisEnterpriseCloudConfiguration')
@@ -817,6 +823,19 @@ module Aws::BedrockAgent
     PromptOverrideConfiguration.add_member(:override_lambda, Shapes::ShapeRef.new(shape: LambdaArn, location_name: "overrideLambda"))
     PromptOverrideConfiguration.struct_class = Types::PromptOverrideConfiguration
 
+    RdsConfiguration.add_member(:resource_arn, Shapes::ShapeRef.new(shape: RdsArn, required: true, location_name: "resourceArn"))
+    RdsConfiguration.add_member(:credentials_secret_arn, Shapes::ShapeRef.new(shape: SecretArn, required: true, location_name: "credentialsSecretArn"))
+    RdsConfiguration.add_member(:database_name, Shapes::ShapeRef.new(shape: RdsDatabaseName, required: true, location_name: "databaseName"))
+    RdsConfiguration.add_member(:table_name, Shapes::ShapeRef.new(shape: RdsTableName, required: true, location_name: "tableName"))
+    RdsConfiguration.add_member(:field_mapping, Shapes::ShapeRef.new(shape: RdsFieldMapping, required: true, location_name: "fieldMapping"))
+    RdsConfiguration.struct_class = Types::RdsConfiguration
+
+    RdsFieldMapping.add_member(:primary_key_field, Shapes::ShapeRef.new(shape: ColumnName, required: true, location_name: "primaryKeyField"))
+    RdsFieldMapping.add_member(:vector_field, Shapes::ShapeRef.new(shape: ColumnName, required: true, location_name: "vectorField"))
+    RdsFieldMapping.add_member(:text_field, Shapes::ShapeRef.new(shape: ColumnName, required: true, location_name: "textField"))
+    RdsFieldMapping.add_member(:metadata_field, Shapes::ShapeRef.new(shape: ColumnName, required: true, location_name: "metadataField"))
+    RdsFieldMapping.struct_class = Types::RdsFieldMapping
+
     RecommendedActions.member = Shapes::ShapeRef.new(shape: RecommendedAction)
 
     RedisEnterpriseCloudConfiguration.add_member(:endpoint, Shapes::ShapeRef.new(shape: RedisEnterpriseCloudEndpoint, required: true, location_name: "endpoint"))
@@ -864,6 +883,7 @@ module Aws::BedrockAgent
     StorageConfiguration.add_member(:opensearch_serverless_configuration, Shapes::ShapeRef.new(shape: OpenSearchServerlessConfiguration, location_name: "opensearchServerlessConfiguration"))
     StorageConfiguration.add_member(:pinecone_configuration, Shapes::ShapeRef.new(shape: PineconeConfiguration, location_name: "pineconeConfiguration"))
     StorageConfiguration.add_member(:redis_enterprise_cloud_configuration, Shapes::ShapeRef.new(shape: RedisEnterpriseCloudConfiguration, location_name: "redisEnterpriseCloudConfiguration"))
+    StorageConfiguration.add_member(:rds_configuration, Shapes::ShapeRef.new(shape: RdsConfiguration, location_name: "rdsConfiguration"))
     StorageConfiguration.struct_class = Types::StorageConfiguration
 
     TagKeyList.member = Shapes::ShapeRef.new(shape: TagKey)

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

@@ -2646,6 +2646,69 @@ module Aws::BedrockAgent
       include Aws::Structure
     end
 
+    # Contains the configurations to use RDS to store knowledge base data.
+    #
+    # @!attribute [rw] resource_arn
+    #   Arn of a RDS Resource.
+    #   @return [String]
+    #
+    # @!attribute [rw] credentials_secret_arn
+    #   Arn of a SecretsManager Secret.
+    #   @return [String]
+    #
+    # @!attribute [rw] database_name
+    #   Name of the database within RDS
+    #   @return [String]
+    #
+    # @!attribute [rw] table_name
+    #   Name of the table within RDS
+    #   @return [String]
+    #
+    # @!attribute [rw] field_mapping
+    #   A mapping of Bedrock Knowledge Base fields to RDS column names
+    #   @return [Types::RdsFieldMapping]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/RdsConfiguration AWS API Documentation
+    #
+    class RdsConfiguration < Struct.new(
+      :resource_arn,
+      :credentials_secret_arn,
+      :database_name,
+      :table_name,
+      :field_mapping)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
+    # A mapping of Bedrock Knowledge Base fields to RDS column names
+    #
+    # @!attribute [rw] primary_key_field
+    #   Name of the column
+    #   @return [String]
+    #
+    # @!attribute [rw] vector_field
+    #   Name of the column
+    #   @return [String]
+    #
+    # @!attribute [rw] text_field
+    #   Name of the column
+    #   @return [String]
+    #
+    # @!attribute [rw] metadata_field
+    #   Name of the column
+    #   @return [String]
+    #
+    # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/RdsFieldMapping AWS API Documentation
+    #
+    class RdsFieldMapping < Struct.new(
+      :primary_key_field,
+      :vector_field,
+      :text_field,
+      :metadata_field)
+      SENSITIVE = []
+      include Aws::Structure
+    end
+
     # Contains the configurations to use Redis Enterprise Cloud to store
     # knowledge base data.
     #
@@ -2846,13 +2909,18 @@ module Aws::BedrockAgent
     #   knowledge base data.
     #   @return [Types::RedisEnterpriseCloudConfiguration]
     #
+    # @!attribute [rw] rds_configuration
+    #   Contains the configurations to use RDS to store knowledge base data.
+    #   @return [Types::RdsConfiguration]
+    #
     # @see http://docs.aws.amazon.com/goto/WebAPI/bedrock-agent-2023-06-05/StorageConfiguration AWS API Documentation
     #
     class StorageConfiguration < Struct.new(
       :type,
       :opensearch_serverless_configuration,
       :pinecone_configuration,
-      :redis_enterprise_cloud_configuration)
+      :redis_enterprise_cloud_configuration,
+      :rds_configuration)
       SENSITIVE = []
       include Aws::Structure
     end

data/lib/aws-sdk-bedrockagent.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-sdk-bedrockagent
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Amazon Web Services
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-11-28 00:00:00.000000000 Z
+date: 2023-12-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core