neuro-simulator 0.3.1__tar.gz → 0.3.2__tar.gz

{neuro_simulator-0.3.1 → neuro_simulator-0.3.2}/PKG-INFO

@@ -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 → neuro_simulator-0.3.2}/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-0.3.1 → neuro_simulator-0.3.2}/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-0.3.1 → neuro_simulator-0.3.2}/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 → neuro_simulator-0.3.2}/neuro_simulator.egg-info/PKG-INFO

@@ -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 → neuro_simulator-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "neuro_simulator"
-version = "0.3.1"
+version = "0.3.2"
 description = "Neuro Simulator Server"
 readme = "README.md"
 authors = [{ name = "Moha-Master", email = "hongkongreporter@outlook.com" }]