neuro-simulator 0.3.1__py3-none-any.whl → 0.3.2__py3-none-any.whl

neuro_simulator/api/system.py

@@ -6,6 +6,7 @@ import time
 
 from ..core.config import config_manager
 
+
 router = APIRouter(tags=["System & Utilities"])
 
 
@@ -57,4 +58,6 @@ async def root():
         "message": "Neuro-Sama Simulator Backend",
         "version": "2.0",
         "api_docs": "/docs",
-    }
+    }
+
+

neuro_simulator/core/application.py

@@ -287,6 +287,10 @@ async def websocket_admin_endpoint(websocket: WebSocket):
         initial_context = await agent.get_message_history()
         await websocket.send_json({"type": "agent_context", "action": "update", "messages": initial_context})
         
+        # Send initial stream status
+        status = {"is_running": process_manager.is_running, "backend_status": "running" if process_manager.is_running else "stopped"}
+        await websocket.send_json({"type": "stream_status", "payload": status})
+        
         # Main loop for receiving messages from the client and pushing log updates
         while websocket.client_state == WebSocketState.CONNECTED:
             # Check for incoming messages
@@ -391,17 +395,29 @@ async def handle_admin_ws_message(websocket: WebSocket, data: dict):
             if not process_manager.is_running:
                 process_manager.start_live_processes()
             response["payload"] = {"status": "success", "message": "Stream started"}
+            # Broadcast stream status update
+            from ..utils.websocket import connection_manager
+            status = {"is_running": process_manager.is_running, "backend_status": "running" if process_manager.is_running else "stopped"}
+            await connection_manager.broadcast_to_admins({"type": "stream_status", "payload": status})
 
         elif action == "stop_stream":
             if process_manager.is_running:
                 await process_manager.stop_live_processes()
             response["payload"] = {"status": "success", "message": "Stream stopped"}
+            # Broadcast stream status update
+            from ..utils.websocket import connection_manager
+            status = {"is_running": process_manager.is_running, "backend_status": "running" if process_manager.is_running else "stopped"}
+            await connection_manager.broadcast_to_admins({"type": "stream_status", "payload": status})
 
         elif action == "restart_stream":
             await process_manager.stop_live_processes()
             await asyncio.sleep(1)
             process_manager.start_live_processes()
             response["payload"] = {"status": "success", "message": "Stream restarted"}
+            # Broadcast stream status update
+            from ..utils.websocket import connection_manager
+            status = {"is_running": process_manager.is_running, "backend_status": "running" if process_manager.is_running else "stopped"}
+            await connection_manager.broadcast_to_admins({"type": "stream_status", "payload": status})
 
         elif action == "get_stream_status":
             status = {"is_running": process_manager.is_running, "backend_status": "running" if process_manager.is_running else "stopped"}
@@ -432,6 +448,24 @@ async def handle_admin_ws_message(websocket: WebSocket, data: dict):
             context = await agent.get_message_history()
             response["payload"] = context
 
+        elif action == "get_last_prompt":
+            # Check if the agent supports prompt generation introspection
+            agent_instance = getattr(agent, 'agent_instance', None) if hasattr(agent, 'agent_instance') else agent
+            if not hasattr(agent_instance, 'memory_manager') or not hasattr(agent_instance.memory_manager, 'get_recent_chat') or not hasattr(agent_instance, '_build_neuro_prompt'):
+                response["payload"] = {"status": "error", "message": "The active agent does not support prompt generation introspection."}
+            else:
+                recent_history = await agent_instance.memory_manager.get_recent_chat(entries=10)
+                messages_for_prompt = []
+                for entry in recent_history:
+                    if entry.get('role') == 'user':
+                        parts = entry.get('content', '').split(':', 1)
+                        if len(parts) == 2:
+                            messages_for_prompt.append({'username': parts[0].strip(), 'text': parts[1].strip()})
+                        else:
+                            messages_for_prompt.append({'username': 'user', 'text': entry.get('content', '')})
+                prompt = await agent_instance._build_neuro_prompt(messages_for_prompt)
+                response["payload"] = {"prompt": prompt}
+
         elif action == "reset_agent_memory":
             await agent.reset_memory()
             response["payload"] = {"status": "success"}

neuro_simulator/utils/process.py

@@ -33,6 +33,7 @@ class ProcessManager:
         from ..core.application import generate_audience_chat_task, neuro_response_cycle, broadcast_events_task
         from ..utils.queue import clear_all_queues
         from ..core.agent_factory import create_agent
+        from ..utils.websocket import connection_manager
         
         asyncio.create_task(create_agent())
         
@@ -45,6 +46,9 @@ class ProcessManager:
         self._tasks.append(asyncio.create_task(neuro_response_cycle()))
         
         self._is_running = True
+        # Broadcast stream status update
+        status = {"is_running": self._is_running, "backend_status": "running" if self._is_running else "stopped"}
+        asyncio.create_task(connection_manager.broadcast_to_admins({"type": "stream_status", "payload": status}))
         logger.info(f"Core processes started: {len(self._tasks)} tasks.")
 
     async def stop_live_processes(self):
@@ -66,6 +70,10 @@ class ProcessManager:
         
         live_stream_manager.reset_stream_state()
         
+        # Broadcast stream status update
+        status = {"is_running": self._is_running, "backend_status": "running" if self._is_running else "stopped"}
+        await connection_manager.broadcast_to_admins({"type": "stream_status", "payload": status})
+        
         logger.info("All core tasks have been stopped.")
 
 # Global singleton instance

{neuro_simulator-0.3.1.dist-info → neuro_simulator-0.3.2.dist-info}/METADATA

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

{neuro_simulator-0.3.1.dist-info → neuro_simulator-0.3.2.dist-info}/RECORD

@@ -11,11 +11,11 @@ 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/stream.py,sha256=hM66flSUygpE-NH9X-ZOV6SiGipBzN1-wjd_wZRpQm4,94
-neuro_simulator/api/system.py,sha256=TV_3DzP-VgLnHdue0pKYFFQNOV_V7tr4aMuxL802vQg,1783
+neuro_simulator/api/system.py,sha256=W05Q41BYAFrw-MTnJ5YJrBG2S1SmTcByoex77GfUaFQ,1787
 neuro_simulator/core/__init__.py,sha256=-ojq25c8XA0CU25b0OxcGjH4IWFEDHR-HXSRSZIuKe8,57
 neuro_simulator/core/agent_factory.py,sha256=qMFidwT5IrOkyNHwmpO8_fRv20KLbaIBfWF-VTFCLNA,1742
 neuro_simulator/core/agent_interface.py,sha256=ZXUCtkQUvsBQ5iCb0gTILJaShn5KmSrEgKhd7PK18HE,2794
-neuro_simulator/core/application.py,sha256=lhTn5KR7ek8-kuyD1qWtppImUUz8pOExH9KRNW1SQ3M,21115
+neuro_simulator/core/application.py,sha256=zQ6QRUU-dyFW2GDKyBxiTO1BKh-Yt6j169YS-MLpalk,23734
 neuro_simulator/core/config.py,sha256=brA8kiekV_995mpz04JiGj1swIWbZZuWWLNYtbroMyE,14884
 neuro_simulator/services/__init__.py,sha256=s3ZrAHg5TpJakadAAGY1h0wDw_xqN4Je4aJwJyRBmk4,61
 neuro_simulator/services/audience.py,sha256=0phlhsujh_GMXm_UMiyOntY-ZMtoseRa_FroIfc5A6w,5028
@@ -25,12 +25,12 @@ neuro_simulator/services/letta.py,sha256=6jBvOTsLMlRILDv-fvX9fhHMONSYeu-ImJGFcKU
 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=9w2JQH59Wy6L8ADrig2QF88iajdykGPZIYJVJe6Al2Y,2603
+neuro_simulator/utils/process.py,sha256=9OYWx8fzaJZqmFUcjQX37AnBhl7YWvrLxDWBa30vqwU,3192
 neuro_simulator/utils/queue.py,sha256=vSkh-BrgfsGN_gDAx2mfK44ydmMapdVyLsDG-7LIJZQ,1643
 neuro_simulator/utils/state.py,sha256=DdBqSAYfjOFtJfB1hEGhYPh32r1ZvFuVlN_-29_-luA,664
 neuro_simulator/utils/websocket.py,sha256=1gtVoH1hafBUfVYmwkVDAbjwETeqyC3sWx706nQzSRo,2085
-neuro_simulator-0.3.1.dist-info/METADATA,sha256=lQv9x0KvOZJAA8NwZ8byIaHYTVnSxuHRPyhUyqLC6vw,8643
-neuro_simulator-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-neuro_simulator-0.3.1.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
-neuro_simulator-0.3.1.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
-neuro_simulator-0.3.1.dist-info/RECORD,,
+neuro_simulator-0.3.2.dist-info/METADATA,sha256=HBAqQoRRJmPcm1IxQwderrq7cTO-UhF-ulzqixLNPh8,8643
+neuro_simulator-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+neuro_simulator-0.3.2.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
+neuro_simulator-0.3.2.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
+neuro_simulator-0.3.2.dist-info/RECORD,,