There's a pair of config settings to turn off all of the AI features:
features.edit_prediction_provider=none
agent.enabled=false
There's additional config to set if inline assistance is automatic, only when the user presses a key, fully disabled, etc.