| |
|
| | 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):
|
| |
|
| | 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()
|
| |
|
