@poolzin/pool-bot 2026.2.24 → 2026.2.26

package/dist/gateway/server-chat.js

@@ -1,15 +1,39 @@
+import { DEFAULT_HEARTBEAT_ACK_MAX_CHARS, stripHeartbeatToken } from "../auto-reply/heartbeat.js";
 import { normalizeVerboseLevel } from "../auto-reply/thinking.js";
+import { isSilentReplyText, SILENT_REPLY_TOKEN } from "../auto-reply/tokens.js";
 import { loadConfig } from "../config/config.js";
 import { getAgentRunContext } from "../infra/agent-events.js";
 import { resolveHeartbeatVisibility } from "../infra/heartbeat-visibility.js";
+import { stripInlineDirectiveTagsForDisplay } from "../utils/directive-tags.js";
 import { loadSessionEntry } from "./session-utils.js";
 import { formatForLog } from "./ws-log.js";
+function resolveHeartbeatAckMaxChars() {
+    try {
+        const cfg = loadConfig();
+        return Math.max(0, cfg.agents?.defaults?.heartbeat?.ackMaxChars ?? DEFAULT_HEARTBEAT_ACK_MAX_CHARS);
+    }
+    catch {
+        return DEFAULT_HEARTBEAT_ACK_MAX_CHARS;
+    }
+}
+function resolveHeartbeatContext(runId, sourceRunId) {
+    const primary = getAgentRunContext(runId);
+    if (primary?.isHeartbeat) {
+        return primary;
+    }
+    if (sourceRunId && sourceRunId !== runId) {
+        const source = getAgentRunContext(sourceRunId);
+        if (source?.isHeartbeat) {
+            return source;
+        }
+    }
+    return primary;
+}
 /**
- * Check if webchat broadcasts should be suppressed for heartbeat runs.
- * Returns true if the run is a heartbeat and showOk is false.
+ * Check if heartbeat ACK/noise should be hidden from interactive chat surfaces.
  */
-function shouldSuppressHeartbeatBroadcast(runId) {
-    const runContext = getAgentRunContext(runId);
+function shouldHideHeartbeatChatOutput(runId, sourceRunId) {
+    const runContext = resolveHeartbeatContext(runId, sourceRunId);
     if (!runContext?.isHeartbeat) {
         return false;
     }
@@ -23,6 +47,22 @@ function shouldSuppressHeartbeatBroadcast(runId) {
         return true;
     }
 }
+function normalizeHeartbeatChatFinalText(params) {
+    if (!shouldHideHeartbeatChatOutput(params.runId, params.sourceRunId)) {
+        return { suppress: false, text: params.text };
+    }
+    const stripped = stripHeartbeatToken(params.text, {
+        mode: "heartbeat",
+        maxAckChars: resolveHeartbeatAckMaxChars(),
+    });
+    if (!stripped.didStrip) {
+        return { suppress: false, text: params.text };
+    }
+    if (stripped.shouldSkip) {
+        return { suppress: true, text: "" };
+    }
+    return { suppress: false, text: stripped.text };
+}
 export function createChatRunRegistry() {
     const chatRunSessions = new Map();
     const add = (sessionId, entry) => {
@@ -141,8 +181,18 @@ export function createToolEventRecipientRegistry() {
     return { add, get, markFinal };
 }
 export function createAgentEventHandler({ broadcast, broadcastToConnIds, nodeSendToSession, agentRunSeq, chatRunState, resolveSessionKeyForRun, clearAgentRunContext, toolEventRecipients, }) {
-    const emitChatDelta = (sessionKey, clientRunId, seq, text) => {
-        chatRunState.buffers.set(clientRunId, text);
+    const emitChatDelta = (sessionKey, clientRunId, sourceRunId, seq, text) => {
+        const cleaned = stripInlineDirectiveTagsForDisplay(text).text;
+        if (!cleaned) {
+            return;
+        }
+        if (isSilentReplyText(cleaned, SILENT_REPLY_TOKEN)) {
+            return;
+        }
+        chatRunState.buffers.set(clientRunId, cleaned);
+        if (shouldHideHeartbeatChatOutput(clientRunId, sourceRunId)) {
+            return;
+        }
         const now = Date.now();
         const last = chatRunState.deltaSentAt.get(clientRunId) ?? 0;
         if (now - last < 150) {
@@ -156,18 +206,22 @@ export function createAgentEventHandler({ broadcast, broadcastToConnIds, nodeSen
             state: "delta",
             message: {
                 role: "assistant",
-                content: [{ type: "text", text }],
+                content: [{ type: "text", text: cleaned }],
                 timestamp: now,
             },
         };
-        // Suppress webchat broadcast for heartbeat runs when showOk is false
-        if (!shouldSuppressHeartbeatBroadcast(clientRunId)) {
-            broadcast("chat", payload, { dropIfSlow: true });
-        }
+        broadcast("chat", payload, { dropIfSlow: true });
         nodeSendToSession(sessionKey, "chat", payload);
     };
-    const emitChatFinal = (sessionKey, clientRunId, seq, jobState, error) => {
-        const text = chatRunState.buffers.get(clientRunId)?.trim() ?? "";
+    const emitChatFinal = (sessionKey, clientRunId, sourceRunId, seq, jobState, error) => {
+        const bufferedText = stripInlineDirectiveTagsForDisplay(chatRunState.buffers.get(clientRunId) ?? "").text.trim();
+        const normalizedHeartbeatText = normalizeHeartbeatChatFinalText({
+            runId: clientRunId,
+            sourceRunId,
+            text: bufferedText,
+        });
+        const text = normalizedHeartbeatText.text.trim();
+        const shouldSuppressSilent = normalizedHeartbeatText.suppress || isSilentReplyText(text, SILENT_REPLY_TOKEN);
         chatRunState.buffers.delete(clientRunId);
         chatRunState.deltaSentAt.delete(clientRunId);
         if (jobState === "done") {
@@ -176,7 +230,7 @@ export function createAgentEventHandler({ broadcast, broadcastToConnIds, nodeSen
                 sessionKey,
                 seq,
                 state: "final",
-                message: text
+                message: text && !shouldSuppressSilent
                     ? {
                         role: "assistant",
                         content: [{ type: "text", text }],
@@ -184,10 +238,7 @@ export function createAgentEventHandler({ broadcast, broadcastToConnIds, nodeSen
                     }
                     : undefined,
             };
-            // Suppress webchat broadcast for heartbeat runs when showOk is false
-            if (!shouldSuppressHeartbeatBroadcast(clientRunId)) {
-                broadcast("chat", payload);
-            }
+            broadcast("chat", payload);
             nodeSendToSession(sessionKey, "chat", payload);
             return;
         }
@@ -225,11 +276,14 @@ export function createAgentEventHandler({ broadcast, broadcastToConnIds, nodeSen
     };
     return (evt) => {
         const chatLink = chatRunState.registry.peek(evt.runId);
-        const sessionKey = chatLink?.sessionKey ?? resolveSessionKeyForRun(evt.runId);
+        const eventSessionKey = typeof evt.sessionKey === "string" && evt.sessionKey.trim() ? evt.sessionKey : undefined;
+        const sessionKey = chatLink?.sessionKey ?? eventSessionKey ?? resolveSessionKeyForRun(evt.runId);
         const clientRunId = chatLink?.clientRunId ?? evt.runId;
+        const eventRunId = chatLink?.clientRunId ?? evt.runId;
+        const eventForClients = chatLink ? { ...evt, runId: eventRunId } : evt;
         const isAborted = chatRunState.abortedRuns.has(clientRunId) || chatRunState.abortedRuns.has(evt.runId);
         // Include sessionKey so Control UI can filter tool streams per session.
-        const agentPayload = sessionKey ? { ...evt, sessionKey } : evt;
+        const agentPayload = sessionKey ? { ...eventForClients, sessionKey } : eventForClients;
         const last = agentRunSeq.get(evt.runId) ?? 0;
         const isToolEvent = evt.stream === "tool";
         const toolVerbose = isToolEvent ? resolveToolVerboseLevel(evt.runId, sessionKey) : "off";
@@ -239,12 +293,14 @@ export function createAgentEventHandler({ broadcast, broadcastToConnIds, nodeSen
                 const data = evt.data ? { ...evt.data } : {};
                 delete data.result;
                 delete data.partialResult;
-                return sessionKey ? { ...evt, sessionKey, data } : { ...evt, data };
+                return sessionKey
+                    ? { ...eventForClients, sessionKey, data }
+                    : { ...eventForClients, data };
             })()
             : agentPayload;
         if (evt.seq !== last + 1) {
             broadcast("agent", {
-                runId: evt.runId,
+                runId: eventRunId,
                 stream: "error",
                 ts: Date.now(),
                 sessionKey,
@@ -277,7 +333,7 @@ export function createAgentEventHandler({ broadcast, broadcastToConnIds, nodeSen
                 nodeSendToSession(sessionKey, "agent", isToolEvent ? toolPayload : agentPayload);
             }
             if (!isAborted && evt.stream === "assistant" && typeof evt.data?.text === "string") {
-                emitChatDelta(sessionKey, clientRunId, evt.seq, evt.data.text);
+                emitChatDelta(sessionKey, clientRunId, evt.runId, evt.seq, evt.data.text);
             }
             else if (!isAborted && (lifecyclePhase === "end" || lifecyclePhase === "error")) {
                 if (chatLink) {
@@ -286,10 +342,10 @@ export function createAgentEventHandler({ broadcast, broadcastToConnIds, nodeSen
                         clearAgentRunContext(evt.runId);
                         return;
                     }
-                    emitChatFinal(finished.sessionKey, finished.clientRunId, evt.seq, lifecyclePhase === "error" ? "error" : "done", evt.data?.error);
+                    emitChatFinal(finished.sessionKey, finished.clientRunId, evt.runId, evt.seq, lifecyclePhase === "error" ? "error" : "done", evt.data?.error);
                 }
                 else {
-                    emitChatFinal(sessionKey, evt.runId, evt.seq, lifecyclePhase === "error" ? "error" : "done", evt.data?.error);
+                    emitChatFinal(sessionKey, eventRunId, evt.runId, evt.seq, lifecyclePhase === "error" ? "error" : "done", evt.data?.error);
                 }
             }
             else if (isAborted && (lifecyclePhase === "end" || lifecyclePhase === "error")) {
@@ -305,6 +361,8 @@ export function createAgentEventHandler({ broadcast, broadcastToConnIds, nodeSen
         if (lifecyclePhase === "end" || lifecyclePhase === "error") {
             toolEventRecipients.markFinal(evt.runId);
             clearAgentRunContext(evt.runId);
+            agentRunSeq.delete(evt.runId);
+            agentRunSeq.delete(clientRunId);
         }
     };
 }

package/dist/gateway/server-constants.js

@@ -1,11 +1,14 @@
-export const MAX_PAYLOAD_BYTES = 512 * 1024; // cap incoming frame size
-export const MAX_BUFFERED_BYTES = 1.5 * 1024 * 1024; // per-connection send buffer limit
+// Keep server maxPayload aligned with gateway client maxPayload so high-res canvas snapshots
+// don't get disconnected mid-invoke with "Max payload size exceeded".
+export const MAX_PAYLOAD_BYTES = 25 * 1024 * 1024;
+export const MAX_BUFFERED_BYTES = 50 * 1024 * 1024; // per-connection send buffer limit (2x max payload)
 const DEFAULT_MAX_CHAT_HISTORY_MESSAGES_BYTES = 6 * 1024 * 1024; // keep history responses comfortably under client WS limits
 let maxChatHistoryMessagesBytes = DEFAULT_MAX_CHAT_HISTORY_MESSAGES_BYTES;
 export const getMaxChatHistoryMessagesBytes = () => maxChatHistoryMessagesBytes;
 export const __setMaxChatHistoryMessagesBytesForTest = (value) => {
-    if (!process.env.VITEST && process.env.NODE_ENV !== "test")
+    if (!process.env.VITEST && process.env.NODE_ENV !== "test") {
         return;
+    }
     if (value === undefined) {
         maxChatHistoryMessagesBytes = DEFAULT_MAX_CHAT_HISTORY_MESSAGES_BYTES;
         return;
@@ -16,13 +19,11 @@ export const __setMaxChatHistoryMessagesBytesForTest = (value) => {
 };
 export const DEFAULT_HANDSHAKE_TIMEOUT_MS = 10_000;
 export const getHandshakeTimeoutMs = () => {
-    if (process.env.VITEST &&
-        (process.env.POOLBOT_TEST_HANDSHAKE_TIMEOUT_MS ||
-            process.env.CLAWDBOT_TEST_HANDSHAKE_TIMEOUT_MS)) {
-        const parsed = Number(process.env.POOLBOT_TEST_HANDSHAKE_TIMEOUT_MS ||
-            process.env.CLAWDBOT_TEST_HANDSHAKE_TIMEOUT_MS);
-        if (Number.isFinite(parsed) && parsed > 0)
+    if (process.env.VITEST && process.env.POOLBOT_TEST_HANDSHAKE_TIMEOUT_MS) {
+        const parsed = Number(process.env.POOLBOT_TEST_HANDSHAKE_TIMEOUT_MS);
+        if (Number.isFinite(parsed) && parsed > 0) {
             return parsed;
+        }
     }
     return DEFAULT_HANDSHAKE_TIMEOUT_MS;
 };

package/dist/gateway/server-cron.js

@@ -235,6 +235,7 @@ export function buildGatewayCronService(params) {
                     status: evt.status,
                     error: evt.error,
                     summary: evt.summary,
+                    delivered: evt.delivered,
                     sessionId: evt.sessionId,
                     sessionKey: evt.sessionKey,
                     runAtMs: evt.runAtMs,