@musashishao/agent-kit 1.9.0 → 1.9.1

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

@@ -0,0 +1,516 @@
+#!/usr/bin/env node
+/**
+ * NotebookLM MCP Server for Agent Kit
+ * 
+ * Custom implementation with:
+ * - Chrome OAuth for Google login
+ * - Local library with tags and skill-linking
+ * - Agent Kit native integration
+ * 
+ * @author Agent Kit
+ */
+
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+
+import {
+    login,
+    loadAuthState,
+    logout as authLogout,
+    AuthState
+} from "./auth/index.js";
+
+import {
+    listNotebooks,
+    queryNotebook,
+    createNotebook,
+    addSource,
+    getSources,
+} from "./services/notebooklm.js";
+
+import {
+    addToLibrary,
+    searchLibrary,
+    removeFromLibrary,
+    linkToSkill,
+    getRecentlyUsed,
+    markAsUsed,
+    LibraryEntry,
+} from "./services/library.js";
+
+// ============================================================================
+// MCP Server
+// ============================================================================
+
+const server = new McpServer({
+    name: "notebooklm",
+    version: "1.0.0",
+});
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_login
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_login",
+    "Đăng nhập NotebookLM bằng tài khoản Google. Mở Chrome để bạn đăng nhập.",
+    {},
+    async () => {
+        try {
+            const state = await login();
+
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            success: true,
+                            email: state.email,
+                            message: "Đăng nhập NotebookLM thành công!",
+                        }, null, 2),
+                    },
+                ],
+            };
+        } catch (error) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            error: `Đăng nhập thất bại: ${error}`,
+                        }),
+                    },
+                ],
+            };
+        }
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_status
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_status",
+    "Kiểm tra trạng thái đăng nhập NotebookLM",
+    {},
+    async () => {
+        const state = loadAuthState();
+
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        authenticated: state.isAuthenticated,
+                        email: state.email,
+                        lastChecked: state.lastChecked,
+                    }, null, 2),
+                },
+            ],
+        };
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_logout
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_logout",
+    "Đăng xuất khỏi NotebookLM và xóa dữ liệu đăng nhập",
+    {},
+    async () => {
+        authLogout();
+
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        success: true,
+                        message: "Đã đăng xuất và xóa dữ liệu đăng nhập",
+                    }, null, 2),
+                },
+            ],
+        };
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_list_notebooks
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_list_notebooks",
+    "Liệt kê tất cả notebooks trong NotebookLM của bạn",
+    {},
+    async () => {
+        try {
+            const notebooks = await listNotebooks();
+
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            count: notebooks.length,
+                            notebooks,
+                        }, null, 2),
+                    },
+                ],
+            };
+        } catch (error) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({ error: `${error}` }),
+                    },
+                ],
+            };
+        }
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_query
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_query",
+    "Hỏi đáp dựa trên nội dung notebook. Trả về câu trả lời với citations nếu có.",
+    {
+        notebook_id: z.string().describe("ID của notebook (lấy từ list_notebooks)"),
+        question: z.string().describe("Câu hỏi của bạn"),
+        include_citations: z.boolean().optional().describe("Có bao gồm trích dẫn nguồn không"),
+    },
+    async ({ notebook_id, question, include_citations }) => {
+        try {
+            // Mark as used in library
+            markAsUsed(notebook_id);
+
+            const result = await queryNotebook(notebook_id, question, {
+                includeCitations: include_citations
+            });
+
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            question,
+                            answer: result.answer,
+                            citations: result.citations,
+                        }, null, 2),
+                    },
+                ],
+            };
+        } catch (error) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({ error: `${error}` }),
+                    },
+                ],
+            };
+        }
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_create_notebook
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_create_notebook",
+    "Tạo một notebook mới trong NotebookLM",
+    {
+        title: z.string().describe("Tên của notebook"),
+    },
+    async ({ title }) => {
+        try {
+            const notebook = await createNotebook(title);
+
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            success: true,
+                            notebook,
+                            message: `Đã tạo notebook "${title}"`,
+                        }, null, 2),
+                    },
+                ],
+            };
+        } catch (error) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({ error: `${error}` }),
+                    },
+                ],
+            };
+        }
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_add_source
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_add_source",
+    "Thêm nguồn tài liệu (URL hoặc text) vào notebook",
+    {
+        notebook_id: z.string().describe("ID của notebook"),
+        source_type: z.enum(["url", "text"]).describe("Loại nguồn: url hoặc text"),
+        content: z.string().describe("URL hoặc nội dung text"),
+    },
+    async ({ notebook_id, source_type, content }) => {
+        try {
+            const source = await addSource(notebook_id, source_type, content);
+
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            success: true,
+                            source,
+                            message: `Đã thêm nguồn vào notebook`,
+                        }, null, 2),
+                    },
+                ],
+            };
+        } catch (error) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({ error: `${error}` }),
+                    },
+                ],
+            };
+        }
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_get_sources
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_get_sources",
+    "Liệt kê các nguồn trong một notebook",
+    {
+        notebook_id: z.string().describe("ID của notebook"),
+    },
+    async ({ notebook_id }) => {
+        try {
+            const sources = await getSources(notebook_id);
+
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            count: sources.length,
+                            sources,
+                        }, null, 2),
+                    },
+                ],
+            };
+        } catch (error) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({ error: `${error}` }),
+                    },
+                ],
+            };
+        }
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_library_add
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_library_add",
+    "Lưu notebook vào thư viện local với tags và mô tả. Hỗ trợ skill-linking.",
+    {
+        notebook_id: z.string().describe("ID của notebook"),
+        notebook_url: z.string().describe("URL của notebook"),
+        title: z.string().describe("Tiêu đề"),
+        description: z.string().optional().describe("Mô tả ngắn"),
+        tags: z.array(z.string()).optional().describe("Tags (ví dụ: ['react', 'frontend'])"),
+        skills: z.array(z.string()).optional().describe("Agent Kit skills liên quan (ví dụ: ['react-patterns'])"),
+    },
+    async ({ notebook_id, notebook_url, title, description, tags, skills }) => {
+        const entry = addToLibrary(notebook_id, notebook_url, title, {
+            description,
+            tags,
+            skills,
+        });
+
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        success: true,
+                        entry,
+                        message: `Đã lưu "${title}" vào thư viện`,
+                    }, null, 2),
+                },
+            ],
+        };
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_library_search
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_library_search",
+    "Tìm kiếm trong thư viện local theo tags, skills, hoặc từ khóa",
+    {
+        tags: z.array(z.string()).optional().describe("Lọc theo tags"),
+        skills: z.array(z.string()).optional().describe("Lọc theo skills"),
+        query: z.string().optional().describe("Tìm trong title/description"),
+    },
+    async ({ tags, skills, query }) => {
+        const results = searchLibrary({ tags, skills, query });
+
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        count: results.length,
+                        entries: results,
+                    }, null, 2),
+                },
+            ],
+        };
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_library_recent
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_library_recent",
+    "Hiển thị các notebooks đã dùng gần đây",
+    {
+        limit: z.number().optional().describe("Số lượng tối đa (mặc định: 5)"),
+    },
+    async ({ limit }) => {
+        const results = getRecentlyUsed(limit || 5);
+
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        count: results.length,
+                        entries: results,
+                    }, null, 2),
+                },
+            ],
+        };
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Tool: notebooklm_skill_link
+// ---------------------------------------------------------------------------
+
+server.tool(
+    "notebooklm_skill_link",
+    "Liên kết notebook với một Agent Kit skill (ví dụ: react-patterns)",
+    {
+        entry_id: z.string().describe("ID trong thư viện local"),
+        skill_name: z.string().describe("Tên skill (từ .agent/skills/)"),
+    },
+    async ({ entry_id, skill_name }) => {
+        const entry = linkToSkill(entry_id, skill_name);
+
+        if (!entry) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({ error: "Entry không tồn tại" }),
+                    },
+                ],
+            };
+        }
+
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        success: true,
+                        entry,
+                        message: `Đã liên kết với skill "${skill_name}"`,
+                    }, null, 2),
+                },
+            ],
+        };
+    }
+);
+
+// ---------------------------------------------------------------------------
+// Resource: library
+// ---------------------------------------------------------------------------
+
+server.resource(
+    "library-list",
+    "notebooklm://library",
+    async () => {
+        const entries = searchLibrary();
+
+        return {
+            contents: [
+                {
+                    uri: "notebooklm://library",
+                    mimeType: "application/json",
+                    text: JSON.stringify(entries, null, 2),
+                },
+            ],
+        };
+    }
+);
+
+// ============================================================================
+// Main
+// ============================================================================
+
+async function main() {
+    const state = loadAuthState();
+
+    if (!state.isAuthenticated) {
+        console.error("⚠️  Chưa đăng nhập NotebookLM.");
+        console.error('   Dùng tool "notebooklm_login" để đăng nhập.');
+    } else {
+        console.error(`✅ Đã đăng nhập: ${state.email || "Unknown"}`);
+    }
+
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+
+    console.error("🎓 NotebookLM MCP Server for Agent Kit running on stdio");
+}
+
+main().catch(console.error);

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

@@ -0,0 +1,3 @@
+export * from "./notebooklm.js";
+export * from "./library.js";
+