Agentpprof Flamegraph

agentpprof Flamegraph Generation

Goal

Generate meaningful flamegraphs from local Codex/Claude Code sessions by iteratively developing tag rules that achieve high prompt coverage.

Workflow

1. Initial Discovery

Run agentpprof without rules to see diagnostics:

agentpprof \
  --project-root /path/to/project \
  --view tokens \
  -o initial.json \
  --format json \
  --include-previews

The output includes:

• tagging.total_prompts: total prompts found
• tagging.unmatched_prompts: prompts without tags
• tagging.unmatched_samples: sample unmatched prompts (up to 20)
• tagging.hint: suggested next step

2. Analyze Unmatched Prompts

Look at unmatched_samples to identify patterns:

• Common keywords or phrases
• Chinese/English patterns
• Action types (review, debug, git, etc.)
• Project-specific terminology

3. Develop Tag Rules

Add --tag-rule arguments iteratively:

agentpprof \
  --project-root /path/to/project \
  --tag-rule 'prompt:review=(?i)review|审核|check' \
  --tag-rule 'prompt:debug=(?i)fix|bug|error|broken' \
  --tag-rule 'prompt:git=(?i)commit|push|pull|git' \
  --view tokens \
  -o iter1.folded

Rule syntax: KIND:TAG=REGEX

• KIND: prompt, session, llm, or all
• TAG: lowercase word, 3-12 letters (semantic, not vague)
• REGEX: case-insensitive patterns with (?i)

Avoid vague tags like task, work, misc, thing, stuff, other — they don't convey semantic meaning and won't aggregate well. Use specific tags like debug, review, paper, naming that describe the activity.

Never use catch-all rules like prompt:misc=. or llm:other=. — they defeat the purpose of semantic tagging by lumping everything together. If you can't classify an item, leave it unmatched and add more specific rules.

Never use placeholder tags like llm:placeholder, llm:response, prompt:other — they indicate that the tagging rules are incomplete. If you see placeholder tags dominating the distribution, investigate why the content isn't being classified properly. Common causes:

• Parser limitations (e.g., "claude response" preview means the actual response content wasn't extracted)
• Rules ordered incorrectly (more specific rules should come before general ones)
• Missing rules for common patterns

4. Check Coverage

Each run shows diagnostics and warnings:

Warning: 150/1000 prompts unmatched. Add prompt tag rules.

Check detailed coverage (tagging diagnostics are on stdout status JSON, not the profile file):

agentpprof --project-root . -o out.json --format json | jq '.tagging'

Definition of "well-tagged":

• Target: All three categories must have < 5% unmatched
  • prompts.unmatched / prompts.total < 5%
  • sessions.unmatched / sessions.total < 5%
  • llm_calls.unmatched / llm_calls.total < 5%

Distribution quality metrics:

• Tag count: 10-20 categories ideal (<5 too coarse, >25 too fragmented)
• Top-1 share: <40% (no single tag should dominate)
• Top-3 share: <70% (diversity across categories)
• Entropy: >0.7 normalized (evenly distributed)

The tool prints detailed distribution analysis:

Distribution (12 prompt tags, 2620 total): top1=35.5%, top3=58.2%, top5=72.1%, entropy=0.78
  Top tags:
    1. prompt:review = 931 (35.5%)
    2. prompt:query = 228 (8.7%)
    3. prompt:discuss = 173 (6.6%)
    4. prompt:edit = 156 (6.0%)
    5. prompt:code = 194 (7.4%)

Warnings are shown if metrics are poor:

  Warning: prompt:misc dominates (55.2%). Target: top1 < 40%. Split into sub-categories.
  Warning: low entropy (0.45). Distribution is uneven. Target: entropy > 0.7.

Spot-check unmatched samples:

agentpprof --project-root . -o out.json --format json \
  | jq '.tagging.unmatched_samples | map(select(.kind == "prompt")) | .[0:10]'

If unmatched prompts share patterns, add rules. Continue iterating until ALL categories have < 5% unmatched. Avoid vague catch-all tags like misc — use specific semantic tags that describe the activity.

5. Generate Final Flamegraphs

for view in tokens files network time operations; do
  agentpprof \
    --project-root /path/to/project \
    "${TAG_RULES[@]}" \
    --view "$view" \
    --svg-width 1200 \
    -o "project-${view}.svg"
done

SVG width: Use --svg-width to adjust flamegraph width (default: 1200px). Narrower widths (800-1000) improve readability for deep flamegraphs; wider (1600-2000) for shallow ones with many tags.

Output formats still work via --format or extension: .pb.gz / .pb (pprof), .folded, .svg, .json. Keep -o/--output required; use --include-previews only on sanitized sessions when writing JSON.

Views

Default stack frames: task,skill,phase,action,object,repeat,result,outcome (plus token for the tokens view). project/agent/session are pprof sample labels (go tool pprof -tags), not default stack frames.

Operation Stack Controls

# Rewrite fields before stacking, filter the subset, then choose frames.
# Use --view operations (or files/network): --op-map patterns that match
# tool fields (cmd/effect) do not apply to tokens samples, which are LLM-only.
agentpprof \
  --project-root /path/to/project \
  --op-map 'task:verify=(?i)cmd=cargo|effect=test' \
  --where 'task=verify' \
  --stack task,action,result,object \
  --stack-rule 'action:run_tests=(?i)cmd=cargo' \
  --view operations \
  -o verify.folded --format folded

# Optional closed-set LLM task taxonomy (requires --tagger llm)
agentpprof \
  --project-root /path/to/project \
  --tagger llm \
  --task-choice 'debug=fix bugs and test failures' \
  --task-choice 'review=code review and diffs' \
  -o declared.pb.gz

• --op-map / --op-map-file: derive or overwrite operation fields (first match wins per field)
• --where FIELD=REGEX or FIELD!=REGEX: AND filters after mapping
• --stack / --stack-rule: choose and rewrite flamegraph frames
• --task-choice TAG=DESCRIPTION: additional closed-set session.task_tag (does not replace free-form tags)

Common Tag Patterns

# Paper writing
--tag-rule 'prompt:paper=(?i)paper|arxiv|latex|abstract|intro|section'

# Code review
--tag-rule 'prompt:review=(?i)review|审核|check|diff|pr'

# Git operations  
--tag-rule 'prompt:git=(?i)commit|push|pull|git|merge|rebase'

# Debugging
--tag-rule 'prompt:debug=(?i)fix|bug|error|broken|为啥|failed'

# Testing
--tag-rule 'prompt:test=(?i)test|cargo test|pytest|verify'

# Formatting/style
--tag-rule 'prompt:format=(?i)格式|style|format|font|图'

# Confirmations (short responses)
--tag-rule 'prompt:confirm=(?i)^嗯$|^是$|^好$|^ok$'

# Context continuations
--tag-rule 'prompt:context=(?i)session is being continued'

# Subagent delegations
--tag-rule 'prompt:delegate=(?i)subagent|task-notification'

Explicit Session Files

For repeatable analysis, use --session-file instead of --project-root:

agentpprof \
  --session-file ~/.claude/projects/.../session1.jsonl \
  --session-file ~/.claude/projects/.../session2.jsonl \
  --project-name my-project \
  "${TAG_RULES[@]}" \
  --view tokens \
  -o output.svg

What Can You Learn?

From tokens view: Which activities consumed the most LLM budget, cache effectiveness, unexpectedly expensive sessions.

From time view: Wall-clock time distribution, longest prompts, workflow bottlenecks.

From files view: Codebase access patterns, security audit for unexpected file access.

From network view: External service contacts, process chains, security audit for domain access.

Notes

• --preset enables built-in keyword rules for quick testing, but they are generic and unlikely to match your project well
• Without --tag-rule or --preset, all prompts are marked unmatched
• Flamegraphs require semantic tags to aggregate meaningfully
• Warnings are printed for any unmatched items — address them by adding rules
• Iterate on rules until coverage is acceptable, then save the command as a script

Example Script

See docs/flamegraph-example/agentsight.sh for a complete example with AgentSight's own development traces.