diff --git a/src/backend/langflow/field_typing/__init__.py b/src/backend/langflow/field_typing/__init__.py index 9a2161d3f..5ba3ec020 100644 --- a/src/backend/langflow/field_typing/__init__.py +++ b/src/backend/langflow/field_typing/__init__.py @@ -31,6 +31,7 @@ from .constants import ( AgentExecutor, NestedDict, Data, + Object, ) __all__ = [ @@ -50,4 +51,5 @@ __all__ = [ "TextSplitter", "Document", "AgentExecutor", + "Object", ] diff --git a/src/backend/langflow/field_typing/constants.py b/src/backend/langflow/field_typing/constants.py index 3ce429548..66abe46bd 100644 --- a/src/backend/langflow/field_typing/constants.py +++ b/src/backend/langflow/field_typing/constants.py @@ -1,3 +1,5 @@ +from typing import Dict, Union + from langchain.agents.agent import AgentExecutor from langchain.chains.base import Chain from langchain.document_loaders.base import BaseLoader @@ -10,12 +12,15 @@ from langchain.schema.memory import BaseMemory from langchain.text_splitter import TextSplitter from langchain.tools import Tool from langchain.vectorstores.base import VectorStore -from typing import Union, Dict # Type alias for more complex dicts NestedDict = Dict[str, Union[str, Dict]] +class Object: + pass + + class Data: pass @@ -47,4 +52,5 @@ CUSTOM_COMPONENT_SUPPORTED_TYPES = { "dict": dict, "NestedDict": NestedDict, "Data": Data, + "Object": Object, }