test-bot: capture send_message tool calls in output

This commit is contained in:
daniel
2026-05-07 16:53:15 -05:00
parent d773985191
commit 04c0aeeb8a

View File

@@ -73,7 +73,7 @@ def main():
continue continue
if not isinstance(event, dict): if not isinstance(event, dict):
continue continue
# Text delta lives in event.contentBlockDelta.delta.text # Text delta in contentBlockDelta
delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {}) delta = event.get('event', {}).get('contentBlockDelta', {}).get('delta', {})
token = delta.get('text', '') or event.get('data', '') token = delta.get('text', '') or event.get('data', '')
if token: if token:
@@ -87,6 +87,15 @@ def main():
messages_sent.append(msg) messages_sent.append(msg)
print(f"\n[MSG {len(messages_sent)}] {msg}") print(f"\n[MSG {len(messages_sent)}] {msg}")
text_buffer = '' 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(): if text_buffer.strip():
msg = text_buffer.strip() msg = text_buffer.strip()