notelm-mcp 1.2.1

package/dist/utils/page-utils.js

@@ -0,0 +1,381 @@
+/**
+ * Page utilities for extracting responses from NotebookLM web UI
+ *
+ * This module provides functions to:
+ * - Extract latest assistant responses from the page
+ * - Wait for new responses with streaming detection
+ * - Detect placeholders and loading states
+ * - Snapshot existing responses for comparison
+ *
+ * Based on the Python implementation from page_utils.py
+ */
+import { log } from "./logger.js";
+import { RESPONSE_SELECTORS, THINKING_SELECTOR } from "../constants.js";
+// ============================================================================
+// Helper Functions
+// ============================================================================
+/**
+ * Simple string hash function (for efficient comparison)
+ */
+function hashString(str) {
+    let hash = 0;
+    for (let i = 0; i < str.length; i++) {
+        const char = str.charCodeAt(i);
+        hash = (hash << 5) - hash + char;
+        hash = hash & hash; // Convert to 32bit integer
+    }
+    return hash;
+}
+// ============================================================================
+// Main Functions
+// ============================================================================
+/**
+ * Snapshot the latest response text currently visible
+ * Returns null if no response found
+ */
+export async function snapshotLatestResponse(page) {
+    return await extractLatestText(page, new Set(), false, 0);
+}
+/**
+ * Snapshot ALL existing assistant response texts
+ * Used to capture visible responses BEFORE submitting a new question
+ */
+export async function snapshotAllResponses(page) {
+    const allTexts = [];
+    const primarySelector = ".to-user-container";
+    try {
+        const containers = await page.$$(primarySelector);
+        if (containers.length > 0) {
+            for (const container of containers) {
+                try {
+                    const textElement = await container.$(".message-text-content");
+                    if (textElement) {
+                        const text = await textElement.innerText();
+                        if (text && text.trim()) {
+                            allTexts.push(text.trim());
+                        }
+                    }
+                }
+                catch {
+                    continue;
+                }
+            }
+            log.info(`📸 [SNAPSHOT] Captured ${allTexts.length} existing responses`);
+        }
+    }
+    catch (error) {
+        log.warning(`⚠️ [SNAPSHOT] Failed to snapshot responses: ${error}`);
+    }
+    return allTexts;
+}
+/**
+ * Count the number of visible assistant response elements
+ */
+export async function countResponseElements(page) {
+    let count = 0;
+    for (const selector of RESPONSE_SELECTORS) {
+        try {
+            const elements = await page.$$(selector);
+            if (elements.length > 0) {
+                // Count only visible elements
+                for (const el of elements) {
+                    try {
+                        const isVisible = await el.isVisible();
+                        if (isVisible) {
+                            count++;
+                        }
+                    }
+                    catch {
+                        continue;
+                    }
+                }
+                // If we found elements with this selector, stop trying others
+                if (count > 0) {
+                    break;
+                }
+            }
+        }
+        catch {
+            continue;
+        }
+    }
+    return count;
+}
+/**
+ * Wait for a new assistant response with streaming detection
+ *
+ * This function:
+ * 1. Polls the page for new response text
+ * 2. Detects streaming (text changes) vs. complete (text stable)
+ * 3. Requires text to be stable for 3 consecutive polls before returning
+ * 4. Ignores placeholders, question echoes, and known responses
+ *
+ * @param page Playwright page instance
+ * @param options Options for waiting
+ * @returns The new response text, or null if timeout
+ */
+export async function waitForLatestAnswer(page, options = {}) {
+    const { question = "", timeoutMs = 120000, pollIntervalMs = 1000, ignoreTexts = [], debug = false, } = options;
+    const deadline = Date.now() + timeoutMs;
+    const sanitizedQuestion = question.trim().toLowerCase();
+    // Track ALL known texts as HASHES (memory efficient!)
+    const knownHashes = new Set();
+    for (const text of ignoreTexts) {
+        if (typeof text === "string" && text.trim()) {
+            knownHashes.add(hashString(text.trim()));
+        }
+    }
+    if (debug) {
+        log.debug(`🔍 [DEBUG] Waiting for NEW answer. Ignoring ${knownHashes.size} known responses`);
+    }
+    let pollCount = 0;
+    let lastCandidate = null;
+    let stableCount = 0; // Track how many times we see the same text
+    const requiredStablePolls = 3; // Text must be stable for 3 consecutive polls
+    while (Date.now() < deadline) {
+        pollCount++;
+        // Check if NotebookLM is still "thinking" (most reliable indicator)
+        try {
+            const thinkingElement = await page.$(THINKING_SELECTOR);
+            if (thinkingElement) {
+                const isVisible = await thinkingElement.isVisible();
+                if (isVisible) {
+                    if (debug && pollCount % 5 === 0) {
+                        log.debug(`🔍 [DEBUG] NotebookLM still thinking (${THINKING_SELECTOR} visible)...`);
+                    }
+                    await page.waitForTimeout(pollIntervalMs);
+                    continue;
+                }
+            }
+        }
+        catch {
+            // Ignore errors checking thinking state
+        }
+        // Extract latest NEW text
+        const candidate = await extractLatestText(page, knownHashes, debug, pollCount);
+        if (candidate) {
+            const normalized = candidate.trim();
+            if (normalized) {
+                const lower = normalized.toLowerCase();
+                // Check if it's the question echo
+                if (lower === sanitizedQuestion) {
+                    if (debug) {
+                        log.debug("🔍 [DEBUG] Found question echo, ignoring");
+                    }
+                    knownHashes.add(hashString(normalized)); // Mark as seen
+                    await page.waitForTimeout(pollIntervalMs);
+                    continue;
+                }
+                // ========================================
+                // STREAMING DETECTION: Check if text is stable
+                // ========================================
+                if (normalized === lastCandidate) {
+                    // Text hasn't changed - it's stable
+                    stableCount++;
+                    if (debug && stableCount === requiredStablePolls) {
+                        log.debug(`✅ [DEBUG] Text stable for ${stableCount} polls (${normalized.length} chars)`);
+                    }
+                }
+                else {
+                    // Text changed - streaming in progress
+                    if (debug && lastCandidate) {
+                        log.debug(`🔄 [DEBUG] Text changed (${normalized.length} chars, was ${lastCandidate.length})`);
+                    }
+                    stableCount = 1;
+                    lastCandidate = normalized;
+                }
+                // Only return once text is stable
+                if (stableCount >= requiredStablePolls) {
+                    if (debug) {
+                        log.debug(`✅ [DEBUG] Returning stable answer (${normalized.length} chars)`);
+                    }
+                    return normalized;
+                }
+            }
+        }
+        await page.waitForTimeout(pollIntervalMs);
+    }
+    if (debug) {
+        log.debug(`⏱️ [DEBUG] Timeout after ${pollCount} polls`);
+    }
+    return null;
+}
+/**
+ * Extract the latest NEW response text from the page
+ * Uses hash-based comparison for efficiency
+ *
+ * @param page Playwright page instance
+ * @param knownHashes Set of hashes of already-seen response texts
+ * @param debug Enable debug logging
+ * @param pollCount Current poll number (for conditional logging)
+ * @returns First NEW response text found, or null
+ */
+async function extractLatestText(page, knownHashes, debug, pollCount) {
+    // Try the primary selector first (most specific for NotebookLM)
+    const primarySelector = ".to-user-container";
+    try {
+        const containers = await page.$$(primarySelector);
+        const totalContainers = containers.length;
+        // Early exit if no new containers possible
+        if (totalContainers <= knownHashes.size) {
+            if (debug && pollCount % 5 === 0) {
+                log.dim(`⏭️ [EXTRACT] No new containers (${totalContainers} total, ${knownHashes.size} known)`);
+            }
+            return null;
+        }
+        if (containers.length > 0) {
+            // Only log every 5th poll to reduce noise
+            if (debug && pollCount % 5 === 0) {
+                log.dim(`🔍 [EXTRACT] Scanning ${totalContainers} containers (${knownHashes.size} known)`);
+            }
+            let skipped = 0;
+            let empty = 0;
+            // Scan ALL containers to find the FIRST with NEW text
+            for (let idx = 0; idx < containers.length; idx++) {
+                const container = containers[idx];
+                try {
+                    const textElement = await container.$(".message-text-content");
+                    if (textElement) {
+                        const text = await textElement.innerText();
+                        if (text && text.trim()) {
+                            // Hash-based comparison (faster & less memory)
+                            const textHash = hashString(text.trim());
+                            if (!knownHashes.has(textHash)) {
+                                log.success(`✅ [EXTRACT] Found NEW text in container[${idx}]: ${text.trim().length} chars`);
+                                return text.trim();
+                            }
+                            else {
+                                skipped++;
+                            }
+                        }
+                        else {
+                            empty++;
+                        }
+                    }
+                }
+                catch {
+                    continue;
+                }
+            }
+            // Only log summary if debug enabled
+            if (debug && pollCount % 5 === 0) {
+                log.dim(`⏭️ [EXTRACT] No NEW text (skipped ${skipped} known, ${empty} empty)`);
+            }
+            return null; // Don't fall through to fallback!
+        }
+        else {
+            if (debug) {
+                log.warning("⚠️ [EXTRACT] No containers found");
+            }
+        }
+    }
+    catch (error) {
+        log.error(`❌ [EXTRACT] Primary selector failed: ${error}`);
+    }
+    // Fallback: Try other selectors (only if primary selector failed/found nothing)
+    if (debug) {
+        log.dim("🔄 [EXTRACT] Trying fallback selectors...");
+    }
+    for (const selector of RESPONSE_SELECTORS) {
+        try {
+            const elements = await page.$$(selector);
+            if (elements.length === 0)
+                continue;
+            // Scan ALL elements to find the first with NEW text
+            for (const element of elements) {
+                try {
+                    // Prefer full container text when available
+                    let container = element;
+                    try {
+                        const closest = await element.evaluateHandle((el) => {
+                            return el.closest("[data-message-author], [data-message-role], [data-author], " +
+                                "[data-testid*='assistant'], [data-automation-id*='response'], article, section");
+                        });
+                        if (closest) {
+                            const closestElement = closest.asElement();
+                            if (closestElement) {
+                                container = closestElement;
+                            }
+                        }
+                    }
+                    catch {
+                        container = element;
+                    }
+                    const text = await container.innerText();
+                    if (text && text.trim() && !knownHashes.has(hashString(text.trim()))) {
+                        return text.trim();
+                    }
+                }
+                catch {
+                    continue;
+                }
+            }
+        }
+        catch {
+            continue;
+        }
+    }
+    // Final fallback: JavaScript evaluation
+    try {
+        const fallbackText = await page.evaluate(() => {
+            const unique = new Set();
+            const isVisible = (el) => {
+                if (!el || !el.isConnected)
+                    return false;
+                const rect = el.getBoundingClientRect();
+                if (rect.width === 0 || rect.height === 0)
+                    return false;
+                const style = window.getComputedStyle(el);
+                if (style.visibility === "hidden" ||
+                    style.display === "none" ||
+                    parseFloat(style.opacity || "1") === 0) {
+                    return false;
+                }
+                return true;
+            };
+            const selectors = [
+                "[data-message-author]",
+                "[data-message-role]",
+                "[data-author]",
+                "[data-renderer*='assistant']",
+                "[data-testid*='assistant']",
+                "[data-automation-id*='response']",
+            ];
+            const candidates = [];
+            for (const selector of selectors) {
+                for (const el of Array.from(document.querySelectorAll(selector))) {
+                    if (!isVisible(el))
+                        continue;
+                    if (unique.has(el))
+                        continue;
+                    unique.add(el);
+                    const text = el.innerText || el.textContent || "";
+                    if (!text.trim())
+                        continue;
+                    candidates.push(text.trim());
+                }
+            }
+            if (candidates.length > 0) {
+                return candidates[candidates.length - 1];
+            }
+            return null;
+        });
+        if (typeof fallbackText === "string" && fallbackText.trim()) {
+            return fallbackText.trim();
+        }
+    }
+    catch {
+        // Ignore evaluation errors
+    }
+    return null;
+}
+// ============================================================================
+// Exports
+// ============================================================================
+export default {
+    snapshotLatestResponse,
+    snapshotAllResponses,
+    countResponseElements,
+    waitForLatestAnswer,
+};
+//# sourceMappingURL=page-utils.js.map

package/dist/utils/page-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"page-utils.js","sourceRoot":"","sources":["../../src/utils/page-utils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAGH,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAExE,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;GAEG;AACH,SAAS,UAAU,CAAC,GAAW;IAC7B,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,MAAM,IAAI,GAAG,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAC/B,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;QACjC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,2BAA2B;IACjD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAGD,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,IAAU;IACrD,OAAO,MAAM,iBAAiB,CAAC,IAAI,EAAE,IAAI,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,IAAU;IACnD,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,eAAe,GAAG,oBAAoB,CAAC;IAE7C,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC;QAClD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC;oBAC/D,IAAI,WAAW,EAAE,CAAC;wBAChB,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,SAAS,EAAE,CAAC;wBAC3C,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;4BACxB,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;wBAC7B,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAS;gBACX,CAAC;YACH,CAAC;YAED,GAAG,CAAC,IAAI,CAAC,0BAA0B,QAAQ,CAAC,MAAM,qBAAqB,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,OAAO,CAAC,+CAA+C,KAAK,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,IAAU;IACpD,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,QAAQ,IAAI,kBAAkB,EAAE,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;YACzC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,8BAA8B;gBAC9B,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;oBAC1B,IAAI,CAAC;wBACH,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,SAAS,EAAE,CAAC;wBACvC,IAAI,SAAS,EAAE,CAAC;4BACd,KAAK,EAAE,CAAC;wBACV,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,SAAS;oBACX,CAAC;gBACH,CAAC;gBACD,8DAA8D;gBAC9D,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;oBACd,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,IAAU,EACV,UAMI,EAAE;IAEN,MAAM,EACJ,QAAQ,GAAG,EAAE,EACb,SAAS,GAAG,MAAM,EAClB,cAAc,GAAG,IAAI,EACrB,WAAW,GAAG,EAAE,EAChB,KAAK,GAAG,KAAK,GACd,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;IACxC,MAAM,iBAAiB,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAExD,sDAAsD;IACtD,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QAC/B,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YAC5C,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,IAAI,KAAK,EAAE,CAAC;QACV,GAAG,CAAC,KAAK,CACP,+CAA+C,WAAW,CAAC,IAAI,kBAAkB,CAClF,CAAC;IACJ,CAAC;IAED,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,IAAI,aAAa,GAAkB,IAAI,CAAC;IACxC,IAAI,WAAW,GAAG,CAAC,CAAC,CAAC,4CAA4C;IACjE,MAAM,mBAAmB,GAAG,CAAC,CAAC,CAAC,8CAA8C;IAE7E,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC7B,SAAS,EAAE,CAAC;QAEZ,oEAAoE;QACpE,IAAI,CAAC;YACH,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC;YACxD,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,SAAS,GAAG,MAAM,eAAe,CAAC,SAAS,EAAE,CAAC;gBACpD,IAAI,SAAS,EAAE,CAAC;oBACd,IAAI,KAAK,IAAI,SAAS,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;wBACjC,GAAG,CAAC,KAAK,CAAC,yCAAyC,iBAAiB,cAAc,CAAC,CAAC;oBACtF,CAAC;oBACD,MAAM,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;oBAC1C,SAAS;gBACX,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,wCAAwC;QAC1C,CAAC;QAED,0BAA0B;QAC1B,MAAM,SAAS,GAAG,MAAM,iBAAiB,CACvC,IAAI,EACJ,WAAW,EACX,KAAK,EACL,SAAS,CACV,CAAC;QAEF,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,UAAU,GAAG,SAAS,CAAC,IAAI,EAAE,CAAC;YACpC,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,KAAK,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC;gBAEvC,kCAAkC;gBAClC,IAAI,KAAK,KAAK,iBAAiB,EAAE,CAAC;oBAChC,IAAI,KAAK,EAAE,CAAC;wBACV,GAAG,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;oBACxD,CAAC;oBACD,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,eAAe;oBACxD,MAAM,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;oBAC1C,SAAS;gBACX,CAAC;gBAED,2CAA2C;gBAC3C,+CAA+C;gBAC/C,2CAA2C;gBAC3C,IAAI,UAAU,KAAK,aAAa,EAAE,CAAC;oBACjC,oCAAoC;oBACpC,WAAW,EAAE,CAAC;oBACd,IAAI,KAAK,IAAI,WAAW,KAAK,mBAAmB,EAAE,CAAC;wBACjD,GAAG,CAAC,KAAK,CACP,6BAA6B,WAAW,WAAW,UAAU,CAAC,MAAM,SAAS,CAC9E,CAAC;oBACJ,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,uCAAuC;oBACvC,IAAI,KAAK,IAAI,aAAa,EAAE,CAAC;wBAC3B,GAAG,CAAC,KAAK,CACP,4BAA4B,UAAU,CAAC,MAAM,eAAe,aAAa,CAAC,MAAM,GAAG,CACpF,CAAC;oBACJ,CAAC;oBACD,WAAW,GAAG,CAAC,CAAC;oBAChB,aAAa,GAAG,UAAU,CAAC;gBAC7B,CAAC;gBAED,kCAAkC;gBAClC,IAAI,WAAW,IAAI,mBAAmB,EAAE,CAAC;oBACvC,IAAI,KAAK,EAAE,CAAC;wBACV,GAAG,CAAC,KAAK,CAAC,sCAAsC,UAAU,CAAC,MAAM,SAAS,CAAC,CAAC;oBAC9E,CAAC;oBACD,OAAO,UAAU,CAAC;gBACpB,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;IAC5C,CAAC;IAED,IAAI,KAAK,EAAE,CAAC;QACV,GAAG,CAAC,KAAK,CAAC,4BAA4B,SAAS,QAAQ,CAAC,CAAC;IAC3D,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;;;;;;GASG;AACH,KAAK,UAAU,iBAAiB,CAC9B,IAAU,EACV,WAAwB,EACxB,KAAc,EACd,SAAiB;IAEjB,gEAAgE;IAChE,MAAM,eAAe,GAAG,oBAAoB,CAAC;IAC7C,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC;QAClD,MAAM,eAAe,GAAG,UAAU,CAAC,MAAM,CAAC;QAE1C,2CAA2C;QAC3C,IAAI,eAAe,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC;YACxC,IAAI,KAAK,IAAI,SAAS,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,GAAG,CAAC,GAAG,CACL,mCAAmC,eAAe,WAAW,WAAW,CAAC,IAAI,SAAS,CACvF,CAAC;YACJ,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,0CAA0C;YAC1C,IAAI,KAAK,IAAI,SAAS,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,GAAG,CAAC,GAAG,CACL,yBAAyB,eAAe,gBAAgB,WAAW,CAAC,IAAI,SAAS,CAClF,CAAC;YACJ,CAAC;YAED,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,IAAI,KAAK,GAAG,CAAC,CAAC;YAEd,sDAAsD;YACtD,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,UAAU,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;gBACjD,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;gBAClC,IAAI,CAAC;oBACH,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC;oBAC/D,IAAI,WAAW,EAAE,CAAC;wBAChB,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,SAAS,EAAE,CAAC;wBAC3C,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;4BACxB,+CAA+C;4BAC/C,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;4BACzC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gCAC/B,GAAG,CAAC,OAAO,CACT,2CAA2C,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,QAAQ,CAC/E,CAAC;gCACF,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;4BACrB,CAAC;iCAAM,CAAC;gCACN,OAAO,EAAE,CAAC;4BACZ,CAAC;wBACH,CAAC;6BAAM,CAAC;4BACN,KAAK,EAAE,CAAC;wBACV,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAS;gBACX,CAAC;YACH,CAAC;YAED,oCAAoC;YACpC,IAAI,KAAK,IAAI,SAAS,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,GAAG,CAAC,GAAG,CACL,qCAAqC,OAAO,WAAW,KAAK,SAAS,CACtE,CAAC;YACJ,CAAC;YACD,OAAO,IAAI,CAAC,CAAC,kCAAkC;QACjD,CAAC;aAAM,CAAC;YACN,IAAI,KAAK,EAAE,CAAC;gBACV,GAAG,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,KAAK,CAAC,wCAAwC,KAAK,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,gFAAgF;IAChF,IAAI,KAAK,EAAE,CAAC;QACV,GAAG,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,MAAM,QAAQ,IAAI,kBAAkB,EAAE,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;YACzC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAEpC,oDAAoD;YACpD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,4CAA4C;oBAC5C,IAAI,SAAS,GAAG,OAAO,CAAC;oBACxB,IAAI,CAAC;wBACH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;4BAClD,OAAO,EAAE,CAAC,OAAO,CACf,6DAA6D;gCAC7D,gFAAgF,CACjF,CAAC;wBACJ,CAAC,CAAC,CAAC;wBACH,IAAI,OAAO,EAAE,CAAC;4BACZ,MAAM,cAAc,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;4BAC3C,IAAI,cAAc,EAAE,CAAC;gCACnB,SAAS,GAAG,cAAgC,CAAC;4BAC/C,CAAC;wBACH,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,SAAS,GAAG,OAAO,CAAC;oBACtB,CAAC;oBAED,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,SAAS,EAAE,CAAC;oBACzC,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;wBACrE,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;oBACrB,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAS;gBACX,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;IACH,CAAC;IAED,wCAAwC;IACxC,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAkB,EAAE;YAC3D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAW,CAAC;YAClC,MAAM,SAAS,GAAG,CAAC,EAAW,EAAW,EAAE;gBACzC,IAAI,CAAC,EAAE,IAAI,CAAE,EAAkB,CAAC,WAAW;oBAAE,OAAO,KAAK,CAAC;gBAC1D,MAAM,IAAI,GAAG,EAAE,CAAC,qBAAqB,EAAE,CAAC;gBACxC,IAAI,IAAI,CAAC,KAAK,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO,KAAK,CAAC;gBACxD,MAAM,KAAK,GAAG,MAAM,CAAC,gBAAgB,CAAC,EAAiB,CAAC,CAAC;gBACzD,IACE,KAAK,CAAC,UAAU,KAAK,QAAQ;oBAC7B,KAAK,CAAC,OAAO,KAAK,MAAM;oBACxB,UAAU,CAAC,KAAK,CAAC,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,EACtC,CAAC;oBACD,OAAO,KAAK,CAAC;gBACf,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC,CAAC;YAEF,MAAM,SAAS,GAAG;gBAChB,uBAAuB;gBACvB,qBAAqB;gBACrB,eAAe;gBACf,8BAA8B;gBAC9B,4BAA4B;gBAC5B,kCAAkC;aACnC,CAAC;YAEF,MAAM,UAAU,GAAa,EAAE,CAAC;YAChC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBACjC,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;oBACjE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;wBAAE,SAAS;oBAC7B,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;wBAAE,SAAS;oBAC7B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBAEf,MAAM,IAAI,GAAI,EAAkB,CAAC,SAAS,IAAK,EAAkB,CAAC,WAAW,IAAI,EAAE,CAAC;oBACpF,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;wBAAE,SAAS;oBAE3B,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;YAED,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,OAAO,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC3C,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,CAAC,IAAI,EAAE,EAAE,CAAC;YAC5D,OAAO,YAAY,CAAC,IAAI,EAAE,CAAC;QAC7B,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,2BAA2B;IAC7B,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,eAAe;IACb,sBAAsB;IACtB,oBAAoB;IACpB,qBAAqB;IACrB,mBAAmB;CACpB,CAAC"}

package/dist/utils/rate-limit-handler.d.ts

@@ -0,0 +1,42 @@
+/**
+ * Rate Limit Handler
+ *
+ * Provides automatic retry logic with exponential backoff for rate-limited operations.
+ * NotebookLM free accounts have a 50 queries/day limit.
+ *
+ * Features:
+ * - Configurable max retries (default: 3)
+ * - Exponential backoff with jitter
+ * - RateLimitError-specific handling
+ */
+export interface RetryOptions {
+    maxRetries?: number;
+    baseDelayMs?: number;
+    maxDelayMs?: number;
+    onRetry?: (attempt: number, delay: number, error: Error) => void;
+}
+/**
+ * Rate Limit Handler with exponential backoff
+ */
+export declare class RateLimitHandler {
+    private options;
+    constructor(options?: RetryOptions);
+    /**
+     * Execute a function with automatic retry on RateLimitError
+     *
+     * @param fn - Async function to execute
+     * @returns Result of the function
+     * @throws RateLimitError if all retries exhausted
+     */
+    executeWithRetry<T>(fn: () => Promise<T>): Promise<T>;
+    /**
+     * Calculate delay using exponential backoff with jitter
+     */
+    private calculateDelay;
+}
+/**
+ * Convenience function to execute with retry using default options
+ */
+export declare function withRateLimitRetry<T>(fn: () => Promise<T>, options?: RetryOptions): Promise<T>;
+export default RateLimitHandler;
+//# sourceMappingURL=rate-limit-handler.d.ts.map

package/dist/utils/rate-limit-handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limit-handler.d.ts","sourceRoot":"","sources":["../../src/utils/rate-limit-handler.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAMH,MAAM,WAAW,YAAY;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CACpE;AAQD;;GAEG;AACH,qBAAa,gBAAgB;IACzB,OAAO,CAAC,OAAO,CAA0E;gBAE7E,OAAO,GAAE,YAAiB;IAOtC;;;;;;OAMG;IACG,gBAAgB,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAwC3D;;OAEG;IACH,OAAO,CAAC,cAAc;CAUzB;AAED;;GAEG;AACH,wBAAsB,kBAAkB,CAAC,CAAC,EACtC,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EACpB,OAAO,CAAC,EAAE,YAAY,GACvB,OAAO,CAAC,CAAC,CAAC,CAGZ;AAED,eAAe,gBAAgB,CAAC"}

package/dist/utils/rate-limit-handler.js

@@ -0,0 +1,88 @@
+/**
+ * Rate Limit Handler
+ *
+ * Provides automatic retry logic with exponential backoff for rate-limited operations.
+ * NotebookLM free accounts have a 50 queries/day limit.
+ *
+ * Features:
+ * - Configurable max retries (default: 3)
+ * - Exponential backoff with jitter
+ * - RateLimitError-specific handling
+ */
+import { RateLimitError } from "../errors.js";
+import { log } from "./logger.js";
+import { sleep } from "./stealth-utils.js";
+const DEFAULT_OPTIONS = {
+    maxRetries: 3,
+    baseDelayMs: 1000,
+    maxDelayMs: 30000,
+};
+/**
+ * Rate Limit Handler with exponential backoff
+ */
+export class RateLimitHandler {
+    options;
+    constructor(options = {}) {
+        this.options = {
+            ...DEFAULT_OPTIONS,
+            ...options,
+        };
+    }
+    /**
+     * Execute a function with automatic retry on RateLimitError
+     *
+     * @param fn - Async function to execute
+     * @returns Result of the function
+     * @throws RateLimitError if all retries exhausted
+     */
+    async executeWithRetry(fn) {
+        let lastError = null;
+        for (let attempt = 0; attempt <= this.options.maxRetries; attempt++) {
+            try {
+                return await fn();
+            }
+            catch (error) {
+                if (!(error instanceof RateLimitError)) {
+                    // Non-rate-limit errors are thrown immediately
+                    throw error;
+                }
+                lastError = error;
+                if (attempt === this.options.maxRetries) {
+                    // All retries exhausted
+                    log.error(`❌ Rate limit: All ${this.options.maxRetries} retries exhausted`);
+                    break;
+                }
+                // Calculate delay with exponential backoff + jitter
+                const delay = this.calculateDelay(attempt);
+                log.warning(`⚠️  Rate limit hit (attempt ${attempt + 1}/${this.options.maxRetries + 1}). ` +
+                    `Retrying in ${Math.round(delay / 1000)}s...`);
+                // Call optional retry callback
+                if (this.options.onRetry) {
+                    this.options.onRetry(attempt + 1, delay, error);
+                }
+                await sleep(delay);
+            }
+        }
+        throw lastError ?? new RateLimitError("Rate limit exceeded after all retries");
+    }
+    /**
+     * Calculate delay using exponential backoff with jitter
+     */
+    calculateDelay(attempt) {
+        // Exponential: baseDelay * 2^attempt
+        const exponentialDelay = this.options.baseDelayMs * Math.pow(2, attempt);
+        // Add random jitter (±25% of delay)
+        const jitter = exponentialDelay * 0.25 * (Math.random() * 2 - 1);
+        // Clamp to max delay
+        return Math.min(exponentialDelay + jitter, this.options.maxDelayMs);
+    }
+}
+/**
+ * Convenience function to execute with retry using default options
+ */
+export async function withRateLimitRetry(fn, options) {
+    const handler = new RateLimitHandler(options);
+    return handler.executeWithRetry(fn);
+}
+export default RateLimitHandler;
+//# sourceMappingURL=rate-limit-handler.js.map

package/dist/utils/rate-limit-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limit-handler.js","sourceRoot":"","sources":["../../src/utils/rate-limit-handler.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAS3C,MAAM,eAAe,GAA4C;IAC7D,UAAU,EAAE,CAAC;IACb,WAAW,EAAE,IAAI;IACjB,UAAU,EAAE,KAAK;CACpB,CAAC;AAEF;;GAEG;AACH,MAAM,OAAO,gBAAgB;IACjB,OAAO,CAA0E;IAEzF,YAAY,UAAwB,EAAE;QAClC,IAAI,CAAC,OAAO,GAAG;YACX,GAAG,eAAe;YAClB,GAAG,OAAO;SACb,CAAC;IACN,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,gBAAgB,CAAI,EAAoB;QAC1C,IAAI,SAAS,GAAiB,IAAI,CAAC;QAEnC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;YAClE,IAAI,CAAC;gBACD,OAAO,MAAM,EAAE,EAAE,CAAC;YACtB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,CAAC,KAAK,YAAY,cAAc,CAAC,EAAE,CAAC;oBACrC,+CAA+C;oBAC/C,MAAM,KAAK,CAAC;gBAChB,CAAC;gBAED,SAAS,GAAG,KAAK,CAAC;gBAElB,IAAI,OAAO,KAAK,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;oBACtC,wBAAwB;oBACxB,GAAG,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,OAAO,CAAC,UAAU,oBAAoB,CAAC,CAAC;oBAC5E,MAAM;gBACV,CAAC;gBAED,oDAAoD;gBACpD,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;gBAE3C,GAAG,CAAC,OAAO,CACP,+BAA+B,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,GAAG,CAAC,KAAK;oBAC9E,eAAe,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAChD,CAAC;gBAEF,+BAA+B;gBAC/B,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;oBACvB,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,GAAG,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;gBACpD,CAAC;gBAED,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;QACL,CAAC;QAED,MAAM,SAAS,IAAI,IAAI,cAAc,CAAC,uCAAuC,CAAC,CAAC;IACnF,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,OAAe;QAClC,qCAAqC;QACrC,MAAM,gBAAgB,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAEzE,oCAAoC;QACpC,MAAM,MAAM,GAAG,gBAAgB,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QAEjE,qBAAqB;QACrB,OAAO,IAAI,CAAC,GAAG,CAAC,gBAAgB,GAAG,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACxE,CAAC;CACJ;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACpC,EAAoB,EACpB,OAAsB;IAEtB,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC9C,OAAO,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;AACxC,CAAC;AAED,eAAe,gBAAgB,CAAC"}

package/dist/utils/rate-limit-handler.test.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Tests for rate-limit-handler.ts
+ *
+ * Unit tests for rate limit retry logic with exponential backoff.
+ */
+export {};
+//# sourceMappingURL=rate-limit-handler.test.d.ts.map

package/dist/utils/rate-limit-handler.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limit-handler.test.d.ts","sourceRoot":"","sources":["../../src/utils/rate-limit-handler.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG"}

package/dist/utils/rate-limit-handler.test.js

@@ -0,0 +1,91 @@
+/**
+ * Tests for rate-limit-handler.ts
+ *
+ * Unit tests for rate limit retry logic with exponential backoff.
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { RateLimitHandler, withRateLimitRetry } from './rate-limit-handler.js';
+import { RateLimitError } from '../errors.js';
+describe('RateLimitHandler', () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+    });
+    afterEach(() => {
+        vi.useRealTimers();
+    });
+    describe('executeWithRetry', () => {
+        it('should execute function successfully on first attempt', async () => {
+            const handler = new RateLimitHandler();
+            const fn = vi.fn().mockResolvedValue('success');
+            const resultPromise = handler.executeWithRetry(fn);
+            await vi.runAllTimersAsync();
+            const result = await resultPromise;
+            expect(result).toBe('success');
+            expect(fn).toHaveBeenCalledTimes(1);
+        });
+        it('should retry on RateLimitError and succeed', async () => {
+            const handler = new RateLimitHandler({ maxRetries: 3, baseDelayMs: 100 });
+            const fn = vi.fn()
+                .mockRejectedValueOnce(new RateLimitError())
+                .mockResolvedValue('success after retry');
+            const resultPromise = handler.executeWithRetry(fn);
+            await vi.runAllTimersAsync();
+            const result = await resultPromise;
+            expect(result).toBe('success after retry');
+            expect(fn).toHaveBeenCalledTimes(2);
+        });
+        it('should throw non-RateLimitError immediately', async () => {
+            const handler = new RateLimitHandler();
+            const error = new Error('Other error');
+            const fn = vi.fn().mockRejectedValue(error);
+            await expect(handler.executeWithRetry(fn)).rejects.toThrow('Other error');
+            expect(fn).toHaveBeenCalledTimes(1);
+        });
+        it('should throw RateLimitError after all retries exhausted', async () => {
+            const handler = new RateLimitHandler({ maxRetries: 2, baseDelayMs: 100 });
+            const fn = vi.fn().mockRejectedValue(new RateLimitError());
+            const resultPromise = handler.executeWithRetry(fn);
+            // Run all timers and expect rejection
+            await vi.runAllTimersAsync();
+            // Awaiting the promise rejection properly
+            let caughtError = null;
+            try {
+                await resultPromise;
+            }
+            catch (e) {
+                caughtError = e;
+            }
+            expect(caughtError).toBeInstanceOf(RateLimitError);
+            // Initial attempt + 2 retries = 3 calls
+            expect(fn).toHaveBeenCalledTimes(3);
+        });
+        it('should call onRetry callback on each retry', async () => {
+            const onRetry = vi.fn();
+            const handler = new RateLimitHandler({
+                maxRetries: 2,
+                baseDelayMs: 100,
+                onRetry,
+            });
+            const fn = vi.fn()
+                .mockRejectedValueOnce(new RateLimitError())
+                .mockRejectedValueOnce(new RateLimitError())
+                .mockResolvedValue('success');
+            const resultPromise = handler.executeWithRetry(fn);
+            await vi.runAllTimersAsync();
+            await resultPromise;
+            expect(onRetry).toHaveBeenCalledTimes(2);
+            expect(onRetry).toHaveBeenCalledWith(1, expect.any(Number), expect.any(RateLimitError));
+            expect(onRetry).toHaveBeenCalledWith(2, expect.any(Number), expect.any(RateLimitError));
+        });
+    });
+    describe('withRateLimitRetry convenience function', () => {
+        it('should work with default options', async () => {
+            const fn = vi.fn().mockResolvedValue('success');
+            const resultPromise = withRateLimitRetry(fn);
+            await vi.runAllTimersAsync();
+            const result = await resultPromise;
+            expect(result).toBe('success');
+        });
+    });
+});
+//# sourceMappingURL=rate-limit-handler.test.js.map

package/dist/utils/rate-limit-handler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limit-handler.test.js","sourceRoot":"","sources":["../../src/utils/rate-limit-handler.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC/E,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAC9B,UAAU,CAAC,GAAG,EAAE;QACZ,EAAE,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACX,EAAE,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,OAAO,GAAG,IAAI,gBAAgB,EAAE,CAAC;YACvC,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEhD,MAAM,aAAa,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;YAEnC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/B,MAAM,CAAC,EAAE,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;YAC1E,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE;iBACb,qBAAqB,CAAC,IAAI,cAAc,EAAE,CAAC;iBAC3C,iBAAiB,CAAC,qBAAqB,CAAC,CAAC;YAE9C,MAAM,aAAa,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;YAEnC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YAC3C,MAAM,CAAC,EAAE,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,OAAO,GAAG,IAAI,gBAAgB,EAAE,CAAC;YACvC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;YACvC,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAC1E,MAAM,CAAC,EAAE,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;YAC1E,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,cAAc,EAAE,CAAC,CAAC;YAE3D,MAAM,aAAa,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;YAEnD,sCAAsC;YACtC,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAE7B,0CAA0C;YAC1C,IAAI,WAAW,GAAiB,IAAI,CAAC;YACrC,IAAI,CAAC;gBACD,MAAM,aAAa,CAAC;YACxB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,WAAW,GAAG,CAAU,CAAC;YAC7B,CAAC;YAED,MAAM,CAAC,WAAW,CAAC,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;YACnD,wCAAwC;YACxC,MAAM,CAAC,EAAE,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACxB,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC;gBACjC,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,GAAG;gBAChB,OAAO;aACV,CAAC,CAAC;YACH,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE;iBACb,qBAAqB,CAAC,IAAI,cAAc,EAAE,CAAC;iBAC3C,qBAAqB,CAAC,IAAI,cAAc,EAAE,CAAC;iBAC3C,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAElC,MAAM,aAAa,GAAG,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAC7B,MAAM,aAAa,CAAC;YAEpB,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC;YACxF,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACrD,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEhD,MAAM,aAAa,GAAG,kBAAkB,CAAC,EAAE,CAAC,CAAC;YAC7C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;YAEnC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}