jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: check for specific nodes first

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-06 15:40:16 -03:00
commit d921983a00
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/graph/graph.py
View file

@ -120,12 +120,12 @@ class Graph:
**{t: MemoryNode for t in memory_creator.to_list()}, **{t: MemoryNode for t in memory_creator.to_list()},
} }
if node_type in FILE_TOOLS:
return FileToolNode
if node_type in node_type_map: if node_type in node_type_map:
return node_type_map[node_type] return node_type_map[node_type]
if node_lc_type in node_type_map: if node_lc_type in node_type_map:
return node_type_map[node_lc_type] return node_type_map[node_lc_type]
if node_type in FILE_TOOLS:
return FileToolNode
return Node return Node
def _build_nodes(self) -> List[Node]: def _build_nodes(self) -> List[Node]: