While yes that is the ideal policy, it doesn't seem likely that it will be implemented by schools and they will probably continue to have them write essays, even if AI is a thing. Private schools would probably be lobbying the government to regulate AI and prevent it from writing essays and all of that or something rather than just do what you said.