mirror of
https://github.com/norandom/OpenUltraSAST.git
synced 2026-10-16 11:49:28 +00:00
Marius CiepluchandClaude Opus 5.5
d1aabf0f66
CI and the release job check `ruff format --check .` and `ruff check .` over the whole repository, while the local gate checked only src and tests, so eight benchmark scripts drifted and the v2.0.0 release job stopped before building. The fixes are formatting, two noqa comments and one renamed loop variable; no script computes anything differently. The local gate now checks the whole tree. The release workflow gains an images job: Dockerfile -> ghcr.io/<owner>/openultrasast, plane/Dockerfile.runner -> ghcr.io/<owner>/openultrasast-runner, tagged with the version, and the digests written to the job summary. Co-Authored-By: Claude Opus 5.5 <[email protected]> Claude-Session: https://claude.ai/code/session_01EmM1j46qgKMh3zqD1QMwEo
68 lines
2.7 KiB
Python
68 lines
2.7 KiB
Python
"""Bulk reader: answer a question about one or more files with the cheap chat model, citing line numbers.
|
|
|
|
The Portal pattern (Spotify, 2026-09): reading five files to answer one question is I/O, not reasoning, so a
|
|
cheap model reads and the expensive one gets the answer. Files are sent in numbered chunks; the answer must cite
|
|
`path:line`, which is what a summary otherwise lacks. Reasoning stays with the caller: this reports what the
|
|
files say, it does not decide anything.
|
|
|
|
Usage: .venv/bin/python benchmarks/dev/bulk_read.py "<question>" FILE [FILE ...] [--max-usd 0.05]
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
CHUNK_LINES = 400
|
|
PROMPT = (
|
|
"Answer the question using only the files below (numbered lines). Be specific and brief; cite every claim as "
|
|
"path:line. If the files do not answer it, say so.\n\nQuestion: {question}\n\n{files}"
|
|
)
|
|
|
|
|
|
def numbered(path: Path) -> list[str]:
|
|
lines = path.read_text(errors="ignore").splitlines()
|
|
chunks = []
|
|
for start in range(0, max(len(lines), 1), CHUNK_LINES):
|
|
body = "\n".join(f"{n + 1:5d}| {line}" for n, line in enumerate(lines[start : start + CHUNK_LINES], start=start))
|
|
chunks.append(f"--- {path} (lines {start + 1}-{min(start + CHUNK_LINES, len(lines))})\n{body}")
|
|
return chunks
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)
|
|
parser.add_argument("question")
|
|
parser.add_argument("files", nargs="+", type=Path)
|
|
parser.add_argument("--max-usd", type=float, default=0.05)
|
|
args = parser.parse_args()
|
|
from openultrasast import tool_hunter
|
|
from openultrasast.config import load_config
|
|
from openultrasast.model.endpoint import DEFAULT_DETECTOR_MODEL
|
|
|
|
load_config()
|
|
client = tool_hunter.resolve_hunter_client()
|
|
if client is None:
|
|
print("no chat client: set DEEPSEEK_API_KEY", file=sys.stderr)
|
|
return 2
|
|
chunks = [chunk for path in args.files for chunk in numbered(path)]
|
|
answers = []
|
|
for chunk in chunks:
|
|
if float(client.cost_usd()) >= args.max_usd:
|
|
answers.append(f"[stopped at the ${args.max_usd} ceiling before: {chunk.splitlines()[0]}]")
|
|
break
|
|
response = client.complete(
|
|
model=DEFAULT_DETECTOR_MODEL,
|
|
messages=[{"role": "user", "content": PROMPT.format(question=args.question, files=chunk)}],
|
|
tools=[],
|
|
timeout_seconds=90,
|
|
)
|
|
answers.append((response.content or "").strip())
|
|
print("\n\n".join(a for a in answers if a))
|
|
print(f"\n[bulk_read: {len(chunks)} chunk(s), ${float(client.cost_usd()):.4f}]", file=sys.stderr)
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|