forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
1.4 KiB
Python
28 lines
1.4 KiB
Python
from typing import Dict, Type, Union
|
|
|
|
from langchain.agents.agent import BaseSingleActionAgent
|
|
from langchain.agents.agent_types import AgentType
|
|
from langchain.agents.chat.base import ChatAgent
|
|
from langchain.agents.conversational.base import ConversationalAgent
|
|
from langchain.agents.conversational_chat.base import ConversationalChatAgent
|
|
from langchain.agents.mrkl.base import 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 ReActDocstoreAgent
|
|
from langchain.agents.self_ask_with_search.base import SelfAskWithSearchAgent
|
|
from langchain.agents.structured_chat.base import StructuredChatAgent
|
|
|
|
AGENT_TYPE = Union[Type[BaseSingleActionAgent], Type[OpenAIMultiFunctionsAgent]]
|
|
|
|
AGENT_TO_CLASS: Dict[AgentType, AGENT_TYPE] = {
|
|
AgentType.ZERO_SHOT_REACT_DESCRIPTION: ZeroShotAgent,
|
|
AgentType.REACT_DOCSTORE: ReActDocstoreAgent,
|
|
AgentType.SELF_ASK_WITH_SEARCH: SelfAskWithSearchAgent,
|
|
AgentType.CONVERSATIONAL_REACT_DESCRIPTION: ConversationalAgent,
|
|
AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION: ChatAgent,
|
|
AgentType.CHAT_CONVERSATIONAL_REACT_DESCRIPTION: ConversationalChatAgent,
|
|
AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION: StructuredChatAgent,
|
|
AgentType.OPENAI_FUNCTIONS: OpenAIFunctionsAgent,
|
|
AgentType.OPENAI_MULTI_FUNCTIONS: OpenAIMultiFunctionsAgent,
|
|
}
|