github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-15 22:44:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Bagatur
2023-12-11 14:19:25 -08:00
committed by GitHub
parent 3b5b0f16c6
commit a844b495c4
5 changed files with 33 additions and 195 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/community/langchain_community/agent_toolkits/__init__.py
View File

@@ -18,9 +18,6 @@ from langchain_community.agent_toolkits.amadeus.toolkit import AmadeusToolkit
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 @@ __all__ = [
"create_pbi_chat_agent",
"create_spark_sql_agent",
"create_sql_agent",
"create_conversational_retrieval_agent",
]

0
libs/community/tests/unit_tests/agent_toolkits/__init__.py Normal file
View File

33
libs/community/tests/unit_tests/agent_toolkits/test_imports.py Normal file
View File

@@ -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)