Ministers are eyeing a levy on milk-based drinks to treat them the same as soft drinks which are subject to a sugar tax.