Update typing import in base.py

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-28 19:38:52 -03:00
commit 77227b887b

View file

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