Add logging to streaming path in agent-runner

This commit is contained in:
daniel
2026-05-07 18:12:31 -05:00
parent ae5e0df884
commit 89d0819189

View File

@@ -228,6 +228,8 @@ def handler(event, context):
continue
# Extract text delta from contentBlockDelta
delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {})
if not isinstance(delta, dict):
continue
token = delta.get('text', '') or event.get('data', '')
if token:
text_buffer += token
@@ -236,11 +238,14 @@ def handler(event, context):
or len(text_buffer) > 800
)
if flush and text_buffer.strip():
print(f'[agent-runner] send chunk {len(text_buffer)}c to {chat_id}')
send_telegram_direct(str(chat_id), bot_token, text_buffer.strip())
text_buffer = ''
# Flush any remaining text
print(f'[agent-runner] stream done buffer={len(text_buffer)} bot_token_set={bool(bot_token)}')
if text_buffer.strip() and bot_token:
print(f'[agent-runner] flushing {len(text_buffer)}c to {chat_id}')
send_telegram_direct(str(chat_id), bot_token, text_buffer.strip())
print(f"[agent-runner] Completed session={session_id} actor={actor_id}")