diff --git a/langchain/agents/agent_toolkits/pandas/base.py b/langchain/agents/agent_toolkits/pandas/base.py index 2d4b9fb5..e6a4326e 100644 --- a/langchain/agents/agent_toolkits/pandas/base.py +++ b/langchain/agents/agent_toolkits/pandas/base.py @@ -35,7 +35,7 @@ def create_pandas_dataframe_agent( prompt = ZeroShotAgent.create_prompt( tools, prefix=prefix, suffix=suffix, input_variables=input_variables ) - partial_prompt = prompt.partial(df=str(df.head())) + partial_prompt = prompt.partial(df=str(df.head().to_markdown())) llm_chain = LLMChain( llm=llm, prompt=partial_prompt,