Function query_engine_state_from_task_context 

fn query_engine_state_from_task_context(
    context: &TaskContext,
) -> Option<Arc<QueryEngineState>>