Skip to content

Commit f0e514f

Browse files
committed
simplify: MemoryControlPlaneClient uses boto3 endpoint resolution
Remove dependency on endpoints.py. Let boto3 resolve the endpoint natively (includes its own region validation). The BEDROCK_AGENTCORE_CONTROL_ENDPOINT env var override is still respected.
1 parent 7c02e0c commit f0e514f

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

src/bedrock_agentcore/memory/controlplane.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,6 @@
1313
import boto3
1414
from botocore.exceptions import ClientError
1515

16-
from bedrock_agentcore._utils.endpoints import _validate_endpoint_url
17-
1816
from .constants import (
1917
MemoryStatus,
2018
)
@@ -35,11 +33,10 @@ def __init__(self, region_name: str = "us-west-2", environment: str = "prod"):
3533
self.region_name = region_name
3634
self.environment = environment
3735

38-
control_endpoint = os.getenv("BEDROCK_AGENTCORE_CONTROL_ENDPOINT")
3936
service_name = os.getenv("BEDROCK_AGENTCORE_CONTROL_SERVICE", "bedrock-agentcore-control")
4037
cp_kwargs: dict = {"region_name": self.region_name}
38+
control_endpoint = os.getenv("BEDROCK_AGENTCORE_CONTROL_ENDPOINT")
4139
if control_endpoint:
42-
_validate_endpoint_url(control_endpoint)
4340
cp_kwargs["endpoint_url"] = control_endpoint
4441
self.client = boto3.client(service_name, **cp_kwargs)
4542

0 commit comments

Comments
 (0)