Search Traces

1"""Search for a keyword across all event properties in a trace."""
2
3import json
4import os
5import sys
6
7
8def load_trace_file(path):
9    with open(path) as f:
10        raw = json.load(f)
11    # Claude Code persists large MCP tool results as [{"type": "text", "text": "<json>"}] — unwrap to get the actual trace data.
12    if isinstance(raw, list) and raw and raw[0].get("type") == "text":
13        raw = json.loads(raw[0]["text"])
14    # Both query-llm-trace and query-llm-traces-list return {"results": [...]}, but handle a bare trace object too.
15    results = raw.get("results", raw)
16    return [results] if isinstance(results, dict) else results
17
18
19def search_obj(obj, term, path=""):
20    if isinstance(obj, str):
21        if term in obj.lower():
22            idx = obj.lower().index(term)
23            start, end = max(0, idx - 80), min(len(obj), idx + len(term) + 80)
24            yield path, obj[start:end]
25    elif isinstance(obj, dict):
26        for k, v in obj.items():
27            yield from search_obj(v, term, f"{path}.{k}")
28    elif isinstance(obj, list):
29        for i, v in enumerate(obj):
30            yield from search_obj(v, term, f"{path}[{i}]")
31
32
33term = os.environ.get("SEARCH", "").lower()
34if not term:
35    print("Usage: SEARCH='keyword' python3 search.py file.json", file=sys.stderr)
36    sys.exit(1)
37
38traces = load_trace_file(sys.argv[1])
39for trace in traces:
40    for ev in trace.get("events", []):
41        p = ev.get("properties", {})
42        name = p.get("$ai_span_name", p.get("$ai_model", ev.get("event", "?")))
43        for path, snippet in search_obj(p, term):
44            print(f"\n[{ev.get('createdAt', '?')}] {name} -> {path}")
45            print(f"  ...{snippet}...")