KeyError: 'type' when make a bar plot

[jiangok-open]

hi,

I am running your example on databricks and colab but both throw error for llm.query("Make a bar plot of the result and use a log scale.", yolo=True)

Error:

KeyError Traceback (most recent call last)
File /databricks/python/lib/python3.10/site-packages/pandas/core/indexes/base.py:3629, in Index.get_loc(self, key, method, tolerance)
3628 try:
-> 3629 return self._engine.get_loc(casted_key)
3630 except KeyError as err:

File /databricks/python/lib/python3.10/site-packages/pandas/_libs/index.pyx:136, in pandas._libs.index.IndexEngine.get_loc()

File /databricks/python/lib/python3.10/site-packages/pandas/_libs/index.pyx:163, in pandas._libs.index.IndexEngine.get_loc()

File pandas/_libs/hashtable_class_helper.pxi:5198, in pandas._libs.hashtable.PyObjectHashTable.get_item()

File pandas/_libs/hashtable_class_helper.pxi:5206, in pandas._libs.hashtable.PyObjectHashTable.get_item()

KeyError: 'type'

any idea? Thanks a lot!

[ccurme]

I can't reproduce this. That particular cell in the notebook has yolo=True. Can you try setting it to False? This will let you see what code it's trying to execute.