Bedrock-Kb.yml

AWSTemplateFormatVersion: 2010-09-09
Description: 'Serverless RAG Q&A application using Knowledge Base , Agents, Opensearch Serverless'

Parameters:
  KnowledgeBaseName:
    Default: rag-demo-kb
    Type: String
    Description: The name of the knowledge base.
  KnowledgeBaseDescription:
    Default: Answer based only on information contained in knowledge base. 
    Type: String
    Description: The description of the knowledge base.
  AmazonBedrockExecutionRoleForKnowledgeBasearn:
    Type: String
    Description: Execution Role for Knowledge Base Arn.
  
  AgentName:
    Default: rag-demo-agent
    Type: String
    Description: The name of the agent.
  AOSSIndexName:
    Default: rag-bedrock-index
    Type: String
    Description: Name of the vector index in the Amazon OpenSearch Service Serverless (AOSS) collection. You can get the name from the output section of the previous stack
  DataSource:
    Type: String
    Description: S3 bucket name from the previous stack.

  S3BucketArn:
    Type: String
    Description: S3 bucket arn from the previous stack.

  CollectionArn:
    Type: String
    Description: Collection Arn from the previous stack.

Resources:

  KnowledgeBaseWithAoss:
    Type: AWS::Bedrock::KnowledgeBase
    Properties:
      Name: !Ref KnowledgeBaseName
      Description: !Ref KnowledgeBaseDescription
      RoleArn: !Ref AmazonBedrockExecutionRoleForKnowledgeBasearn
      KnowledgeBaseConfiguration:
        Type: "VECTOR"
        VectorKnowledgeBaseConfiguration:
          EmbeddingModelArn: !Sub "arn:${AWS::Partition}:bedrock:${AWS::Region}::foundation-model/cohere.embed-english-v3"
      StorageConfiguration:
        Type: "OPENSEARCH_SERVERLESS"
        OpensearchServerlessConfiguration:
          CollectionArn: !Ref CollectionArn
          VectorIndexName: !Ref AOSSIndexName
          FieldMapping:
            VectorField: "vector"
            TextField: "text"
            MetadataField: "metadata"

  SampleDataSource:
    Type: AWS::Bedrock::DataSource
    Properties:
      KnowledgeBaseId: !Ref KnowledgeBaseWithAoss
      Name: !Ref DataSource
      DataSourceConfiguration:
        Type: "S3"
        S3Configuration:
          BucketArn: !Ref S3BucketArn

    

  # AmazonBedrockExecutionRoleForAgentsQA:
  #   Type: AWS::IAM::Role
  #   Properties:
  #     RoleName: AmazonBedrockExecutionRoleForAgents
  #     AssumeRolePolicyDocument:
  #       Statement:
  #       - Effect: Allow
  #         Principal:
  #           Service: bedrock.amazonaws.com
  #         Action: sts:AssumeRole
  #     ManagedPolicyArns:
  #       - arn:aws:iam::aws:policy/AmazonBedrockFullAccess

  
  # AgentResource:
  #     Type: AWS::Bedrock::Agent
  #     Properties:
  #       AgentName: !Ref AgentName
  #       AgentResourceRoleArn: !GetAtt AmazonBedrockExecutionRoleForAgentsQA.Arn
  #       AutoPrepare: true
  #       FoundationModel: "anthropic.claude-v2"
  #       Instruction: "You are a Q&A bot to answer questions on Amazon SageMaker"
  #       Description: "Description is here"
  #       IdleSessionTTLInSeconds: 900
  #       KnowledgeBases:
  #         - KnowledgeBaseId: !Ref KnowledgeBaseWithAoss
  #           Description: !Ref KnowledgeBaseDescription
  #           KnowledgeBaseState: ENABLED