Skip to main content
`Ctrl`+`K`

#

# Source code for langchain.agents.agent_toolkits.csv.base

    """Agent for working with csvs."""
    from typing import Any, Optional

    from langchain.agents.agent import AgentExecutor
    from langchain.agents.agent_toolkits.pandas.base import create_pandas_dataframe_agent
    from langchain.llms.base import BaseLLM

    def create_csv_agent(
        llm: BaseLLM, path: str, pandas_kwargs: Optional[dict] = None, **kwargs: Any
    ) -> AgentExecutor:
        """Create csv agent by loading to a dataframe and using pandas agent."""
        import pandas as pd

        _kwargs = pandas_kwargs or {}
        df = pd.read_csv(path, **_kwargs)
        return create_pandas_dataframe_agent(llm, df, **kwargs)