I am trying to use the PandasQueryEngine but it can't get the column name right when I query the data. I have tried updating the instruction_str or the pandas_prompt. (the name of the column in the df is averageRating and it keeps using rating or imdb_rating as the name of the column.) Does anyone have any idea how to fix it?
**update: When I use the PandasQueryEngine by itself on the df, it works fine, but when I use the get_chat_engine with it, it has the above problem.
Hey there! It sounds like you're running into a frustrating issue with the PandasQueryEngine not recognizing your column name correctly when using it in conjunction with the chat engine. The tricky part here is likely due to how the chat engine interprets the instructions and prompts—it's probably not aligning with the actual column names in your DataFrame. I suggest double-checking the mapping of your DataFrame's columns in the chat engine's context to ensure it's pulling the right names. If you need a hand getting this sorted out, I’d be more than happy to help! Cheers!