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
|
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()
|
||||||
|
|||||||
Reference in New Issue
Block a user