Improve test coverage for vertex_ai_models

What does this merge request do and why?

Improve test coverage for vertex_ai_models.

How to set up and validate locally

Numbered steps to set up and validate the change are strongly suggested.

Merge request checklist

• I've ran the affected pipeline(s) to validate that nothing is broken.
• Tests added for new functionality. If not, please raise an issue to follow up.
• Documentation added/updated, if needed.

Closes #270 (closed)