Merge pull request #14 from jhakulin/jhakulin/minor-fix-stop

Clear consume event queue in stop
jhakulin · Nov 29, 2024 · 185fa7c · 185fa7c
2 parents b3225d1 + dc3b185
commit 185fa7c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/realtime_ai/aio/realtime_ai_client.py b/src/realtime_ai/aio/realtime_ai_client.py
@@ -54,6 +54,8 @@ async def stop(self):
                         await self._consume_task
                     except asyncio.CancelledError:
                         logger.info("RealtimeAIClient: consume_events task cancelled.")
+
+                await self.service_manager.clear_event_queue()
             except Exception as e:
                 logger.error(f"RealtimeAIClient: Error during client stop: {e}")
 

diff --git a/src/realtime_ai/realtime_ai_client.py b/src/realtime_ai/realtime_ai_client.py
@@ -85,6 +85,7 @@ def stop(self, timeout: float = 5.0):
                     logger.info("RealtimeAIClient: ThreadPoolExecutor shut down.")
                     self.executor = None
 
+                self.service_manager.clear_event_queue()
                 logger.info("RealtimeAIClient: Services stopped.")
             except Exception as e:
                 logger.error(f"RealtimeAIClient: Error during client stop: {e}")