Update ResultDict to ResultData in Vertex class

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-20 16:59:08 -03:00
commit 35a6b8956a

View file

@ -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):