@poolzin/pool-bot 2026.2.25 → 2026.2.26

package/dist/agents/pi-embedded-subscribe.tools.js

@@ -2,6 +2,8 @@ import { getChannelPlugin, normalizeChannelId } from "../channels/plugins/index.
 import { normalizeTargetForProvider } from "../infra/outbound/target-normalization.js";
 import { MEDIA_TOKEN_RE } from "../media/parse.js";
 import { truncateUtf16Safe } from "../utils.js";
+import { collectTextContentBlocks } from "./content-blocks.js";
+import { normalizeToolName } from "./tool-policy.js";
 const TOOL_RESULT_MAX_CHARS = 8000;
 const TOOL_ERROR_MAX_CHARS = 400;
 function truncateToolText(text) {
@@ -23,6 +25,20 @@ function normalizeToolErrorText(text) {
         ? `${truncateUtf16Safe(firstLine, TOOL_ERROR_MAX_CHARS)}…`
         : firstLine;
 }
+function isErrorLikeStatus(status) {
+    const normalized = status.trim().toLowerCase();
+    if (!normalized) {
+        return false;
+    }
+    if (normalized === "0" ||
+        normalized === "ok" ||
+        normalized === "success" ||
+        normalized === "completed" ||
+        normalized === "running") {
+        return false;
+    }
+    return /error|fail|timeout|timed[_\s-]?out|denied|cancel|invalid|forbidden/.test(normalized);
+}
 function readErrorCandidate(value) {
     if (typeof value === "string") {
         return normalizeToolErrorText(value);
@@ -51,7 +67,10 @@ function extractErrorField(value) {
         return direct;
     }
     const status = typeof record.status === "string" ? record.status.trim() : "";
-    return status ? normalizeToolErrorText(status) : undefined;
+    if (!status || !isErrorLikeStatus(status)) {
+        return undefined;
+    }
+    return normalizeToolErrorText(status);
 }
 export function sanitizeToolResult(result) {
     if (!result || typeof result !== "object") {
@@ -87,20 +106,9 @@ export function extractToolResultText(result) {
         return undefined;
     }
     const record = result;
-    const content = Array.isArray(record.content) ? record.content : null;
-    if (!content) {
-        return undefined;
-    }
-    const texts = content
+    const texts = collectTextContentBlocks(record.content)
         .map((item) => {
-        if (!item || typeof item !== "object") {
-            return undefined;
-        }
-        const entry = item;
-        if (entry.type !== "text" || typeof entry.text !== "string") {
-            return undefined;
-        }
-        const trimmed = entry.text.trim();
+        const trimmed = item.trim();
         return trimmed ? trimmed : undefined;
     })
         .filter((value) => Boolean(value));
@@ -109,6 +117,52 @@ export function extractToolResultText(result) {
     }
     return texts.join("\n");
 }
+// Core tool names that are allowed to emit local MEDIA: paths.
+// Plugin/MCP tools are intentionally excluded to prevent untrusted file reads.
+const TRUSTED_TOOL_RESULT_MEDIA = new Set([
+    "agents_list",
+    "apply_patch",
+    "browser",
+    "canvas",
+    "cron",
+    "edit",
+    "exec",
+    "gateway",
+    "image",
+    "memory_get",
+    "memory_search",
+    "message",
+    "nodes",
+    "process",
+    "read",
+    "session_status",
+    "sessions_history",
+    "sessions_list",
+    "sessions_send",
+    "sessions_spawn",
+    "subagents",
+    "tts",
+    "web_fetch",
+    "web_search",
+    "write",
+]);
+const HTTP_URL_RE = /^https?:\/\//i;
+export function isToolResultMediaTrusted(toolName) {
+    if (!toolName) {
+        return false;
+    }
+    const normalized = normalizeToolName(toolName);
+    return TRUSTED_TOOL_RESULT_MEDIA.has(normalized);
+}
+export function filterToolResultMediaUrls(toolName, mediaUrls) {
+    if (mediaUrls.length === 0) {
+        return mediaUrls;
+    }
+    if (isToolResultMediaTrusted(toolName)) {
+        return mediaUrls;
+    }
+    return mediaUrls.filter((url) => HTTP_URL_RE.test(url.trim()));
+}
 /**
  * Extract media file paths from a tool result.
  *

package/dist/agents/pi-embedded-utils.js

@@ -1,6 +1,9 @@
 import { stripReasoningTagsFromText } from "../shared/text/reasoning-tags.js";
 import { sanitizeUserFacingText } from "./pi-embedded-helpers.js";
 import { formatToolDetail, resolveToolDisplay } from "./tool-display.js";
+export function isAssistantMessage(msg) {
+    return msg?.role === "assistant";
+}
 /**
  * Strip malformed Minimax tool invocations that leak into text content.
  * Minimax sometimes embeds tool calls as XML in text blocks instead of

package/dist/agents/pi-extensions/compaction-safeguard-runtime.js

@@ -1,20 +1,4 @@
-// Session-scoped runtime registry keyed by object identity.
-// Follows the same WeakMap pattern as context-pruning/runtime.ts.
-const REGISTRY = new WeakMap();
-export function setCompactionSafeguardRuntime(sessionManager, value) {
-    if (!sessionManager || typeof sessionManager !== "object") {
-        return;
-    }
-    const key = sessionManager;
-    if (value === null) {
-        REGISTRY.delete(key);
-        return;
-    }
-    REGISTRY.set(key, value);
-}
-export function getCompactionSafeguardRuntime(sessionManager) {
-    if (!sessionManager || typeof sessionManager !== "object") {
-        return null;
-    }
-    return REGISTRY.get(sessionManager) ?? null;
-}
+import { createSessionManagerRuntimeRegistry } from "./session-manager-runtime-registry.js";
+const registry = createSessionManagerRuntimeRegistry();
+export const setCompactionSafeguardRuntime = registry.set;
+export const getCompactionSafeguardRuntime = registry.get;

package/dist/agents/pi-extensions/compaction-safeguard.js

@@ -1,5 +1,11 @@
-import { BASE_CHUNK_RATIO, MIN_CHUNK_RATIO, SAFETY_MARGIN, computeAdaptiveChunkRatio, estimateMessagesTokens, isOversizedForSummary, pruneHistoryForContextShare, resolveContextWindowTokens, summarizeInStages, } from "../compaction.js";
+import fs from "node:fs";
+import path from "node:path";
+import { extractSections } from "../../auto-reply/reply/post-compaction-context.js";
+import { createSubsystemLogger } from "../../logging/subsystem.js";
+import { BASE_CHUNK_RATIO, MIN_CHUNK_RATIO, SAFETY_MARGIN, SUMMARIZATION_OVERHEAD_TOKENS, computeAdaptiveChunkRatio, estimateMessagesTokens, isOversizedForSummary, pruneHistoryForContextShare, resolveContextWindowTokens, summarizeInStages, } from "../compaction.js";
+import { collectTextContentBlocks } from "../content-blocks.js";
 import { getCompactionSafeguardRuntime } from "./compaction-safeguard-runtime.js";
+const log = createSubsystemLogger("compaction-safeguard");
 const FALLBACK_SUMMARY = "Summary unavailable due to context limits. Older messages were truncated.";
 const TURN_PREFIX_INSTRUCTIONS = "This summary covers the prefix of a split turn. Focus on the original request," +
     " early progress, and any details needed to understand the retained suffix.";
@@ -9,54 +15,51 @@ function normalizeFailureText(text) {
     return text.replace(/\s+/g, " ").trim();
 }
 function truncateFailureText(text, maxChars) {
-    if (text.length <= maxChars)
+    if (text.length <= maxChars) {
         return text;
+    }
     return `${text.slice(0, Math.max(0, maxChars - 3))}...`;
 }
 function formatToolFailureMeta(details) {
-    if (!details || typeof details !== "object")
+    if (!details || typeof details !== "object") {
         return undefined;
+    }
     const record = details;
     const status = typeof record.status === "string" ? record.status : undefined;
     const exitCode = typeof record.exitCode === "number" && Number.isFinite(record.exitCode)
         ? record.exitCode
         : undefined;
     const parts = [];
-    if (status)
+    if (status) {
         parts.push(`status=${status}`);
-    if (exitCode !== undefined)
+    }
+    if (exitCode !== undefined) {
         parts.push(`exitCode=${exitCode}`);
+    }
     return parts.length > 0 ? parts.join(" ") : undefined;
 }
 function extractToolResultText(content) {
-    if (!Array.isArray(content))
-        return "";
-    const parts = [];
-    for (const block of content) {
-        if (!block || typeof block !== "object")
-            continue;
-        const rec = block;
-        if (rec.type === "text" && typeof rec.text === "string") {
-            parts.push(rec.text);
-        }
-    }
-    return parts.join("\n");
+    return collectTextContentBlocks(content).join("\n");
 }
 function collectToolFailures(messages) {
     const failures = [];
     const seen = new Set();
     for (const message of messages) {
-        if (!message || typeof message !== "object")
+        if (!message || typeof message !== "object") {
             continue;
+        }
         const role = message.role;
-        if (role !== "toolResult")
+        if (role !== "toolResult") {
             continue;
+        }
         const toolResult = message;
-        if (toolResult.isError !== true)
+        if (toolResult.isError !== true) {
             continue;
+        }
         const toolCallId = typeof toolResult.toolCallId === "string" ? toolResult.toolCallId : "";
-        if (!toolCallId || seen.has(toolCallId))
+        if (!toolCallId || seen.has(toolCallId)) {
             continue;
+        }
         seen.add(toolCallId);
         const toolName = typeof toolResult.toolName === "string" && toolResult.toolName.trim()
             ? toolResult.toolName
@@ -70,8 +73,9 @@ function collectToolFailures(messages) {
     return failures;
 }
 function formatToolFailuresSection(failures) {
-    if (failures.length === 0)
+    if (failures.length === 0) {
         return "";
+    }
     const lines = failures.slice(0, MAX_TOOL_FAILURES).map((failure) => {
         const meta = failure.meta ? ` (${failure.meta})` : "";
         return `- ${failure.toolName}${meta}: ${failure.summary}`;
@@ -83,8 +87,8 @@ function formatToolFailuresSection(failures) {
 }
 function computeFileLists(fileOps) {
     const modified = new Set([...fileOps.edited, ...fileOps.written]);
-    const readFiles = [...fileOps.read].filter((f) => !modified.has(f)).sort();
-    const modifiedFiles = [...modified].sort();
+    const readFiles = [...fileOps.read].filter((f) => !modified.has(f)).toSorted();
+    const modifiedFiles = [...modified].toSorted();
     return { readFiles, modifiedFiles };
 }
 function formatFileOperations(readFiles, modifiedFiles) {
@@ -95,10 +99,39 @@ function formatFileOperations(readFiles, modifiedFiles) {
     if (modifiedFiles.length > 0) {
         sections.push(`<modified-files>\n${modifiedFiles.join("\n")}\n</modified-files>`);
     }
-    if (sections.length === 0)
+    if (sections.length === 0) {
         return "";
+    }
     return `\n\n${sections.join("\n\n")}`;
 }
+/**
+ * Read and format critical workspace context for compaction summary.
+ * Extracts "Session Startup" and "Red Lines" from AGENTS.md.
+ * Limited to 2000 chars to avoid bloating the summary.
+ */
+async function readWorkspaceContextForSummary() {
+    const MAX_SUMMARY_CONTEXT_CHARS = 2000;
+    const workspaceDir = process.cwd();
+    const agentsPath = path.join(workspaceDir, "AGENTS.md");
+    try {
+        if (!fs.existsSync(agentsPath)) {
+            return "";
+        }
+        const content = await fs.promises.readFile(agentsPath, "utf-8");
+        const sections = extractSections(content, ["Session Startup", "Red Lines"]);
+        if (sections.length === 0) {
+            return "";
+        }
+        const combined = sections.join("\n\n");
+        const safeContent = combined.length > MAX_SUMMARY_CONTEXT_CHARS
+            ? combined.slice(0, MAX_SUMMARY_CONTEXT_CHARS) + "\n...[truncated]..."
+            : combined;
+        return `\n\n<workspace-critical-rules>\n${safeContent}\n</workspace-critical-rules>`;
+    }
+    catch {
+        return "";
+    }
+}
 export default function compactionSafeguardExtension(api) {
     api.on("session_before_compact", async (event, ctx) => {
         const { preparation, customInstructions, signal } = event;
@@ -133,10 +166,11 @@ export default function compactionSafeguardExtension(api) {
             };
         }
         try {
-            const contextWindowTokens = resolveContextWindowTokens(model);
+            const runtime = getCompactionSafeguardRuntime(ctx.sessionManager);
+            const modelContextWindow = resolveContextWindowTokens(model);
+            const contextWindowTokens = runtime?.contextWindowTokens ?? modelContextWindow;
             const turnPrefixMessages = preparation.turnPrefixMessages ?? [];
             let messagesToSummarize = preparation.messagesToSummarize;
-            const runtime = getCompactionSafeguardRuntime(ctx.sessionManager);
             const maxHistoryShare = runtime?.maxHistoryShare ?? 0.5;
             const tokensBefore = typeof preparation.tokensBefore === "number" && Number.isFinite(preparation.tokensBefore)
                 ? preparation.tokensBefore
@@ -156,14 +190,15 @@ export default function compactionSafeguardExtension(api) {
                     });
                     if (pruned.droppedChunks > 0) {
                         const newContentRatio = (newContentTokens / contextWindowTokens) * 100;
-                        console.warn(`Compaction safeguard: new content uses ${newContentRatio.toFixed(1)}% of context; dropped ${pruned.droppedChunks} older chunk(s) ` +
+                        log.warn(`Compaction safeguard: new content uses ${newContentRatio.toFixed(1)}% of context; dropped ${pruned.droppedChunks} older chunk(s) ` +
                             `(${pruned.droppedMessages} messages) to fit history budget.`);
                         messagesToSummarize = pruned.messages;
                         // Summarize dropped messages so context isn't lost
                         if (pruned.droppedMessagesList.length > 0) {
                             try {
                                 const droppedChunkRatio = computeAdaptiveChunkRatio(pruned.droppedMessagesList, contextWindowTokens);
-                                const droppedMaxChunkTokens = Math.max(1, Math.floor(contextWindowTokens * droppedChunkRatio));
+                                const droppedMaxChunkTokens = Math.max(1, Math.floor(contextWindowTokens * droppedChunkRatio) -
+                                    SUMMARIZATION_OVERHEAD_TOKENS);
                                 droppedSummary = await summarizeInStages({
                                     messages: pruned.droppedMessagesList,
                                     model,
@@ -177,16 +212,18 @@ export default function compactionSafeguardExtension(api) {
                                 });
                             }
                             catch (droppedError) {
-                                console.warn(`Compaction safeguard: failed to summarize dropped messages, continuing without: ${droppedError instanceof Error ? droppedError.message : String(droppedError)}`);
+                                log.warn(`Compaction safeguard: failed to summarize dropped messages, continuing without: ${droppedError instanceof Error ? droppedError.message : String(droppedError)}`);
                             }
                         }
                     }
                 }
             }
-            // Use adaptive chunk ratio based on message sizes
+            // Use adaptive chunk ratio based on message sizes, reserving headroom for
+            // the summarization prompt, system prompt, previous summary, and reasoning budget
+            // that generateSummary adds on top of the serialized conversation chunk.
             const allMessages = [...messagesToSummarize, ...turnPrefixMessages];
             const adaptiveRatio = computeAdaptiveChunkRatio(allMessages, contextWindowTokens);
-            const maxChunkTokens = Math.max(1, Math.floor(contextWindowTokens * adaptiveRatio));
+            const maxChunkTokens = Math.max(1, Math.floor(contextWindowTokens * adaptiveRatio) - SUMMARIZATION_OVERHEAD_TOKENS);
             const reserveTokens = Math.max(1, Math.floor(preparation.settings.reserveTokens));
             // Feed dropped-messages summary as previousSummary so the main summarization
             // incorporates context from pruned messages instead of losing it entirely.
@@ -219,6 +256,11 @@ export default function compactionSafeguardExtension(api) {
             }
             summary += toolFailureSection;
             summary += fileOpsSummary;
+            // Append workspace critical context (Session Startup + Red Lines from AGENTS.md)
+            const workspaceContext = await readWorkspaceContextForSummary();
+            if (workspaceContext) {
+                summary += workspaceContext;
+            }
             return {
                 compaction: {
                     summary,
@@ -229,7 +271,7 @@ export default function compactionSafeguardExtension(api) {
             };
         }
         catch (error) {
-            console.warn(`Compaction summarization failed; truncating history: ${error instanceof Error ? error.message : String(error)}`);
+            log.warn(`Compaction summarization failed; truncating history: ${error instanceof Error ? error.message : String(error)}`);
             return {
                 compaction: {
                     summary: fallbackSummary,

package/dist/agents/pi-settings.js

@@ -17,3 +17,43 @@ export function resolveCompactionReserveTokensFloor(cfg) {
     }
     return DEFAULT_PI_COMPACTION_RESERVE_TOKENS_FLOOR;
 }
+function toNonNegativeInt(value) {
+    if (typeof value !== "number" || !Number.isFinite(value) || value < 0) {
+        return undefined;
+    }
+    return Math.floor(value);
+}
+function toPositiveInt(value) {
+    if (typeof value !== "number" || !Number.isFinite(value) || value <= 0) {
+        return undefined;
+    }
+    return Math.floor(value);
+}
+export function applyPiCompactionSettingsFromConfig(params) {
+    const currentReserveTokens = params.settingsManager.getCompactionReserveTokens();
+    const currentKeepRecentTokens = params.settingsManager.getCompactionKeepRecentTokens();
+    const compactionCfg = params.cfg?.agents?.defaults?.compaction;
+    const configuredReserveTokens = toNonNegativeInt(compactionCfg?.reserveTokens);
+    const configuredKeepRecentTokens = toPositiveInt(compactionCfg?.keepRecentTokens);
+    const reserveTokensFloor = resolveCompactionReserveTokensFloor(params.cfg);
+    const targetReserveTokens = Math.max(configuredReserveTokens ?? currentReserveTokens, reserveTokensFloor);
+    const targetKeepRecentTokens = configuredKeepRecentTokens ?? currentKeepRecentTokens;
+    const overrides = {};
+    if (targetReserveTokens !== currentReserveTokens) {
+        overrides.reserveTokens = targetReserveTokens;
+    }
+    if (targetKeepRecentTokens !== currentKeepRecentTokens) {
+        overrides.keepRecentTokens = targetKeepRecentTokens;
+    }
+    const didOverride = Object.keys(overrides).length > 0;
+    if (didOverride) {
+        params.settingsManager.applyOverrides({ compaction: overrides });
+    }
+    return {
+        didOverride,
+        compaction: {
+            reserveTokens: targetReserveTokens,
+            keepRecentTokens: targetKeepRecentTokens,
+        },
+    };
+}

package/dist/agents/pi-tools.policy.js

@@ -5,6 +5,7 @@ import { normalizeMessageChannel } from "../utils/message-channel.js";
 import { resolveAgentConfig, resolveAgentIdFromSessionKey } from "./agent-scope.js";
 import { compileGlobPatterns, matchesAnyGlobPattern } from "./glob-pattern.js";
 import { pickSandboxToolPolicy } from "./sandbox-tool-policy.js";
+import { DEFAULT_SUBAGENT_MAX_SPAWN_DEPTH } from "../config/agent-limits.js";
 import { expandToolGroups, normalizeToolName } from "./tool-policy.js";
 function makeToolPolicyMatcher(policy) {
     const deny = compileGlobPatterns({
@@ -75,7 +76,7 @@ function resolveSubagentDenyList(depth, maxSpawnDepth) {
 }
 export function resolveSubagentToolPolicy(cfg, depth) {
     const configured = cfg?.tools?.subagents?.tools;
-    const maxSpawnDepth = cfg?.agents?.defaults?.subagents?.maxSpawnDepth ?? 1;
+    const maxSpawnDepth = cfg?.agents?.defaults?.subagents?.maxSpawnDepth ?? DEFAULT_SUBAGENT_MAX_SPAWN_DEPTH;
     const effectiveDepth = typeof depth === "number" && depth >= 0 ? depth : 1;
     const baseDeny = resolveSubagentDenyList(effectiveDepth, maxSpawnDepth);
     const deny = [...baseDeny, ...(Array.isArray(configured?.deny) ? configured.deny : [])];

package/dist/agents/provider/config-loader.js

@@ -46,7 +46,7 @@ export function loadPoolConfig(cfg) {
             }
             try {
                 TokenPool.addToken(providerID, tokenCfg.id, tokenCfg.key, {
-                    tier: tokenCfg.tier ?? "paid",
+                    tier: (tokenCfg.tier ?? "paid"),
                     label: tokenCfg.label,
                     enabled: tokenCfg.enabled ?? true,
                 });