fix: sts missing region

Author: yanbasic

src/emd/models/model.py

@@ -15,6 +15,7 @@
 )
 import boto3
 from .utils.text_utilities import normalize
+from emd.utils.aws_service_utils import get_current_region
 from emd.constants import (
     MODEL_STACK_NAME_PREFIX,
     MODEL_DEFAULT_TAG
@@ -374,15 +375,15 @@ def get_deploy_version_from_stack_name(cls,stack_name):
             raise ValueError(f"stack_name:{stack_name} is not a valid model stack name")
 
     def get_image_build_account_id(self):
-        current_account_id = boto3.client("sts").get_caller_identity()["Account"]
+        current_account_id = boto3.client("sts", region_name=get_current_region()).get_caller_identity()["Account"]
         build_image_account_id = (
             self.executable_config.current_engine.base_image_account_id or \
             current_account_id
         )
         return build_image_account_id
 
     def get_image_push_account_id(self):
-        current_account_id = boto3.client("sts").get_caller_identity()["Account"]
+        current_account_id = boto3.client("sts", region_name=get_current_region()).get_caller_identity()["Account"]
         return current_account_id
 
     def get_image_uri(

src/emd/utils/decorators.py

@@ -97,7 +97,7 @@ def inner(*args, **kwargs):
 @catch_aws_credential_errors
 def print_aws_profile():
     console = Console()
-    sts = boto3.client("sts")
+    sts = boto3.client("sts", region_name=get_current_region())
     response = sts.get_caller_identity()
     profile_name = os.environ.get("AWS_PROFILE","default")
     account_id = response["Account"]

src/pipeline/utils/aws_service_utils.py

@@ -6,6 +6,7 @@
 import json
 import hashlib
 import boto3
+from emd.utils.aws_service_utils import get_current_region
 logger = get_logger(__name__)
 
 
@@ -22,7 +23,7 @@ def check_aws_environment():
     """
     try:
         # Try to create a boto3 client and make a simple API call
-        sts = boto3.client('sts')
+        sts = boto3.client("sts", region_name=get_current_region())
         response = sts.get_caller_identity()
         logger.info("AWS environment is properly configured.")
         account_id = response['Account']

src/pipeline/utils/common.py

@@ -2,6 +2,7 @@
 import logging
 import argparse
 import boto3
+from emd.utils.aws_service_utils import get_current_region
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -54,7 +55,7 @@ def download_file_from_s3_by_s5cmd(s3_file_path, local_file_path):
 
 def upload_dir_to_s3(bucket_name, local_dir_name):
     logger.info(f"Uploading {local_dir_name} to {bucket_name} bucket")
-    s3 = boto3.client('s3')
+    s3 = boto3.client('s3', region_name=get_current_region())
     for root, dirs, files in os.walk(local_dir_name):
         for file in files:
             local_path = os.path.join(root, file)