From 7d8830f707d4698dcefeef8162483b5ba2d65534 Mon Sep 17 00:00:00 2001 From: Tahjyei Thompson <45900878+Tajcore@users.noreply.github.com> Date: Thu, 29 Jun 2023 20:34:26 -0500 Subject: [PATCH] Add `OpenAIMultiFunctionsAgent` to import list in agents directory (#6824) - Added OpenAIMultiFunctionsAgent to the import list of the Agents directory --------- Co-authored-by: Dev 2049 --- langchain/agents/__init__.py | 2 ++ tests/unit_tests/agents/test_public_api.py | 1 + 2 files changed, 3 insertions(+) diff --git a/langchain/agents/__init__.py b/langchain/agents/__init__.py index 87bfd254b9..37d6e0c019 100644 --- a/langchain/agents/__init__.py +++ b/langchain/agents/__init__.py @@ -32,6 +32,7 @@ from langchain.agents.load_tools import ( from langchain.agents.loading import load_agent from langchain.agents.mrkl.base import MRKLChain, ZeroShotAgent from langchain.agents.openai_functions_agent.base import OpenAIFunctionsAgent +from langchain.agents.openai_functions_multi_agent.base import OpenAIMultiFunctionsAgent from langchain.agents.react.base import ReActChain, ReActTextWorldAgent from langchain.agents.self_ask_with_search.base import SelfAskWithSearchChain from langchain.agents.structured_chat.base import StructuredChatAgent @@ -49,6 +50,7 @@ __all__ = [ "LLMSingleActionAgent", "MRKLChain", "OpenAIFunctionsAgent", + "OpenAIMultiFunctionsAgent", "ReActChain", "ReActTextWorldAgent", "SelfAskWithSearchChain", diff --git a/tests/unit_tests/agents/test_public_api.py b/tests/unit_tests/agents/test_public_api.py index 19ed7ffba8..b22367a6e3 100644 --- a/tests/unit_tests/agents/test_public_api.py +++ b/tests/unit_tests/agents/test_public_api.py @@ -12,6 +12,7 @@ _EXPECTED = [ "LLMSingleActionAgent", "MRKLChain", "OpenAIFunctionsAgent", + "OpenAIMultiFunctionsAgent", "ReActChain", "ReActTextWorldAgent", "SelfAskWithSearchChain",