File size: 801 Bytes
8fd5450 | 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 | """
Save before/after episodes to disk for comparison.
"""
import json
import os
from datetime import datetime
def log_transcript(
episode: dict,
decision: dict,
result: dict,
label: str,
out_dir: str = "./transcripts"
) -> str:
os.makedirs(out_dir, exist_ok=True)
timestamp = datetime.utcnow().strftime("%Y%m%dT%H%M%SZ")
filename = f"{label}_{timestamp}.json"
path = os.path.join(out_dir, filename)
payload = {
"label": label,
"timestamp": timestamp,
"episode": episode,
"decision": decision,
"result": result,
}
with open(path, "w") as f:
json.dump(payload, f, indent=2, default=str)
return path
def load_transcript(path: str) -> dict:
with open(path, "r") as f:
return json.load(f)
|