I think govs will employ AI for enhanced capabilities and in doing so make legacy functions more and more obsolete.

Whether headcount follows a leaner more capable stack is yet to be seen, but I think what's more likely is new departments for AI being stood up that eventually render old departments inert, and finally shut down with lofty pensions.