add stopped flag and async for queue put

TEN-framework · Nov 5, 2024 · 2fc5766 · 2fc5766
1 parent c5db5dd
commit 2fc5766
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/agents/ten_packages/extension/glue_python_async/extension.py b/agents/ten_packages/extension/glue_python_async/extension.py
@@ -109,7 +109,8 @@ async def on_start(self, ten_env: AsyncTenEnv) -> None:
     async def on_stop(self, ten_env: AsyncTenEnv) -> None:
         ten_env.log_debug("on_stop")
 
-        self.queue.put(None)
+        self.stopped = True
+        await self.queue.put(None)
         await self._flush()
 
         ten_env.on_stop_done()