Skip to content

Commit 8c21282

Browse files
authored
Generalize SQS region (#355)
* Generalize SQS region * Fix lint * Fix hmi_config * Remove DD change
1 parent d29faff commit 8c21282

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

model-engine/model_engine_server/infra/gateways/resources/k8s_resource_types.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -496,9 +496,9 @@ def get_endpoint_resource_arguments_from_request(
496496

497497
change_cause_message = (
498498
f"Deployment at {datetime.utcnow()} UTC. "
499-
f"Using deployment constructed from model bundle ID: {model_bundle.id}, "
500-
f"model bundle name: {model_bundle.name}, "
501-
f"endpoint ID: {model_endpoint_record.id}"
499+
f"Using deployment constructed from model bundle ID {model_bundle.id}, "
500+
f"model bundle name {model_bundle.name}, "
501+
f"endpoint ID {model_endpoint_record.id}"
502502
)
503503

504504
priority = LAUNCH_DEFAULT_PRIORITY_CLASS

model-engine/model_engine_server/infra/gateways/resources/live_sqs_endpoint_resource_delegate.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from aiobotocore.client import AioBaseClient
88
from model_engine_server.common.config import hmi_config
99
from model_engine_server.core.aws.roles import session
10+
from model_engine_server.core.config import infra_config
1011
from model_engine_server.core.loggers import logger_name, make_logger
1112
from model_engine_server.domain.exceptions import EndpointResourceInfraException
1213
from model_engine_server.infra.gateways.resources.sqs_endpoint_resource_delegate import (
@@ -21,7 +22,9 @@
2122

2223

2324
def _create_async_sqs_client(sqs_profile: Optional[str]) -> AioBaseClient:
24-
return session(role=sqs_profile, session_type=AioSession).client("sqs", region_name="us-west-2")
25+
return session(role=sqs_profile, session_type=AioSession).client(
26+
"sqs", region_name=infra_config().default_region
27+
)
2528

2629

2730
def _get_queue_policy(queue_name: str) -> str:

0 commit comments

Comments
 (0)