Add successors_ids property to Vertex class
This commit is contained in:
parent
a48b497b28
commit
8a4e9a7226
1 changed files with 6 additions and 1 deletions
|
|
@ -3,6 +3,8 @@ import inspect
|
|||
import types
|
||||
from typing import TYPE_CHECKING, Any, Callable, Coroutine, Dict, List, Optional
|
||||
|
||||
from loguru import logger
|
||||
|
||||
from langflow.graph.schema import InterfaceComponentTypes
|
||||
from langflow.graph.utils import UnbuiltObject, UnbuiltResult
|
||||
from langflow.graph.vertex.utils import generate_result
|
||||
|
|
@ -11,7 +13,6 @@ from langflow.interface.listing import lazy_load_dict
|
|||
from langflow.services.deps import get_storage_service
|
||||
from langflow.utils.constants import DIRECT_TYPES
|
||||
from langflow.utils.util import sync_to_async
|
||||
from loguru import logger
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from langflow.api.v1.schemas import ResultData
|
||||
|
|
@ -121,6 +122,10 @@ class Vertex:
|
|||
def successors(self) -> List["Vertex"]:
|
||||
return self.graph.get_successors(self)
|
||||
|
||||
@property
|
||||
def successors_ids(self) -> List[str]:
|
||||
return self.graph.successor_map.get(self.id, [])
|
||||
|
||||
def __getstate__(self):
|
||||
return {
|
||||
"_data": self._data,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue