chore: removes hardcoded text value

Author: bencegadanyi1-nhs

packages/cdk/resources/BedrockPromptResources.ts

@@ -14,6 +14,8 @@ export interface BedrockPromptResourcesProps {
 export class BedrockPromptResources extends Construct {
   public readonly queryReformulationPrompt: Prompt
   public readonly ragResponsePrompt: Prompt
+  public readonly ragModelId: string
+  public readonly queryReformulationModelId: string
 
   constructor(scope: Construct, id: string, props: BedrockPromptResourcesProps) {
     super(scope, id)
@@ -56,6 +58,10 @@ export class BedrockPromptResources extends Construct {
       variants: [ragResponsePromptVariant]
     })
 
+    // expose model IDs for use in Lambda environment variables
+    this.ragModelId = novaProModel.modelId
+    this.queryReformulationModelId = novaLiteModel.modelId
+
     this.queryReformulationPrompt = queryReformulationPrompt
     this.ragResponsePrompt = ragPrompt
   }

packages/cdk/resources/Functions.ts

@@ -5,10 +5,6 @@ import {StringParameter} from "aws-cdk-lib/aws-ssm"
 import {Secret} from "aws-cdk-lib/aws-secretsmanager"
 import {TableV2} from "aws-cdk-lib/aws-dynamodb"
 
-// Claude model for RAG responses
-const RAG_MODEL_ID = "anthropic.claude-3-sonnet-20240229-v1:0"
-// Claude model for query reformulation
-const QUERY_REFORMULATION_MODEL_ID = "anthropic.claude-3-haiku-20240307-v1:0"
 const BEDROCK_KB_DATA_SOURCE = "eps-assist-kb-ds"
 const LAMBDA_MEMORY_SIZE = "265"
 
@@ -38,6 +34,8 @@ export interface FunctionsProps {
   readonly ragResponsePromptVersion: string
   readonly isPullRequest: boolean
   readonly mainSlackBotLambdaExecutionRoleArn : string
+  readonly ragModelId: string
+  readonly queryReformulationModelId: string
 }
 
 export class Functions extends Construct {
@@ -58,8 +56,8 @@ export class Functions extends Construct {
       additionalPolicies: [props.slackBotManagedPolicy],
       dependencyLocation: ".dependencies/slackBotFunction",
       environmentVariables: {
-        "RAG_MODEL_ID": RAG_MODEL_ID,
-        "QUERY_REFORMULATION_MODEL_ID": QUERY_REFORMULATION_MODEL_ID,
+        "RAG_MODEL_ID": props.ragModelId,
+        "QUERY_REFORMULATION_MODEL_ID": props.queryReformulationModelId,
         "KNOWLEDGEBASE_ID": props.knowledgeBaseId,
         "BEDROCK_KB_DATA_SOURCE": BEDROCK_KB_DATA_SOURCE,
         "LAMBDA_MEMORY_SIZE": LAMBDA_MEMORY_SIZE,

packages/cdk/stacks/EpsAssistMeStack.ts

@@ -157,6 +157,8 @@ export class EpsAssistMeStack extends Stack {
       ragResponsePromptName: bedrockPromptResources.ragResponsePrompt.promptName,
       reformulationPromptVersion: bedrockPromptResources.queryReformulationPrompt.promptVersion,
       ragResponsePromptVersion: bedrockPromptResources.ragResponsePrompt.promptVersion,
+      ragModelId: bedrockPromptResources.ragModelId,
+      queryReformulationModelId: bedrockPromptResources.queryReformulationModelId,
       isPullRequest: isPullRequest,
       mainSlackBotLambdaExecutionRoleArn: mainSlackBotLambdaExecutionRoleArn
     })