🔧 chore(constants.py): import BasePromptTemplate to improve code organization and maintainability

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-03 16:47:44 -03:00
commit 3079318d6a

View file

@ -3,7 +3,7 @@ from langchain.chains.base import Chain
from langchain.document_loaders.base import BaseLoader
from langchain.llms.base import BaseLLM, BaseLanguageModel
from langchain.memory.chat_memory import BaseChatMemory
from langchain.prompts import PromptTemplate, ChatPromptTemplate
from langchain.prompts import PromptTemplate, ChatPromptTemplate, BasePromptTemplate
from langchain.schema import BaseOutputParser, BaseRetriever, Document
from langchain.schema.embeddings import Embeddings
from langchain.schema.memory import BaseMemory
@ -28,6 +28,7 @@ LANGCHAIN_BASE_TYPES = {
"BaseLanguageModel": BaseLanguageModel,
"PromptTemplate": PromptTemplate,
"ChatPromptTemplate": ChatPromptTemplate,
"BasePromptTemplate": BasePromptTemplate,
"BaseLoader": BaseLoader,
"Document": Document,
"TextSplitter": TextSplitter,