fix(bedrock): fix region inference when aws_profile is used

Author: Ashutosh0x

src/anthropic/lib/bedrock/_client.py

@@ -69,20 +69,20 @@ def _prepare_options(input_options: FinalRequestOptions) -> FinalRequestOptions:
     return options
 
 
-def _infer_region(aws_session: boto3.Session | None = None) -> str:
+def _infer_region(aws_session: boto3.Session | None = None, aws_profile: str | None = None) -> str:
     """
     Infer the AWS region from the environment variables or
     from the boto3 session if available.
     """
     if aws_session is not None and aws_session.region_name:
         return aws_session.region_name
 
-    aws_region = os.environ.get("AWS_REGION")
+    aws_region = os.environ.get("AWS_REGION") or os.environ.get("AWS_DEFAULT_REGION")
     if aws_region is None:
         try:
             import boto3
 
-            session = boto3.Session()
+            session = boto3.Session(profile_name=aws_profile)
             if session.region_name:
                 aws_region = session.region_name
         except ImportError:
@@ -168,7 +168,7 @@ def __init__(
         self.aws_access_key = aws_access_key
 
         self.aws_session = aws_session
-        self.aws_region = aws_region or _infer_region(aws_session)
+        self.aws_region = aws_region or _infer_region(aws_session, aws_profile=aws_profile)
         self.aws_profile = aws_profile
 
         self.aws_session_token = aws_session_token
@@ -322,7 +322,7 @@ def __init__(
         self.aws_access_key = aws_access_key
 
         self.aws_session = aws_session
-        self.aws_region = aws_region or _infer_region(aws_session)
+        self.aws_region = aws_region or _infer_region(aws_session, aws_profile=aws_profile)
         self.aws_profile = aws_profile
 
         self.aws_session_token = aws_session_token