From 2fe0a2750bc26272ef41e6096551aa20a9528d79 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 3 Jun 2025 20:58:09 -0700
Subject: [PATCH] test: ensure aws region correctly set

---
 .../logging_callback_tests/test_bedrock_knowledgebase_hook.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py b/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py
index 96efbea7ad..a8afe9ebba 100644
--- a/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py
+++ b/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py
@@ -34,6 +34,10 @@ class TestCustomLogger(CustomLogger):
         self.standard_logging_payload = kwargs.get("standard_logging_object")
         pass
 
+@pytest.fixture(autouse=True)
+def add_aws_region_to_env(monkeypatch):
+    monkeypatch.setenv("AWS_REGION", "us-west-2")
+
 
 @pytest.fixture
 def setup_vector_store_registry():