Function persist_trajectory

src/observability/persistence.py:39–86

Write a per-run evaluation record when ``AGENT_TRAJECTORY_DIR`` is set. Reads ``run_id`` / ``scenario_id`` from the same contextvars used by :func:`agent_run_span`, so CLI-level wiring doesn't have to touch the runner's public signature. Returns the output path, or ``None`` when pe

(
    *,
    runner_name: str,
    model: str,
    question: str,
    answer: str,
    trajectory: Any,
)

def persist_trajectory(
    *,
    runner_name: str,
    model: str,
    question: str,
    answer: str,
    trajectory: Any,
) -> Path | None:
    """Write a per-run evaluation record when ``AGENT_TRAJECTORY_DIR`` is set.

    Reads ``run_id`` / ``scenario_id`` from the same contextvars used by
    :func:`agent_run_span`, so CLI-level wiring doesn't have to touch the
    runner's public signature.

    Returns the output path, or ``None`` when persistence is disabled.
    """
    dir_env = os.environ.get(_TRAJECTORY_DIR_ENV)
    if not dir_env:
        return None

    run_id = _run_id_var.get()
    if not run_id:
        _log.warning(
            "%s is set but no run_id in context; skipping trajectory persist",
            _TRAJECTORY_DIR_ENV,
        )
        return None

    out_dir = Path(dir_env)
    out_dir.mkdir(parents=True, exist_ok=True)
    out_path = out_dir / f"{run_id}.json"

    record = {
        "run_id": run_id,
        "scenario_id": _scenario_id_var.get(),
        "runner": runner_name,
        "model": model,
        "question": question,
        "answer": answer,
        "trajectory": _serialize_trajectory(trajectory),
    }

    try:
        out_path.write_text(json.dumps(record, indent=2, default=str), encoding="utf-8")
    except OSError:
        _log.exception("persist_trajectory: write failed at %s", out_path)
        return None
    return out_path


def _serialize_trajectory(trajectory: Any) -> Any: