How to write a function with multiple agg like sum, first, collect_set, count_distinct in pyspark?

Question

data = data.groupby(columns_for_groupby).agg(first(\'AGE\').alias(\'AGE\'),
    first(\'WEIGHT\').alias(\'WEIGHT\'),
    first(\'MOBILE\').alias(\'MOBILE\'),
    sum