diff --git a/src/backend/langflow/field_typing/__init__.py b/src/backend/langflow/field_typing/__init__.py index fb21a238e..c6680662f 100644 --- a/src/backend/langflow/field_typing/__init__.py +++ b/src/backend/langflow/field_typing/__init__.py @@ -1,27 +1,9 @@ -from .constants import ( - AgentExecutor, - BaseChatMemory, - BaseLanguageModel, - BaseLLM, - BaseLoader, - BaseMemory, - BaseOutputParser, - BasePromptTemplate, - BaseRetriever, - Callable, - Chain, - ChatPromptTemplate, - Data, - Document, - Embeddings, - NestedDict, - Object, - PromptTemplate, - TextSplitter, - Tool, - VectorStore, -) - +from .constants import (AgentExecutor, BaseChatMemory, BaseLanguageModel, + BaseLLM, BaseLoader, BaseMemory, BaseOutputParser, + BasePromptTemplate, BaseRetriever, Callable, Chain, + ChatPromptTemplate, Data, Document, Embeddings, + NestedDict, Object, Prompt, PromptTemplate, + TextSplitter, Tool, VectorStore) __all__ = [ "NestedDict", @@ -45,4 +27,5 @@ __all__ = [ "Callable", "BasePromptTemplate", "ChatPromptTemplate", + "Prompt" ] diff --git a/src/backend/langflow/field_typing/constants.py b/src/backend/langflow/field_typing/constants.py index 76ef3aa5b..028ab6aff 100644 --- a/src/backend/langflow/field_typing/constants.py +++ b/src/backend/langflow/field_typing/constants.py @@ -25,6 +25,8 @@ class Object: class Data: pass +class Prompt + LANGCHAIN_BASE_TYPES = { "Chain": Chain, @@ -44,6 +46,7 @@ LANGCHAIN_BASE_TYPES = { "BaseOutputParser": BaseOutputParser, "BaseMemory": BaseMemory, "BaseChatMemory": BaseChatMemory, + } # Langchain base types plus Python base types CUSTOM_COMPONENT_SUPPORTED_TYPES = { @@ -58,4 +61,5 @@ CUSTOM_COMPONENT_SUPPORTED_TYPES = { "Data": Data, "Object": Object, "Callable": Callable, + "Prompt": Prompt, }