Open tytung2020 opened 4 days ago
The GoogleVertexAI
class in pandas-ai currently supports the "gemini-pro"
model. The model "gemini-1.5-flash"
is not listed as a supported model, which is why you are encountering the "Unsupported model" error.
Here is the relevant part of the code that lists the supported generative models:
_supported_generative_models = [
"gemini-pro",
]
To resolve this issue, you should use the "gemini-pro"
model instead of "gemini-1.5-flash"
:
llm = GoogleVertexAI(
project_id="some_location", location="us-central1", model="gemini-pro"
)
For more details, you can refer to the GoogleVertexAI class implementation [1].
To continue talking to Dosu, mention @dosu.
ok gemini-pro works, but I need it to based on the latest gemini-1.5-flash or gemini-1.5-pro. gemini-pro is based on a Nov 2023 version which is not that accurate.
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:
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.