See - https://cloud.google.com/vertex-ai/docs/generative-ai/sdk-for-llm/llm-sdk-overview-ref#load_a_foundation_model - https://cloud.google.com/vertex-ai/docs/generative-ai/chat/test-chat-prompts