From 670304a8b344bc706e80b77dd2be3faa0d3d94ff Mon Sep 17 00:00:00 2001 From: Leonid Ganeline Date: Sun, 23 Jul 2023 18:02:20 -0700 Subject: [PATCH] simplified nmspace (#8152) recreated #7894 (it is easy to recreate than resolve conflicts) A small refactoring to improve the API Reference Agents table @baskaryan --- .../agents/agent_toolkits/__init__.py | 42 +++++++++---------- ...toolkit.py => azure_cognitive_services.py} | 0 .../azure_cognitive_services/__init__.py | 7 ---- 3 files changed, 21 insertions(+), 28 deletions(-) rename libs/langchain/langchain/agents/agent_toolkits/{azure_cognitive_services/toolkit.py => azure_cognitive_services.py} (100%) delete mode 100644 libs/langchain/langchain/agents/agent_toolkits/azure_cognitive_services/__init__.py diff --git a/libs/langchain/langchain/agents/agent_toolkits/__init__.py b/libs/langchain/langchain/agents/agent_toolkits/__init__.py index 05704a7183..0e1680e1e0 100644 --- a/libs/langchain/langchain/agents/agent_toolkits/__init__.py +++ b/libs/langchain/langchain/agents/agent_toolkits/__init__.py @@ -1,6 +1,6 @@ """Agent toolkits.""" from langchain.agents.agent_toolkits.amadeus.toolkit import AmadeusToolkit -from langchain.agents.agent_toolkits.azure_cognitive_services.toolkit import ( +from langchain.agents.agent_toolkits.azure_cognitive_services import ( AzureCognitiveServicesToolkit, ) from langchain.agents.agent_toolkits.csv.base import create_csv_agent @@ -40,33 +40,33 @@ from langchain.agents.agent_toolkits.zapier.toolkit import ZapierToolkit __all__ = [ "AmadeusToolkit", - "create_json_agent", - "create_sql_agent", - "create_openapi_agent", - "create_pbi_agent", - "create_pbi_chat_agent", - "create_python_agent", - "create_vectorstore_agent", + "AzureCognitiveServicesToolkit", + "FileManagementToolkit", + "GmailToolkit", + "JiraToolkit", "JsonToolkit", - "SQLDatabaseToolkit", - "SparkSQLToolkit", "NLAToolkit", - "PowerBIToolkit", + "O365Toolkit", "OpenAPIToolkit", - "VectorStoreToolkit", - "create_vectorstore_router_agent", + "PlayWrightBrowserToolkit", + "PowerBIToolkit", + "SQLDatabaseToolkit", + "SparkSQLToolkit", "VectorStoreInfo", "VectorStoreRouterToolkit", + "VectorStoreToolkit", + "ZapierToolkit", + "create_csv_agent", + "create_json_agent", + "create_openapi_agent", "create_pandas_dataframe_agent", + "create_pbi_agent", + "create_pbi_chat_agent", + "create_python_agent", "create_spark_dataframe_agent", "create_spark_sql_agent", - "create_csv_agent", - "ZapierToolkit", - "GmailToolkit", - "JiraToolkit", - "FileManagementToolkit", - "PlayWrightBrowserToolkit", - "AzureCognitiveServicesToolkit", - "O365Toolkit", + "create_sql_agent", + "create_vectorstore_agent", + "create_vectorstore_router_agent", "create_xorbits_agent", ] diff --git a/libs/langchain/langchain/agents/agent_toolkits/azure_cognitive_services/toolkit.py b/libs/langchain/langchain/agents/agent_toolkits/azure_cognitive_services.py similarity index 100% rename from libs/langchain/langchain/agents/agent_toolkits/azure_cognitive_services/toolkit.py rename to libs/langchain/langchain/agents/agent_toolkits/azure_cognitive_services.py diff --git a/libs/langchain/langchain/agents/agent_toolkits/azure_cognitive_services/__init__.py b/libs/langchain/langchain/agents/agent_toolkits/azure_cognitive_services/__init__.py deleted file mode 100644 index 0535dfbe91..0000000000 --- a/libs/langchain/langchain/agents/agent_toolkits/azure_cognitive_services/__init__.py +++ /dev/null @@ -1,7 +0,0 @@ -"""Azure Cognitive Services Toolkit.""" - -from langchain.agents.agent_toolkits.azure_cognitive_services.toolkit import ( - AzureCognitiveServicesToolkit, -) - -__all__ = ["AzureCognitiveServicesToolkit"]