Using GPT-4

[wenger9]

Hi! Thanks for putting together such a great library.

I am testing pandasai on a google collar notebook. How can I switch between the currently supported models? For example, I'd like to try gpt-4.

Thanks!

[mspronesti]

just set the model parameter in the OpenAI class when instiating it, e.g.

from pandasai import PandasAI
from pandasai.llm import OpenAI

llm = OpenAI(api_key='YOUR_KEY_HERE', model='gpt-4')
pandas_ai = PandasAI(llm, conversational=False)
pandas_ai(df, prompt='YOUR_PROMPT_HERE')

if you use AzureOpenAI just create a deployment of GPT-4 and set deployment_name to the deployment name.

Hope this helps! If it does, please accept my answer 😄

[wenger9]

Thank you very much!

[mspronesti]

Glad it helped :)