import json


def build_onboarding_context(
    session,
    state,
    conversation_history,
):

    return {
        "patient": {
            "id": session.patient_id,
            "name": session.patient_name,
            "age": session.age,
            "gender": session.gender,
        },

        "category": {
            "type": session.category_type,
            "code": session.category_code,
            "name": session.category_name,
        },

        "language": session.language,

        "current_stage": session.stage,

        "current_state": json.loads(
            state.state_json
        ),

        "conversation": conversation_history,
    }