community[patch]: Fix agenttoolkits imports (#14559)

langchain-ai · Dec 11, 2023 · a844b49 · a844b49
1 parent 3b5b0f1
commit a844b49
Show file tree

Hide file tree

Showing 5 changed files with 33 additions and 195 deletions.
diff --git a/...community/langchain_community/agent_toolkits/conversational_retrieval/openai_functions.py b/...community/langchain_community/agent_toolkits/conversational_retrieval/openai_functions.py
diff --git a/...pts/community_split/libs/community/langchain_community/agent_toolkits/vectorstore/base.py b/...pts/community_split/libs/community/langchain_community/agent_toolkits/vectorstore/base.py
diff --git a/libs/community/langchain_community/agent_toolkits/__init__.py b/libs/community/langchain_community/agent_toolkits/__init__.py
@@ -18,9 +18,6 @@
 from langchain_community.agent_toolkits.azure_cognitive_services import (
     AzureCognitiveServicesToolkit,
 )
-from langchain_community.agent_toolkits.conversational_retrieval.openai_functions import (  # noqa: E501
-    create_conversational_retrieval_agent,
-)
 from langchain_community.agent_toolkits.file_management.toolkit import (
     FileManagementToolkit,
 )
@@ -74,5 +71,4 @@
     "create_pbi_chat_agent",
     "create_spark_sql_agent",
     "create_sql_agent",
-    "create_conversational_retrieval_agent",
 ]
diff --git a/libs/community/tests/unit_tests/agent_toolkits/__init__.py b/libs/community/tests/unit_tests/agent_toolkits/__init__.py
diff --git a/libs/community/tests/unit_tests/agent_toolkits/test_imports.py b/libs/community/tests/unit_tests/agent_toolkits/test_imports.py
@@ -0,0 +1,33 @@
+from langchain_community.agent_toolkits import __all__
+
+EXPECTED_ALL = [
+    "AINetworkToolkit",
+    "AmadeusToolkit",
+    "AzureCognitiveServicesToolkit",
+    "FileManagementToolkit",
+    "GmailToolkit",
+    "JiraToolkit",
+    "JsonToolkit",
+    "MultionToolkit",
+    "NasaToolkit",
+    "NLAToolkit",
+    "O365Toolkit",
+    "OpenAPIToolkit",
+    "PlayWrightBrowserToolkit",
+    "PowerBIToolkit",
+    "SlackToolkit",
+    "SteamToolkit",
+    "SQLDatabaseToolkit",
+    "SparkSQLToolkit",
+    "ZapierToolkit",
+    "create_json_agent",
+    "create_openapi_agent",
+    "create_pbi_agent",
+    "create_pbi_chat_agent",
+    "create_spark_sql_agent",
+    "create_sql_agent",
+]
+
+
+def test_all_imports() -> None:
+    assert set(__all__) == set(EXPECTED_ALL)