From eff5eb6af28ecdbde4a00a23c9ac6f013aa91386 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 24 Jan 2025 08:26:47 -0300 Subject: [PATCH] fix: remove lock from log transaction handling in Vertex (#5906) --- .../base/langflow/graph/vertex/base.py | 20 +++++++++---------- 1 file changed, 9 insertions(+), 11 deletions(-) diff --git a/src/backend/base/langflow/graph/vertex/base.py b/src/backend/base/langflow/graph/vertex/base.py index 08e8f441e..d21a1f598 100644 --- a/src/backend/base/langflow/graph/vertex/base.py +++ b/src/backend/base/langflow/graph/vertex/base.py @@ -15,7 +15,7 @@ from loguru import logger from langflow.exceptions.component import ComponentBuildError from langflow.graph.schema import INPUT_COMPONENTS, OUTPUT_COMPONENTS, InterfaceComponentTypes, ResultData -from langflow.graph.utils import UnbuiltObject, UnbuiltResult +from langflow.graph.utils import UnbuiltObject, UnbuiltResult, log_transaction from langflow.interface import initialize from langflow.interface.listing import lazy_load_dict from langflow.schema.artifact import ArtifactType @@ -632,17 +632,15 @@ class Vertex: target: Optional target vertex error: Optional error information """ - # Commenting this out for now - # async with self._lock: - # if self.log_transaction_tasks: - # # Safely await and remove completed tasks - # task = self.log_transaction_tasks.pop() - # await task + if self.log_transaction_tasks: + # Safely await and remove completed tasks + task = self.log_transaction_tasks.pop() + await task - # # Create and track new task - # task = asyncio.create_task(log_transaction(flow_id, source, status, target, error)) - # self.log_transaction_tasks.add(task) - # task.add_done_callback(self.log_transaction_tasks.discard) + # Create and track new task + task = asyncio.create_task(log_transaction(flow_id, source, status, target, error)) + self.log_transaction_tasks.add(task) + task.add_done_callback(self.log_transaction_tasks.discard) async def _get_result( self,