From 3079318d6a621f76b1cc922b25c0b514a0eb5d68 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 3 Nov 2023 16:47:44 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(constants.py):=20import=20?= =?UTF-8?q?BasePromptTemplate=20to=20improve=20code=20organization=20and?= =?UTF-8?q?=20maintainability?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/field_typing/constants.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/field_typing/constants.py b/src/backend/langflow/field_typing/constants.py index 2cbc09bb2..4e508ba61 100644 --- a/src/backend/langflow/field_typing/constants.py +++ b/src/backend/langflow/field_typing/constants.py @@ -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,