diff --git a/langchain/agents/agent_toolkits/pandas/base.py b/langchain/agents/agent_toolkits/pandas/base.py index fec09514d73..3c8a4615c06 100644 --- a/langchain/agents/agent_toolkits/pandas/base.py +++ b/langchain/agents/agent_toolkits/pandas/base.py @@ -118,6 +118,8 @@ def _get_prompt_and_tools( ) -> Tuple[BasePromptTemplate, List[PythonAstREPLTool]]: try: import pandas as pd + + pd.set_option("display.max_columns", None) except ImportError: raise ValueError( "pandas package not found, please install with `pip install pandas`" @@ -227,6 +229,8 @@ def _get_functions_prompt_and_tools( ) -> Tuple[BasePromptTemplate, List[PythonAstREPLTool]]: try: import pandas as pd + + pd.set_option("display.max_columns", None) except ImportError: raise ValueError( "pandas package not found, please install with `pip install pandas`"