Update query_reformulator test suite

kris-szlapa · kris-szlapa · commit 4bb466b8da7f · 2025-08-18T16:38:38.000Z
diff --git a/packages/slackBotFunction/tests/test_query_reformulator.py b/packages/slackBotFunction/tests/test_query_reformulator.py
@@ -8,7 +8,7 @@
     "os.environ",
     {
         "AWS_REGION": "eu-west-2",
-        "QUERY_REFORMULATION_PROMPT_ARN": "arn:aws:bedrock:eu-west-2:123456789012:prompt/test-prompt",
+        "QUERY_REFORMULATION_MODEL_ID": "anthropic.claude-3-haiku-20240307-v1:0",
     },
 )
 def test_reformulate_query_success(mock_boto_client):
@@ -34,7 +34,7 @@ def test_reformulate_query_success(mock_boto_client):
     "os.environ",
     {
         "AWS_REGION": "eu-west-2",
-        "QUERY_REFORMULATION_PROMPT_ARN": "arn:aws:bedrock:eu-west-2:123456789012:prompt/test-prompt",
+        "QUERY_REFORMULATION_MODEL_ID": "anthropic.claude-3-haiku-20240307-v1:0",
     },
 )
 def test_reformulate_query_fallback_on_error(mock_boto_client):

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@`
`8`	`8`	`"os.environ",`
`9`	`9`	`{`
`10`	`10`	`"AWS_REGION": "eu-west-2",`
`11`		`- "QUERY_REFORMULATION_PROMPT_ARN": "arn:aws:bedrock:eu-west-2:123456789012:prompt/test-prompt",`
	`11`	`+ "QUERY_REFORMULATION_MODEL_ID": "anthropic.claude-3-haiku-20240307-v1:0",`
`12`	`12`	`},`
`13`	`13`	`)`
`14`	`14`	`def test_reformulate_query_success(mock_boto_client):`
`@@ -34,7 +34,7 @@ def test_reformulate_query_success(mock_boto_client):`
`34`	`34`	`"os.environ",`
`35`	`35`	`{`
`36`	`36`	`"AWS_REGION": "eu-west-2",`
`37`		`- "QUERY_REFORMULATION_PROMPT_ARN": "arn:aws:bedrock:eu-west-2:123456789012:prompt/test-prompt",`
	`37`	`+ "QUERY_REFORMULATION_MODEL_ID": "anthropic.claude-3-haiku-20240307-v1:0",`
`38`	`38`	`},`
`39`	`39`	`)`
`40`	`40`	`def test_reformulate_query_fallback_on_error(mock_boto_client):`