Function restore_persisted_session_state

core/src/agent_api/session_persistence.rs:203–243

(
    session: &AgentSession,
    store: &Arc<dyn SessionStore>,
    data: SessionData,
)

}

pub(super) fn restore_persisted_session_state(
    session: &AgentSession,
    store: &Arc<dyn SessionStore>,
    data: SessionData,
) -> Result<()> {
    let session_id = data.id.clone();
    *write_or_recover(&session.history) = data.messages;

    if let Some(artifacts) = load_artifacts(store, &session_id)? {
        let target_store = session.tool_executor.artifact_store();
        for artifact in artifacts.artifacts() {
            target_store.put(artifact);
        }
    }

    if let Some(events) = load_trace_events(store, &session_id)? {
        session.trace_sink.replace_events(events);
    }

    if let Some(records) = load_run_records(store, &session_id)? {
        if let Ok(handle) = tokio::runtime::Handle::try_current() {
            tokio::task::block_in_place(|| {
                handle.block_on(session.run_store.replace_records(records))
            });
        }
    }

    if let Some(reports) = load_verification_reports(store, &session_id)? {
        *write_or_recover(&session.verification_reports) = reports;
    }

    if let Some(tasks) = load_subagent_tasks(store, &session_id)? {
        if let Ok(handle) = tokio::runtime::Handle::try_current() {
            tokio::task::block_in_place(|| {
                handle.block_on(session.subagent_tasks.replace_snapshots(tasks))
            });
        }
    }

    Ok(())
}

struct SessionDataSnapshotInput<'a> {
    session_id: &'a str,