91 lines
4.1 KiB
Python
91 lines
4.1 KiB
Python
"""Extrahiert Subagenten-Aufrufe (Prompt, Typ, Ergebnislaenge) aus dem persistierten
|
|
Session-Transkript eines Headless-Laufs und legt sie im Laufverzeichnis unter _meta ab.
|
|
|
|
Aufruf: python subagenten.py <laufverzeichnis> [<transkript-wurzel>]
|
|
Die Session-ID wird aus <laufverzeichnis>/RawResult.json gelesen.
|
|
"""
|
|
import io, json, os, sys, glob
|
|
|
|
lauf = sys.argv[1]
|
|
wurzel = sys.argv[2] if len(sys.argv) > 2 else os.path.expandvars(r'%USERPROFILE%\.claude\projects')
|
|
|
|
roh = json.load(io.open(os.path.join(lauf, 'RawResult.json'), encoding='utf-8-sig'))
|
|
sid = roh.get('session_id')
|
|
stats = roh.get('subagent_stats') or {}
|
|
gesamt = stats.get('spawned', 0)
|
|
verschachtelt = stats.get('spawned_by_subagents', 0)
|
|
# Im Haupttranskript stehen nur die direkt vom Hauptagenten gestarteten Subagenten.
|
|
# Von Subagenten gestartete liegen in deren eigenen Transkripten.
|
|
erwartet = gesamt - verschachtelt
|
|
|
|
treffer = glob.glob(os.path.join(wurzel, '*', sid + '.jsonl'))
|
|
if not treffer:
|
|
print('KEIN TRANSKRIPT gefunden fuer Session', sid)
|
|
sys.exit(1)
|
|
pfad = treffer[0]
|
|
|
|
aufrufe, ergebnisse = [], {}
|
|
for zeile in io.open(pfad, encoding='utf-8'):
|
|
try:
|
|
o = json.loads(zeile)
|
|
except Exception:
|
|
continue
|
|
cont = (o.get('message') or {}).get('content')
|
|
if not isinstance(cont, list):
|
|
continue
|
|
for c in cont:
|
|
if not isinstance(c, dict):
|
|
continue
|
|
if c.get('type') == 'tool_use' and c.get('name') in ('Task', 'Agent'):
|
|
ein = c.get('input') or {}
|
|
aufrufe.append({
|
|
'id': c.get('id'),
|
|
'werkzeug': c.get('name'),
|
|
'subagent_type': ein.get('subagent_type'),
|
|
'description': ein.get('description'),
|
|
'run_in_background': ein.get('run_in_background'),
|
|
'model': ein.get('model'),
|
|
'prompt': ein.get('prompt') or '',
|
|
})
|
|
elif c.get('type') == 'tool_result':
|
|
inhalt = c.get('content')
|
|
if isinstance(inhalt, list):
|
|
inhalt = ' '.join(str(x.get('text', '')) for x in inhalt if isinstance(x, dict))
|
|
ergebnisse[c.get('tool_use_id')] = len(str(inhalt or ''))
|
|
|
|
for a in aufrufe:
|
|
a['ergebnis_zeichen'] = ergebnisse.get(a['id'])
|
|
|
|
meta = os.path.join(lauf, '_meta')
|
|
os.makedirs(meta, exist_ok=True)
|
|
io.open(os.path.join(meta, 'subagenten.json'), 'w', encoding='utf-8').write(
|
|
json.dumps(aufrufe, indent=1, ensure_ascii=False))
|
|
|
|
md = ['# Subagenten-Aufrufe', '',
|
|
'Session `%s`, Transkript `%s`.' % (sid, os.path.basename(pfad)),
|
|
'',
|
|
'`subagent_stats`: **%s** Subagenten gesamt, davon **%s** von Subagenten gestartet '
|
|
'(max_depth %s). Direkt vom Hauptagenten erwartet: **%s**. Im Transkript gefunden: **%s**.'
|
|
% (gesamt, verschachtelt, stats.get('max_depth'), erwartet, len(aufrufe)), '']
|
|
if verschachtelt:
|
|
md += ['> Die %s von Subagenten gestarteten Aufrufe stehen in deren eigenen Transkripten und'
|
|
' sind hier **nicht** enthalten.' % verschachtelt, '']
|
|
if erwartet != len(aufrufe):
|
|
md += ['> **Abweichung** zwischen erwarteter und gefundener Anzahl.',
|
|
'> Ursache pruefen, bevor die Prompts ausgewertet werden.', '']
|
|
for i, a in enumerate(aufrufe, 1):
|
|
md += ['## %d. %s' % (i, a['description'] or '(ohne Beschreibung)'),
|
|
'',
|
|
'- **Werkzeug:** `%s` **Typ:** `%s` **Hintergrund:** %s'
|
|
% (a['werkzeug'], a['subagent_type'], a['run_in_background']),
|
|
'- **Prompt-Zeichen:** %d **Ergebnis-Zeichen:** %s'
|
|
% (len(a['prompt']), a['ergebnis_zeichen']),
|
|
'', '### Prompt', '', '```', a['prompt'].rstrip(), '```', '']
|
|
io.open(os.path.join(meta, 'subagenten.md'), 'w', encoding='utf-8').write('\n'.join(md))
|
|
|
|
print('Subagenten gefunden: %d (direkt erwartet: %s, gesamt: %s, davon verschachtelt: %s)' % (len(aufrufe), erwartet, gesamt, verschachtelt))
|
|
for a in aufrufe:
|
|
print(' - %-42s %s Prompt %6d Z. Ergebnis %s Z.'
|
|
% ((a['description'] or '')[:42], a['subagent_type'], len(a['prompt']), a['ergebnis_zeichen']))
|
|
print('geschrieben:', os.path.join(meta, 'subagenten.md'))
|