jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-03 16:47:44 -03:00
commit 3079318d6a
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/field_typing/constants.py
View file

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