From 5277798cca08223fd53b1a2acfc18f9c7a36e463 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 26 Jan 2024 17:07:07 -0300 Subject: [PATCH] Refactor Vertex class to make _run method asynchronous --- src/backend/langflow/graph/vertex/base.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index 024017b2e..64e005756 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -4,13 +4,12 @@ import types from enum import Enum from typing import TYPE_CHECKING, Any, Callable, Coroutine, Dict, List, Optional -from loguru import logger - from langflow.graph.utils import UnbuiltObject, UnbuiltResult from langflow.interface.initialize import loading from langflow.interface.listing import lazy_load_dict from langflow.utils.constants import DIRECT_TYPES from langflow.utils.util import sync_to_async +from loguru import logger if TYPE_CHECKING: from langflow.graph.edge.base import ContractEdge @@ -284,7 +283,7 @@ class Vertex: self._built = True - def _run(self, user_id: str, inputs: Optional[dict] = None): + async def _run(self, user_id: str, inputs: Optional[dict] = None): # user_id is just for compatibility with the other build methods inputs = inputs or {} # inputs = {key: value or "" for key, value in inputs.items()}