From 1df99008a10a8d1701956cd49660dac955a5afd1 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 26 Feb 2024 23:03:16 -0300 Subject: [PATCH] Fix vertex building error and improve vertex data comparison --- src/backend/langflow/graph/graph/base.py | 5 ++++- src/backend/langflow/graph/vertex/base.py | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py index 400ae99b9..2f94f150a 100644 --- a/src/backend/langflow/graph/graph/base.py +++ b/src/backend/langflow/graph/graph/base.py @@ -162,7 +162,10 @@ class Graph: self.edges = new_edges def vertex_data_is_identical(self, vertex: Vertex, other_vertex: Vertex) -> bool: - return vertex.__repr__() == other_vertex.__repr__() + data_is_equivalent = vertex.__repr__() == other_vertex.__repr__() + if not data_is_equivalent: + return False + return self.vertex_edges_are_identical(vertex, other_vertex) def vertex_edges_are_identical(self, vertex: Vertex, other_vertex: Vertex) -> bool: same_length = len(vertex.edges) == len(other_vertex.edges) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index 8addb1ac7..874d4bd21 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -516,6 +516,7 @@ class Vertex: self._update_built_object_and_artifacts(result) except Exception as exc: logger.exception(exc) + raise ValueError( f"Error building node {self.display_name}: {str(exc)}" ) from exc