The AI integrations, AFAIK, work only if you provide an API key.
But yes, your point is valid. For Tor, if you enter an API key, you could be identified. Still, does the Tor Browser prevent you from installing addons which are no more secure than these AI features? It didn't years ago - not sure if that's changed.