From 0b2a07ec040dfe6a1704920b7a9c4a414ae4e581 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 16 Aug 2024 15:19:59 -0300 Subject: [PATCH] fix: Improve logic to consider target handle name in ComponentVertex. Fixes #3380 --- src/backend/base/langflow/graph/vertex/types.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/graph/vertex/types.py b/src/backend/base/langflow/graph/vertex/types.py index 6eb0dae9e99f..1c60b54618dc 100644 --- a/src/backend/base/langflow/graph/vertex/types.py +++ b/src/backend/base/langflow/graph/vertex/types.py @@ -114,7 +114,11 @@ async def _get_result(self, requester: "Vertex", target_handle_name: str | None edges = self.get_edge_with_target(requester.id) result = UNDEFINED for edge in edges: - if edge is not None and edge.source_handle.name in self.results: + if ( + edge is not None + and edge.source_handle.name in self.results + and edge.target_handle.field_name == target_handle_name + ): # Get the result from the output instead of the results dict try: output = self.get_output(edge.source_handle.name)