test-bot: use env vars for credentials when available

This commit is contained in:
daniel
2026-05-08 09:38:20 -05:00
parent c3432649c0
commit f90171cb43

View File

@@ -13,7 +13,12 @@ BOT_TOKEN_SECRET_ARN = 'arn:aws:secretsmanager:us-east-1:495395224548:secret:age
def main(): def main():
prompt = ' '.join(sys.argv[1:]) if len(sys.argv) > 1 else 'Say hello.' prompt = ' '.join(sys.argv[1:]) if len(sys.argv) > 1 else 'Say hello.'
session = boto3.Session(profile_name='ai1') # Use env vars if present (export-credentials), fall back to profile
import os
if os.environ.get('AWS_ACCESS_KEY_ID'):
session = boto3.Session()
else:
session = boto3.Session(profile_name='ai1')
client = session.client( client = session.client(
'bedrock-agentcore', region_name='us-east-1', 'bedrock-agentcore', region_name='us-east-1',
config=Config(read_timeout=600, connect_timeout=10) config=Config(read_timeout=600, connect_timeout=10)