From 152f0254c0c1ef90641c3a22827e09b17fb9e9be Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 23 Jun 2024 22:34:05 -0300 Subject: [PATCH] refactor: Stop tracing service gracefully --- src/backend/base/langflow/services/tracing/service.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/services/tracing/service.py b/src/backend/base/langflow/services/tracing/service.py index 9c73890e8..c23dba24b 100644 --- a/src/backend/base/langflow/services/tracing/service.py +++ b/src/backend/base/langflow/services/tracing/service.py @@ -66,9 +66,12 @@ class TracingService(Service): try: self.running = False await self.flush() + # check the qeue is empty + if not self.logs_queue.empty(): + await self.logs_queue.join() self.worker_task.cancel() - if self.worker_task: - await self.worker_task + self.worker_task = None + except Exception as e: logger.error(f"Error stopping tracing service: {e}")