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