🚀 feat(constants.py): import AgentExecutor from langchain.agents.agent module to use it in LANGCHAIN_BASE_TYPES dictionary

🔀 chore(constants.py): remove unnecessary import statements
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-10 13:45:58 -03:00
commit 7933667973

View file

@ -8,10 +8,11 @@ from langchain.text_splitter import TextSplitter
from langchain.tools import Tool from langchain.tools import Tool
from langchain.vectorstores.base import VectorStore from langchain.vectorstores.base import VectorStore
from langchain.schema import BaseOutputParser from langchain.schema import BaseOutputParser
from langchain.agents.agent import AgentExecutor
LANGCHAIN_BASE_TYPES = { LANGCHAIN_BASE_TYPES = {
"Chain": Chain, "Chain": Chain,
"AgentExecutor": AgentExecutor,
"Tool": Tool, "Tool": Tool,
"BaseLLM": BaseLLM, "BaseLLM": BaseLLM,
"PromptTemplate": PromptTemplate, "PromptTemplate": PromptTemplate,