It's (at least to me) fairly obvious from both the feature's dedicated page and DataGrip's purchasing page that the AI Assistant is an add-on cost.