jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Stop tracing service gracefully

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-23 22:34:05 -03:00
commit 152f0254c0
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/services/tracing/service.py
View file

@ -66,9 +66,12 @@ class TracingService(Service):
try: try:
self.running = False self.running = False
await self.flush() await self.flush()
# check the qeue is empty
if not self.logs_queue.empty():
await self.logs_queue.join()
self.worker_task.cancel() self.worker_task.cancel()
if self.worker_task: self.worker_task = None
await self.worker_task
except Exception as e: except Exception as e:
logger.error(f"Error stopping tracing service: {e}") logger.error(f"Error stopping tracing service: {e}")