From 77227b887b702d27b9736b8cd5dafc240e022fe9 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 Nov 2023 19:38:52 -0300 Subject: [PATCH] Update typing import in base.py --- src/backend/langflow/graph/vertex/base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index 8d25e502c..29cf55293 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -1,7 +1,7 @@ import ast import inspect import types -from typing import TYPE_CHECKING, Any, Dict, List, Optional +from typing import TYPE_CHECKING, Any, Coroutine, Dict, List, Optional from langflow.graph.utils import UnbuiltObject from langflow.interface.initialize import loading @@ -245,7 +245,10 @@ class Vertex: if self.is_task and self.task_id is not None: task = self.get_task() + result = task.get(timeout=timeout) + if isinstance(result, Coroutine): + result = await result if result is not None: # If result is ready self._update_built_object_and_artifacts(result) return self._built_object