How to get or return the output data used to generate the visualization?

[rrk390]

I am trying to get the output data which has been used to generate the visualization based on the user query. I am working with LIDA now a days with Azure Open AI API.

It will be helpful if you let me know on this.
Few observation from my side to achieve the above goal:

• Do I need to change any prompt messages for role system. assistant or user?
• Do I need to implement any custom functions which will perform to generate the data only?
• Do you already added the same requirement code in the LIDA library and I missed that by any chance?

Waiting for your response.