feat(plan): add count_by_dataset function to query grouped counts by dataset