45555bc4aa
User pushback: live-graph showed only "main" node, no pulses on agents.
Root cause: hook stdin doesn't carry parent_tool_use_id for sub-agent
tool calls — we only get the sub-agent's own session_id, which doesn't
link back to the spawn's tool_use_id.
Sequential heuristic via shared state file:
- agent-event-spawn.sh appends tool_use_id to /tmp/kei-active-children.tsv
- tool-use-event.sh reads the LAST line of that file → uses that
tool_use_id as agent_id for the emitted event
- agent-event-done.sh removes the spawn's line (grep -v + atomic mv)
Verified end-to-end: a code-implementer agent ran 5 Bash calls during
its lifetime — all 5 tool_use events were correctly attributed to the
spawn's tool_use_id. After agent_done, subsequent orchestrator-direct
tool calls correctly fall back to agent_id="main".
Limitation: parallel agents may misattribute. The "most recent live
spawn" heuristic works for single-agent-at-a-time which is the common
case. Parallel spawns share /tmp/kei-active-children.tsv and a sub-
agent's tool calls all attribute to whichever spawn appended last.
Acceptable for v1 demo; proper parent-tool-use-id propagation requires
Claude Code to expose it in sub-agent stdin (upstream change).
The `mv` after `grep -v` runs UNCONDITIONALLY (not gated on grep's
exit code) — grep -v returns 1 when ALL lines match, which would
otherwise leave the stale file in place.
Bypass: `KEI_EVENTS_BYPASS=1` (existing) covers all 3 hooks.
Override path: `KEI_ACTIVE_SPAWNS_FILE=/path/to/file`.
=== STATUS-TRUTH MARKER ===
shipped: functional
stubs: 0
cargo-check: NOT-RUN
behaviour-verified: yes
follow-up-required:
- Parallel-agent attribution would need parent_tool_use_id from
Claude Code sub-agent stdin (not currently exposed).
- Race condition window between spawn append and done remove is
millisecond-scale; observed clean in single-agent demo.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
53 lines
2 KiB
Bash
Executable file
53 lines
2 KiB
Bash
Executable file
#!/bin/sh
|
|
# tool-use-event.sh — PreToolUse hook for Bash/Read/Edit/Write/Grep/Glob/NotebookEdit.
|
|
#
|
|
# Emits `tool_use` event to ~/.claude/memory/agent-events.jsonl.
|
|
# Attributes the call to the parent agent via /tmp/kei-active-children.tsv
|
|
# (most-recent-spawn heuristic) so the live-graph viewer can pulse the
|
|
# correct node.
|
|
#
|
|
# Agent tools (spawns) are excluded — handled by agent-event-spawn.sh.
|
|
# Defensive: never blocks, exits 0 on every path.
|
|
# Bypass via `KEI_EVENTS_BYPASS=1`.
|
|
set -u
|
|
|
|
[ "${KEI_EVENTS_BYPASS:-0}" = "1" ] && exit 0
|
|
command -v jq >/dev/null 2>&1 || exit 0
|
|
|
|
PAYLOAD=$(cat 2>/dev/null || true)
|
|
[ -n "$PAYLOAD" ] || exit 0
|
|
|
|
TOOL=$(printf '%s' "$PAYLOAD" | jq -r '.tool_name // empty' 2>/dev/null)
|
|
case "$TOOL" in
|
|
Bash|Read|Edit|Write|Grep|Glob|NotebookEdit) ;;
|
|
*) exit 0 ;;
|
|
esac
|
|
|
|
EVENTS_FILE="$HOME/.claude/memory/agent-events.jsonl"
|
|
mkdir -p "$(dirname "$EVENTS_FILE")" 2>/dev/null || true
|
|
|
|
TOOL_USE_ID=$(printf '%s' "$PAYLOAD" | jq -r '.tool_use_id // .toolUseId // "unknown"' 2>/dev/null)
|
|
|
|
# Parent agent attribution. Claude Code stdin carries session_id of WHOEVER
|
|
# is running (orchestrator OR sub-agent), but does NOT give parent spawn's
|
|
# tool_use_id. We consult the active-spawns ledger written by
|
|
# agent-event-spawn.sh / removed by agent-event-done.sh:
|
|
# - non-empty file → attribute to the MOST RECENT live spawn
|
|
# (sequential heuristic — works for single-agent-at-a-time)
|
|
# - empty file → fall back to "main" (orchestrator)
|
|
ACTIVE_FILE="${KEI_ACTIVE_SPAWNS_FILE:-/tmp/kei-active-children.tsv}"
|
|
AGENT_ID="main"
|
|
if [ -s "$ACTIVE_FILE" ]; then
|
|
LAST_SPAWN=$(tail -1 "$ACTIVE_FILE" 2>/dev/null | awk '{print $2}')
|
|
[ -n "$LAST_SPAWN" ] && AGENT_ID="$LAST_SPAWN"
|
|
fi
|
|
|
|
jq -cn \
|
|
--arg ts "$(date -u +%Y-%m-%dT%H:%M:%S.000Z 2>/dev/null)" \
|
|
--arg id "$TOOL_USE_ID" \
|
|
--arg agent_id "$AGENT_ID" \
|
|
--arg tool "$TOOL" \
|
|
'{ts:$ts,event:"tool_use",id:$id,agent_id:$agent_id,tool:$tool}' \
|
|
>> "$EVENTS_FILE" 2>/dev/null || true
|
|
|
|
exit 0
|