You can just add your extra columns to the input table. Vertex AI will ignore them and they will be included in the output table.