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"]