From c88d63546fbb107e1b805f296a6df58e2065dd12 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 14 Nov 2023 21:00:43 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(field=5Ftyping):=20import=20?= =?UTF-8?q?Object=20class=20in=20=5F=5Finit=5F=5F.py=20to=20fix=20missing?= =?UTF-8?q?=20import=20error=20=F0=9F=94=A7=20fix(constants.py):=20add=20O?= =?UTF-8?q?bject=20class=20to=20CUSTOM=5FCOMPONENT=5FSUPPORTED=5FTYPES=20t?= =?UTF-8?q?o=20support=20object=20type=20in=20field=20typing?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/field_typing/__init__.py | 2 ++ src/backend/langflow/field_typing/constants.py | 8 +++++++- 2 files changed, 9 insertions(+), 1 deletion(-) 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, }