which Google models are you using? I assume they are from Vertex AI, but there is Palm2 generic and 3 coding specific models for different types of tasks (auto complete, code completion, code chat)