Can I only use the advice the IELTSi teacher gives me for the part of the paper they talk about?

The advice about what the examiners are looking for, given to you by the IELTSi teacher, is applicable for both Writing papers and all types of tasks. So e.g., Task Response is important in all Task 2 papers, the teacher is just using the Cause and Solutions Essay as an example.