From d921983a00670e5b0c5c2a3bee084a7296b988ea Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Thu, 6 Apr 2023 15:40:16 -0300 Subject: [PATCH] fix: check for specific nodes first --- src/backend/langflow/graph/graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/graph/graph.py b/src/backend/langflow/graph/graph.py index 1e4555fc8..81c488881 100644 --- a/src/backend/langflow/graph/graph.py +++ b/src/backend/langflow/graph/graph.py @@ -120,12 +120,12 @@ class Graph: **{t: MemoryNode for t in memory_creator.to_list()}, } + if node_type in FILE_TOOLS: + return FileToolNode if node_type in node_type_map: return node_type_map[node_type] if node_lc_type in node_type_map: return node_type_map[node_lc_type] - if node_type in FILE_TOOLS: - return FileToolNode return Node def _build_nodes(self) -> List[Node]: