@musashishao/agent-kit 1.9.0 → 1.9.1

package/.agent/mcp/config/registry.json

@@ -1,54 +1,68 @@
 {
-    "registries": {
-        "npm": {
-            "name": "NPM Registry",
-            "searchUrl": "https://registry.npmjs.org/-/v1/search?text=mcp-server",
-            "pattern": "mcp-server-*"
-        },
-        "awesome": {
-            "name": "Awesome MCP Servers",
-            "url": "https://github.com/punkpeye/awesome-mcp-servers",
-            "categories": [
-                "browser-automation",
-                "databases",
-                "developer-tools",
-                "file-systems",
-                "version-control",
-                "cloud-platforms"
-            ]
-        }
+  "registries": {
+    "npm": {
+      "name": "NPM Registry",
+      "searchUrl": "https://registry.npmjs.org/-/v1/search?text=mcp-server",
+      "pattern": "mcp-server-*"
     },
-    "featured": [
-        {
-            "name": "@anthropic/mcp-server-filesystem",
-            "description": "Secure file system access",
-            "source": "npm"
-        },
-        {
-            "name": "@anthropic/mcp-server-sqlite",
-            "description": "SQLite database integration",
-            "source": "npm"
-        },
-        {
-            "name": "@anthropic/mcp-server-github",
-            "description": "GitHub API integration",
-            "source": "npm"
-        },
-        {
-            "name": "@anthropic/mcp-server-memory",
-            "description": "Persistent memory for AI",
-            "source": "npm"
-        },
-        {
-            "name": "playwright-mcp",
-            "description": "Browser automation with Playwright",
-            "source": "npm"
-        },
-        {
-            "name": "mcp-server-fetch",
-            "description": "HTTP fetch capabilities",
-            "source": "npm"
-        }
-    ],
-    "installed": []
+    "awesome": {
+      "name": "Awesome MCP Servers",
+      "url": "https://github.com/punkpeye/awesome-mcp-servers",
+      "categories": [
+        "browser-automation",
+        "databases",
+        "developer-tools",
+        "file-systems",
+        "version-control",
+        "cloud-platforms"
+      ]
+    }
+  },
+  "featured": [
+    {
+      "name": "@anthropic/mcp-server-filesystem",
+      "description": "Secure file system access",
+      "source": "npm"
+    },
+    {
+      "name": "@anthropic/mcp-server-sqlite",
+      "description": "SQLite database integration",
+      "source": "npm"
+    },
+    {
+      "name": "@anthropic/mcp-server-github",
+      "description": "GitHub API integration",
+      "source": "npm"
+    },
+    {
+      "name": "@anthropic/mcp-server-memory",
+      "description": "Persistent memory for AI",
+      "source": "npm"
+    },
+    {
+      "name": "playwright-mcp",
+      "description": "Browser automation with Playwright",
+      "source": "npm"
+    },
+    {
+      "name": "mcp-server-fetch",
+      "description": "HTTP fetch capabilities",
+      "source": "npm"
+    }
+  ],
+  "installed": [
+    {
+      "name": "notebooklm-mcp",
+      "serverName": "notebooklm",
+      "config": {
+        "command": "node",
+        "args": [
+          ".agent/mcp/servers/notebooklm/dist/index.js"
+        ]
+      },
+      "installedAt": "2026-01-30T16:50:20.478Z",
+      "claudeConfigured": false,
+      "cursorConfigured": true
+    }
+  ]
 }

package/.agent/mcp/servers/notebooklm/README.md

@@ -0,0 +1,114 @@
+# NotebookLM MCP Server - Agent Kit Custom
+
+> 🎓 Custom NotebookLM MCP server với Chrome OAuth, skill-linking, và Agent Kit native integration.
+
+## ✨ Tính năng
+
+| Feature | Community | Agent Kit Custom |
+|---------|-----------|------------------|
+| Chrome OAuth | ✅ | ✅ Persistent sessions |
+| Query Notebook | ✅ | ✅ + markAsUsed tracking |
+| Library | Tags only | ✅ **Skill-linking** |
+| Agent Integration | ❌ | ✅ Native |
+
+## 🛠️ Tools (12)
+
+### Authentication
+| Tool | Description |
+|------|-------------|
+| `notebooklm_login` | Đăng nhập via Chrome OAuth |
+| `notebooklm_status` | Kiểm tra trạng thái |
+| `notebooklm_logout` | Đăng xuất + xóa data |
+
+### Notebook Operations
+| Tool | Description |
+|------|-------------|
+| `notebooklm_list_notebooks` | Liệt kê tất cả notebooks |
+| `notebooklm_query` | Hỏi đáp với citations |
+| `notebooklm_create_notebook` | Tạo notebook mới |
+| `notebooklm_add_source` | Thêm URL/text |
+| `notebooklm_get_sources` | List sources |
+
+### Library (Unique)
+| Tool | Description |
+|------|-------------|
+| `notebooklm_library_add` | Lưu vào library local |
+| `notebooklm_library_search` | Tìm theo tags/skills |
+| `notebooklm_library_recent` | Notebooks dùng gần đây |
+| `notebooklm_skill_link` | **Link notebook ↔ Agent Kit skill** |
+
+## 🚀 Cài đặt
+
+```bash
+cd .agent/mcp/servers/notebooklm
+npm install
+npm run build
+```
+
+## ⚙️ Cấu hình
+
+### Cursor (`.cursor/mcp.json`)
+```json
+{
+  "mcpServers": {
+    "notebooklm": {
+      "command": "node",
+      "args": [".agent/mcp/servers/notebooklm/dist/index.js"]
+    }
+  }
+}
+```
+
+### Claude Desktop
+```json
+{
+  "mcpServers": {
+    "notebooklm": {
+      "command": "node",
+      "args": ["/path/to/.agent/mcp/servers/notebooklm/dist/index.js"]
+    }
+  }
+}
+```
+
+## 📖 Sử dụng
+
+### 1. Đăng nhập
+```
+User: "Log me in to NotebookLM"
+→ Chrome opens → Login Google → Done
+```
+
+### 2. Query notebook
+```
+User: "Query notebook abc123: How to implement auth?"
+→ Returns answer with citations
+```
+
+### 3. Skill-linking (Unique feature)
+```
+User: "Add my React docs notebook to library, link to react-patterns skill"
+
+# Sau đó
+User: "I'm working on React, find relevant notebooks"
+→ Auto-suggests notebooks linked to react-patterns
+```
+
+## 📁 Data Locations
+
+| Data | Path |
+|------|------|
+| Chrome Profile | `~/.agent-kit/notebooklm/chrome-profile/` |
+| Library | `~/.agent-kit/notebooklm/library/notebooks.json` |
+
+## 🔧 Environment Variables
+
+| Variable | Description |
+|----------|-------------|
+| `NOTEBOOKLM_USER_DATA` | Chrome profile path |
+| `NOTEBOOKLM_LIBRARY_DIR` | Library storage path |
+
+---
+
+**Version:** 1.0.0  
+**Made for:** Agent Kit

package/.agent/mcp/servers/notebooklm/package.json

@@ -0,0 +1,35 @@
+{
+    "name": "@agent-kit/mcp-notebooklm",
+    "version": "1.0.0",
+    "description": "Custom NotebookLM MCP server for Agent Kit - Chrome OAuth, skill-aware research, codebase context",
+    "type": "module",
+    "main": "dist/index.js",
+    "bin": {
+        "mcp-notebooklm": "./dist/index.js"
+    },
+    "scripts": {
+        "build": "tsc",
+        "dev": "tsx src/index.ts",
+        "start": "node dist/index.js"
+    },
+    "keywords": [
+        "mcp",
+        "notebooklm",
+        "agent-kit",
+        "ai",
+        "research"
+    ],
+    "dependencies": {
+        "@modelcontextprotocol/sdk": "^1.12.0",
+        "playwright": "^1.50.0",
+        "zod": "^3.23.8"
+    },
+    "devDependencies": {
+        "@types/node": "^22.0.0",
+        "tsx": "^4.19.0",
+        "typescript": "^5.6.0"
+    },
+    "engines": {
+        "node": ">=20.0.0"
+    }
+}

package/.agent/mcp/servers/notebooklm/src/auth/chrome.ts

@@ -0,0 +1,225 @@
+/**
+ * Chrome Authentication Module
+ * 
+ * Handles Google OAuth via Chrome browser.
+ * Uses Playwright for browser automation with persistent context.
+ */
+
+import { chromium, BrowserContext, Page } from "playwright";
+import * as fs from "fs";
+import * as path from "path";
+import * as os from "os";
+
+const USER_DATA_DIR = process.env.NOTEBOOKLM_USER_DATA ||
+    path.join(os.homedir(), ".agent-kit", "notebooklm", "chrome-profile");
+
+const COOKIES_FILE = path.join(USER_DATA_DIR, "cookies.json");
+const AUTH_STATE_FILE = path.join(USER_DATA_DIR, "auth-state.json");
+
+export interface AuthState {
+    isAuthenticated: boolean;
+    email?: string;
+    lastChecked: string;
+}
+
+/**
+ * Ensure user data directory exists
+ */
+function ensureDataDir(): void {
+    if (!fs.existsSync(USER_DATA_DIR)) {
+        fs.mkdirSync(USER_DATA_DIR, { recursive: true });
+    }
+}
+
+/**
+ * Load saved authentication state
+ */
+export function loadAuthState(): AuthState {
+    ensureDataDir();
+
+    if (fs.existsSync(AUTH_STATE_FILE)) {
+        try {
+            return JSON.parse(fs.readFileSync(AUTH_STATE_FILE, "utf-8"));
+        } catch {
+            return { isAuthenticated: false, lastChecked: new Date().toISOString() };
+        }
+    }
+
+    return { isAuthenticated: false, lastChecked: new Date().toISOString() };
+}
+
+/**
+ * Save authentication state
+ */
+export function saveAuthState(state: AuthState): void {
+    ensureDataDir();
+    fs.writeFileSync(AUTH_STATE_FILE, JSON.stringify(state, null, 2));
+}
+
+/**
+ * Launch browser with persistent context
+ */
+export async function launchBrowser(): Promise<BrowserContext> {
+    ensureDataDir();
+
+    const context = await chromium.launchPersistentContext(USER_DATA_DIR, {
+        headless: false,
+        args: [
+            "--disable-blink-features=AutomationControlled",
+            "--no-first-run",
+            "--no-default-browser-check",
+        ],
+        viewport: { width: 1280, height: 800 },
+    });
+
+    return context;
+}
+
+/**
+ * Check if user is logged in to NotebookLM
+ */
+export async function checkLogin(page: Page): Promise<{ loggedIn: boolean; email?: string }> {
+    try {
+        await page.goto("https://notebooklm.google.com", {
+            waitUntil: "networkidle",
+            timeout: 30000,
+        });
+
+        // Check for sign-in prompt
+        const signInButton = await page.$('text="Sign in"');
+        if (signInButton) {
+            return { loggedIn: false };
+        }
+
+        // Check for account menu (indicates logged in)
+        const accountButton = await page.$('[aria-label*="Account"], [data-ogsr-up]');
+        if (accountButton) {
+            // Try to get email
+            const email = await page.$eval(
+                '[data-email], [aria-label*="@"]',
+                (el) => el.getAttribute("data-email") || el.getAttribute("aria-label")?.match(/[\w.-]+@[\w.-]+/)?.[0]
+            ).catch(() => undefined);
+
+            return { loggedIn: true, email };
+        }
+
+        // If we see notebooks list, we're logged in
+        const notebooksList = await page.$('[role="listbox"], [class*="notebook"]');
+        if (notebooksList) {
+            return { loggedIn: true };
+        }
+
+        return { loggedIn: false };
+    } catch (error) {
+        return { loggedIn: false };
+    }
+}
+
+/**
+ * Perform login flow
+ * Opens browser and waits for user to complete Google OAuth
+ */
+export async function login(): Promise<AuthState> {
+    const context = await launchBrowser();
+    const page = await context.newPage();
+
+    try {
+        // Navigate to NotebookLM
+        await page.goto("https://notebooklm.google.com", { waitUntil: "networkidle" });
+
+        // Check if already logged in
+        const status = await checkLogin(page);
+
+        if (status.loggedIn) {
+            const state: AuthState = {
+                isAuthenticated: true,
+                email: status.email,
+                lastChecked: new Date().toISOString(),
+            };
+            saveAuthState(state);
+            await context.close();
+            return state;
+        }
+
+        // Wait for user to complete login (max 5 minutes)
+        console.error("🔐 Waiting for you to complete Google login...");
+        console.error("   Please log in to your Google account in the browser window.");
+
+        // Poll for login completion
+        let attempts = 0;
+        const maxAttempts = 60; // 5 minutes with 5 second intervals
+
+        while (attempts < maxAttempts) {
+            await page.waitForTimeout(5000);
+            const loginStatus = await checkLogin(page);
+
+            if (loginStatus.loggedIn) {
+                const state: AuthState = {
+                    isAuthenticated: true,
+                    email: loginStatus.email,
+                    lastChecked: new Date().toISOString(),
+                };
+                saveAuthState(state);
+
+                // Save cookies for future sessions
+                const cookies = await context.cookies();
+                fs.writeFileSync(COOKIES_FILE, JSON.stringify(cookies, null, 2));
+
+                console.error(`✅ Login successful! ${loginStatus.email || ""}`);
+                await context.close();
+                return state;
+            }
+
+            attempts++;
+        }
+
+        // Timeout
+        await context.close();
+        throw new Error("Login timeout - please try again");
+
+    } catch (error) {
+        await context.close();
+        throw error;
+    }
+}
+
+/**
+ * Get authenticated browser context
+ * Returns existing session or throws if not authenticated
+ */
+export async function getAuthenticatedContext(): Promise<BrowserContext> {
+    const state = loadAuthState();
+
+    if (!state.isAuthenticated) {
+        throw new Error("Not authenticated. Please run notebooklm_login first.");
+    }
+
+    const context = await launchBrowser();
+
+    // Load saved cookies if available
+    if (fs.existsSync(COOKIES_FILE)) {
+        try {
+            const cookies = JSON.parse(fs.readFileSync(COOKIES_FILE, "utf-8"));
+            await context.addCookies(cookies);
+        } catch {
+            // Ignore cookie errors
+        }
+    }
+
+    return context;
+}
+
+/**
+ * Logout and clear saved credentials
+ */
+export function logout(): void {
+    if (fs.existsSync(COOKIES_FILE)) fs.unlinkSync(COOKIES_FILE);
+    if (fs.existsSync(AUTH_STATE_FILE)) fs.unlinkSync(AUTH_STATE_FILE);
+
+    // Clear Chrome profile data
+    if (fs.existsSync(USER_DATA_DIR)) {
+        fs.rmSync(USER_DATA_DIR, { recursive: true, force: true });
+    }
+}
+
+export { USER_DATA_DIR };

package/.agent/mcp/servers/notebooklm/src/auth/index.ts

@@ -0,0 +1 @@
+export * from "./chrome.js";