StdoutCallback() deprecation - alternative method

[tos-valantic]

Hi,
in pandasai==1.4.7 I have used the now deprecated StdoutCallback() to catch the llm-generated function and display it in a flask app. Is there any alternative to get the function? explain() or rephrase_query() did not help with that.
Any help is welcome!

@contextmanager
def capture_stdout():
    old_stdout = sys.stdout
    sys.stdout = mystdout = StringIO()
    yield mystdout
    sys.stdout = old_stdout
    
    
pandasai_sdf = SmartDataframe(
    df_pandas,
    config={"llm": llm, "verbose": False, "callback": StdoutCallback()},
)

prompt_extension = "\nPrevent devide by zero."
with capture_stdout() as stdout_buffer:
    result = pandasai_sdf.chat(
        query=user_input + prompt_extension, output_type="dataframe"
    )

stdout_content = stdout_buffer.getvalue()

[gventuri]

@tos-wdp, If the goal is to get the last code executed, you can simply use df.last_code_executed or lake.last_code_executed where df and lake are either the variable of your SmartDataframe or SmartDatalake.

[tos-valantic]

Thank you, this is a nice solution!

Minor downside is that the comments are missing.

stdout_content:

last_code_executed: