mirror of
https://github.com/hwchase17/langchain
synced 2024-10-29 17:07:25 +00:00
c7ca350cd3
In LangChain, all module classes are enumerated in the `__init__.py` file of the correspondent module. But some classes were missed and were not included in the module `__init__.py` This PR: - added the missed classes to the module `__init__.py` files - `__init__.py:__all_` variable value (a list of the class names) was sorted - `langchain.tools.sql_database.tool.QueryCheckerTool` was renamed into the `QuerySQLCheckerTool` because it conflicted with `langchain.tools.spark_sql.tool.QueryCheckerTool` - changes to `pyproject.toml`: - added `pgvector` to `pyproject.toml:extended_testing` - added `pandas` to `pyproject.toml:[tool.poetry.group.test.dependencies]` - commented out the `streamlit` from `collbacks/__init__.py`, It is because now the `streamlit` requires Python >=3.7, !=3.9.7 - fixed duplicate names in `tools` - fixed correspondent ut-s #### Who can review? @hwchase17 @dev2049
44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
from langchain.agents import __all__ as agents_all
|
|
|
|
_EXPECTED = [
|
|
"Agent",
|
|
"AgentExecutor",
|
|
"AgentOutputParser",
|
|
"AgentType",
|
|
"BaseMultiActionAgent",
|
|
"BaseSingleActionAgent",
|
|
"ConversationalAgent",
|
|
"ConversationalChatAgent",
|
|
"LLMSingleActionAgent",
|
|
"MRKLChain",
|
|
"OpenAIFunctionsAgent",
|
|
"ReActChain",
|
|
"ReActTextWorldAgent",
|
|
"SelfAskWithSearchChain",
|
|
"StructuredChatAgent",
|
|
"Tool",
|
|
"ZeroShotAgent",
|
|
"create_csv_agent",
|
|
"create_json_agent",
|
|
"create_openapi_agent",
|
|
"create_pandas_dataframe_agent",
|
|
"create_pbi_agent",
|
|
"create_pbi_chat_agent",
|
|
"create_spark_dataframe_agent",
|
|
"create_spark_sql_agent",
|
|
"create_sql_agent",
|
|
"create_vectorstore_agent",
|
|
"create_vectorstore_router_agent",
|
|
"get_all_tool_names",
|
|
"initialize_agent",
|
|
"load_agent",
|
|
"load_huggingface_tool",
|
|
"load_tools",
|
|
"tool",
|
|
]
|
|
|
|
|
|
def test_public_api() -> None:
|
|
"""Test for regressions or changes in the agents public API."""
|
|
assert sorted(agents_all) == sorted(_EXPECTED)
|