From de796b7d7e02052f4cea4c9d2fac20b69a4312f5 Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Fri, 13 Feb 2026 15:46:17 +0800
Subject: [PATCH] bugfix

---
 dagflow/dagflow.py |  1 -
 dagflow/init.py    | 11 +++++++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/dagflow/dagflow.py b/dagflow/dagflow.py
index e962a77..3bf8d89 100644
--- a/dagflow/dagflow.py
+++ b/dagflow/dagflow.py
@@ -108,7 +108,6 @@ class FlowEngine:
 		env = ServerEnv()
 		self.aio = []
 		async with (env, 'workflow') as sor:
-			fid = env.uuid()
 			self.aio = await sor.R('flow_instance', {'status': 'running'})
 		while True:
 			for r in self.aio:
diff --git a/dagflow/init.py b/dagflow/init.py
index e3b201f..efa55fb 100644
--- a/dagflow/init.py
+++ b/dagflow/init.py
@@ -1,10 +1,17 @@
+from functools import partial
 from ahserver.serverenv import ServerEnv
-from ahserver.configuredServer import add_startup
+from ahserver.configuredServer import add_cleanupctx
 from .dagflow import FlowEngine
 
+async def dagbacktask(engine, app):
+	task = asyncio.create_task(engine.bgtask())
+	yield
+	task.cancel()
+
 def load_dagflow():
 	engine = FlowEngine()
-	add_startup(engine.bgtask)
+	f = partial(dagbacktask, engine)
+	add_cleanupctx(f)
 	env = ServerEnv()
 	env.workflow_engine = engine