🧰 New Function: Download conversation with your agent as PDF!

[igor-sadalski]

As per @serena2z #206 issue

Its a heavy PR so still bugs are present, help me spot them and tag me! Ill try to fixed them as they come ;). Feel free to fork and help be debug/add better styling etc.

Supported Features

PDF available via simple flag

agent = A1(
    path="your/dataset/path/here",
    llm="your-llm-model",
    source="your-llm-source",
)
prompt = f"prompt"
log = agent.go(prompt)
agent.save_conversation_history("your_output_filename", save_pdf=True)

Automatically Extract human prompt

Show plans and steps

Notify user about potential parsing errors

Show code execution and detailed tool use

Show update plan as the rollout continues

Show Observations

Automatically embed images after code that generates them

Show concise summary

Testing

example

path = "/home/jupyter/test_genomics_data.h5ad"

log = agent.go(
    f"Read in {path} and create a UMAP plot colored by cell_types_names column, then import the tool to create embeddings of the data using UCE plot the embeddigns color by the cell_types_names column show this, then create a scatter plot of sin and cosine functions plot them as well as print them their vales of the sines and exit."
)

example PDF output I got:
test_pdf.pdf

[kexinhuang12345]

oh wow - this is amazing! thank you - we will test it out and merge!

[amehrjou]

Nice work! Just a note, there is an open PR (#177) on reasoning trace tracking, happy to help resolve any conflicts if both get merged.