neuro-simulator 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

neuro_simulator/api/stream.py

@@ -33,7 +33,7 @@ async def start_stream():
 async def stop_stream():
     """Stops the live stream processes."""
     if process_manager.is_running:
-        process_manager.stop_live_processes()
+        await process_manager.stop_live_processes()
         return {"status": "success", "message": "Stream stopped"}
     else:
         return {"status": "info", "message": "Stream is not running"}
@@ -41,7 +41,7 @@ async def stop_stream():
 @router.post("/restart", dependencies=[Depends(get_api_token)])
 async def restart_stream():
     """Restarts the live stream processes."""
-    process_manager.stop_live_processes()
+    await process_manager.stop_live_processes()
     await asyncio.sleep(1) # Give time for tasks to cancel
     process_manager.start_live_processes()
     return {"status": "success", "message": "Stream restarted"}

neuro_simulator/utils/process.py

@@ -2,6 +2,9 @@
 import asyncio
 import logging
 
+from ..services.stream import live_stream_manager
+from .websocket import connection_manager
+
 logger = logging.getLogger(__name__.replace("neuro_simulator", "server", 1))
 
 class ProcessManager:
@@ -28,7 +31,6 @@ class ProcessManager:
 
         logger.info("Starting core stream processes...")
         from ..core.application import generate_audience_chat_task, neuro_response_cycle, broadcast_events_task
-        from ..services.stream import live_stream_manager
         from ..utils.queue import clear_all_queues
         from ..core.agent_factory import create_agent
         
@@ -45,10 +47,14 @@ class ProcessManager:
         self._is_running = True
         logger.info(f"Core processes started: {len(self._tasks)} tasks.")
 
-    def stop_live_processes(self):
+    async def stop_live_processes(self):
         """Stops and cleans up all running background tasks."""
         if not self.is_running:
             return
+        
+        logger.info("Broadcasting offline message before stopping tasks...")
+        await connection_manager.broadcast({"type": "offline"})
+        await asyncio.sleep(0.1) # Give a brief moment for the message to be sent
             
         logger.info(f"Stopping {len(self._tasks)} core tasks...")
         for task in self._tasks:
@@ -58,7 +64,6 @@ class ProcessManager:
         self._tasks.clear()
         self._is_running = False
         
-        from ..services.stream import live_stream_manager
         live_stream_manager.reset_stream_state()
         
         logger.info("All core tasks have been stopped.")

{neuro_simulator-0.2.1.dist-info → neuro_simulator-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuro_simulator
-Version: 0.2.1
+Version: 0.2.2
 Summary: Neuro Simulator Server
 Author-email: Moha-Master <hongkongreporter@outlook.com>
 License-Expression: MIT

{neuro_simulator-0.2.1.dist-info → neuro_simulator-0.2.2.dist-info}/RECORD

@@ -11,7 +11,7 @@ neuro_simulator/agent/tools/__init__.py,sha256=1WZy6PADfi6o1avyy1y-ThWBFAPJ_bBqt
 neuro_simulator/agent/tools/core.py,sha256=o6Oyis-HFD-g6Z_u3T--tkmr9ylKJvybKqMRSMUwi1Q,5555
 neuro_simulator/api/__init__.py,sha256=5LWyDSayPGdQS8Rv13nmAKLyhPnMVPyTYDdvoMPB4xw,56
 neuro_simulator/api/agent.py,sha256=ABl_JoIxB4wW_J2J52bWndmTXkfGJBS5LZmbGuh7zv8,6343
-neuro_simulator/api/stream.py,sha256=Pg793dXLnJ0oydrIwWFgWD0cVdFAx_QOw6L5LjE0uIo,2069
+neuro_simulator/api/stream.py,sha256=Yg-cwjVI2HTLAt7moWbDQ1ZbbP6QE1ZB8LrW6A3tcQk,2081
 neuro_simulator/api/system.py,sha256=hXznMcThuFhwopYWgpzrRxwtBuFnF_b_vinkOaE5XOs,3712
 neuro_simulator/core/__init__.py,sha256=-ojq25c8XA0CU25b0OxcGjH4IWFEDHR-HXSRSZIuKe8,57
 neuro_simulator/core/agent_factory.py,sha256=qMFidwT5IrOkyNHwmpO8_fRv20KLbaIBfWF-VTFCLNA,1742
@@ -26,12 +26,12 @@ neuro_simulator/services/letta.py,sha256=iXyzFyPVty3SjYCHOJAF1QqhUFmxTQbqk8lYJOw
 neuro_simulator/services/stream.py,sha256=dG7RuNI_ICohPkqKZ-zlBppo54BgWm_KYBs-ezzc73E,5907
 neuro_simulator/utils/__init__.py,sha256=xSEFzjT827W81mNyQ_DLtr00TgFlttqfFgpz9pSxFXQ,58
 neuro_simulator/utils/logging.py,sha256=BO-q_cCcoeamsc8eJqq2-L3Z8nhApze_v6LnmD-O8Ww,3411
-neuro_simulator/utils/process.py,sha256=FmwIuOsp19jJBEbAj-WeUpdUsQN7B9_J1rlU0Q1TXuo,2388
+neuro_simulator/utils/process.py,sha256=9w2JQH59Wy6L8ADrig2QF88iajdykGPZIYJVJe6Al2Y,2603
 neuro_simulator/utils/queue.py,sha256=7SSnUnrqxHkqSeYVjp1S5kpr8qlo4IVdU2NsGwFQAVw,1546
 neuro_simulator/utils/state.py,sha256=E1ilecgMTOAhiw_kNvLG0akkhdZKhzKLrA3oB4NNVTA,538
 neuro_simulator/utils/websocket.py,sha256=yOdFvJzbNhcUn5EAuyS55G_R8q-snas5OvkOtS8g19E,2292
-neuro_simulator-0.2.1.dist-info/METADATA,sha256=bbJ6VZxoq8fqq0qVzjm0nYUKIhwfoNPNQd8hDKPSREY,8676
-neuro_simulator-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-neuro_simulator-0.2.1.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
-neuro_simulator-0.2.1.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
-neuro_simulator-0.2.1.dist-info/RECORD,,
+neuro_simulator-0.2.2.dist-info/METADATA,sha256=lBjb1G0VhfbDCQhRWQOxJzlt9Ut07XL6zApn6X6G_jo,8676
+neuro_simulator-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+neuro_simulator-0.2.2.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
+neuro_simulator-0.2.2.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
+neuro_simulator-0.2.2.dist-info/RECORD,,