Remove hardcoded region parameters from BDA blueprint classes

Author: Bob Strahan

lib/idp_common_pkg/idp_common/bda/bda_blueprint_creator.py

@@ -12,11 +12,9 @@
 
 
 class BDABlueprintCreator:
-    def __init__(self, region_name="us-west-2"):
+    def __init__(self):
         """Initialize Bedrock client."""
-        self.bedrock_client = boto3.client(
-            service_name="bedrock-data-automation", region_name=region_name
-        )
+        self.bedrock_client = boto3.client(service_name="bedrock-data-automation")
 
     def update_data_automation_project(self, projectArn: str, blueprint):
         """
@@ -209,7 +207,6 @@ def create_blueprint(self, document_type, blueprint_name, schema=None):
         Args:
             document_type (str): Type of document
             blueprint_name (str): Name for the blueprint
-            region (str): AWS region
             labels (list, optional): List of labels for the document
 
         Returns:

lib/idp_common_pkg/idp_common/bda/bda_blueprint_service.py

@@ -16,14 +16,9 @@
 
 
 class BdaBlueprintService:
-    def __init__(
-        self,
-        dataAutomationProjectArn: Optional[str] = None,
-        region: Optional[str] = "us-west-2",
-    ):
+    def __init__(self, dataAutomationProjectArn: Optional[str] = None):
         self.dataAutomationProjectArn = dataAutomationProjectArn
-        self.region = region or os.environ.get("AWS_REGION", "us-east-1")
-        self.blueprint_creator = BDABlueprintCreator(region_name=self.region)
+        self.blueprint_creator = BDABlueprintCreator()
         self.configuration_table_name = os.environ.get("CONFIGURATION_TABLE_NAME", "")
         dynamodb = boto3.resource("dynamodb")
         self.configuration_table = dynamodb.Table(self.configuration_table_name)

lib/idp_common_pkg/tests/unit/bda/test_bda_blueprint_service.py

@@ -136,8 +136,7 @@ def service(self):
             mock_boto_client.return_value = mock_bedrock_client
 
             service = BdaBlueprintService(
-                dataAutomationProjectArn="arn:aws:bedrock:us-west-2:123456789012:project/test-project",
-                region="us-west-2",
+                dataAutomationProjectArn="arn:aws:bedrock:us-west-2:123456789012:project/test-project"
             )
 
             # Store mocks for access in tests
@@ -164,42 +163,21 @@ def test_init(self):
             ),
         ):
             service = BdaBlueprintService(
-                dataAutomationProjectArn="arn:aws:bedrock:us-west-2:123456789012:project/test-project",
-                region="us-west-2",
+                dataAutomationProjectArn="arn:aws:bedrock:us-west-2:123456789012:project/test-project"
             )
 
             assert (
                 service.dataAutomationProjectArn
                 == "arn:aws:bedrock:us-west-2:123456789012:project/test-project"
             )
-            assert service.region == "us-west-2"
             assert service.configuration_table_name == "test-config-table"
 
             # Verify boto3 client was called for BDABlueprintCreator
-            mock_boto_client.assert_called_with(
-                service_name="bedrock-data-automation", region_name="us-west-2"
-            )
+            mock_boto_client.assert_called_with(service_name="bedrock-data-automation")
 
             # Verify DynamoDB table was set up
             mock_dynamodb.assert_called_once_with("dynamodb")
 
-    def test_init_with_default_region(self):
-        """Test initialization with default region from environment."""
-        with (
-            patch("boto3.resource"),
-            patch("boto3.client"),
-            patch.dict(
-                "os.environ",
-                {"AWS_REGION": "us-east-1", "CONFIGURATION_TABLE_NAME": "test-table"},
-            ),
-        ):
-            service = BdaBlueprintService(
-                dataAutomationProjectArn="arn:aws:bedrock:us-west-2:123456789012:project/test-project",
-                region=None,  # Explicitly pass None to trigger environment lookup
-            )
-
-            assert service.region == "us-east-1"
-
     def test_stringify_values(self, service):
         """Test the _stringify_values method."""
         # Test with mixed data types