I think of government as Legislative, Executive and Judiciary. AI can support help all of these branches not replace them. Especially the legislative and judiciary need humans in the decision loop. The executive (esp. for administrative tasks) will probably have the most day-to-day actions replaced by AI.