feat: release bedrock agents sample (#23)

* feat: release bedrock agents sample

---------

Signed-off-by: Scott Schreckengaust <[email protected]>

Author: scottschreckengaust

samples/bedrock-agent/README.md

@@ -96,6 +96,8 @@ This project is built using [Terraform](https://www.terraform.io/). See [Getting
 5. Enable Access to Amazon Bedrock Models
 > You must explicitly enable access to models before they can be used with the Amazon Bedrock service. Please follow these steps in the [Amazon Bedrock User Guide](https://docs.aws.amazon.com/bedrock/latest/userguide/model-access.html) to enable access to the models (```Anthropic::Claude```):.
 
+_NOTE: The default architecture is `x86_64`, feel free to add `-var="architecture=$(python -c "import platform;print('x86_64' if platform.machine().upper() in ['X86', 'AMD64'] else 'arm64')")"` to inspect and use your platform's architecture or set it directly with the `arm64.tfvars`_
+
 6. Check the plan.
 
     ```shell

samples/bedrock-agent/arm64.tfvars

@@ -0,0 +1 @@
+architecture = "arm64"

samples/bedrock-agent/data.tf

@@ -1 +1 @@
-# data "aws_bedrock_foundation_models" "test" {}
+data "aws_bedrock_foundation_models" "test" {}

samples/bedrock-agent/main.tf

@@ -1,9 +1,3 @@
-variable "region" {
-  type        = string
-  description = "AWS region to deploy the resources"
-  default     = "us-east-1"
-}
-
 provider "aws" {
   region = var.region
 }
@@ -19,13 +13,12 @@ provider "opensearch" {
 
 module "bedrock" {
   #checkov:skip=CKV_TF_1:Terraform registry has no ability to use a commit hash
-  # source            = "aws-ia/bedrock/aws"
-  # version           = "0.0.3"
-  source                       = "github.com/aws-ia/terraform-aws-bedrock//?ref=58798f1fcf95db48ce3f336c7e93d50c83e27ce8"
+  source                       = "aws-ia/bedrock/aws"
+  version                      = "0.0.3"
   create_kb                    = true
   create_default_kb            = true
   create_agent                 = true
-  foundation_model             = "anthropic.claude-3-5-haiku-20241022-v1:0" # "anthropic.claude-v2"
+  foundation_model             = var.foundation_model
   instruction                  = "You are a helpful and friendly agent that answers questions about literature."
   create_ag                    = true
   action_group_name            = "bedrock-agent-action-group"
@@ -36,25 +29,20 @@ module "bedrock" {
 }
 
 module "lambda" {
+  #checkov:skip=CKV_TF_1:Terraform registry has no ability to use a commit hash
   source        = "terraform-aws-modules/lambda/aws"
   version       = "7.15.0"
   function_name = "bedrock-agent-action"
   handler       = "index.handler"
   runtime       = "python3.12"
   publish       = true
   timeout       = 15
+  architectures = ["${var.architecture}", ]
+  layers        = ["arn:aws:lambda:${var.region}:017000801446:layer:AWSLambdaPowertoolsPythonV3-python312-${var.architecture}:4", ]
   source_path = [
     {
       path           = "${path.module}/lambda/action-group"
       poetry_install = true
     }
   ]
-  architectures = [
-    "arm64",
-    # "x86_64",
-  ]
-  layers = [
-    "arn:aws:lambda:${var.region}:017000801446:layer:AWSLambdaPowertoolsPythonV3-python312-arm64:4",
-    # "arn:aws:lambda:${var.region}:017000801446:layer:AWSLambdaPowertoolsPythonV3-python312-x86_64:4",
-  ]
 }

samples/bedrock-agent/outputs.tf

@@ -16,9 +16,4 @@ output "knowledge_base_id" {
 output "data_source_id" {
   description = "The Bedrock Agent's data source"
   value       = module.bedrock.datasource_identifier
-}
-
-# output "foundation_models" {
-#   description = "The Bedrock Agent's foundation models"
-#   value       = data.aws_bedrock_foundation_models.test
-# }
+}

samples/bedrock-agent/variables.tf

@@ -0,0 +1,37 @@
+variable "region" {
+  type        = string
+  description = "AWS region to deploy the resources"
+  default     = "us-east-1"
+  validation {
+    condition = contains([
+      "us-east-1", "us-west-2",
+      "ap-southeast-1", "ap-southeast-2", "ap-northeast-1", "ap-south-1",
+      "ca-central-1",
+      "eu-central-1", "eu-west-1", "eu-west-2", "eu-west-3",
+      "sa-east-1",
+    ], var.region)
+    error_message = "Please use a region supporting Agents for Amazon Bedrock https://docs.aws.amazon.com/general/latest/gr/bedrock.html."
+  }
+}
+
+variable "architecture" {
+  type        = string
+  description = "The runtime architecture"
+  nullable    = false
+  default     = "x86_64"
+  validation {
+    condition     = contains(["x86_64", "arm64"], var.architecture)
+    error_message = "The architecture must be either 'x86_64' or 'arm64'."
+  }
+}
+
+variable "foundation_model" {
+  type        = string
+  description = "The foundation model to use for the agent"
+  nullable    = false
+  default     = "anthropic.claude-3-5-haiku-20241022-v1:0"
+  validation {
+    condition     = contains(data.aws_bedrock_foundation_models.test.model_summaries[*].model_id, var.foundation_model)
+    error_message = "The foundational model is not found"
+  }
+}