File size: 1,491 Bytes
d466b7d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
from __future__ import annotations
import time, json, argparse
from pathlib import Path
from dir2md.core import Config, generate_markdown_report
def run_case(root: Path, preset: str, mode: str | None, budget: int, file_budget: int) -> dict:
cfg = Config(
root=root, output=root/"_BENCH.md", include_globs=[], exclude_globs=[], omit_globs=[],
respect_gitignore=True, follow_symlinks=False, max_bytes=200_000, max_lines=2000,
include_contents=True, only_ext=None, add_stats=True, add_toc=False,
llm_mode=(mode or "ref"), budget_tokens=budget, max_file_tokens=file_budget,
dedup_bits=16, sample_head=120, sample_tail=40, strip_comments=False,
emit_manifest=False, preset=preset, explain_capsule=True,
)
t0 = time.perf_counter()
md = generate_markdown_report(cfg)
dt = time.perf_counter() - t0
est = md.split("Estimated tokens (prompt): `")[-1].split("`")[0]
return {"preset": preset, "mode": cfg.llm_mode, "elapsed_sec": round(dt,3), "est_tokens": int(est)}
def main():
ap = argparse.ArgumentParser()
ap.add_argument("path", nargs="?", default=".")
ns = ap.parse_args()
root = Path(ns.path).resolve()
cases = [
("iceberg", None, 6000, 1000),
("pro", "summary", 6000, 1000),
("pro", "ref", 4000, 1000),
("pro", "inline", 8000, 1200),
]
rows = [run_case(root, *c) for c in cases]
print(json.dumps(rows, indent=2))
if __name__ == "__main__":
main()
|