agentgui 1.0.392 → 1.0.393

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentgui",
-  "version": "1.0.392",
+  "version": "1.0.393",
   "description": "Multi-agent ACP client with real-time communication",
   "type": "module",
   "main": "server.js",

package/server.js

@@ -1825,6 +1825,12 @@ const server = http.createServer(async (req, res) => {
       return;
     }
 
+    if (pathOnly === '/api/ws-stats' && req.method === 'GET') {
+      const stats = wsOptimizer.getStats();
+      sendJSON(req, res, 200, stats);
+      return;
+    }
+
     if (pathOnly === '/api/agents/search' && req.method === 'POST') {
       const body = await parseBody(req);
       const result = queries.searchAgents(discoveredAgents, body);
@@ -3732,6 +3738,7 @@ wss.on('connection', (ws, req) => {
     ws.on('close', () => {
       if (ws.terminalProc) { try { ws.terminalProc.kill(); } catch(e) {} ws.terminalProc = null; }
       syncClients.delete(ws);
+      wsOptimizer.removeClient(ws);
       for (const sub of ws.subscriptions) {
         const idx = subscriptionIndex.get(sub);
         if (idx) { idx.delete(ws); if (idx.size === 0) subscriptionIndex.delete(sub); }
@@ -3785,6 +3792,7 @@ const heartbeatInterval = setInterval(() => {
   syncClients.forEach(ws => {
     if (!ws.isAlive) {
       syncClients.delete(ws);
+      wsOptimizer.removeClient(ws);
       return ws.terminate();
     }
     ws.isAlive = false;