How I declare and assign a function to one of my agent? Can you give me a sample agent node please?

[mraguth]

This is my function:

import pandas as pd
import pyarrow as pa
import pyarrow.parquet as pq

def dataframe_to_parquet(dataframe, filename, compression='gzip'):
"""
Writes a pandas DataFrame to a Parquet file.

Args:
    dataframe (pd.DataFrame): The DataFrame to be written.
    filename (str): The name of the output Parquet file.
    compression (str, optional): The compression method to use (default: 'gzip').

Returns:
    None
"""
dataframe.to_parquet(filename, compression=compression)

[mraguth]

I added @tool at the top of the function and created an agent like below. But that new agent is not invoking in the flow

sdf_agent = create_agent(
llm,
[dataframe_to_parquet],
#system_message="Any charts you display will be visible by the user.",
system_message=""" XXXXXX
"""
)
sdf_node = functools.partial(agent_node, agent=sdf_agent, name="Parq Generator")