def METHOD_NAME(self, unit: Unit) -> Dict[str, Any]:
    """
    Return a dict containing all data associated with the given
    unit, including its status, data, and start and end time.
    Also includes the DB ids for the worker, the unit, and the
    relevant assignment this unit was a part of.
    """
    agent = unit.get_assigned_agent()
    assert agent is not None, f"Trying to get completed data from unassigned unit {unit}"
    return {
        "worker_id": agent.worker_id,
        "unit_id": unit.db_id,
        "assignment_id": unit.assignment_id,
        "status": agent.db_status,
        "data": agent.state.get_parsed_data(),
        "task_start": agent.state.get_task_start(),
        "task_end": agent.state.get_task_end(),
        "tips": agent.state.get_tips(),
        "feedback": agent.state.get_feedback(),
    }