From f90171cb43fbb461de7c8060fdf0e6296d1cb4cc Mon Sep 17 00:00:00 2001
From: daniel <daniel@ness.family>
Date: Fri, 8 May 2026 09:38:20 -0500
Subject: [PATCH] test-bot: use env vars for credentials when available

---
 test-bot.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/test-bot.py b/test-bot.py
index d638f53..21b7996 100644
--- a/test-bot.py
+++ b/test-bot.py
@@ -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)