70 lines
2.3 KiB
Python
Executable File
70 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import sys
|
|
import json
|
|
|
|
|
|
def summarize(txt):
|
|
try:
|
|
obj = json.loads(txt)
|
|
except Exception:
|
|
return f"tool_result: {len(txt)} bytes"
|
|
data = obj.get("data", {}) or {}
|
|
msg = obj.get("message") or obj.get("status") or ""
|
|
# Common tool shapes
|
|
if "sha256" in str(data):
|
|
ln = data.get("lengthBytes") or data.get("length") or ""
|
|
return f"len={ln}".strip()
|
|
if "diagnostics" in data:
|
|
diags = data["diagnostics"] or []
|
|
w = sum(d.get("severity", "").lower() == "warning" for d in diags)
|
|
e = sum(d.get("severity", "").lower() in ("error", "fatal")
|
|
for d in diags)
|
|
ok = "OK" if not e else "FAIL"
|
|
return f"validate: {ok} (warnings={w}, errors={e})"
|
|
if "matches" in data:
|
|
m = data["matches"] or []
|
|
if m:
|
|
first = m[0]
|
|
return f"find_in_file: {len(m)} match(es) first@{first.get('line', 0)}:{first.get('col', 0)}"
|
|
return "find_in_file: 0 matches"
|
|
if "lines" in data: # console
|
|
lines = data["lines"] or []
|
|
lvls = {"info": 0, "warning": 0, "error": 0}
|
|
for L in lines:
|
|
lvls[L.get("level", "").lower()] = lvls.get(
|
|
L.get("level", "").lower(), 0)+1
|
|
return f"console: {len(lines)} lines (info={lvls.get('info', 0)},warn={lvls.get('warning', 0)},err={lvls.get('error', 0)})"
|
|
# Fallback: short status
|
|
return (msg or "tool_result")[:80]
|
|
|
|
|
|
def prune_message(msg):
|
|
if "content" not in msg:
|
|
return msg
|
|
newc = []
|
|
for c in msg["content"]:
|
|
if c.get("type") == "tool_result" and c.get("content"):
|
|
out = []
|
|
for chunk in c["content"]:
|
|
if chunk.get("type") == "text":
|
|
out.append(
|
|
{"type": "text", "text": summarize(chunk.get("text", ""))})
|
|
newc.append({"type": "tool_result", "tool_use_id": c.get(
|
|
"tool_use_id"), "content": out})
|
|
else:
|
|
newc.append(c)
|
|
msg["content"] = newc
|
|
return msg
|
|
|
|
|
|
def main():
|
|
convo = json.load(sys.stdin)
|
|
if isinstance(convo, dict) and "messages" in convo:
|
|
convo["messages"] = [prune_message(m) for m in convo["messages"]]
|
|
elif isinstance(convo, list):
|
|
convo = [prune_message(m) for m in convo]
|
|
json.dump(convo, sys.stdout, ensure_ascii=False)
|
|
|
|
|
|
main()
|