Support for Gemini-1.5-flash and pro

[tytung2020]

🚀 The feature

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.

Motivation, pitch

These are better Gemini models and should be supported.

Alternatives

No response

Additional context

No response