test-bot: use env vars for credentials when available
This commit is contained in:
@@ -13,6 +13,11 @@ 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.'
|
||||||
|
# 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')
|
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',
|
||||||
|
|||||||
Reference in New Issue
Block a user