From 25a1d7f3cacdb2f9d8a71683da4c6355fbec7571 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 Nov 2023 14:35:04 -0300 Subject: [PATCH] Fix source and target id assignment in Edge constructor --- src/backend/langflow/graph/edge/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/graph/edge/base.py b/src/backend/langflow/graph/edge/base.py index c9c1077f6..399638e18 100644 --- a/src/backend/langflow/graph/edge/base.py +++ b/src/backend/langflow/graph/edge/base.py @@ -22,8 +22,8 @@ class TargetHandle(BaseModel): class Edge: def __init__(self, source: "Vertex", target: "Vertex", edge: dict): - self.source_id: str = source.id - self.target_id: str = target.id + self.source_id: str = source.id if source else "" + self.target_id: str = target.id if target else "" if data := edge.get("data", {}): self._source_handle = data.get("sourceHandle", {}) self._target_handle = data.get("targetHandle", {})