@yeaft/webchat-agent 1.0.415 → 1.0.417

package/local-runtime/server/handlers/agent-browser.js

@@ -0,0 +1,308 @@
+import { sendToAgent, sendToWebClient } from '../ws-utils.js';
+import {
+  browserClientForPeer,
+  browserPeers,
+  browserRoutes,
+  completeBrowserRequest,
+  deleteBrowserPeer,
+  deleteBrowserRoute,
+  getBrowserPeer,
+  getBrowserRoute,
+  installBrowserRoute,
+} from '../browser-runtime-routes.js';
+
+const AGENT_BROWSER_TYPES = new Set([
+  'browser_session_created',
+  'browser_session_error',
+  'browser_session_snapshot',
+  'browser_session_list_result',
+  'browser_peer_prepared',
+  'browser_peer_offer',
+  'browser_peer_ice_candidate',
+  'browser_peer_state',
+  'browser_peer_error',
+]);
+
+function clean(value, max = 512) {
+  return typeof value === 'string' ? value.trim().slice(0, max) : '';
+}
+
+function sessionSnapshot(msg) {
+  return {
+    browserSessionId: clean(msg?.browserSessionId),
+    revision: Number(msg?.revision) || 1,
+    state: clean(msg?.state, 32) || 'unknown',
+    activeUrl: clean(msg?.activeUrl, 4096) || 'about:blank',
+    title: clean(msg?.title, 512),
+    pageRevision: Number(msg?.pageRevision) || 1,
+    captureMode: clean(msg?.captureMode, 32) || null,
+    viewport: msg?.viewport && typeof msg.viewport === 'object' ? {
+      width: Number(msg.viewport.width) || 0,
+      height: Number(msg.viewport.height) || 0,
+      deviceScaleFactor: Number(msg.viewport.deviceScaleFactor) || 1,
+    } : null,
+    viewerCount: Math.max(0, Number(msg?.viewerCount) || 0),
+    interactivePeerCount: Math.max(0, Number(msg?.interactivePeerCount) || 0),
+    authorizedProducerCount: Math.max(0, Number(msg?.authorizedProducerCount) || 0),
+    expiresAt: Number(msg?.expiresAt) || null,
+    terminalReason: clean(msg?.terminalReason, 128) || null,
+    safeError: clean(msg?.safeError, 500) || null,
+    sourceRef: msg?.sourceRef && typeof msg.sourceRef === 'object' ? {
+      kind: clean(msg.sourceRef.kind, 32),
+      sessionId: clean(msg.sourceRef.sessionId) || undefined,
+      conversationId: clean(msg.sourceRef.conversationId) || undefined,
+      workItemId: clean(msg.sourceRef.workItemId) || undefined,
+    } : null,
+  };
+}
+
+function publicSessionMessage(agentId, msg) {
+  if (msg?.type === 'browser_session_error') {
+    return {
+      type: 'browser_session_error',
+      agentId,
+      requestId: clean(msg.requestId) || null,
+      browserSessionId: clean(msg.browserSessionId) || null,
+      code: clean(msg.code, 128) || 'browser_runtime_error',
+      safeError: clean(msg.safeError, 500) || null,
+    };
+  }
+  if (msg?.type === 'browser_session_list_result') {
+    return {
+      type: msg.type,
+      agentId,
+      requestId: clean(msg.requestId) || null,
+      sessions: (Array.isArray(msg.sessions) ? msg.sessions : []).slice(0, 64).map(sessionSnapshot),
+    };
+  }
+  return {
+    type: clean(msg?.type, 64),
+    agentId,
+    requestId: clean(msg?.requestId) || null,
+    ...sessionSnapshot(msg),
+  };
+}
+
+async function sendCreateResponse(agentId, msg) {
+  const request = completeBrowserRequest(agentId, msg);
+  if (!request) return true;
+  const client = browserClientForPeer({
+    clientId: request.clientId,
+    webConnectionId: request.webConnectionId,
+    webConnectionGeneration: request.webConnectionGeneration,
+  });
+  if (!client || client.userId !== request.ownerUserId) return true;
+  if (msg.type === 'browser_session_created') {
+    const route = installBrowserRoute({ agentId, ownerUserId: request.ownerUserId, msg });
+    if (!route) {
+      await sendToWebClient(client, {
+        type: 'browser_session_error',
+        agentId,
+        requestId: request.requestId,
+        browserSessionId: clean(msg.browserSessionId) || null,
+        code: 'browser_route_capacity',
+        safeError: 'browser_route_capacity',
+      });
+      await sendToAgent(agent, {
+        type: 'browser_session_close',
+        agentId,
+        requestId: `server-cleanup-${request.serverRequestId}`,
+        browserSessionId: clean(msg.browserSessionId),
+        expectedRevision: Number(msg.revision) || 1,
+        serverIdentity: {
+          ownerUserId: request.ownerUserId,
+          clientId: request.clientId,
+          webConnectionId: request.webConnectionId,
+          webConnectionGeneration: request.webConnectionGeneration,
+        },
+      }).catch(() => {});
+      return true;
+    }
+  }
+  const publicResponse = publicSessionMessage(agentId, request.response);
+  // Create request replay is served from the ledger. Persist only the explicit
+  // public projection, never the Agent-originated object.
+  request.response = publicResponse;
+  await sendToWebClient(client, publicResponse);
+  return true;
+}
+
+/** Agent-authenticated Browser events routed only through Server-owned ledgers. */
+export async function handleAgentBrowser(agentId, agent, msg) {
+  if (!AGENT_BROWSER_TYPES.has(msg?.type)) return false;
+  if (msg.type === 'browser_session_created'
+      || (msg.type === 'browser_session_error' && msg.requestId)) {
+    return sendCreateResponse(agentId, msg);
+  }
+
+  if (msg.type === 'browser_session_list_result') {
+    const request = completeBrowserRequest(agentId, msg, { consume: true });
+    if (!request) return true;
+    for (const snapshot of Array.isArray(msg.sessions) ? msg.sessions : []) {
+      if (!snapshot?.browserSessionId) continue;
+      installBrowserRoute({ agentId, ownerUserId: request.ownerUserId, msg: snapshot });
+    }
+    const client = browserClientForPeer({
+      clientId: request.clientId,
+      webConnectionId: request.webConnectionId,
+      webConnectionGeneration: request.webConnectionGeneration,
+    });
+    if (client?.userId === request.ownerUserId) {
+      await sendToWebClient(client, publicSessionMessage(agentId, request.response));
+    }
+    return true;
+  }
+
+  if (msg.type === 'browser_session_snapshot' && msg.requestId) {
+    const request = completeBrowserRequest(agentId, msg, { consume: true });
+    if (!request) return true;
+    const client = browserClientForPeer({
+      clientId: request.clientId,
+      webConnectionId: request.webConnectionId,
+      webConnectionGeneration: request.webConnectionGeneration,
+    });
+    if (client?.userId === request.ownerUserId) {
+      const route = getBrowserRoute(agentId, msg.browserSessionId);
+      if (route) {
+        route.revision = Number(msg.revision) || route.revision;
+        route.state = clean(msg.state, 32) || route.state;
+        if (route.state === 'closed' || route.state === 'failed') {
+          deleteBrowserRoute(agentId, route.browserSessionId);
+        }
+      }
+      await sendToWebClient(client, publicSessionMessage(agentId, request.response));
+    }
+    return true;
+  }
+
+  if (msg.type === 'browser_session_snapshot') {
+    const route = getBrowserRoute(agentId, msg.browserSessionId);
+    if (!route) return true;
+    route.revision = Number(msg.revision) || route.revision;
+    route.state = clean(msg.state, 32) || route.state;
+    route.updatedAt = Date.now();
+    const recipients = [];
+    for (const peer of browserPeers.values()) {
+      if (peer.agentId !== agentId || peer.browserSessionId !== route.browserSessionId) continue;
+      const client = browserClientForPeer(peer);
+      if (client && !recipients.includes(client)) recipients.push(client);
+    }
+    if (route.state === 'closed' || route.state === 'failed') {
+      deleteBrowserRoute(agentId, route.browserSessionId);
+    }
+    const snapshot = publicSessionMessage(agentId, msg);
+    for (const client of recipients) await sendToWebClient(client, snapshot);
+    return true;
+  }
+
+  const peer = getBrowserPeer(msg.peerId);
+  if (!peer || peer.agentId !== agentId
+      || peer.browserSessionId !== msg.browserSessionId
+      || peer.connectionGeneration !== Number(msg.connectionGeneration)) return true;
+  const route = getBrowserRoute(agentId, peer.browserSessionId);
+  if (!route || route.ownerUserId !== peer.ownerUserId) {
+    deleteBrowserPeer(peer.peerId);
+    return true;
+  }
+  const client = browserClientForPeer(peer);
+  if (!client || client.userId !== route.ownerUserId) {
+    deleteBrowserPeer(peer.peerId);
+    return true;
+  }
+
+  if (msg.type === 'browser_peer_prepared') {
+    peer.state = 'prepared';
+    await sendToWebClient(client, {
+      type: msg.type,
+      agentId,
+      browserSessionId: peer.browserSessionId,
+      peerId: peer.peerId,
+      requestId: peer.requestId,
+      connectionGeneration: peer.connectionGeneration,
+      iceTransportPolicy: peer.iceTransportPolicy,
+      iceServers: peer.webIceServers || [],
+      role: peer.role,
+    });
+    if (peer.pendingOffer) {
+      await sendToWebClient(client, peer.pendingOffer);
+      peer.pendingOffer = null;
+      peer.state = 'offered';
+    }
+    for (const candidate of peer.pendingCandidates.splice(0)) {
+      await sendToWebClient(client, candidate);
+    }
+    return true;
+  }
+  if (msg.type === 'browser_peer_offer') {
+    const description = msg.description;
+    if (description?.type !== 'offer' || typeof description.sdp !== 'string' || description.sdp.length > 96 * 1024) {
+      deleteBrowserPeer(peer.peerId);
+      return true;
+    }
+    const offer = {
+      type: msg.type,
+      agentId,
+      browserSessionId: peer.browserSessionId,
+      peerId: peer.peerId,
+      requestId: peer.requestId,
+      connectionGeneration: peer.connectionGeneration,
+      description: { type: 'offer', sdp: description.sdp },
+      iceServers: peer.webIceServers || [],
+      role: peer.role,
+    };
+    if (peer.state !== 'prepared' && peer.state !== 'offered') {
+      peer.pendingOffer = offer;
+      return true;
+    }
+    peer.state = 'offered';
+    await sendToWebClient(client, offer);
+    return true;
+  }
+  if (msg.type === 'browser_peer_ice_candidate') {
+    if (peer.agentCandidateCount >= 128) return true;
+    const raw = msg.candidate;
+    if (raw != null && (typeof raw !== 'object'
+      || typeof raw.candidate !== 'string'
+      || raw.candidate.length > 4096)) return true;
+    peer.agentCandidateCount += 1;
+    const candidate = {
+      type: msg.type,
+      agentId,
+      browserSessionId: peer.browserSessionId,
+      peerId: peer.peerId,
+      connectionGeneration: peer.connectionGeneration,
+      candidate: raw == null ? null : {
+        candidate: raw.candidate,
+        sdpMid: clean(raw.sdpMid, 256) || null,
+        sdpMLineIndex: Number.isInteger(raw.sdpMLineIndex) ? raw.sdpMLineIndex : null,
+        usernameFragment: clean(raw.usernameFragment, 256) || null,
+      },
+    };
+    if (peer.state !== 'prepared' && peer.state !== 'offered') {
+      if (peer.pendingCandidates.length < 128) peer.pendingCandidates.push(candidate);
+      return true;
+    }
+    await sendToWebClient(client, candidate);
+    return true;
+  }
+  if (msg.type === 'browser_peer_state') {
+    peer.state = clean(msg.state, 32) || peer.state;
+    if (peer.state === 'connected') peer.expiresAt = null;
+    if (['failed', 'disconnected', 'closed'].includes(peer.state)) deleteBrowserPeer(peer.peerId);
+  }
+  if (msg.type === 'browser_peer_error') deleteBrowserPeer(peer.peerId);
+  await sendToWebClient(client, {
+    type: msg.type,
+    agentId,
+    browserSessionId: peer.browserSessionId,
+    peerId: peer.peerId,
+    connectionGeneration: peer.connectionGeneration,
+    requestId: peer.requestId,
+    state: clean(msg.state, 32) || null,
+    code: clean(msg.code, 128) || null,
+    safeError: clean(msg.safeError, 500) || null,
+  });
+  return true;
+}
+
+export { AGENT_BROWSER_TYPES };

package/local-runtime/server/handlers/client-browser.js

@@ -0,0 +1,290 @@
+import { createHash } from 'node:crypto';
+import { CONFIG } from '../config.js';
+import { agents } from '../context.js';
+import { sendToAgent, sendToWebClient } from '../ws-utils.js';
+import {
+  browserPeerMatchesClient,
+  browserServerIdentity,
+  deleteBrowserPeer,
+  getBrowserPeer,
+  getBrowserRoute,
+  mintBrowserIceServers,
+  registerBrowserCreateRequest,
+  registerBrowserRequest,
+  reserveBrowserPeer,
+} from '../browser-runtime-routes.js';
+
+const CLIENT_BROWSER_TYPES = new Set([
+  'browser_session_create',
+  'browser_session_get',
+  'browser_session_list',
+  'browser_session_close',
+  'browser_peer_attach',
+  'browser_peer_answer',
+  'browser_peer_ice_candidate',
+  'browser_peer_detach',
+]);
+
+function clean(value, max = 512) {
+  return typeof value === 'string' ? value.trim().slice(0, max) : '';
+}
+
+function digest(value) {
+  return createHash('sha256').update(JSON.stringify(value || null)).digest('hex');
+}
+
+async function fail(client, msg, code, safeError = code) {
+  await sendToWebClient(client, {
+    type: String(msg.type || '').startsWith('browser_peer_') ? 'browser_peer_error' : 'browser_session_error',
+    requestId: clean(msg.requestId) || null,
+    browserSessionId: clean(msg.browserSessionId) || null,
+    peerId: clean(msg.peerId) || null,
+    connectionGeneration: Number(msg.connectionGeneration) || null,
+    code,
+    safeError,
+  });
+  return true;
+}
+
+function agentSupportsBrowser(agent) {
+  const capabilities = new Set(agent?.capabilities || []);
+  return capabilities.has('browser_runtime')
+    && capabilities.has('browser_webrtc')
+    && (capabilities.has('browser_capture_tab') || capabilities.has('browser_capture_cdp'));
+}
+
+function safeInitialUrl(value) {
+  const raw = clean(value, 4096) || 'about:blank';
+  if (raw === 'about:blank') return raw;
+  try {
+    const url = new URL(raw);
+    if (!['http:', 'https:'].includes(url.protocol) || url.username || url.password) return null;
+    return url.href;
+  } catch {
+    return null;
+  }
+}
+
+function canonicalCreate(msg) {
+  const initialUrl = safeInitialUrl(msg.options?.initialUrl);
+  if (!initialUrl) return null;
+  return {
+    sourceRef: msg.sourceRef && typeof msg.sourceRef === 'object' ? {
+      kind: clean(msg.sourceRef.kind, 32),
+      sessionId: clean(msg.sourceRef.sessionId),
+      conversationId: clean(msg.sourceRef.conversationId),
+      workItemId: clean(msg.sourceRef.workItemId),
+    } : null,
+    options: {
+      initialUrl,
+      viewport: {
+        width: Math.min(1920, Math.max(320, Math.floor(Number(msg.options?.viewport?.width) || 1280))),
+        height: Math.min(1080, Math.max(240, Math.floor(Number(msg.options?.viewport?.height) || 720))),
+        deviceScaleFactor: Math.min(2, Math.max(1, Number(msg.options?.viewport?.deviceScaleFactor) || 1)),
+      },
+      locale: clean(msg.options?.locale, 32) || 'en-US',
+      capturePreference: ['auto', 'tab'].includes(msg.options?.capturePreference)
+        ? msg.options.capturePreference : 'auto',
+    },
+  };
+}
+
+function ownerRoute(client, msg) {
+  const route = getBrowserRoute(msg.agentId, msg.browserSessionId);
+  return route && route.ownerUserId === client.userId ? route : null;
+}
+
+/** Owner-checked Web → Server → Agent Browser lifecycle and signaling relay. */
+export async function handleClientBrowser(client, msg, checkAgentAccess) {
+  if (!CLIENT_BROWSER_TYPES.has(msg?.type)) return false;
+  if (!CONFIG.browserRuntime.enabled) return fail(client, msg, 'browser_runtime_disabled');
+  if (client.browserRuntimeProtocol !== 1) return fail(client, msg, 'browser_protocol_required');
+  const agentId = clean(msg.agentId);
+  if (!await checkAgentAccess(agentId)) return true;
+  const agent = agents.get(agentId);
+  if (!agentSupportsBrowser(agent)) return fail(client, msg, 'browser_runtime_unavailable');
+  const requestId = clean(msg.requestId);
+  const requestRequired = msg.type !== 'browser_peer_answer'
+    && msg.type !== 'browser_peer_ice_candidate';
+  if (requestRequired && !requestId) return fail(client, msg, 'browser_request_id_required');
+  const identity = browserServerIdentity(client);
+
+  if (msg.type === 'browser_session_create') {
+    const canonical = canonicalCreate(msg);
+    if (!canonical) return fail(client, msg, 'browser_url_invalid');
+    const registration = registerBrowserCreateRequest({
+      agentId,
+      client,
+      requestId,
+      digest: digest(canonical),
+    });
+    if (registration.conflict) return fail(client, msg, 'browser_request_conflict');
+    if (registration.capacity) return fail(client, msg, 'browser_request_capacity');
+    if (registration.duplicate) {
+      if (registration.request.response) {
+        await sendToWebClient(client, registration.request.response);
+      }
+      return true;
+    }
+    await sendToAgent(agent, {
+      type: msg.type,
+      agentId,
+      requestId: registration.request.serverRequestId,
+      ...canonical,
+      serverIdentity: identity,
+    });
+    return true;
+  }
+
+  if (msg.type === 'browser_session_list') {
+    const request = registerBrowserRequest({ agentId, client, requestId, kind: msg.type });
+    if (!request) return fail(client, msg, 'browser_request_conflict');
+    await sendToAgent(agent, {
+      type: msg.type,
+      agentId,
+      requestId: request.serverRequestId,
+      serverIdentity: identity,
+    });
+    return true;
+  }
+
+  if (msg.type === 'browser_peer_attach') {
+    if (!ownerRoute(client, msg)) return fail(client, msg, 'browser_session_not_found');
+    const connectionGeneration = Number(msg.connectionGeneration);
+    const reserved = reserveBrowserPeer({
+      agentId,
+      browserSessionId: clean(msg.browserSessionId),
+      client,
+      requestId,
+      connectionGeneration,
+      role: 'viewer',
+    });
+    if (reserved.error) return fail(client, msg, reserved.error);
+    const peer = reserved.peer;
+    if (reserved.duplicate) {
+      if (peer.state === 'prepared' || peer.state === 'offered' || peer.state === 'connected') {
+        await sendToWebClient(client, {
+          type: 'browser_peer_prepared',
+          agentId,
+          browserSessionId: peer.browserSessionId,
+          peerId: peer.peerId,
+          requestId: peer.requestId,
+          connectionGeneration: peer.connectionGeneration,
+          iceTransportPolicy: peer.iceTransportPolicy,
+          iceServers: peer.webIceServers || [],
+          role: peer.role,
+        });
+        if (peer.pendingOffer) await sendToWebClient(client, peer.pendingOffer);
+      }
+      return true;
+    }
+    const commonScope = {
+      ownerUserId: client.userId,
+      agentId,
+      browserSessionId: peer.browserSessionId,
+      peerId: peer.peerId,
+      connectionGeneration: peer.connectionGeneration,
+    };
+    try {
+      await sendToAgent(agent, {
+        type: 'browser_peer_prepare',
+        agentId,
+        browserSessionId: peer.browserSessionId,
+        peerId: peer.peerId,
+        requestId,
+        connectionGeneration: peer.connectionGeneration,
+        serverIdentity: identity,
+        routeExpiresAt: peer.expiresAt,
+        iceTransportPolicy: CONFIG.browserRuntime.iceTransportPolicy,
+        agentIceServers: mintBrowserIceServers({ ...commonScope, endpointRole: 'agent' }),
+      });
+      peer.webIceServers = mintBrowserIceServers({ ...commonScope, endpointRole: 'web' });
+      peer.state = 'preparing';
+    } catch (error) {
+      deleteBrowserPeer(peer.peerId);
+      return fail(client, msg, 'browser_peer_prepare_failed', String(error?.message || error).slice(0, 500));
+    }
+    return true;
+  }
+
+  if (msg.type === 'browser_session_get' || msg.type === 'browser_session_close') {
+    const route = ownerRoute(client, msg);
+    if (!route) return fail(client, msg, 'browser_session_not_found');
+    const request = registerBrowserRequest({
+      agentId,
+      client,
+      requestId,
+      kind: msg.type,
+      browserSessionId: route.browserSessionId,
+    });
+    if (!request) return fail(client, msg, 'browser_request_conflict');
+    await sendToAgent(agent, {
+      type: msg.type,
+      agentId,
+      requestId: request.serverRequestId,
+      browserSessionId: route.browserSessionId,
+      ...(msg.type === 'browser_session_close' ? { expectedRevision: Number(msg.expectedRevision) || route.revision } : {}),
+      serverIdentity: identity,
+    });
+    return true;
+  }
+
+  const peer = getBrowserPeer(msg.peerId);
+  if (!browserPeerMatchesClient(peer, client, msg)) return fail(client, msg, 'browser_peer_stale');
+  if (msg.type === 'browser_peer_answer') {
+    const description = msg.description;
+    if (description?.type !== 'answer' || typeof description.sdp !== 'string' || description.sdp.length > 96 * 1024) {
+      return fail(client, msg, 'browser_sdp_invalid');
+    }
+    peer.state = 'answering';
+    await sendToAgent(agent, {
+      type: msg.type,
+      agentId,
+      browserSessionId: peer.browserSessionId,
+      peerId: peer.peerId,
+      connectionGeneration: peer.connectionGeneration,
+      description: { type: 'answer', sdp: description.sdp },
+      serverIdentity: identity,
+    });
+    return true;
+  }
+  if (msg.type === 'browser_peer_ice_candidate') {
+    if (peer.webCandidateCount >= 128) return fail(client, msg, 'browser_candidate_limit');
+    const candidate = msg.candidate;
+    if (candidate != null && (typeof candidate !== 'object'
+      || typeof candidate.candidate !== 'string'
+      || candidate.candidate.length > 4096)) return fail(client, msg, 'browser_candidate_invalid');
+    peer.webCandidateCount += 1;
+    await sendToAgent(agent, {
+      type: msg.type,
+      agentId,
+      browserSessionId: peer.browserSessionId,
+      peerId: peer.peerId,
+      connectionGeneration: peer.connectionGeneration,
+      candidate: candidate == null ? null : {
+        candidate: candidate.candidate,
+        sdpMid: clean(candidate.sdpMid, 256) || null,
+        sdpMLineIndex: Number.isInteger(candidate.sdpMLineIndex) ? candidate.sdpMLineIndex : null,
+        usernameFragment: clean(candidate.usernameFragment, 256) || null,
+      },
+      serverIdentity: identity,
+    });
+    return true;
+  }
+  if (msg.type === 'browser_peer_detach') {
+    deleteBrowserPeer(peer.peerId);
+    await sendToAgent(agent, {
+      type: msg.type,
+      agentId,
+      requestId,
+      browserSessionId: peer.browserSessionId,
+      peerId: peer.peerId,
+      connectionGeneration: peer.connectionGeneration,
+      serverIdentity: identity,
+    });
+    return true;
+  }
+  return false;
+}
+
+export { CLIENT_BROWSER_TYPES };

package/local-runtime/server/ws-agent.js

@@ -16,6 +16,8 @@ import { handleAgentFileTerminal } from './handlers/agent-file-terminal.js';
 import { handleAgentSync } from './handlers/agent-sync.js';
 import { recordPerfTraceEvent } from './perf-trace.js';
 import { clearWorkbenchCorrelationsForAgent } from './workbench-correlation.js';
+import { clearBrowserRuntimeForAgent } from './browser-runtime-routes.js';
+import { handleAgentBrowser } from './handlers/agent-browser.js';
 import { markAgentHeartbeatSeen } from './heartbeat-policy.js';
 
 /**
@@ -229,6 +231,7 @@ function handleAgentDisconnect(agentId, agentName, ws) {
   // Phase 4: 清理目录缓存
   clearAgentDirCache(agentId);
   clearWorkbenchCorrelationsForAgent(agentId);
+  clearBrowserRuntimeForAgent(agentId);
   // Phase 1: 清理同步超时
   if (agent._syncTimeout) {
     clearTimeout(agent._syncTimeout);
@@ -294,6 +297,7 @@ function completeAgentRegistration(ws, agentId, agentName, workDir, sessionKey,
   agents.get(agentId)._syncTimeout = syncTimeout;
 
   if (existingAgent?.ws && existingAgent.ws !== ws) {
+    clearBrowserRuntimeForAgent(agentId);
     existingAgent.ws.close(1008, 'Superseded by a newer Agent connection');
   }
 
@@ -347,7 +351,9 @@ async function handleAgentMessage(agentId, msg, ws) {
     'yeaft_history_chunk', 'yeaft_history_outline', 'yeaft_history_search_result', 'yeaft_history_window', 'slash_commands_update', 'agent_metrics',
     'file_content', 'file_saved', 'file_op_result', 'file_search_result',
     'git_status_result', 'git_diff_result', 'git_op_result',
-    'terminal_created', 'terminal_output', 'terminal_closed', 'terminal_error'
+    'terminal_created', 'terminal_output', 'terminal_closed', 'terminal_error',
+    'browser_session_created', 'browser_session_error', 'browser_session_snapshot', 'browser_session_list_result',
+    'browser_peer_prepared', 'browser_peer_offer', 'browser_peer_ice_candidate', 'browser_peer_state', 'browser_peer_error'
   ]);
   if (msg.conversationId && !CONV_EXEMPT_TYPES.has(msg.type)) {
     if (!agent.conversations.has(msg.conversationId)) {
@@ -358,6 +364,7 @@ async function handleAgentMessage(agentId, msg, ws) {
 
   // Dispatch to handler sub-modules
   if (await handleAgentConversation(agentId, agent, msg)) return;
+  if (await handleAgentBrowser(agentId, agent, msg)) return;
   if (await handleAgentWorkCenter(agentId, msg)) return;
   if (await handleAgentOutput(agentId, agent, msg)) return;
   if (await handleAgentFileTerminal(agentId, agent, msg)) return;