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

Add successors_ids property to Vertex class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-22 18:26:32 -03:00
commit 8a4e9a7226
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/graph/vertex/base.py
View file

@ -3,6 +3,8 @@ import inspect
import types import types
from typing import TYPE_CHECKING, Any, Callable, Coroutine, Dict, List, Optional from typing import TYPE_CHECKING, Any, Callable, Coroutine, Dict, List, Optional
from loguru import logger
from langflow.graph.schema import InterfaceComponentTypes from langflow.graph.schema import InterfaceComponentTypes
from langflow.graph.utils import UnbuiltObject, UnbuiltResult from langflow.graph.utils import UnbuiltObject, UnbuiltResult
from langflow.graph.vertex.utils import generate_result 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.services.deps import get_storage_service
from langflow.utils.constants import DIRECT_TYPES from langflow.utils.constants import DIRECT_TYPES
from langflow.utils.util import sync_to_async from langflow.utils.util import sync_to_async
from loguru import logger
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.api.v1.schemas import ResultData from langflow.api.v1.schemas import ResultData
@ -121,6 +122,10 @@ class Vertex:
def successors(self) -> List["Vertex"]: def successors(self) -> List["Vertex"]:
return self.graph.get_successors(self) return self.graph.get_successors(self)
@property
def successors_ids(self) -> List[str]:
return self.graph.successor_map.get(self.id, [])
def __getstate__(self): def __getstate__(self):
return { return {
"_data": self._data, "_data": self._data,