remove claude

Author: anthony-nhs

packages/cdk/resources/Constants.ts

@@ -0,0 +1,4 @@
+// Nova model for RAG responses
+export const RAG_MODEL_ID = "amazon.nova-pro-v1:0"
+// Nova model for query reformulation
+export const QUERY_REFORMULATION_MODEL_ID = "amazon.nova-lite-v1:0"

packages/cdk/resources/Functions.ts

@@ -4,12 +4,8 @@ import {ManagedPolicy, PolicyStatement, Role} from "aws-cdk-lib/aws-iam"
 import {StringParameter} from "aws-cdk-lib/aws-ssm"
 import {Secret} from "aws-cdk-lib/aws-secretsmanager"
 import {TableV2} from "aws-cdk-lib/aws-dynamodb"
+import {QUERY_REFORMULATION_MODEL_ID, RAG_MODEL_ID} from "./Constants"
 
-// 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"
 
 export interface FunctionsProps {
@@ -61,7 +57,6 @@ export class Functions extends Construct {
         "RAG_MODEL_ID": RAG_MODEL_ID,
         "QUERY_REFORMULATION_MODEL_ID": QUERY_REFORMULATION_MODEL_ID,
         "KNOWLEDGEBASE_ID": props.knowledgeBaseId,
-        "BEDROCK_KB_DATA_SOURCE": BEDROCK_KB_DATA_SOURCE,
         "LAMBDA_MEMORY_SIZE": LAMBDA_MEMORY_SIZE,
         "SLACK_BOT_TOKEN_PARAMETER": props.slackBotTokenParameter.parameterName,
         "SLACK_SIGNING_SECRET_PARAMETER": props.slackSigningSecretParameter.parameterName,

packages/cdk/resources/RuntimePolicies.ts

@@ -1,10 +1,6 @@
 import {Construct} from "constructs"
 import {PolicyStatement, ManagedPolicy} from "aws-cdk-lib/aws-iam"
-
-// 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"
+import {QUERY_REFORMULATION_MODEL_ID, RAG_MODEL_ID} from "./Constants"
 
 export interface RuntimePoliciesProps {
   readonly region: string

packages/slackBotFunction/app/services/query_reformulator.py

@@ -13,10 +13,10 @@
 
 def reformulate_query(user_query: str) -> str:
     """
-    Reformulate user query using Claude Haiku for better RAG retrieval.
+    Reformulate user query using Nova for better RAG retrieval.
 
     Loads prompt template from Bedrock Prompt Management, formats it with the user's
-    query, and uses Claude to generate a reformulated version optimized for vector search.
+    query, and uses Nova to generate a reformulated version optimized for vector search.
     """
     try:
         client: BedrockRuntimeClient = boto3.client("bedrock-runtime", region_name=os.environ["AWS_REGION"])