AI assistant for data analysis