From 9f00421d64fe958bd0e1ca1e1eef58917355a1fe Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 23 Jun 2023 09:13:09 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A5=20chore(constants.py):=20remove=20?= =?UTF-8?q?DIRECT=5FTYPES=20constant=20from=20vertex=20and=20graph=20modul?= =?UTF-8?q?es=20The=20DIRECT=5FTYPES=20constant=20was=20removed=20from=20t?= =?UTF-8?q?he=20vertex=20and=20graph=20modules=20as=20it=20is=20now=20defi?= =?UTF-8?q?ned=20in=20the=20utils=20module.=20This=20change=20improves=20c?= =?UTF-8?q?ode=20organization=20and=20reduces=20duplication.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/graph/graph/constants.py | 3 --- src/backend/langflow/graph/vertex/constants.py | 2 +- src/backend/langflow/utils/constants.py | 1 + 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/graph/graph/constants.py b/src/backend/langflow/graph/graph/constants.py index ff1317d39..3398f253f 100644 --- a/src/backend/langflow/graph/graph/constants.py +++ b/src/backend/langflow/graph/graph/constants.py @@ -30,9 +30,6 @@ from langflow.interface.wrappers.base import wrapper_creator from typing import Dict, Type -DIRECT_TYPES = ["str", "bool", "code", "int", "float", "Any", "prompt"] - - VERTEX_TYPE_MAP: Dict[str, Type[Vertex]] = { **{t: PromptVertex for t in prompt_creator.to_list()}, **{t: AgentVertex for t in agent_creator.to_list()}, diff --git a/src/backend/langflow/graph/vertex/constants.py b/src/backend/langflow/graph/vertex/constants.py index 8372e13a7..8b1378917 100644 --- a/src/backend/langflow/graph/vertex/constants.py +++ b/src/backend/langflow/graph/vertex/constants.py @@ -1 +1 @@ -DIRECT_TYPES = ["str", "bool", "code", "int", "float", "Any", "prompt"] + diff --git a/src/backend/langflow/utils/constants.py b/src/backend/langflow/utils/constants.py index bf59c6cc4..44103c2b7 100644 --- a/src/backend/langflow/utils/constants.py +++ b/src/backend/langflow/utils/constants.py @@ -36,3 +36,4 @@ def python_function(text: str) -> str: \"\"\"This is a default python function that returns the input text\"\"\" return text """ +DIRECT_TYPES = ["str", "bool", "code", "int", "float", "Any", "prompt"]