Skip to content

Commit

Permalink
fix: Change error handling to return default Vertex for unknown node …
Browse files Browse the repository at this point in the history
…types in graph class
  • Loading branch information
ogabrielluiz committed Aug 6, 2024
1 parent b381e01 commit 1bc4403
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/backend/base/langflow/graph/graph/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -1422,7 +1422,7 @@ def _get_vertex_class(self, node_type: str, node_base_type: str, node_id: str) -

if node_type in lazy_load_vertex_dict.VERTEX_TYPE_MAP:
return lazy_load_vertex_dict.VERTEX_TYPE_MAP[node_type]
raise ValueError(f"Vertex type {node_type} not found")
return Vertex

def _build_vertices(self) -> List["Vertex"]:
"""Builds the vertices of the graph."""
Expand Down

0 comments on commit 1bc4403

Please sign in to comment.