notebooklm-mcp-server 3.0.1 → 3.0.3

package/dist/auth.js

@@ -77,10 +77,27 @@ export class AuthManager {
         }
         if (onStatus)
             onStatus('Extracting secure session cookies...');
-        const cookies = await context.cookies();
-        const cookieString = cookies
-            .map(c => `${c.name}=${c.value}`)
+        // CRITICAL: Filter cookies to only those that match notebooklm.google.com
+        // Playwright's context.cookies() without URL returns ALL cookies from ALL domains
+        // (including accounts.google.com, youtube.com, etc.), causing duplicate cookie names
+        // and conflicting session values. CDP's Network.getCookies (used by Python) only
+        // returns cookies for the current page URL. We match that behavior here.
+        const allCookies = await context.cookies('https://notebooklm.google.com');
+        // Deduplicate by name (keep last value, like browsers do)
+        const cookieMap = new Map();
+        for (const c of allCookies) {
+            cookieMap.set(c.name, c.value);
+        }
+        const cookieString = Array.from(cookieMap.entries())
+            .map(([name, value]) => `${name}=${value}`)
             .join('; ');
+        // Validate that required session cookies are present
+        const REQUIRED_COOKIES = ['SID', 'HSID', 'SSID', 'APISID', 'SAPISID'];
+        const missingCookies = REQUIRED_COOKIES.filter(name => !cookieMap.has(name));
+        if (missingCookies.length > 0) {
+            console.error(`Warning: Missing required cookies: ${missingCookies.join(', ')}`);
+        }
+        console.error(`Extracted ${cookieMap.size} unique cookies for notebooklm.google.com`);
         const authData = {
             cookies: cookieString,
             updatedAt: new Date().toISOString()

package/dist/client.d.ts

@@ -68,6 +68,16 @@ export declare class NotebookLMClient {
      */
     private extractRpcResult;
     listNotebooks(): Promise<Notebook[]>;
+    /**
+     * Get a single notebook's data including all source IDs.
+     * Uses the GET_NOTEBOOK RPC (rLM1Ne).
+     */
+    getNotebook(notebookId: string): Promise<any>;
+    /**
+     * Extract source IDs from raw notebook data returned by getNotebook().
+     * Matches Python _extract_source_ids_from_notebook().
+     */
+    private extractSourceIdsFromNotebook;
     createNotebook(title: string): Promise<string>;
     deleteNotebook(notebookId: string): Promise<boolean>;
     renameNotebook(notebookId: string, newTitle: string): Promise<boolean>;

package/dist/client.js

@@ -340,6 +340,47 @@ export class NotebookLMClient {
         }
         return notebooks;
     }
+    /**
+     * Get a single notebook's data including all source IDs.
+     * Uses the GET_NOTEBOOK RPC (rLM1Ne).
+     */
+    async getNotebook(notebookId) {
+        // Python: params = [notebook_id, None, [2], None, 0]
+        const result = await this.callRpc(RPC_IDS.GET_NOTEBOOK, [notebookId, null, [2], null, 0], `/notebook/${notebookId}`);
+        return result;
+    }
+    /**
+     * Extract source IDs from raw notebook data returned by getNotebook().
+     * Matches Python _extract_source_ids_from_notebook().
+     */
+    extractSourceIdsFromNotebook(notebookData) {
+        const sourceIds = [];
+        if (!notebookData || !Array.isArray(notebookData))
+            return sourceIds;
+        try {
+            // Structure: notebookData[0] = [title, sources_array, notebook_id, ...]
+            const notebookInfo = Array.isArray(notebookData[0]) ? notebookData[0] : notebookData;
+            if (notebookInfo.length > 1 && Array.isArray(notebookInfo[1])) {
+                const sources = notebookInfo[1];
+                for (const source of sources) {
+                    // Each source: [[source_id], title, metadata, ...]
+                    if (Array.isArray(source) && source.length > 0) {
+                        const sourceIdWrapper = source[0];
+                        if (Array.isArray(sourceIdWrapper) && sourceIdWrapper.length > 0) {
+                            const sourceId = sourceIdWrapper[0];
+                            if (typeof sourceId === 'string') {
+                                sourceIds.push(sourceId);
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        catch {
+            // ignore parse errors
+        }
+        return sourceIds;
+    }
     async createNotebook(title) {
         // Python: params = [title, None, None, [2], [1, None, None, None, None, None, None, None, None, None, [1]]]
         const params = [title, null, null, [2], [1, null, null, null, null, null, null, null, null, null, [1]]];
@@ -973,9 +1014,21 @@ export class NotebookLMClient {
     // =========================================================================
     async query(notebookId, queryText, sourceIds, conversationId, _retryCount = 0) {
         await this.init();
+        // Auto-fetch source_ids from notebook if not provided (matches Python behavior)
+        if (!sourceIds || sourceIds.length === 0) {
+            try {
+                const notebookData = await this.getNotebook(notebookId);
+                sourceIds = this.extractSourceIdsFromNotebook(notebookData);
+                console.error(`[NotebookLM] Auto-fetched ${sourceIds.length} source IDs from notebook`);
+            }
+            catch (e) {
+                console.error(`[NotebookLM] Warning: Could not auto-fetch source IDs: ${e.message}`);
+                sourceIds = [];
+            }
+        }
         const cid = conversationId || uuidv4();
         // Python: sources_array = [[[sid]] for sid in source_ids] (triple nested)
-        const sources = sourceIds ? sourceIds.map(id => [[id]]) : [];
+        const sources = sourceIds.length > 0 ? sourceIds.map(id => [[id]]) : [];
         // Python: params = [sources_array, query_text, None, [2, null, [1]], conversation_id]
         const params = [
             sources,

package/dist/server.js

@@ -4,7 +4,7 @@ import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextpro
 import { NotebookLMClient } from "./client.js";
 import { AuthManager } from "./auth.js";
 import chalk from "chalk";
-const VERSION = "3.0.1";
+const VERSION = "3.0.3";
 const server = new Server({
     name: "notebooklm-mcp-server",
     version: VERSION,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "notebooklm-mcp-server",
-  "version": "3.0.1",
+  "version": "3.0.3",
   "description": "Node.js Model Context Protocol server for Google NotebookLM",
   "type": "module",
   "repository": {