Sempre que tentei utilizar o system proposto:
system = f"""Você tem acesso a um dataframe pandas `df`. \
Aqui está a saída de `df.head().to_markdown()`:
\`\`\`
{df.head().to_markdown()}
\`\`\`
Dada uma pergunta do usuário, escreva o código Python para respondê-la. \
Retorne SOMENTE o código Python válido e nada mais. \
Não presuma que você tem acesso a nenhuma biblioteca além das bibliotecas Python integradas e pandas."""
recebi erro, consegui mudando e deixando da seguinte forma:
system = f"""Você tem acesso a um dataframe pandas `df`. \
O dataframe contém as seguintes colunas: {df.columns}. \
Dada uma pergunta do usuário, escreva o código Python para respondê-la. \
Retorne SOMENTE o código Python válido e nada mais. \
Não presuma que você tem acesso a nenhuma biblioteca além das bibliotecas Python integradas e pandas."""