Skip to main content
ClaudeWave
Skill78.6k repo starsupdated today

agent-runtime-hooks

Agent runtime lifecycle hooks enable observing and intercepting agent execution across 16 hook types spanning tool calls, human interventions, context compression, sub-agent calls, and step completion. Use these hooks to implement tool mocking, require human approval before actions, compress conversation context, trace execution flow, evaluate agent behavior, or call sub-agents within a parent agent's workflow.

Install in Claude Code
Copy
git clone --depth 1 https://github.com/lobehub/lobehub /tmp/agent-runtime-hooks && cp -r /tmp/agent-runtime-hooks/.agents/skills/agent-runtime-hooks ~/.claude/skills/agent-runtime-hooks
Then start a new Claude Code session; the skill loads automatically.

SKILL.md

# Agent Runtime Hooks

Lifecycle hooks for observing and intercepting agent execution. Hooks are registered per-operation via `execAgent({ hooks })` and dispatched by `HookDispatcher`.

## Hook Types

16 hook types across 5 categories:

```
execAgent({ hooks })
  │
  ├─ beforeStep ──────────── Before each step executes
  │     │
  │     ├─ [call_llm]        LLM inference
  │     │
  │     ├─ [call_tool]
  │     │     ├─ beforeToolCall ── Before tool executes (supports mocking)
  │     │     ├─ (tool execution)
  │     │     ├─ afterToolCall ─── After tool completes (observation only)
  │     │     └─ onToolCallError ─ Tool threw an exception
  │     │
  │     ├─ [request_human_approve]
  │     │     ├─ beforeHumanIntervention ── Before agent pauses
  │     │     ├─ afterHumanIntervention ─── After approve/reject + resume
  │     │     └─ onStopByHumanIntervention ── User rejected, agent halted
  │     │
  │     ├─ [compress_context]
  │     │     ├─ beforeCompact ──── Before compression starts
  │     │     ├─ afterCompact ───── After compression completes
  │     │     └─ onCompactError ─── Compression failed
  │     │
  │     ├─ [callAgent] (via execSubAgentTask)
  │     │     ├─ beforeCallAgent ── Before sub-agent starts
  │     │     ├─ afterCallAgent ─── After sub-agent completes
  │     │     └─ onCallAgentError ── Sub-agent failed
  │     │
  │     └─ afterStep ──────────── After step completes
  │
  ├─ (next step...)
  │
  ├─ onComplete ───────────── Operation reaches terminal state
  └─ onError ──────────────── Error during execution
```

## Key Files

| File                                                            | Role                                                   |
| --------------------------------------------------------------- | ------------------------------------------------------ |
| `packages/agent-runtime/src/types/hooks.ts`                     | Type definitions (AgentHookType, all event interfaces) |
| `apps/server/src/services/agentRuntime/hooks/types.ts`          | Server-side types (AgentHook, re-exports)              |
| `apps/server/src/services/agentRuntime/hooks/HookDispatcher.ts` | Registration, dispatch, dispatchBeforeToolCall         |
| `apps/server/src/modules/AgentRuntime/RuntimeExecutors.ts`      | Tool/Compact/HumanIntervention hook dispatch           |
| `apps/server/src/services/agentRuntime/AgentRuntimeService.ts`  | Step hooks + HumanIntervention resume/reject           |
| `apps/server/src/services/aiAgent/index.ts`                     | CallAgent hook dispatch                                |

## Registration Flow

```ts
const hooks: AgentHook[] = [
  { id: 'my-hook', type: 'afterStep', handler: async (event) => { ... } },
];
await aiAgentService.execAgent({ agentId, prompt, hooks });
// Internally: hookDispatcher.register(operationId, hooks)
// Cleanup:    hookDispatcher.unregister(operationId)
```

## Hook Reference

### Step Level

**`beforeStep`** — Before each step. `event: AgentHookEvent`
**`afterStep`** — After each step. `event: AgentHookEvent` (content, toolsCalling, totalCost, etc.)
**`onComplete`** — Terminal state. `event: AgentHookEvent` (reason: done/error/interrupted/max_steps/cost_limit)
**`onError`** — Error occurred. `event: AgentHookEvent` (errorMessage, errorDetail)

### Tool Call Level

**`beforeToolCall`** — Before tool executes. **Supports mocking** via `event.mock()`.

```ts
// event: ToolCallHookEvent
{
  (identifier, apiName, args, callIndex, stepIndex, operationId, mock);
}
// Mock example:
event.mock({ content: '{"error":"rate limited"}' });
```

Dispatch method: `hookDispatcher.dispatchBeforeToolCall()` (returns mock result or null).

**`afterToolCall`** — After tool completes. Observation only.

```ts
// event: AfterToolCallHookEvent
{
  (identifier, apiName, args, callIndex, content, success, mocked, executionTimeMs, stepIndex);
}
```

**`onToolCallError`** — Tool threw an exception (catch block, not just `success=false`).

```ts
// event: ToolCallErrorHookEvent
{
  (identifier, apiName, args, callIndex, error, stepIndex);
}
```

### Human Intervention

**`beforeHumanIntervention`** — Before agent pauses for approval.

```ts
// event: BeforeHumanInterventionHookEvent
{ operationId, stepIndex, pendingTools: [{ identifier, apiName }] }
```

**`afterHumanIntervention`** — After approve/reject, agent resumes.

```ts
// event: AfterHumanInterventionHookEvent
{ operationId, action: 'approve' | 'reject' | 'rejectAndContinue', toolCallId?, rejectionReason? }
```

**`onStopByHumanIntervention`** — User rejected, agent halted.

```ts
// event: StopByHumanInterventionHookEvent
{ operationId, toolCallId?, rejectionReason? }
```

### Context Compression

**`beforeCompact`** — Before compression starts.

```ts
// event: BeforeCompactHookEvent
{
  (operationId, stepIndex, messageCount, tokenCount);
}
```

**`afterCompact`** — After compression completes.

```ts
// event: AfterCompactHookEvent
{
  (operationId, stepIndex, groupId, messagesBefore, messagesAfter, summary);
}
```

**`onCompactError`** — Compression failed.

```ts
// event: CompactErrorHookEvent
{
  (operationId, stepIndex, tokenCount, error);
}
```

### Sub-Agent (CallAgent)

**`beforeCallAgent`** — Before calling sub-agent. Dispatched on **parent** operation.

```ts
// event: BeforeCallAgentHookEvent
{
  (operationId, agentId, instruction);
}
```

**`afterCallAgent`** — Sub-agent completed. Dispatched on **parent** operation.

```ts
// event: AfterCallAgentHookEvent
{
  (operationId, agentId, subOperationId, threadId, success);
}
```

**`onCallAgentError`** — Sub-agent failed. Dispatched on **parent** operation.

```ts
// event: CallAgentErrorHookEvent
{
  (operationId, agentId, error);
}
```

Note: CallAgent hooks require `parentOperationId` in `ExecSubAgentTaskParams`.

## Design Notes

- **Fire-and-forget**: All handlers return `Promise<void>`. Errors are non-fatal.
- **Exception**: `beforeToolCall` supports mock via `event.mock()` — uses `dispatchBeforeT