@roomi-fields/notebooklm-mcp 1.3.5 → 1.5.0

package/dist/utils/citation-extractor.js

@@ -0,0 +1,492 @@
+/**
+ * Citation Extractor for NotebookLM
+ *
+ * Extracts source citations by hovering over citation markers [1], [2], etc.
+ * and capturing the tooltip content.
+ *
+ * Features:
+ * - Hover-based extraction (no additional API calls)
+ * - Multiple output formats (inline, footnotes, json, expanded)
+ * - Graceful fallback if citations can't be extracted
+ */
+import { log } from './logger.js';
+import { randomDelay } from './stealth-utils.js';
+// ============================================================================
+// CSS Selectors for NotebookLM Citations
+// ============================================================================
+/**
+ * Selectors to find citation markers in the response
+ * NotebookLM typically uses superscript numbers or bracketed references
+ */
+const CITATION_SELECTORS = [
+    // Primary: Citation links/buttons
+    '.citation-link',
+    '.citation-marker',
+    '[data-citation]',
+    '[data-citation-id]',
+    '[data-source-id]',
+    // Superscript numbers
+    'sup.citation',
+    'sup[data-citation]',
+    'sup a',
+    // Bracketed references [1], [2]
+    '.reference-marker',
+    '[role="button"][aria-label*="citation"]',
+    '[role="button"][aria-label*="source"]',
+    // NotebookLM specific (may need adjustment based on actual DOM)
+    '.source-citation',
+    '.inline-citation',
+    'button.citation',
+    // Generic clickable citations
+    '[class*="citation"]',
+    '[class*="source-ref"]',
+];
+/**
+ * Selectors for the tooltip/popover that appears on hover
+ */
+const TOOLTIP_SELECTORS = [
+    // Common tooltip patterns
+    '[role="tooltip"]',
+    '.tooltip',
+    '.popover',
+    '.citation-tooltip',
+    '.citation-popover',
+    '.source-preview',
+    '.source-tooltip',
+    // Material Design / Google patterns
+    '.mdc-tooltip',
+    '.mat-tooltip',
+    '[class*="tooltip"]',
+    '[class*="popover"]',
+    // NotebookLM specific
+    '.citation-preview',
+    '.source-card',
+    '.source-snippet',
+    '[data-tooltip]',
+    '[aria-describedby]',
+];
+// ============================================================================
+// Main Extraction Function
+// ============================================================================
+/**
+ * Extract citations from a NotebookLM response by hovering over citation markers
+ *
+ * @param page Playwright page instance
+ * @param answerText The answer text to process
+ * @param responseContainer The container element holding the response
+ * @param format Desired output format
+ * @returns Extraction result with formatted answer and citations
+ */
+export async function extractCitations(page, answerText, responseContainer, format = 'none') {
+    // Early return if no extraction requested
+    if (format === 'none') {
+        return {
+            originalAnswer: answerText,
+            formattedAnswer: answerText,
+            citations: [],
+            format,
+            success: true,
+        };
+    }
+    log.info(`📚 [CITATIONS] Extracting sources (format: ${format})...`);
+    const citations = [];
+    try {
+        // Find citation markers in the response
+        const citationElements = await findCitationElements(page, responseContainer);
+        if (citationElements.length === 0) {
+            log.info(`📚 [CITATIONS] No citation markers found in response`);
+            return {
+                originalAnswer: answerText,
+                formattedAnswer: answerText,
+                citations: [],
+                format,
+                success: true,
+            };
+        }
+        log.info(`📚 [CITATIONS] Found ${citationElements.length} citation markers`);
+        // Extract source text for each citation via hover
+        for (const { element, marker, number } of citationElements) {
+            try {
+                const sourceText = await extractSourceViaHover(page, element);
+                if (sourceText) {
+                    citations.push({
+                        marker,
+                        number,
+                        sourceText,
+                    });
+                    log.success(`  ✅ [${marker}] Extracted: "${sourceText.substring(0, 50)}..."`);
+                }
+                else {
+                    log.warning(`  ⚠️  [${marker}] Could not extract source text`);
+                }
+            }
+            catch (error) {
+                log.warning(`  ⚠️  [${marker}] Error extracting: ${error}`);
+            }
+            // Small delay between hovers to avoid detection
+            await randomDelay(100, 200);
+        }
+        // Format the answer based on requested format
+        const formattedAnswer = formatAnswerWithSources(answerText, citations, format);
+        log.success(`📚 [CITATIONS] Extracted ${citations.length}/${citationElements.length} sources`);
+        return {
+            originalAnswer: answerText,
+            formattedAnswer,
+            citations,
+            format,
+            success: true,
+        };
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        log.error(`❌ [CITATIONS] Extraction failed: ${errorMessage}`);
+        return {
+            originalAnswer: answerText,
+            formattedAnswer: answerText,
+            citations: [],
+            format,
+            success: false,
+            error: errorMessage,
+        };
+    }
+}
+/**
+ * Find citation marker elements in the response
+ */
+async function findCitationElements(page, container) {
+    const results = [];
+    const seenNumbers = new Set();
+    // Search context: container if available, otherwise full page
+    const searchContext = container || page;
+    // Try each selector
+    for (const selector of CITATION_SELECTORS) {
+        try {
+            const elements = await searchContext.$$(selector);
+            for (const element of elements) {
+                try {
+                    // Get the text content to identify the citation number
+                    const text = await element.innerText();
+                    const match = text.match(/\[?(\d+)\]?/);
+                    if (match) {
+                        const number = parseInt(match[1], 10);
+                        // Skip if we've already found this citation
+                        if (seenNumbers.has(number))
+                            continue;
+                        seenNumbers.add(number);
+                        const marker = `[${number}]`;
+                        results.push({ element, marker, number });
+                    }
+                }
+                catch {
+                    continue;
+                }
+            }
+        }
+        catch {
+            continue;
+        }
+    }
+    // Also try to find citations by regex pattern in the page
+    if (results.length === 0) {
+        const regexResults = await findCitationsByRegex(page, container);
+        results.push(...regexResults);
+    }
+    // Sort by citation number
+    results.sort((a, b) => a.number - b.number);
+    return results;
+}
+/**
+ * Find citations using regex matching on text content
+ * Fallback method when CSS selectors don't work
+ *
+ * This improved version:
+ * 1. Detects citation markers [n] in the text
+ * 2. Finds clickable elements containing those markers
+ * 3. Returns element handles for hover extraction
+ */
+async function findCitationsByRegex(page, container) {
+    const results = [];
+    const seenNumbers = new Set();
+    try {
+        // First, detect unique citation numbers in the response text
+        const citationNumbers = await page.evaluate((containerSelector) => {
+            const searchRoot = containerSelector
+                ? // @ts-expect-error - document available in browser context
+                    document.querySelector(containerSelector)
+                : // @ts-expect-error - document available in browser context
+                    document.body;
+            if (!searchRoot)
+                return [];
+            const text = searchRoot.textContent || '';
+            const matches = text.matchAll(/\[(\d+)\]/g);
+            const numbers = new Set();
+            for (const match of matches) {
+                numbers.add(parseInt(match[1], 10));
+            }
+            return Array.from(numbers).sort((a, b) => a - b);
+        }, container ? '.to-user-container .message-text-content' : null);
+        log.info(`📚 [CITATIONS] Regex found ${citationNumbers.length} unique citation markers: [${citationNumbers.join(', ')}]`);
+        // For each citation number, try to find a clickable element
+        // NotebookLM typically wraps citations in <a>, <button>, or <span> with click handlers
+        const searchContext = container || page;
+        for (const num of citationNumbers) {
+            if (seenNumbers.has(num))
+                continue;
+            // Try multiple strategies to find the citation element
+            const strategies = [
+                // Strategy 1: Look for links with the citation number
+                `a:has-text("[${num}]")`,
+                `button:has-text("[${num}]")`,
+                `span:has-text("[${num}]")`,
+                // Strategy 2: Look for elements with citation-related attributes
+                `[data-citation="${num}"]`,
+                `[data-source="${num}"]`,
+                `[aria-label*="${num}"]`,
+                // Strategy 3: Look for superscript elements
+                `sup:has-text("${num}")`,
+                // Strategy 4: Generic clickable elements with the number
+                `[role="button"]:has-text("[${num}]")`,
+                `[role="link"]:has-text("[${num}]")`,
+            ];
+            for (const selector of strategies) {
+                try {
+                    const element = await searchContext.$(selector);
+                    if (element) {
+                        const isVisible = await element.isVisible();
+                        if (isVisible) {
+                            seenNumbers.add(num);
+                            results.push({
+                                element,
+                                marker: `[${num}]`,
+                                number: num,
+                            });
+                            log.info(`  📌 Found element for [${num}] using selector: ${selector}`);
+                            break;
+                        }
+                    }
+                }
+                catch {
+                    continue;
+                }
+            }
+            // If no element found with specific selectors, try XPath as last resort
+            if (!seenNumbers.has(num)) {
+                try {
+                    // Find any element that contains exactly [n]
+                    const xpath = `//*[contains(text(), '[${num}]')]`;
+                    const elements = await page.$$(`::-p-xpath(${xpath})`);
+                    for (const element of elements) {
+                        try {
+                            const text = await element.innerText();
+                            // Make sure it's the citation marker, not just any text with [n]
+                            if (text.includes(`[${num}]`) && text.length < 50) {
+                                const isVisible = await element.isVisible();
+                                if (isVisible) {
+                                    seenNumbers.add(num);
+                                    results.push({
+                                        element,
+                                        marker: `[${num}]`,
+                                        number: num,
+                                    });
+                                    log.info(`  📌 Found element for [${num}] using XPath fallback`);
+                                    break;
+                                }
+                            }
+                        }
+                        catch {
+                            continue;
+                        }
+                    }
+                }
+                catch {
+                    // XPath failed, continue without this citation
+                }
+            }
+            if (!seenNumbers.has(num)) {
+                log.warning(`  ⚠️  Could not find DOM element for [${num}]`);
+            }
+        }
+        log.info(`📚 [CITATIONS] Found ${results.length}/${citationNumbers.length} citation elements`);
+    }
+    catch (error) {
+        log.warning(`⚠️  [CITATIONS] Regex search failed: ${error}`);
+    }
+    // Sort by citation number
+    results.sort((a, b) => a.number - b.number);
+    return results;
+}
+/**
+ * Extract source text by hovering over a citation element
+ */
+async function extractSourceViaHover(page, element) {
+    try {
+        // Scroll element into view
+        await element.scrollIntoViewIfNeeded();
+        await randomDelay(50, 100);
+        // Hover over the element
+        await element.hover();
+        // Wait for tooltip to appear
+        await randomDelay(300, 500);
+        // Try to find and read the tooltip content
+        for (const tooltipSelector of TOOLTIP_SELECTORS) {
+            try {
+                const tooltip = await page.$(tooltipSelector);
+                if (tooltip) {
+                    const isVisible = await tooltip.isVisible();
+                    if (isVisible) {
+                        const text = await tooltip.innerText();
+                        if (text && text.trim()) {
+                            // Move mouse away to dismiss tooltip
+                            await page.mouse.move(0, 0);
+                            await randomDelay(100, 150);
+                            return text.trim();
+                        }
+                    }
+                }
+            }
+            catch {
+                continue;
+            }
+        }
+        // Fallback: Check for aria-describedby attribute
+        try {
+            const describedBy = await element.getAttribute('aria-describedby');
+            if (describedBy) {
+                const tooltipById = await page.$(`#${describedBy}`);
+                if (tooltipById) {
+                    const text = await tooltipById.innerText();
+                    if (text && text.trim()) {
+                        await page.mouse.move(0, 0);
+                        return text.trim();
+                    }
+                }
+            }
+        }
+        catch {
+            // Ignore
+        }
+        // Move mouse away even if we didn't find content
+        await page.mouse.move(0, 0);
+        return null;
+    }
+    catch (error) {
+        log.warning(`⚠️  [CITATIONS] Hover extraction failed: ${error}`);
+        return null;
+    }
+}
+// ============================================================================
+// Formatting Functions
+// ============================================================================
+/**
+ * Format the answer with extracted sources based on requested format
+ */
+export function formatAnswerWithSources(answer, citations, format) {
+    if (citations.length === 0 || format === 'none') {
+        return answer;
+    }
+    switch (format) {
+        case 'inline':
+            return formatInline(answer, citations);
+        case 'footnotes':
+            return formatFootnotes(answer, citations);
+        case 'expanded':
+            return formatExpanded(answer, citations);
+        case 'json':
+            // For JSON format, we return the original answer
+            // The citations are available in the result object
+            return answer;
+        default:
+            return answer;
+    }
+}
+/**
+ * Format with inline source excerpts: "text [1: source excerpt]"
+ *
+ * NotebookLM returns citations in different formats:
+ * - Superscript numbers without brackets: "text1,2" or "text3"
+ * - Sometimes with brackets: "text[1]"
+ * - Sometimes stuck together: "text123" (meaning citations 1, 2, 3)
+ *
+ * This function handles all formats.
+ */
+function formatInline(answer, citations) {
+    let result = answer;
+    // Sort citations by number in DESCENDING order to avoid replacing [1] before [10]
+    const sortedCitations = [...citations].sort((a, b) => b.number - a.number);
+    for (const citation of sortedCitations) {
+        const shortSource = truncateSource(citation.sourceText, 100);
+        const num = citation.number;
+        const inlineReplacement = `[${num}: "${shortSource}"]`;
+        // Pattern 1: Bracketed format [n]
+        const bracketedPattern = `\\[${num}\\]`;
+        if (new RegExp(bracketedPattern).test(result)) {
+            result = result.replace(new RegExp(bracketedPattern, 'g'), inlineReplacement);
+            continue;
+        }
+        // Pattern 2: Superscript format - number followed by comma/period/space/newline/end
+        // Also handles citations stuck together by using lookahead for next digit or punctuation
+        // Match: "word1," or "word1." or "word12" (where we want to match the 1)
+        const superscriptPattern = `(\\D)(${num})(?=[,\\.;:\\s\\n]|(?=\\d)|$)`;
+        if (new RegExp(superscriptPattern).test(result)) {
+            result = result.replace(new RegExp(superscriptPattern, 'g'), `$1${inlineReplacement}`);
+        }
+    }
+    return result;
+}
+/**
+ * Format with footnotes at the end
+ */
+function formatFootnotes(answer, citations) {
+    if (citations.length === 0)
+        return answer;
+    const footnotes = citations
+        .map((c) => {
+        const source = c.sourceName ? `${c.sourceName}: ` : '';
+        return `${c.marker} ${source}${c.sourceText}`;
+    })
+        .join('\n\n');
+    return `${answer}\n\n---\n**Sources:**\n${footnotes}`;
+}
+/**
+ * Format with expanded inline quotes replacing markers
+ *
+ * Handles bracketed [n], superscript n, and stuck-together formats.
+ */
+function formatExpanded(answer, citations) {
+    let result = answer;
+    // Sort citations by number in DESCENDING order to avoid replacing 1 before 10
+    const sortedCitations = [...citations].sort((a, b) => b.number - a.number);
+    for (const citation of sortedCitations) {
+        const shortSource = truncateSource(citation.sourceText, 150);
+        const replacement = `"${shortSource}"`;
+        const num = citation.number;
+        // Pattern 1: Bracketed format [n]
+        const bracketedPattern = `\\[${num}\\]`;
+        if (new RegExp(bracketedPattern).test(result)) {
+            result = result.replace(new RegExp(bracketedPattern, 'g'), replacement);
+            continue;
+        }
+        // Pattern 2: Superscript format - also handles stuck-together citations
+        const superscriptPattern = `(\\D)(${num})(?=[,\\.;:\\s\\n]|(?=\\d)|$)`;
+        if (new RegExp(superscriptPattern).test(result)) {
+            result = result.replace(new RegExp(superscriptPattern, 'g'), `$1${replacement}`);
+        }
+    }
+    return result;
+}
+/**
+ * Truncate source text to a reasonable length
+ */
+function truncateSource(text, maxLength) {
+    if (text.length <= maxLength)
+        return text;
+    return text.substring(0, maxLength - 3) + '...';
+}
+// ============================================================================
+// Exports
+// ============================================================================
+export default {
+    extractCitations,
+    formatAnswerWithSources,
+};
+//# sourceMappingURL=citation-extractor.js.map

package/dist/utils/citation-extractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"citation-extractor.js","sourceRoot":"","sources":["../../src/utils/citation-extractor.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAGH,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAgDjD,+EAA+E;AAC/E,yCAAyC;AACzC,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,kBAAkB,GAAG;IACzB,kCAAkC;IAClC,gBAAgB;IAChB,kBAAkB;IAClB,iBAAiB;IACjB,oBAAoB;IACpB,kBAAkB;IAClB,sBAAsB;IACtB,cAAc;IACd,oBAAoB;IACpB,OAAO;IACP,gCAAgC;IAChC,mBAAmB;IACnB,yCAAyC;IACzC,uCAAuC;IACvC,gEAAgE;IAChE,kBAAkB;IAClB,kBAAkB;IAClB,iBAAiB;IACjB,8BAA8B;IAC9B,qBAAqB;IACrB,uBAAuB;CACxB,CAAC;AAEF;;GAEG;AACH,MAAM,iBAAiB,GAAG;IACxB,0BAA0B;IAC1B,kBAAkB;IAClB,UAAU;IACV,UAAU;IACV,mBAAmB;IACnB,mBAAmB;IACnB,iBAAiB;IACjB,iBAAiB;IACjB,oCAAoC;IACpC,cAAc;IACd,cAAc;IACd,oBAAoB;IACpB,oBAAoB;IACpB,sBAAsB;IACtB,mBAAmB;IACnB,cAAc;IACd,iBAAiB;IACjB,gBAAgB;IAChB,oBAAoB;CACrB,CAAC;AAEF,+EAA+E;AAC/E,2BAA2B;AAC3B,+EAA+E;AAE/E;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,IAAU,EACV,UAAkB,EAClB,iBAAuC,EACvC,SAAuB,MAAM;IAE7B,0CAA0C;IAC1C,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;QACtB,OAAO;YACL,cAAc,EAAE,UAAU;YAC1B,eAAe,EAAE,UAAU;YAC3B,SAAS,EAAE,EAAE;YACb,MAAM;YACN,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,GAAG,CAAC,IAAI,CAAC,8CAA8C,MAAM,MAAM,CAAC,CAAC;IAErE,MAAM,SAAS,GAAe,EAAE,CAAC;IAEjC,IAAI,CAAC;QACH,wCAAwC;QACxC,MAAM,gBAAgB,GAAG,MAAM,oBAAoB,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;QAE7E,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,GAAG,CAAC,IAAI,CAAC,sDAAsD,CAAC,CAAC;YACjE,OAAO;gBACL,cAAc,EAAE,UAAU;gBAC1B,eAAe,EAAE,UAAU;gBAC3B,SAAS,EAAE,EAAE;gBACb,MAAM;gBACN,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,wBAAwB,gBAAgB,CAAC,MAAM,mBAAmB,CAAC,CAAC;QAE7E,kDAAkD;QAClD,KAAK,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,gBAAgB,EAAE,CAAC;YAC3D,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,MAAM,qBAAqB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAE9D,IAAI,UAAU,EAAE,CAAC;oBACf,SAAS,CAAC,IAAI,CAAC;wBACb,MAAM;wBACN,MAAM;wBACN,UAAU;qBACX,CAAC,CAAC;oBACH,GAAG,CAAC,OAAO,CAAC,QAAQ,MAAM,iBAAiB,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;gBAChF,CAAC;qBAAM,CAAC;oBACN,GAAG,CAAC,OAAO,CAAC,UAAU,MAAM,iCAAiC,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,OAAO,CAAC,UAAU,MAAM,uBAAuB,KAAK,EAAE,CAAC,CAAC;YAC9D,CAAC;YAED,gDAAgD;YAChD,MAAM,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC9B,CAAC;QAED,8CAA8C;QAC9C,MAAM,eAAe,GAAG,uBAAuB,CAAC,UAAU,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAE/E,GAAG,CAAC,OAAO,CAAC,4BAA4B,SAAS,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,UAAU,CAAC,CAAC;QAE/F,OAAO;YACL,cAAc,EAAE,UAAU;YAC1B,eAAe;YACf,SAAS;YACT,MAAM;YACN,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,GAAG,CAAC,KAAK,CAAC,oCAAoC,YAAY,EAAE,CAAC,CAAC;QAE9D,OAAO;YACL,cAAc,EAAE,UAAU;YAC1B,eAAe,EAAE,UAAU;YAC3B,SAAS,EAAE,EAAE;YACb,MAAM;YACN,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,YAAY;SACpB,CAAC;IACJ,CAAC;AACH,CAAC;AAYD;;GAEG;AACH,KAAK,UAAU,oBAAoB,CACjC,IAAU,EACV,SAA+B;IAE/B,MAAM,OAAO,GAAsB,EAAE,CAAC;IACtC,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IAEtC,8DAA8D;IAC9D,MAAM,aAAa,GAAG,SAAS,IAAI,IAAI,CAAC;IAExC,oBAAoB;IACpB,KAAK,MAAM,QAAQ,IAAI,kBAAkB,EAAE,CAAC;QAC1C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;YAElD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,uDAAuD;oBACvD,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC;oBACvC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;oBAExC,IAAI,KAAK,EAAE,CAAC;wBACV,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAEtC,4CAA4C;wBAC5C,IAAI,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC;4BAAE,SAAS;wBACtC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;wBAExB,MAAM,MAAM,GAAG,IAAI,MAAM,GAAG,CAAC;wBAC7B,OAAO,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;oBAC5C,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAS;gBACX,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;IACH,CAAC;IAED,0DAA0D;IAC1D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,YAAY,GAAG,MAAM,oBAAoB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACjE,OAAO,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;IAChC,CAAC;IAED,0BAA0B;IAC1B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;IAE5C,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;;;;GAQG;AACH,KAAK,UAAU,oBAAoB,CACjC,IAAU,EACV,SAA+B;IAE/B,MAAM,OAAO,GAAsB,EAAE,CAAC;IACtC,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IAEtC,IAAI,CAAC;QACH,6DAA6D;QAC7D,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,QAAQ,CACzC,CAAC,iBAAgC,EAAE,EAAE;YACnC,MAAM,UAAU,GAAG,iBAAiB;gBAClC,CAAC,CAAC,2DAA2D;oBAC3D,QAAQ,CAAC,aAAa,CAAC,iBAAiB,CAAC;gBAC3C,CAAC,CAAC,2DAA2D;oBAC3D,QAAQ,CAAC,IAAI,CAAC;YAElB,IAAI,CAAC,UAAU;gBAAE,OAAO,EAAE,CAAC;YAE3B,MAAM,IAAI,GAAG,UAAU,CAAC,WAAW,IAAI,EAAE,CAAC;YAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YAC5C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;YAElC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YACtC,CAAC;YAED,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACnD,CAAC,EACD,SAAS,CAAC,CAAC,CAAC,0CAA0C,CAAC,CAAC,CAAC,IAAI,CAC9D,CAAC;QAEF,GAAG,CAAC,IAAI,CACN,8BAA8B,eAAe,CAAC,MAAM,8BAA8B,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAChH,CAAC;QAEF,4DAA4D;QAC5D,uFAAuF;QACvF,MAAM,aAAa,GAAG,SAAS,IAAI,IAAI,CAAC;QAExC,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;YAClC,IAAI,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,SAAS;YAEnC,uDAAuD;YACvD,MAAM,UAAU,GAAG;gBACjB,sDAAsD;gBACtD,gBAAgB,GAAG,KAAK;gBACxB,qBAAqB,GAAG,KAAK;gBAC7B,mBAAmB,GAAG,KAAK;gBAC3B,iEAAiE;gBACjE,mBAAmB,GAAG,IAAI;gBAC1B,iBAAiB,GAAG,IAAI;gBACxB,iBAAiB,GAAG,IAAI;gBACxB,4CAA4C;gBAC5C,iBAAiB,GAAG,IAAI;gBACxB,yDAAyD;gBACzD,8BAA8B,GAAG,KAAK;gBACtC,4BAA4B,GAAG,KAAK;aACrC,CAAC;YAEF,KAAK,MAAM,QAAQ,IAAI,UAAU,EAAE,CAAC;gBAClC,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;oBAChD,IAAI,OAAO,EAAE,CAAC;wBACZ,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC;wBAC5C,IAAI,SAAS,EAAE,CAAC;4BACd,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;4BACrB,OAAO,CAAC,IAAI,CAAC;gCACX,OAAO;gCACP,MAAM,EAAE,IAAI,GAAG,GAAG;gCAClB,MAAM,EAAE,GAAG;6BACZ,CAAC,CAAC;4BACH,GAAG,CAAC,IAAI,CAAC,2BAA2B,GAAG,qBAAqB,QAAQ,EAAE,CAAC,CAAC;4BACxE,MAAM;wBACR,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAS;gBACX,CAAC;YACH,CAAC;YAED,wEAAwE;YACxE,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,6CAA6C;oBAC7C,MAAM,KAAK,GAAG,0BAA0B,GAAG,MAAM,CAAC;oBAClD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,cAAc,KAAK,GAAG,CAAC,CAAC;oBAEvD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;wBAC/B,IAAI,CAAC;4BACH,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC;4BACvC,iEAAiE;4BACjE,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;gCAClD,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC;gCAC5C,IAAI,SAAS,EAAE,CAAC;oCACd,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;oCACrB,OAAO,CAAC,IAAI,CAAC;wCACX,OAAO;wCACP,MAAM,EAAE,IAAI,GAAG,GAAG;wCAClB,MAAM,EAAE,GAAG;qCACZ,CAAC,CAAC;oCACH,GAAG,CAAC,IAAI,CAAC,2BAA2B,GAAG,wBAAwB,CAAC,CAAC;oCACjE,MAAM;gCACR,CAAC;4BACH,CAAC;wBACH,CAAC;wBAAC,MAAM,CAAC;4BACP,SAAS;wBACX,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,+CAA+C;gBACjD,CAAC;YACH,CAAC;YAED,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC1B,GAAG,CAAC,OAAO,CAAC,yCAAyC,GAAG,GAAG,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,GAAG,CAAC,IAAI,CAAC,wBAAwB,OAAO,CAAC,MAAM,IAAI,eAAe,CAAC,MAAM,oBAAoB,CAAC,CAAC;IACjG,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,OAAO,CAAC,wCAAwC,KAAK,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,0BAA0B;IAC1B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;IAE5C,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,qBAAqB,CAAC,IAAU,EAAE,OAAsB;IACrE,IAAI,CAAC;QACH,2BAA2B;QAC3B,MAAM,OAAO,CAAC,sBAAsB,EAAE,CAAC;QACvC,MAAM,WAAW,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QAE3B,yBAAyB;QACzB,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;QAEtB,6BAA6B;QAC7B,MAAM,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAE5B,2CAA2C;QAC3C,KAAK,MAAM,eAAe,IAAI,iBAAiB,EAAE,CAAC;YAChD,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;gBAC9C,IAAI,OAAO,EAAE,CAAC;oBACZ,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC;oBAC5C,IAAI,SAAS,EAAE,CAAC;wBACd,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC;wBACvC,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;4BACxB,qCAAqC;4BACrC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;4BAC5B,MAAM,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;4BAC5B,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;wBACrB,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;QAED,iDAAiD;QACjD,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC;YACnE,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,CAAC,CAAC,IAAI,WAAW,EAAE,CAAC,CAAC;gBACpD,IAAI,WAAW,EAAE,CAAC;oBAChB,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,SAAS,EAAE,CAAC;oBAC3C,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;wBACxB,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;wBAC5B,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;oBACrB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;QAED,iDAAiD;QACjD,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,OAAO,CAAC,4CAA4C,KAAK,EAAE,CAAC,CAAC;QACjE,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,uBAAuB,CACrC,MAAc,EACd,SAAqB,EACrB,MAAoB;IAEpB,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;QAChD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,QAAQ;YACX,OAAO,YAAY,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QACzC,KAAK,WAAW;YACd,OAAO,eAAe,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC5C,KAAK,UAAU;YACb,OAAO,cAAc,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC3C,KAAK,MAAM;YACT,iDAAiD;YACjD,mDAAmD;YACnD,OAAO,MAAM,CAAC;QAChB;YACE,OAAO,MAAM,CAAC;IAClB,CAAC;AACH,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,YAAY,CAAC,MAAc,EAAE,SAAqB;IACzD,IAAI,MAAM,GAAG,MAAM,CAAC;IAEpB,kFAAkF;IAClF,MAAM,eAAe,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;IAE3E,KAAK,MAAM,QAAQ,IAAI,eAAe,EAAE,CAAC;QACvC,MAAM,WAAW,GAAG,cAAc,CAAC,QAAQ,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QAC7D,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC;QAC5B,MAAM,iBAAiB,GAAG,IAAI,GAAG,MAAM,WAAW,IAAI,CAAC;QAEvD,kCAAkC;QAClC,MAAM,gBAAgB,GAAG,MAAM,GAAG,KAAK,CAAC;QACxC,IAAI,IAAI,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,gBAAgB,EAAE,GAAG,CAAC,EAAE,iBAAiB,CAAC,CAAC;YAC9E,SAAS;QACX,CAAC;QAED,oFAAoF;QACpF,yFAAyF;QACzF,yEAAyE;QACzE,MAAM,kBAAkB,GAAG,SAAS,GAAG,+BAA+B,CAAC;QACvE,IAAI,IAAI,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAChD,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,kBAAkB,EAAE,GAAG,CAAC,EAAE,KAAK,iBAAiB,EAAE,CAAC,CAAC;QACzF,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,MAAc,EAAE,SAAqB;IAC5D,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,MAAM,CAAC;IAE1C,MAAM,SAAS,GAAG,SAAS;SACxB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACT,MAAM,MAAM,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;QACvD,OAAO,GAAG,CAAC,CAAC,MAAM,IAAI,MAAM,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC;IAChD,CAAC,CAAC;SACD,IAAI,CAAC,MAAM,CAAC,CAAC;IAEhB,OAAO,GAAG,MAAM,0BAA0B,SAAS,EAAE,CAAC;AACxD,CAAC;AAED;;;;GAIG;AACH,SAAS,cAAc,CAAC,MAAc,EAAE,SAAqB;IAC3D,IAAI,MAAM,GAAG,MAAM,CAAC;IAEpB,8EAA8E;IAC9E,MAAM,eAAe,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;IAE3E,KAAK,MAAM,QAAQ,IAAI,eAAe,EAAE,CAAC;QACvC,MAAM,WAAW,GAAG,cAAc,CAAC,QAAQ,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QAC7D,MAAM,WAAW,GAAG,IAAI,WAAW,GAAG,CAAC;QACvC,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC;QAE5B,kCAAkC;QAClC,MAAM,gBAAgB,GAAG,MAAM,GAAG,KAAK,CAAC;QACxC,IAAI,IAAI,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,gBAAgB,EAAE,GAAG,CAAC,EAAE,WAAW,CAAC,CAAC;YACxE,SAAS;QACX,CAAC;QAED,wEAAwE;QACxE,MAAM,kBAAkB,GAAG,SAAS,GAAG,+BAA+B,CAAC;QACvE,IAAI,IAAI,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAChD,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,kBAAkB,EAAE,GAAG,CAAC,EAAE,KAAK,WAAW,EAAE,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAY,EAAE,SAAiB;IACrD,IAAI,IAAI,CAAC,MAAM,IAAI,SAAS;QAAE,OAAO,IAAI,CAAC;IAC1C,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;AAClD,CAAC;AAED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,eAAe;IACb,gBAAgB;IAChB,uBAAuB;CACxB,CAAC"}

package/dist/utils/page-utils.d.ts

@@ -10,6 +10,14 @@
  * Based on the Python implementation from page_utils.py
  */
 import type { Page } from 'patchright';
+/**
+ * Check if text is an error message from NotebookLM
+ */
+export declare function isErrorMessage(text: string): boolean;
+/**
+ * Check if text indicates a rate limit error (should trigger account rotation)
+ */
+export declare function isRateLimitMessage(text: string): boolean;
 /**
  * Snapshot the latest response text currently visible
  * Returns null if no response found

package/dist/utils/page-utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"page-utils.d.ts","sourceRoot":"","sources":["../../src/utils/page-utils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAwEvC;;;GAGG;AACH,wBAAsB,sBAAsB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAE/E;AAED;;;GAGG;AACH,wBAAsB,oBAAoB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CA4BxE;AAED;;GAEG;AACH,wBAAsB,qBAAqB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CA2BvE;AAED;;;;;;;;;;;;GAYG;AACH,wBAAsB,mBAAmB,CACvC,IAAI,EAAE,IAAI,EACV,OAAO,GAAE;IACP,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,KAAK,CAAC,EAAE,OAAO,CAAC;CACZ,GACL,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CA2GxB;;;;;;;AAmMD,wBAKE"}
+{"version":3,"file":"page-utils.d.ts","sourceRoot":"","sources":["../../src/utils/page-utils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAkFvC;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAGpD;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAGxD;AA4CD;;;GAGG;AACH,wBAAsB,sBAAsB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAE/E;AAED;;;GAGG;AACH,wBAAsB,oBAAoB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CA4BxE;AAED;;GAEG;AACH,wBAAsB,qBAAqB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CA2BvE;AAED;;;;;;;;;;;;GAYG;AACH,wBAAsB,mBAAmB,CACvC,IAAI,EAAE,IAAI,EACV,OAAO,GAAE;IACP,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,KAAK,CAAC,EAAE,OAAO,CAAC;CACZ,GACL,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CA8IxB;;;;;;;AA+MD,wBAKE"}