test-bot: capture send_message tool calls in output
This commit is contained in:
11
test-bot.py
11
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()
|
||||
|
||||
Reference in New Issue
Block a user