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.
22 lines
1.1 KiB
Python
22 lines
1.1 KiB
Python
from typing import Dict, Type
|
|
|
|
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.react.base import ReActDocstoreAgent
|
|
from langchain.agents.self_ask_with_search.base import SelfAskWithSearchAgent
|
|
from langchain.agents.structured_chat.base import StructuredChatAgent
|
|
|
|
AGENT_TO_CLASS: Dict[AgentType, Type[BaseSingleActionAgent]] = {
|
|
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,
|
|
}
|