Support for Gemini models via VertexAI?

[tytung2020]

System Info

import os import pandas as pd from pandasai import Agent from pandasai.llm import GoogleVertexAI

🐛 Describe the bug

any update on this issue? I saw an earlier issue and it was closed, but after I tested it still not able to use Google VertexAI with these parameters:

llm = GoogleVertexAI(
    project_id="some_location", location="us-central1", model="gemini-1.5-flash"
)

with this error: ERROR:pandasai.helpers.logger:Pipeline failed on step 3: Unsupported model: The model 'gemini-1.5-flash' doesn't exist or is not supported yet. Unfortunately, I was not able to answer your question, because of the following error:

Unsupported model: The model 'gemini-1.5-flash' doesn't exist or is not supported yet.