Using BambooLLM as default LLM

[wschimitsch]

I'm attempting to use the BambooLLM as the default LLM for PandasAI (3.0.0b18). Per set up LLM, pai.api_key.set("api-key") should do this (I've signed up and gotten my own api key). However, the following code raises a ValueError, stating No LLM is provided.

import pandasai as pai
from sklearn.datasets import load_iris

pai.api_key.set("my-api-key") # real api key here

iris = load_iris()

df = pai.DataFrame(data=iris.data, columns=iris.feature_names)

response = df.chat("What is the average sepal width for setosa flowers?")

print(response)

Returns

ValueError: No LLM provided. Please provide an LLM to the agent.

Is support for BambooLLM as the default LLM deprecated? Or am I missing something simple? It doesn't seem as though the code sets any LLM as default in this version. Thanks.