File size: 1,819 Bytes
fba140f |
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 39 40 41 42 43 44 45 46 47 |
# -*- coding: utf-8 -*-
import json, argparse, uuid
from pathlib import Path
SYS = "You output ONLY JSON, no explanation. Validate against AttackPlan v1.1 semantics."
def plan_to_prompt(p):
# small, human prompt summarizing the requested edits
bits=[]
for it in p.get("plan", [])[:6]:
op = it.get("op","set"); pt = it.get("point",""); val = it.get("attack_value")
nm = it.get("name","")
scope = (it.get("scope") or {})
area = scope.get("mg"); mim = scope.get("mim")
s = f"{op} {pt} on {nm} to {val}"
if mim: s += f" in {mim}"
if area: s += f" ({area})"
bits.append(s)
return "; ".join(bits) if bits else "Generate an AttackPlan JSON v1.1 for no-op."
def main():
ap = argparse.ArgumentParser()
ap.add_argument("--src", required=True, help="scripts/train_attackplan.aug.jsonl (or filtered)")
ap.add_argument("--out", default="datasets/chat_attackplan.jsonl")
args = ap.parse_args()
src = Path(args.src).read_text(encoding="utf-8-sig").splitlines()
out = []
for ln in src:
if not ln.strip(): continue
obj = json.loads(ln)
prompt = plan_to_prompt(obj)
out.append({
"id": str(uuid.uuid4()),
"messages": [
{"role":"system","content":SYS},
{"role":"user","content":f"Task: {prompt}\nReturn ONLY the JSON."},
{"role":"assistant","content": json.dumps(obj, ensure_ascii=False)}
]
})
Path(args.out).parent.mkdir(parents=True, exist_ok=True)
Path(args.out).write_text("\n".join(json.dumps(x, ensure_ascii=False) for x in out)+"\n", encoding="utf-8")
print("[ok] wrote", Path(args.out).resolve(), "rows:", len(out))
if __name__ == "__main__":
main()
|