@poolzin/pool-bot 2026.2.20 → 2026.2.22

package/dist/agents/pi-embedded-helpers/errors.js

@@ -1,5 +1,23 @@
 import { formatSandboxToolPolicyBlockedMessage } from "../sandbox.js";
-export const BILLING_ERROR_USER_MESSAGE = "⚠️ API provider returned a billing error — your API key has run out of credits or has an insufficient balance. Check your provider's billing dashboard and top up or switch to a different API key.";
+export function formatBillingErrorMessage(provider) {
+    const providerName = provider?.trim();
+    if (providerName) {
+        return `⚠️ ${providerName} returned a billing error — your API key has run out of credits or has an insufficient balance. Check your ${providerName} billing dashboard and top up or switch to a different API key.`;
+    }
+    return "⚠️ API provider returned a billing error — your API key has run out of credits or has an insufficient balance. Check your provider's billing dashboard and top up or switch to a different API key.";
+}
+export const BILLING_ERROR_USER_MESSAGE = formatBillingErrorMessage();
+const RATE_LIMIT_ERROR_USER_MESSAGE = "⚠️ API rate limit reached. Please try again later.";
+const OVERLOADED_ERROR_USER_MESSAGE = "The AI service is temporarily overloaded. Please try again in a moment.";
+function formatRateLimitOrOverloadedErrorCopy(raw) {
+    if (isRateLimitErrorMessage(raw)) {
+        return RATE_LIMIT_ERROR_USER_MESSAGE;
+    }
+    if (isOverloadedErrorMessage(raw)) {
+        return OVERLOADED_ERROR_USER_MESSAGE;
+    }
+    return undefined;
+}
 export function isContextOverflowError(errorMessage) {
     if (!errorMessage) {
         return false;
@@ -20,7 +38,8 @@ export function isContextOverflowError(errorMessage) {
         (lower.includes("413") && lower.includes("too large")));
 }
 const CONTEXT_WINDOW_TOO_SMALL_RE = /context window.*(too small|minimum is)/i;
-const CONTEXT_OVERFLOW_HINT_RE = /context.*overflow|context window.*(too (?:large|long)|exceed|over|limit|max(?:imum)?|requested|sent|tokens)|(?:prompt|request|input).*(too (?:large|long)|exceed|over|limit|max(?:imum)?)/i;
+const CONTEXT_OVERFLOW_HINT_RE = /context.*overflow|context window.*(too (?:large|long)|exceed|over|limit|max(?:imum)?|requested|sent|tokens)|prompt.*(too (?:large|long)|exceed|over|limit|max(?:imum)?)|(?:request|input).*(?:context|window|length|token).*(too (?:large|long)|exceed|over|limit|max(?:imum)?)/i;
+const RATE_LIMIT_HINT_RE = /rate limit|too many requests|requests per (?:minute|hour|day)|quota|throttl|429\b/i;
 export function isLikelyContextOverflowError(errorMessage) {
     if (!errorMessage) {
         return false;
@@ -28,9 +47,18 @@ export function isLikelyContextOverflowError(errorMessage) {
     if (CONTEXT_WINDOW_TOO_SMALL_RE.test(errorMessage)) {
         return false;
     }
+    // Rate limit errors can match the broad CONTEXT_OVERFLOW_HINT_RE pattern
+    // (e.g., "request reached organization TPD rate limit" matches request.*limit).
+    // Exclude them before checking context overflow heuristics.
+    if (isRateLimitErrorMessage(errorMessage)) {
+        return false;
+    }
     if (isContextOverflowError(errorMessage)) {
         return true;
     }
+    if (RATE_LIMIT_HINT_RE.test(errorMessage)) {
+        return false;
+    }
     return CONTEXT_OVERFLOW_HINT_RE.test(errorMessage);
 }
 export function isCompactionFailureError(errorMessage) {
@@ -45,15 +73,24 @@ export function isCompactionFailureError(errorMessage) {
     if (!hasCompactionTerm) {
         return false;
     }
-    // For compaction failures, also accept "context overflow" without colon
-    // since the error message itself describes a compaction/summarization failure
-    return isContextOverflowError(errorMessage) || lower.includes("context overflow");
+    // Treat any likely overflow shape as a compaction failure when compaction terms are present.
+    // Providers often vary wording (e.g. "context window exceeded") across APIs.
+    if (isLikelyContextOverflowError(errorMessage)) {
+        return true;
+    }
+    // Keep explicit fallback for bare "context overflow" strings.
+    return lower.includes("context overflow");
 }
 const ERROR_PAYLOAD_PREFIX_RE = /^(?:error|api\s*error|apierror|openai\s*error|anthropic\s*error|gateway\s*error)[:\s-]+/i;
 const FINAL_TAG_RE = /<\s*\/?\s*final\s*>/gi;
 const ERROR_PREFIX_RE = /^(?:error|api\s*error|openai\s*error|anthropic\s*error|gateway\s*error|request failed|failed|exception)[:\s-]+/i;
 const CONTEXT_OVERFLOW_ERROR_HEAD_RE = /^(?:context overflow:|request_too_large\b|request size exceeds\b|request exceeds the maximum size\b|context length exceeded\b|maximum context length\b|prompt is too long\b|exceeds model context window\b)/i;
+const BILLING_ERROR_HEAD_RE = /^(?:error[:\s-]+)?billing(?:\s+error)?(?:[:\s-]+|$)|^(?:error[:\s-]+)?(?:credit balance|insufficient credits?|payment required|http\s*402\b)/i;
 const HTTP_STATUS_PREFIX_RE = /^(?:http\s*)?(\d{3})\s+(.+)$/i;
+const HTTP_STATUS_CODE_PREFIX_RE = /^(?:http\s*)?(\d{3})(?:\s+([\s\S]+))?$/i;
+const HTML_ERROR_PREFIX_RE = /^\s*(?:<!doctype\s+html\b|<html\b)/i;
+const CLOUDFLARE_HTML_ERROR_CODES = new Set([521, 522, 523, 524, 525, 526, 530]);
+const TRANSIENT_HTTP_ERROR_CODES = new Set([500, 502, 503, 521, 522, 523, 524, 529]);
 const HTTP_ERROR_HINTS = [
     "error",
     "bad request",
@@ -71,6 +108,42 @@ const HTTP_ERROR_HINTS = [
     "too many requests",
     "permission",
 ];
+function extractLeadingHttpStatus(raw) {
+    const match = raw.match(HTTP_STATUS_CODE_PREFIX_RE);
+    if (!match) {
+        return null;
+    }
+    const code = Number(match[1]);
+    if (!Number.isFinite(code)) {
+        return null;
+    }
+    return { code, rest: (match[2] ?? "").trim() };
+}
+export function isCloudflareOrHtmlErrorPage(raw) {
+    const trimmed = raw.trim();
+    if (!trimmed) {
+        return false;
+    }
+    const status = extractLeadingHttpStatus(trimmed);
+    if (!status || status.code < 500) {
+        return false;
+    }
+    if (CLOUDFLARE_HTML_ERROR_CODES.has(status.code)) {
+        return true;
+    }
+    return (status.code < 600 && HTML_ERROR_PREFIX_RE.test(status.rest) && /<\/html>/i.test(status.rest));
+}
+export function isTransientHttpError(raw) {
+    const trimmed = raw.trim();
+    if (!trimmed) {
+        return false;
+    }
+    const status = extractLeadingHttpStatus(trimmed);
+    if (!status) {
+        return false;
+    }
+    return TRANSIENT_HTTP_ERROR_CODES.has(status.code);
+}
 function stripFinalTagsFromText(text) {
     if (!text) {
         return text;
@@ -103,6 +176,9 @@ function collapseConsecutiveDuplicateBlocks(text) {
     return result.join("\n\n");
 }
 function isLikelyHttpErrorText(raw) {
+    if (isCloudflareOrHtmlErrorPage(raw)) {
+        return true;
+    }
     const match = raw.match(HTTP_STATUS_PREFIX_RE);
     if (!match) {
         return false;
@@ -123,6 +199,15 @@ function shouldRewriteContextOverflowText(raw) {
         ERROR_PREFIX_RE.test(raw) ||
         CONTEXT_OVERFLOW_ERROR_HEAD_RE.test(raw));
 }
+function shouldRewriteBillingText(raw) {
+    if (!isBillingErrorMessage(raw)) {
+        return false;
+    }
+    return (isRawApiErrorPayload(raw) ||
+        isLikelyHttpErrorText(raw) ||
+        ERROR_PREFIX_RE.test(raw) ||
+        BILLING_ERROR_HEAD_RE.test(raw));
+}
 function isErrorPayloadObject(payload) {
     if (!payload || typeof payload !== "object" || Array.isArray(payload)) {
         return false;
@@ -252,6 +337,10 @@ export function formatRawAssistantErrorForUi(raw) {
     if (!trimmed) {
         return "LLM request failed with an unknown error.";
     }
+    const leadingStatus = extractLeadingHttpStatus(trimmed);
+    if (leadingStatus && isCloudflareOrHtmlErrorPage(trimmed)) {
+        return `The AI service is temporarily unavailable (HTTP ${leadingStatus.code}). Please try again in a moment.`;
+    }
     const httpMatch = trimmed.match(HTTP_STATUS_PREFIX_RE);
     if (httpMatch) {
         const rest = httpMatch[2].trim();
@@ -307,11 +396,15 @@ export function formatAssistantErrorText(msg, opts) {
     if (invalidRequest?.[1]) {
         return `LLM request rejected: ${invalidRequest[1]}`;
     }
-    if (isOverloadedErrorMessage(raw)) {
-        return "The AI service is temporarily overloaded. Please try again in a moment.";
+    const transientCopy = formatRateLimitOrOverloadedErrorCopy(raw);
+    if (transientCopy) {
+        return transientCopy;
+    }
+    if (isTimeoutErrorMessage(raw)) {
+        return "LLM request timed out.";
     }
     if (isBillingErrorMessage(raw)) {
-        return BILLING_ERROR_USER_MESSAGE;
+        return formatBillingErrorMessage(opts?.provider);
     }
     if (isLikelyHttpErrorText(raw) || isRawApiErrorPayload(raw)) {
         return formatRawAssistantErrorForUi(raw);
@@ -330,7 +423,7 @@ export function sanitizeUserFacingText(text, opts) {
     const stripped = stripFinalTagsFromText(text);
     const trimmed = stripped.trim();
     if (!trimmed) {
-        return stripped;
+        return "";
     }
     // Only apply error-pattern rewrites when the caller knows this text is an error payload.
     // Otherwise we risk swallowing legitimate assistant text that merely *mentions* these errors.
@@ -350,8 +443,9 @@ export function sanitizeUserFacingText(text, opts) {
             return formatRawAssistantErrorForUi(trimmed);
         }
         if (ERROR_PREFIX_RE.test(trimmed)) {
-            if (isOverloadedErrorMessage(trimmed) || isRateLimitErrorMessage(trimmed)) {
-                return "The AI service is temporarily overloaded. Please try again in a moment.";
+            const prefixedCopy = formatRateLimitOrOverloadedErrorCopy(trimmed);
+            if (prefixedCopy) {
+                return prefixedCopy;
             }
             if (isTimeoutErrorMessage(trimmed)) {
                 return "LLM request timed out.";
@@ -359,7 +453,16 @@ export function sanitizeUserFacingText(text, opts) {
             return formatRawAssistantErrorForUi(trimmed);
         }
     }
-    return collapseConsecutiveDuplicateBlocks(stripped);
+    // Preserve legacy behavior for explicit billing-head text outside known
+    // error contexts (e.g., "billing: please upgrade your plan"), while
+    // keeping conversational billing mentions untouched.
+    if (shouldRewriteBillingText(trimmed)) {
+        return BILLING_ERROR_USER_MESSAGE;
+    }
+    // Strip leading blank lines (including whitespace-only lines) without clobbering indentation on
+    // the first content line (e.g. markdown/code blocks).
+    const withoutLeadingEmptyLines = stripped.replace(/^(?:[ \t]*\r?\n)+/, "");
+    return collapseConsecutiveDuplicateBlocks(withoutLeadingEmptyLines);
 }
 export function isRateLimitAssistantError(msg) {
     if (!msg || msg.stopReason !== "error") {
@@ -377,9 +480,18 @@ const ERROR_PATTERNS = {
         "usage limit",
     ],
     overloaded: [/overloaded_error|"type"\s*:\s*"overloaded_error"/i, "overloaded"],
-    timeout: ["timeout", "timed out", "deadline exceeded", "context deadline exceeded"],
+    timeout: [
+        "timeout",
+        "timed out",
+        "deadline exceeded",
+        "context deadline exceeded",
+        /without sending (?:any )?chunks?/i,
+        /\bstop reason:\s*abort\b/i,
+        /\breason:\s*abort\b/i,
+        /\bunhandled stop reason:\s*abort\b/i,
+    ],
     billing: [
-        /\b402\b/,
+        /["']?(?:status|code)["']?\s*[:=]\s*402\b|\bhttp\s*402\b|\berror(?:\s+code)?\s*[:=]?\s*402\b|\b(?:got|returned|received)\s+(?:a\s+)?402\b|^\s*402\s+payment/i,
         "payment required",
         "insufficient credits",
         "credit balance",
@@ -434,7 +546,16 @@ export function isBillingErrorMessage(raw) {
     if (!value) {
         return false;
     }
-    return matchesErrorPatterns(value, ERROR_PATTERNS.billing);
+    if (matchesErrorPatterns(value, ERROR_PATTERNS.billing)) {
+        return true;
+    }
+    if (!BILLING_ERROR_HEAD_RE.test(raw)) {
+        return false;
+    }
+    return (value.includes("upgrade") ||
+        value.includes("credits") ||
+        value.includes("payment") ||
+        value.includes("plan"));
 }
 export function isMissingToolCallInputError(raw) {
     if (!raw) {
@@ -510,6 +631,10 @@ export function classifyFailoverReason(raw) {
     if (isImageSizeError(raw)) {
         return null;
     }
+    if (isTransientHttpError(raw)) {
+        // Treat transient 5xx provider failures as retryable transport issues.
+        return "timeout";
+    }
     if (isRateLimitErrorMessage(raw)) {
         return "rate_limit";
     }

package/dist/agents/pi-embedded-helpers/images.js

@@ -3,24 +3,32 @@ import { sanitizeContentBlocksImages } from "../tool-images.js";
 import { stripThoughtSignatures } from "./bootstrap.js";
 export function isEmptyAssistantMessageContent(message) {
     const content = message.content;
-    if (content == null)
+    if (content == null) {
         return true;
-    if (!Array.isArray(content))
+    }
+    if (!Array.isArray(content)) {
         return false;
+    }
     return content.every((block) => {
-        if (!block || typeof block !== "object")
+        if (!block || typeof block !== "object") {
             return true;
+        }
         const rec = block;
-        if (rec.type !== "text")
+        if (rec.type !== "text") {
             return false;
+        }
         return typeof rec.text !== "string" || rec.text.trim().length === 0;
     });
 }
 export async function sanitizeSessionMessagesImages(messages, label, options) {
     const sanitizeMode = options?.sanitizeMode ?? "full";
     const allowNonImageSanitization = sanitizeMode === "full";
+    const imageSanitization = {
+        maxDimensionPx: options?.maxDimensionPx,
+        maxBytes: options?.maxBytes,
+    };
     // We sanitize historical session messages because Anthropic can reject a request
-    // if the transcript contains oversized base64 images (see MAX_IMAGE_DIMENSION_PX).
+    // if the transcript contains oversized base64 images (default max side 1200px).
     const sanitizedIds = allowNonImageSanitization && options?.sanitizeToolCallIds
         ? sanitizeToolCallIdsForCloudCodeAssist(messages, options.toolCallIdMode)
         : messages;
@@ -34,7 +42,7 @@ export async function sanitizeSessionMessagesImages(messages, label, options) {
         if (role === "toolResult") {
             const toolMsg = msg;
             const content = Array.isArray(toolMsg.content) ? toolMsg.content : [];
-            const nextContent = (await sanitizeContentBlocksImages(content, label));
+            const nextContent = (await sanitizeContentBlocksImages(content, label, imageSanitization));
             out.push({ ...toolMsg, content: nextContent });
             continue;
         }
@@ -42,7 +50,7 @@ export async function sanitizeSessionMessagesImages(messages, label, options) {
             const userMsg = msg;
             const content = userMsg.content;
             if (Array.isArray(content)) {
-                const nextContent = (await sanitizeContentBlocksImages(content, label));
+                const nextContent = (await sanitizeContentBlocksImages(content, label, imageSanitization));
                 out.push({ ...userMsg, content: nextContent });
                 continue;
             }
@@ -52,7 +60,7 @@ export async function sanitizeSessionMessagesImages(messages, label, options) {
             if (assistantMsg.stopReason === "error") {
                 const content = assistantMsg.content;
                 if (Array.isArray(content)) {
-                    const nextContent = (await sanitizeContentBlocksImages(content, label));
+                    const nextContent = (await sanitizeContentBlocksImages(content, label, imageSanitization));
                     out.push({ ...assistantMsg, content: nextContent });
                 }
                 else {
@@ -63,7 +71,7 @@ export async function sanitizeSessionMessagesImages(messages, label, options) {
             const content = assistantMsg.content;
             if (Array.isArray(content)) {
                 if (!allowNonImageSanitization) {
-                    const nextContent = (await sanitizeContentBlocksImages(content, label));
+                    const nextContent = (await sanitizeContentBlocksImages(content, label, imageSanitization));
                     out.push({ ...assistantMsg, content: nextContent });
                     continue;
                 }
@@ -71,14 +79,16 @@ export async function sanitizeSessionMessagesImages(messages, label, options) {
                     ? content // Keep signatures for Antigravity Claude
                     : stripThoughtSignatures(content, options?.sanitizeThoughtSignatures); // Strip for Gemini
                 const filteredContent = strippedContent.filter((block) => {
-                    if (!block || typeof block !== "object")
+                    if (!block || typeof block !== "object") {
                         return true;
+                    }
                     const rec = block;
-                    if (rec.type !== "text" || typeof rec.text !== "string")
+                    if (rec.type !== "text" || typeof rec.text !== "string") {
                         return true;
+                    }
                     return rec.text.trim().length > 0;
                 });
-                const finalContent = (await sanitizeContentBlocksImages(filteredContent, label));
+                const finalContent = (await sanitizeContentBlocksImages(filteredContent, label, imageSanitization));
                 if (finalContent.length === 0) {
                     continue;
                 }

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

@@ -1,5 +1,5 @@
-export { buildBootstrapContextFiles, DEFAULT_BOOTSTRAP_MAX_CHARS, ensureSessionHeader, resolveBootstrapMaxChars, stripThoughtSignatures, } from "./pi-embedded-helpers/bootstrap.js";
-export { BILLING_ERROR_USER_MESSAGE, classifyFailoverReason, formatRawAssistantErrorForUi, formatAssistantErrorText, getApiErrorPayloadFingerprint, isAuthAssistantError, isAuthErrorMessage, isBillingAssistantError, parseApiErrorInfo, parseImageSizeError, sanitizeUserFacingText, isBillingErrorMessage, isCloudCodeAssistFormatError, isCompactionFailureError, isContextOverflowError, isLikelyContextOverflowError, isFailoverAssistantError, isFailoverErrorMessage, isImageDimensionErrorMessage, isImageSizeError, isOverloadedErrorMessage, isRawApiErrorPayload, isRateLimitAssistantError, isRateLimitErrorMessage, isTimeoutErrorMessage, parseImageDimensionError, } from "./pi-embedded-helpers/errors.js";
+export { buildBootstrapContextFiles, DEFAULT_BOOTSTRAP_MAX_CHARS, DEFAULT_BOOTSTRAP_TOTAL_MAX_CHARS, ensureSessionHeader, resolveBootstrapMaxChars, resolveBootstrapTotalMaxChars, stripThoughtSignatures, } from "./pi-embedded-helpers/bootstrap.js";
+export { BILLING_ERROR_USER_MESSAGE, formatBillingErrorMessage, classifyFailoverReason, formatRawAssistantErrorForUi, formatAssistantErrorText, getApiErrorPayloadFingerprint, isAuthAssistantError, isAuthErrorMessage, isBillingAssistantError, parseApiErrorInfo, sanitizeUserFacingText, isBillingErrorMessage, isCloudflareOrHtmlErrorPage, isCloudCodeAssistFormatError, isCompactionFailureError, isContextOverflowError, isLikelyContextOverflowError, isFailoverAssistantError, isFailoverErrorMessage, isImageDimensionErrorMessage, isImageSizeError, isOverloadedErrorMessage, isRawApiErrorPayload, isRateLimitAssistantError, isRateLimitErrorMessage, isTransientHttpError, isTimeoutErrorMessage, parseImageDimensionError, parseImageSizeError, } from "./pi-embedded-helpers/errors.js";
 export { isGoogleModelApi, sanitizeGoogleTurnOrdering } from "./pi-embedded-helpers/google.js";
 export { downgradeOpenAIReasoningBlocks } from "./pi-embedded-helpers/openai.js";
 export { isEmptyAssistantMessageContent, sanitizeSessionMessagesImages, } from "./pi-embedded-helpers/images.js";

package/dist/agents/pi-embedded-runner/abort.js

@@ -1,9 +1,16 @@
-export function isAbortError(err) {
-    if (!err || typeof err !== "object")
+/**
+ * Runner abort check. Catches any abort-related message for embedded runners.
+ * More permissive than the core isAbortError since runners need to catch
+ * various abort signals from different sources.
+ */
+export function isRunnerAbortError(err) {
+    if (!err || typeof err !== "object") {
         return false;
+    }
     const name = "name" in err ? String(err.name) : "";
-    if (name === "AbortError")
+    if (name === "AbortError") {
         return true;
+    }
     const message = "message" in err && typeof err.message === "string" ? err.message.toLowerCase() : "";
     return message.includes("aborted");
 }