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

Fix source and target id assignment in Edge

Browse source 
constructor
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-28 14:35:04 -03:00
commit 25a1d7f3ca
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/edge/base.py
View file

@ -22,8 +22,8 @@ class TargetHandle(BaseModel):
class Edge: class Edge:
def __init__(self, source: "Vertex", target: "Vertex", edge: dict): def __init__(self, source: "Vertex", target: "Vertex", edge: dict):
self.source_id: str = source.id self.source_id: str = source.id if source else ""
self.target_id: str = target.id self.target_id: str = target.id if target else ""
if data := edge.get("data", {}): if data := edge.get("data", {}):
self._source_handle = data.get("sourceHandle", {}) self._source_handle = data.get("sourceHandle", {})
self._target_handle = data.get("targetHandle", {}) self._target_handle = data.get("targetHandle", {})