notebooklm-mcp-server 1.0.9 → 1.1.0

package/dist/auth.js

@@ -22,39 +22,25 @@ export class AuthManager {
         if (onStatus)
             onStatus('Waiting for Google Login...');
         // Wait for the user to be logged in. 
-        // We check for either the URL pattern or the presence of session cookies
         let isDone = false;
         try {
             await Promise.race([
-                page.waitForURL('**/notebook/**', { timeout: 300000 }).then(() => { isDone = true; }),
+                // Wait for redirect to dashboard or notebook
+                page.waitForURL(url => url.origin === 'https://notebooklm.google.com' &&
+                    (url.pathname.includes('/notebook') || url.pathname === '/'), { timeout: 300000 }).then(() => { isDone = true; }),
+                // Wait for user icon or other logged-in indicia
+                page.waitForSelector('button[aria-label*="Account"], img[src*="googleusercontent.com"]', { timeout: 300000 })
+                    .then(() => { isDone = true; }),
+                // Safety fallback: if we see the landing page, we might already be logged in
                 page.waitForFunction(() => {
+                    const body = document.body.innerText;
                     return window.location.href.includes('notebooklm.google.com') &&
-                        !document.body.innerText.includes('Sign in');
+                        !body.includes('Sign in') &&
+                        (body.includes('Notebooks') || body.includes('Create new'));
                 }, { polling: 2000, timeout: 300000 }).then(() => { isDone = true; }),
-                // Fallback: check if we have the critical cookies
-                new Promise((resolve) => {
-                    const checkCookies = async () => {
-                        if (isDone)
-                            return;
-                        try {
-                            const cookies = await context.cookies();
-                            const sessionCookie = cookies.find(c => c.name === '__Secure-3PSID' || c.name === 'SID');
-                            if (sessionCookie && page.url().includes('notebooklm.google.com')) {
-                                isDone = true;
-                                resolve(true);
-                            }
-                            else {
-                                if (!isDone)
-                                    setTimeout(checkCookies, 2000);
-                            }
-                        }
-                        catch (e) {
-                            isDone = true;
-                        }
-                    };
-                    checkCookies();
-                })
             ]);
+            // Small delay to ensure all session cookies are synced
+            await page.waitForTimeout(2000);
         }
         catch (e) {
             throw new Error('Authentication timed out or browser was closed.');

package/dist/index.js

@@ -4,7 +4,7 @@ const program = new Command();
 program
     .name('notebooklm-mcp-server')
     .description('NotebookLM MCP Server (Node.js)')
-    .version('1.0.9');
+    .version('1.1.0');
 program
     .command('server')
     .description('Start the MCP server (default)')

package/dist/server.js

@@ -6,7 +6,7 @@ import { AuthManager } from "./auth.js";
 import chalk from "chalk";
 const server = new Server({
     name: "notebooklm-mcp-server",
-    version: "1.0.9",
+    version: "1.1.0",
 }, {
     capabilities: {
         tools: {},
@@ -262,6 +262,11 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
                     required: ["notebook_id"],
                 },
             },
+            {
+                name: "refresh_auth",
+                description: "Open browser to refresh Google Session cookies (interactive)",
+                inputSchema: { type: "object", properties: {} },
+            },
         ],
     };
 });
@@ -271,76 +276,82 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         switch (name) {
             case "notebook_list":
                 const notebooks = await client.listNotebooks();
-                return { content: [{ type: "text", text: JSON.stringify(notebooks, null, 2) }] };
+                return { content: [{ type: "text", text: JSON.stringify({ notebooks }, null, 2) }] };
             case "notebook_create":
                 const newId = await client.createNotebook(args?.title);
-                return { content: [{ type: "text", text: `Created notebook with ID: ${newId}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", notebook_id: newId }) }] };
             case "notebook_delete":
                 await client.deleteNotebook(args?.notebook_id);
-                return { content: [{ type: "text", text: `Deleted notebook ${args?.notebook_id}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", message: `Deleted notebook ${args?.notebook_id}` }) }] };
             case "notebook_rename":
                 await client.renameNotebook(args?.notebook_id, args?.title);
-                return { content: [{ type: "text", text: `Renamed notebook to ${args?.title}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", message: `Renamed notebook to ${args?.title}` }) }] };
             case "notebook_add_url":
                 const sourceIdUrl = await client.addUrlSource(args?.notebook_id, args?.url);
-                return { content: [{ type: "text", text: `Added source with ID: ${sourceIdUrl}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", source_id: sourceIdUrl }) }] };
             case "notebook_add_text":
                 const sourceIdText = await client.addTextSource(args?.notebook_id, args?.title, args?.content);
-                return { content: [{ type: "text", text: `Added text source with ID: ${sourceIdText}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", source_id: sourceIdText }) }] };
             case "notebook_query":
                 const queryResult = await client.query(args?.notebook_id, args?.query, undefined, args?.conversation_id);
                 return {
                     content: [
-                        { type: "text", text: queryResult.answer },
-                        { type: "text", text: `\n\n(Conversation ID: ${queryResult.conversation_id})` }
+                        { type: "text", text: JSON.stringify(queryResult, null, 2) }
                     ]
                 };
             case "research_start":
                 const taskInfo = await client.startResearch(args?.notebook_id, args?.query, args?.source || 'web', args?.mode || 'fast');
-                return { content: [{ type: "text", text: `Started research task: ${JSON.stringify(taskInfo)}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", task: taskInfo }, null, 2) }] };
             case "research_poll":
                 const researchResult = await client.pollResearch(args?.notebook_id);
                 return { content: [{ type: "text", text: JSON.stringify(researchResult, null, 2) }] };
             case "research_import":
                 const imported = await client.importResearchSources(args?.notebook_id, args?.task_id, args?.sources);
-                return { content: [{ type: "text", text: `Imported ${imported.length} sources: ${JSON.stringify(imported)}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", imported_count: imported.length, sources: imported }, null, 2) }] };
             case "mind_map_generate":
                 const mmData = await client.generateMindMap(args?.source_ids);
                 return { content: [{ type: "text", text: JSON.stringify(mmData, null, 2) }] };
             case "mind_map_save":
                 const savedMm = await client.saveMindMap(args?.notebook_id, args?.mind_map_json, args?.source_ids, args?.title);
-                return { content: [{ type: "text", text: `Mind Map saved: ${JSON.stringify(savedMm)}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", mind_map: savedMm }, null, 2) }] };
             case "mind_map_list":
                 const maps = await client.listMindMaps(args?.notebook_id);
-                return { content: [{ type: "text", text: JSON.stringify(maps, null, 2) }] };
+                return { content: [{ type: "text", text: JSON.stringify({ mind_maps: maps }, null, 2) }] };
             case "mind_map_delete":
                 await client.deleteMindMap(args?.notebook_id, args?.mind_map_id);
-                return { content: [{ type: "text", text: "Mind Map deleted." }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", message: "Mind Map deleted." }) }] };
             case "notebook_add_local_file":
                 const fileSourceId = await client.uploadLocalFile(args?.notebook_id, args?.path);
-                return { content: [{ type: "text", text: `Successfully uploaded and added file. Source ID: ${fileSourceId}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", source_id: fileSourceId }) }] };
             case "notebook_add_drive":
                 const driveId = await client.addDriveSource(args?.notebook_id, args?.file_id);
-                return { content: [{ type: "text", text: `Successfully added Drive source. ID: ${driveId}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", source_id: driveId }) }] };
             case "source_delete":
                 await client.deleteSource(args?.notebook_id, args?.source_id);
-                return { content: [{ type: "text", text: `Source ${args?.source_id} deleted successfully.` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", message: `Source ${args?.source_id} deleted successfully.` }) }] };
             case "source_sync":
                 await client.syncDriveSource(args?.notebook_id, args?.source_id);
-                return { content: [{ type: "text", text: `Sync triggered for source ${args?.source_id}.` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", message: `Sync triggered for source ${args?.source_id}.` }) }] };
             case "audio_overview_create":
                 const audioInfo = await client.createAudioOverview(args?.notebook_id, args?.source_ids || [], args?.language || 'en');
-                return { content: [{ type: "text", text: `Audio Overview generation started: ${JSON.stringify(audioInfo)}` }] };
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", audio_info: audioInfo }, null, 2) }] };
             case "studio_poll":
                 const studioStatus = await client.pollStudioStatus(args?.notebook_id);
                 return { content: [{ type: "text", text: JSON.stringify(studioStatus, null, 2) }] };
+            case "refresh_auth":
+                const auth = new AuthManager();
+                await auth.runAuthentication();
+                // After auth is successful, we need to update our client
+                const newCookies = auth.getSavedCookies();
+                client = new NotebookLMClient(newCookies);
+                return { content: [{ type: "text", text: JSON.stringify({ status: "success", message: "Authentication refreshed successfully." }) }] };
             default:
                 throw new Error(`Unknown tool: ${name}`);
         }
     }
     catch (error) {
         return {
-            content: [{ type: "text", text: `Error: ${error.message}` }],
+            content: [{ type: "text", text: JSON.stringify({ status: "error", error: error.message }) }],
             isError: true,
         };
     }

package/package.json

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