test-bot: use env vars for credentials when available
This commit is contained in:
@@ -13,7 +13,12 @@ BOT_TOKEN_SECRET_ARN = 'arn:aws:secretsmanager:us-east-1:495395224548:secret:age
|
||||
|
||||
def main():
|
||||
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(
|
||||
'bedrock-agentcore', region_name='us-east-1',
|
||||
config=Config(read_timeout=600, connect_timeout=10)
|
||||
|
||||
Reference in New Issue
Block a user