From 04c0aeeb8a9a50717c03361daac969c36719b8b2 Mon Sep 17 00:00:00 2001 From: daniel Date: Thu, 7 May 2026 16:53:15 -0500 Subject: [PATCH] test-bot: capture send_message tool calls in output --- test-bot.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/test-bot.py b/test-bot.py index d17166f..d638f53 100644 --- a/test-bot.py +++ b/test-bot.py @@ -73,7 +73,7 @@ def main(): continue if not isinstance(event, dict): continue - # Text delta lives in event.contentBlockDelta.delta.text + # Text delta in contentBlockDelta delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {}) token = delta.get('text', '') or event.get('data', '') if token: @@ -87,6 +87,15 @@ def main(): messages_sent.append(msg) print(f"\n[MSG {len(messages_sent)}] {msg}") text_buffer = '' + # send_message tool calls (captured from message events) + msg_ev = event.get('message', {}) + if isinstance(msg_ev, dict): + for block in msg_ev.get('content', []): + if isinstance(block, dict) and block.get('toolUse', {}).get('name') == 'send_message': + tool_text = block['toolUse'].get('input', {}).get('text', '') + if tool_text: + messages_sent.append(tool_text) + print(f"\n[TOOL MSG {len(messages_sent)}] {tool_text}") if text_buffer.strip(): msg = text_buffer.strip()