Publish complete protocol generations atomically

scripts/run_direct_protocol.py

@@ -17,11 +17,12 @@ REPO_ROOT = Path(__file__).resolve().parents[1]
 if str(REPO_ROOT) not in sys.path:
     sys.path.insert(0, str(REPO_ROOT))
 
+from src.meeting_lab.models.meeting_context import load_meeting_context  # noqa: E402
+from src.meeting_lab.protocol.history import persist_protocol_generation  # noqa: E402
 from src.meeting_lab.llm.ollama import DEFAULT_ENDPOINT  # noqa: E402
 from src.meeting_lab.protocol.generate_direct_protocol import (  # noqa: E402
     DEFAULT_MODEL,
     DEFAULT_SAFE_INPUT_TOKEN_BUDGET,
-    DirectProtocolResult,
     generate_direct_protocol,
 )
 
@@ -64,22 +65,6 @@ def write_json(path: Path, data: Any) -> None:
     path.write_text(json.dumps(data, ensure_ascii=False, indent=2) + "\n", encoding="utf-8")
 
 
-def persist_result(run_dir: Path, result: DirectProtocolResult) -> Path:
-    protocol_dir = run_dir / "protocol"
-    protocol_dir.mkdir()
-    (protocol_dir / "exact_prompt.txt").write_text(result.exact_prompt, encoding="utf-8")
-    write_json(protocol_dir / "raw_response.json", result.raw_response)
-    write_json(protocol_dir / "runtime_metadata.json", result.runtime_metadata)
-    transcript_input = getattr(result, "transcript_input", None)
-    if transcript_input is not None:
-        (protocol_dir / "transcript_input.txt").write_text(
-            transcript_input, encoding="utf-8"
-        )
-    protocol_path = run_dir / "protocol.md"
-    protocol_path.write_text(result.protocol_text, encoding="utf-8")
-    return protocol_path
-
-
 def run(args: argparse.Namespace) -> tuple[int, Path, Path | None]:
     run_dir = create_unique_run_dir(args.output_root, args.transcript.stem)
     timestamp = datetime.now().astimezone().isoformat(timespec="seconds")
@@ -129,7 +114,11 @@ def run(args: argparse.Namespace) -> tuple[int, Path, Path | None]:
             endpoint=args.ollama_endpoint,
             safe_input_token_budget=args.safe_input_token_budget,
         )
-        protocol_path = persist_result(run_dir, result)
+        protocol_path = persist_protocol_generation(
+            run_dir,
+            result,
+            context=load_meeting_context(preserved_context) if preserved_context else None,
+        )
         metadata["status"] = "completed"
         metadata["final_protocol_path"] = str(protocol_path.resolve())
     except Exception as exc: