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

refactor: Fix conditional check in Graph.mark_branch method

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:48:53 -03:00
commit 4eec44e9a8
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/base/langflow/graph/graph/base.py
View file

@ -4,6 +4,7 @@ from collections import defaultdict, deque
from functools import partial from functools import partial
from itertools import chain from itertools import chain
from typing import TYPE_CHECKING, Dict, Generator, List, Optional, Tuple, Type, Union from typing import TYPE_CHECKING, Dict, Generator, List, Optional, Tuple, Type, Union
from loguru import logger from loguru import logger
from langflow.exceptions.component import ComponentBuildException from langflow.exceptions.component import ComponentBuildException
@ -489,7 +490,7 @@ class Graph:
# should be marked # should be marked
if output_name: if output_name:
edge = self.get_edge(vertex_id, child_id) edge = self.get_edge(vertex_id, child_id)
if edge.source_handle.name != output_name: if edge and edge.source_handle.name != output_name:
continue continue
self.mark_branch(child_id, state) self.mark_branch(child_id, state)