mirror of
https://github.com/kennethreitz/langchain.git
synced 2026-06-05 23:00:18 +00:00
Yifei Song
2e47412073
- [Xorbits](https://doc.xorbits.io/en/latest/) is an open-source computing framework that makes it easy to scale data science and machine learning workloads in parallel. Xorbits can leverage multi cores or GPUs to accelerate computation on a single machine, or scale out up to thousands of machines to support processing terabytes of data. - This PR added support for the Xorbits agent, which allows langchain to interact with Xorbits Pandas dataframe and Xorbits Numpy array. - Dependencies: This change requires the Xorbits library to be installed in order to be used. `pip install xorbits` - Request for review: @hinthornw - Twitter handle: https://twitter.com/Xorbitsio
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
from langchain.agents import __all__ as agents_all
|
|
|
|
_EXPECTED = [
|
|
"Agent",
|
|
"AgentExecutor",
|
|
"AgentOutputParser",
|
|
"AgentType",
|
|
"BaseMultiActionAgent",
|
|
"BaseSingleActionAgent",
|
|
"ConversationalAgent",
|
|
"ConversationalChatAgent",
|
|
"LLMSingleActionAgent",
|
|
"MRKLChain",
|
|
"OpenAIFunctionsAgent",
|
|
"OpenAIMultiFunctionsAgent",
|
|
"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",
|
|
"create_xorbits_agent",
|
|
]
|
|
|
|
|
|
def test_public_api() -> None:
|
|
"""Test for regressions or changes in the agents public API."""
|
|
assert sorted(agents_all) == sorted(_EXPECTED)
|