From 7d03d8f586f123e5059cbd0f45cb4c701bf0976f Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Thu, 1 Feb 2024 10:17:26 -0800 Subject: [PATCH] docs: fix docstring examples (#16889) --- .../agents/agent_toolkits/csv/base.py | 18 +++++++-------- .../agents/agent_toolkits/pandas/base.py | 23 +++++++++---------- 2 files changed, 20 insertions(+), 21 deletions(-) diff --git a/libs/experimental/langchain_experimental/agents/agent_toolkits/csv/base.py b/libs/experimental/langchain_experimental/agents/agent_toolkits/csv/base.py index a7fa928c23c..e757714d6d8 100644 --- a/libs/experimental/langchain_experimental/agents/agent_toolkits/csv/base.py +++ b/libs/experimental/langchain_experimental/agents/agent_toolkits/csv/base.py @@ -34,16 +34,16 @@ def create_csv_agent( Example: .. code-block:: python - from langchain_openai import ChatOpenAI - from langchain_experimental.agents import create_csv_agent + from langchain_openai import ChatOpenAI + from langchain_experimental.agents import create_csv_agent - llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0) - agent_executor = create_pandas_dataframe_agent( - llm, - "titanic.csv", - agent_type="openai-tools", - verbose=True - ) + llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0) + agent_executor = create_pandas_dataframe_agent( + llm, + "titanic.csv", + agent_type="openai-tools", + verbose=True + ) """ # noqa: E501 try: import pandas as pd diff --git a/libs/experimental/langchain_experimental/agents/agent_toolkits/pandas/base.py b/libs/experimental/langchain_experimental/agents/agent_toolkits/pandas/base.py index 741bfc7a48f..0970c39595b 100644 --- a/libs/experimental/langchain_experimental/agents/agent_toolkits/pandas/base.py +++ b/libs/experimental/langchain_experimental/agents/agent_toolkits/pandas/base.py @@ -200,21 +200,20 @@ def create_pandas_dataframe_agent( a PythonAstREPLTool with the DataFrame(s) and any user-provided extra_tools. Example: - .. code-block:: python - from langchain_openai import ChatOpenAI - from langchain_experimental.agents import create_pandas_dataframe_agent - import pandas as pd + from langchain_openai import ChatOpenAI + from langchain_experimental.agents import create_pandas_dataframe_agent + import pandas as pd - df = pd.read_csv("titanic.csv") - llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0) - agent_executor = create_pandas_dataframe_agent( - llm, - df, - agent_type="openai-tools", - verbose=True - ) + df = pd.read_csv("titanic.csv") + llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0) + agent_executor = create_pandas_dataframe_agent( + llm, + df, + agent_type="openai-tools", + verbose=True + ) """ # noqa: E501 try: