mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-11-10 21:47:02 +01:00
Krzysztof kuhy Rudnicki
c287f59621
- Subagents that outlived /clear land in a new session's subagents/ dir with no meta: borrow the spawning session's agent-<id>.meta.json, else the first prompt line. Clears the $35.93 '?' row; no launch lacked a description. - bash_tokens_per_call rose 41% with Bash tokens/day flat (cheap calls fell 28%). Compare bash_tokens_per_day and bash_big_result_share (results >2k) instead; the Large-Bash lever keys on the share. Tool counters grouped in ToolPayloads. Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01TjyHmRrjZssaxXDWcGuCeQ
125 lines
4.8 KiB
Python
125 lines
4.8 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from python_pkg.token_audit import discover
|
|
|
|
|
|
def _touch(path: Path, mtime: float, text: str = "{}") -> Path:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
path.write_text(text, encoding="utf-8")
|
|
os.utime(path, (mtime, mtime))
|
|
return path
|
|
|
|
|
|
def test_find_transcripts_prunes_by_mtime_newest_first(tmp_path: Path) -> None:
|
|
old = _touch(tmp_path / "p" / "old.jsonl", 1000)
|
|
mid = _touch(tmp_path / "p" / "mid.jsonl", 20_000)
|
|
new = _touch(tmp_path / "q" / "new.jsonl", 50_000)
|
|
assert discover.find_transcripts(tmp_path, since=10_000) == [new, mid]
|
|
assert discover.find_transcripts(tmp_path, since=0) == [new, mid, old]
|
|
|
|
|
|
def test_find_transcripts_includes_subagent_layout(tmp_path: Path) -> None:
|
|
main = _touch(tmp_path / "p" / "s.jsonl", 100)
|
|
sub = _touch(tmp_path / "p" / "s" / "subagents" / "agent-1.jsonl", 200)
|
|
assert discover.find_transcripts(tmp_path, since=0) == [sub, main]
|
|
|
|
|
|
def test_first_cwd_skips_junk_and_cwdless_records(tmp_path: Path) -> None:
|
|
lines = ["not json", "[1]", '{"a": 1}', '{"cwd": 5}', '{"cwd": "/x"}']
|
|
path = _touch(tmp_path / "t.jsonl", 1, "\n".join(lines))
|
|
assert discover.first_cwd(path) == "/x"
|
|
|
|
|
|
def test_first_cwd_none_when_absent(tmp_path: Path) -> None:
|
|
path = _touch(tmp_path / "t.jsonl", 1, '{"a": 1}')
|
|
assert discover.first_cwd(path) is None
|
|
|
|
|
|
def test_agent_meta_reads_sibling(tmp_path: Path) -> None:
|
|
path = _touch(tmp_path / "agent-1.jsonl", 1)
|
|
_touch(tmp_path / "agent-1.meta.json", 1, '{"agentType": "Explore"}')
|
|
assert discover.agent_meta(path) == {"agentType": "Explore"}
|
|
|
|
|
|
def test_agent_meta_empty_when_missing_or_bad(tmp_path: Path) -> None:
|
|
path = _touch(tmp_path / "agent-1.jsonl", 1)
|
|
assert discover.agent_meta(path) == {}
|
|
_touch(tmp_path / "agent-1.meta.json", 1, "garbage")
|
|
assert discover.agent_meta(path) == {}
|
|
|
|
|
|
def _meta(path: Path, meta: dict[str, object] | str) -> None:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
text = meta if isinstance(meta, str) else json.dumps(meta)
|
|
path.write_text(text, encoding="utf-8")
|
|
|
|
|
|
def test_agent_meta_prefers_own_description(tmp_path: Path) -> None:
|
|
own = tmp_path / "p" / "s" / "subagents" / "agent-1.meta.json"
|
|
_meta(own, {"agentType": "Plan", "description": "mine"})
|
|
_meta(
|
|
tmp_path / "p" / "o" / "subagents" / "agent-1.meta.json", {"description": "x"}
|
|
)
|
|
path = own.parent / "agent-1.jsonl"
|
|
assert discover.agent_meta(path) == {"agentType": "Plan", "description": "mine"}
|
|
|
|
|
|
def test_agent_meta_borrows_missing_fields(tmp_path: Path) -> None:
|
|
sub = tmp_path / "p" / "new" / "subagents"
|
|
_meta(sub / "agent-2.meta.json", {"agentType": "fork", "spawnDepth": 0})
|
|
_meta(tmp_path / "p" / "bad" / "subagents" / "agent-2.meta.json", "not json")
|
|
_meta(tmp_path / "p" / "nod" / "subagents" / "agent-2.meta.json", {"x": 1})
|
|
_meta(
|
|
tmp_path / "q" / "old" / "subagents" / "agent-2.meta.json",
|
|
{"agentType": "general-purpose", "description": "spawned"},
|
|
)
|
|
assert discover.agent_meta(sub / "agent-2.jsonl") == {
|
|
"agentType": "fork",
|
|
"description": "spawned",
|
|
}
|
|
|
|
|
|
def test_agent_meta_empty_when_nothing_found(tmp_path: Path) -> None:
|
|
path = tmp_path / "p" / "s" / "subagents" / "agent-3.jsonl"
|
|
path.parent.mkdir(parents=True)
|
|
assert discover.agent_meta(path) == {}
|
|
|
|
|
|
def test_first_prompt_skips_non_prompts_and_truncates(tmp_path: Path) -> None:
|
|
rows = [
|
|
"junk",
|
|
json.dumps({"type": "assistant", "message": {"content": "no"}}),
|
|
json.dumps({"type": "user", "message": "flat"}),
|
|
json.dumps({"type": "user", "message": {"content": 7}}),
|
|
json.dumps(
|
|
{"type": "user", "message": {"content": ["s", {"type": "tool_result"}]}}
|
|
),
|
|
json.dumps({"type": "user", "message": {"content": " \n"}}),
|
|
json.dumps(
|
|
{
|
|
"type": "user",
|
|
"message": {"content": [{"type": "text", "text": "x" * 80}]},
|
|
}
|
|
),
|
|
]
|
|
path = _touch(tmp_path / "t.jsonl", 1, "\n".join(rows))
|
|
assert discover.first_prompt(path) == "x" * 59 + "…"
|
|
assert discover.first_prompt(path, limit=100) == "x" * 80
|
|
|
|
|
|
def test_first_prompt_text_block_without_text(tmp_path: Path) -> None:
|
|
row = {"type": "user", "message": {"content": [{"type": "text"}]}}
|
|
path = _touch(tmp_path / "t.jsonl", 1, json.dumps(row))
|
|
assert discover.first_prompt(path) == ""
|
|
|
|
|
|
def test_first_prompt_strips_harness_wrappers(tmp_path: Path) -> None:
|
|
text = "<fork-boilerplate>\nYou are a fork <b>x</b>\n</fork-boilerplate>\nFix CI"
|
|
row = {"type": "user", "message": {"content": text}}
|
|
path = _touch(tmp_path / "t.jsonl", 1, json.dumps(row))
|
|
assert discover.first_prompt(path) == "Fix CI"
|