fix(bedrock.py): fix boto3 config init

This commit is contained in:
Krrish Dholakia 2024-05-03 17:22:06 -07:00
parent 7715a9d333
commit defed7c08a
2 changed files with 53 additions and 63 deletions

View file

@ -598,6 +598,8 @@ def init_bedrock_client(
config = boto3.session.Config(
connect_timeout=timeout.connect, read_timeout=timeout.read
)
else:
config = boto3.session.Config()
### CHECK STS ###
if aws_role_name is not None and aws_session_name is not None: