From 35a6b8956a5808545af8a22ed500636b34607aa0 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 20 Feb 2024 16:59:08 -0300 Subject: [PATCH] Update ResultDict to ResultData in Vertex class --- src/backend/langflow/graph/vertex/base.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index a5378b95e..c43b8fd0d 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -14,7 +14,7 @@ from langflow.utils.util import sync_to_async from loguru import logger if TYPE_CHECKING: - from langflow.api.v1.schemas import ResultDict + from langflow.api.v1.schemas import ResultData from langflow.graph.edge.base import ContractEdge from langflow.graph.graph.base import Graph @@ -50,7 +50,7 @@ class Vertex: self.parent_is_top_level = False self.layer = None self.should_run = True - self.result: Optional["ResultDict"] = None + self.result: Optional["ResultData"] = None try: self.is_interface_component = InterfaceComponentTypes(self.vertex_type) except ValueError: @@ -81,7 +81,7 @@ class Vertex: ) return edge_results - def set_result(self, result: "ResultDict") -> None: + def set_result(self, result: "ResultData") -> None: self.result = result def get_built_result(self):