From 1e7ae107b054a9552f0c735fd99a4581fac79f97 Mon Sep 17 00:00:00 2001 From: Rafael Sales Date: Wed, 18 Dec 2024 22:56:05 -0300 Subject: [PATCH] Refactor imports in agent toolkits to streamline CloudWatch logs agent integration --- libs/experimental/langchain_experimental/agents/__init__.py | 3 +-- .../agents/agent_toolkits/__init__.py | 6 +++--- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/libs/experimental/langchain_experimental/agents/__init__.py b/libs/experimental/langchain_experimental/agents/__init__.py index 590888c..16730e6 100644 --- a/libs/experimental/langchain_experimental/agents/__init__.py +++ b/libs/experimental/langchain_experimental/agents/__init__.py @@ -9,11 +9,11 @@ """ from langchain_experimental.agents.agent_toolkits import ( + create_cloudwatch_logs_agent, create_csv_agent, create_pandas_dataframe_agent, create_spark_dataframe_agent, create_xorbits_agent, - create_cloudwatch_logs_agent, ) __all__ = [ @@ -22,5 +22,4 @@ "create_spark_dataframe_agent", "create_xorbits_agent", "create_cloudwatch_logs_agent", - ] diff --git a/libs/experimental/langchain_experimental/agents/agent_toolkits/__init__.py b/libs/experimental/langchain_experimental/agents/agent_toolkits/__init__.py index 68d698b..b14df46 100644 --- a/libs/experimental/langchain_experimental/agents/agent_toolkits/__init__.py +++ b/libs/experimental/langchain_experimental/agents/agent_toolkits/__init__.py @@ -1,3 +1,6 @@ +from langchain_experimental.agents.agent_toolkits.cloudwatch.base import ( + create_cloudwatch_logs_agent, +) from langchain_experimental.agents.agent_toolkits.csv.base import create_csv_agent from langchain_experimental.agents.agent_toolkits.pandas.base import ( create_pandas_dataframe_agent, @@ -9,9 +12,6 @@ from langchain_experimental.agents.agent_toolkits.xorbits.base import ( create_xorbits_agent, ) -from langchain_experimental.agents.agent_toolkits.cloudwatch.base import ( - create_cloudwatch_logs_agent, -) __all__ = [ "create_xorbits_agent",