@khoinguyen2002/doc-mcp 1.0.0 → 1.0.1

package/dist/db/vector.d.ts

@@ -1,8 +1,8 @@
 export declare function initVectorDB(): Promise<void>;
 export declare function embedText(text: string): Promise<number[]>;
-export declare function upsertProjectDocument(projectId: string, text: string, metadata?: Record<string, any>): Promise<void>;
-export declare function searchProjectMemory(projectId: string, query: string, topK?: number): Promise<any[]>;
-export declare function deleteProjectDocument(projectId: string, fileId: string): Promise<void>;
-export declare function checkProjectDocumentExists(projectId: string, fileId: string): Promise<boolean>;
-export declare function getProjectDocumentMetadata(projectId: string): Promise<Record<string, string>>;
+export declare function upsertProjectDocument(folderId: string, text: string, metadata?: Record<string, any>): Promise<void>;
+export declare function searchProjectMemory(folderId: string, query: string, topK?: number): Promise<any[]>;
+export declare function deleteProjectDocument(folderId: string, fileId: string): Promise<void>;
+export declare function checkProjectDocumentExists(folderId: string, fileId: string): Promise<boolean>;
+export declare function getProjectDocumentMetadata(folderId: string): Promise<Record<string, string>>;
 //# sourceMappingURL=vector.d.ts.map

package/dist/db/vector.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vector.d.ts","sourceRoot":"","sources":["../../src/db/vector.ts"],"names":[],"mappings":"AAOA,wBAAsB,YAAY,kBAqCjC;AAED,wBAAsB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAwB/D;AAED,wBAAsB,qBAAqB,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CA0B9H;AAED,wBAAsB,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,MAAU,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,CA4B5G;AAED,wBAAsB,qBAAqB,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAa5F;AAED,wBAAsB,0BAA0B,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAapG;AAED,wBAAsB,0BAA0B,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAwBnG"}
+{"version":3,"file":"vector.d.ts","sourceRoot":"","sources":["../../src/db/vector.ts"],"names":[],"mappings":"AAOA,wBAAsB,YAAY,kBAqCjC;AAED,wBAAsB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAwB/D;AAED,wBAAsB,qBAAqB,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CA0B7H;AAED,wBAAsB,mBAAmB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,MAAU,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,CA4B3G;AAED,wBAAsB,qBAAqB,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAa3F;AAED,wBAAsB,0BAA0B,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAanG;AAED,wBAAsB,0BAA0B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAwBlG"}

package/dist/db/vector.js

@@ -61,7 +61,7 @@ export async function embedText(text) {
     }
     return json.data[0].embedding;
 }
-export async function upsertProjectDocument(projectId, text, metadata = {}) {
+export async function upsertProjectDocument(folderId, text, metadata = {}) {
     await initVectorDB();
     if (!client)
         throw new Error("Qdrant not initialized");
@@ -73,7 +73,7 @@ export async function upsertProjectDocument(projectId, text, metadata = {}) {
                 id: uuidv4(),
                 vector: vector,
                 payload: {
-                    projectId,
+                    folderId,
                     text,
                     source: metadata.source || "user",
                     file_id: metadata.file_id || null,
@@ -84,9 +84,9 @@ export async function upsertProjectDocument(projectId, text, metadata = {}) {
             }
         ]
     });
-    console.error(`Upserted document chunk for project ${projectId}`);
+    console.error(`Upserted document chunk for folder ${folderId}`);
 }
-export async function searchProjectMemory(projectId, query, topK = 3) {
+export async function searchProjectMemory(folderId, query, topK = 3) {
     await initVectorDB();
     if (!client)
         throw new Error("Qdrant not initialized");
@@ -98,9 +98,9 @@ export async function searchProjectMemory(projectId, query, topK = 3) {
         filter: {
             must: [
                 {
-                    key: "projectId",
+                    key: "folderId",
                     match: {
-                        value: projectId
+                        value: folderId
                     }
                 }
             ]
@@ -113,42 +113,42 @@ export async function searchProjectMemory(projectId, query, topK = 3) {
         ...r.payload
     }));
 }
-export async function deleteProjectDocument(projectId, fileId) {
+export async function deleteProjectDocument(folderId, fileId) {
     await initVectorDB();
     if (!client)
         return;
     await client.delete(COLLECTION_NAME, {
         filter: {
             must: [
-                { key: "projectId", match: { value: projectId } },
+                { key: "folderId", match: { value: folderId } },
                 { key: "file_id", match: { value: fileId } }
             ]
         }
     });
-    console.error(`Deleted old chunks from Qdrant for ${projectId} / ${fileId}`);
+    console.error(`Deleted old chunks from Qdrant for ${folderId} / ${fileId}`);
 }
-export async function checkProjectDocumentExists(projectId, fileId) {
+export async function checkProjectDocumentExists(folderId, fileId) {
     await initVectorDB();
     if (!client)
         return false;
     const res = await client.count(COLLECTION_NAME, {
         filter: {
             must: [
-                { key: "projectId", match: { value: projectId } },
+                { key: "folderId", match: { value: folderId } },
                 { key: "file_id", match: { value: fileId } }
             ]
         }
     });
     return res.count > 0;
 }
-export async function getProjectDocumentMetadata(projectId) {
+export async function getProjectDocumentMetadata(folderId) {
     await initVectorDB();
     if (!client)
         return {};
     const res = await client.scroll(COLLECTION_NAME, {
         filter: {
             must: [
-                { key: "projectId", match: { value: projectId } },
+                { key: "folderId", match: { value: folderId } },
                 { key: "source", match: { value: "google_drive" } }
             ]
         },

package/dist/mcp-server.js

@@ -16,15 +16,19 @@ const server = new McpServer({
 });
 // Register tools
 server.registerTool("list_drive_files", {
-    description: "List and search for Google Drive documents in the configured folder.",
+    description: "List and search for Google Drive documents and subfolders in a specific folder.",
     inputSchema: {
         keyword: z
             .string()
             .optional()
             .describe("Optional keyword to search for in document titles"),
+        targetFolderId: z
+            .string()
+            .optional()
+            .describe("Optional Google Drive folder ID to list contents from. Defaults to the root knowledge folder."),
     },
-}, async ({ keyword }) => {
-    const res = await listDriveFiles(keyword);
+}, async ({ keyword, targetFolderId }) => {
+    const res = await listDriveFiles(keyword, targetFolderId);
     if (!res.success) {
         return {
             content: [{ type: "text", text: `Error: ${res.error}` }],

package/dist/tools/driveTools.d.ts

@@ -1,4 +1,4 @@
-export declare function listDriveFiles(keyword?: string): Promise<{
+export declare function listDriveFiles(keyword?: string, targetFolderId?: string): Promise<{
     success: boolean;
     results: import("googleapis").drive_v3.Schema$File[];
     error?: undefined;

package/dist/tools/driveTools.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"driveTools.d.ts","sourceRoot":"","sources":["../../src/tools/driveTools.ts"],"names":[],"mappings":"AAiCA,wBAAsB,cAAc,CAAC,OAAO,CAAC,EAAE,MAAM;;;;;;;;GAoCpD;AAED,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM;;;;;;;;GA6CxE;AAED,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,MAAM;;;;;;;;GAsCrD;AAED,wBAAsB,eAAe,CAAC,QAAQ,EAAE,MAAM;;;;;;GAuCrD"}
+{"version":3,"file":"driveTools.d.ts","sourceRoot":"","sources":["../../src/tools/driveTools.ts"],"names":[],"mappings":"AAiCA,wBAAsB,cAAc,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,cAAc,CAAC,EAAE,MAAM;;;;;;;;GAoC7E;AAED,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM;;;;;;;;GA6CxE;AAED,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,MAAM;;;;;;;;GAsCrD;AAED,wBAAsB,eAAe,CAAC,QAAQ,EAAE,MAAM;;;;;;GAqDrD"}

package/dist/tools/driveTools.js

@@ -19,8 +19,8 @@ function getDriveClient() {
     });
     return google.drive({ version: "v3", auth });
 }
-export async function listDriveFiles(keyword) {
-    const folderId = process.env.DOC_MCP_DRIVE_FOLDER_ID;
+export async function listDriveFiles(keyword, targetFolderId) {
+    const folderId = targetFolderId || config.DOC_MCP_DRIVE_FOLDER_ID;
     if (!folderId) {
         return {
             success: false,
@@ -29,14 +29,14 @@ export async function listDriveFiles(keyword) {
     }
     try {
         const drive = getDriveClient();
-        let q = "mimeType = 'application/vnd.google-apps.document'";
+        let q = "(mimeType = 'application/vnd.google-apps.document' or mimeType = 'application/vnd.google-apps.folder') and trashed = false";
         q = `'${folderId}' in parents and ${q}`;
         if (keyword) {
             q = `name contains '${keyword}' and ${q}`;
         }
         const res = await drive.files.list({
             q,
-            fields: "files(id, name, description)",
+            fields: "files(id, name, description, mimeType)",
             spaces: "drive",
             pageSize: 50,
             supportsAllDrives: true,
@@ -92,7 +92,7 @@ export async function syncSingleDocument(fileId, folderId) {
     return { synced: false, driveModifiedTime };
 }
 export async function readDriveDocument(fileId) {
-    const folderId = process.env.DOC_MCP_DRIVE_FOLDER_ID;
+    const folderId = config.DOC_MCP_DRIVE_FOLDER_ID;
     if (!folderId) {
         return {
             success: false,
@@ -130,17 +130,28 @@ export async function readDriveDocument(fileId) {
 export async function syncFolderState(folderId) {
     try {
         const drive = getDriveClient();
-        let q = "mimeType = 'application/vnd.google-apps.document'";
-        q = `'${folderId}' in parents and ${q}`;
-        const res = await drive.files.list({
-            q,
-            fields: "files(id, name, modifiedTime)",
-            spaces: "drive",
-            pageSize: 100,
-            supportsAllDrives: true,
-            includeItemsFromAllDrives: true,
-        });
-        const driveFiles = res.data.files || [];
+        async function getAllDocumentsFlat() {
+            let allDocs = [];
+            let pageToken = undefined;
+            do {
+                const docsRes = await drive.files.list({
+                    // Chú ý: Đéo check parentId nữa, gom sạch sành sanh mọi file .doc mà Service Account nhìn thấy
+                    q: `mimeType = 'application/vnd.google-apps.document' and trashed = false`,
+                    fields: "nextPageToken, files(id, name, modifiedTime)",
+                    spaces: "drive",
+                    pageSize: 100, // Google API limit mỗi page, tự động nhảy trang nếu nhiều hơn
+                    pageToken,
+                    supportsAllDrives: true,
+                    includeItemsFromAllDrives: true,
+                });
+                if (docsRes.data.files) {
+                    allDocs = allDocs.concat(docsRes.data.files);
+                }
+                pageToken = docsRes.data.nextPageToken || undefined;
+            } while (pageToken);
+            return allDocs;
+        }
+        const driveFiles = await getAllDocumentsFlat();
         const dbMetaMap = await getProjectDocumentMetadata(folderId);
         // Sync updated or new files
         for (const file of driveFiles) {
@@ -153,7 +164,7 @@ export async function syncFolderState(folderId) {
         }
         // Delete removed files from DB
         for (const dbFileId of Object.keys(dbMetaMap)) {
-            if (!driveFiles.find(f => f.id === dbFileId)) {
+            if (!driveFiles.find((f) => f.id === dbFileId)) {
                 await deleteProjectDocument(folderId, dbFileId);
             }
         }

package/dist/tools/knowledgeTools.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"knowledgeTools.d.ts","sourceRoot":"","sources":["../../src/tools/knowledgeTools.ts"],"names":[],"mappings":"AAGA,wBAAsB,aAAa,CAAC,OAAO,EAAE,MAAM;;;;;;;;GAelD;AAED,wBAAsB,eAAe,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,MAAU;;;;;;;;GAuBpE"}
+{"version":3,"file":"knowledgeTools.d.ts","sourceRoot":"","sources":["../../src/tools/knowledgeTools.ts"],"names":[],"mappings":"AAIA,wBAAsB,aAAa,CAAC,OAAO,EAAE,MAAM;;;;;;;;GAmBlD;AAED,wBAAsB,eAAe,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,MAAU;;;;;;;;GAmCpE"}

package/dist/tools/knowledgeTools.js

@@ -1,25 +1,34 @@
+import { config } from "../config.js";
 import { upsertProjectDocument, searchProjectMemory } from "../db/vector.js";
 import { syncFolderState } from "./driveTools.js";
 export async function saveAgentNote(content) {
-    const folderId = process.env.DOC_MCP_DRIVE_FOLDER_ID;
+    const folderId = config.DOC_MCP_DRIVE_FOLDER_ID;
     if (!folderId) {
-        return { success: false, error: "DOC_MCP_DRIVE_FOLDER_ID is not configured." };
+        return {
+            success: false,
+            error: "DOC_MCP_DRIVE_FOLDER_ID is not configured.",
+        };
     }
     try {
-        // We use folderId as the "projectId" parameter for vector-db namespace
         await upsertProjectDocument(folderId, content, {
             source: "agent",
         });
-        return { success: true, message: "Successfully stored note in vector memory." };
+        return {
+            success: true,
+            message: "Successfully stored note in vector memory.",
+        };
     }
     catch (err) {
         return { success: false, error: `Failed to store note: ${err.message}` };
     }
 }
 export async function searchKnowledge(query, topK = 3) {
-    const folderId = process.env.DOC_MCP_DRIVE_FOLDER_ID;
+    const folderId = config.DOC_MCP_DRIVE_FOLDER_ID;
     if (!folderId) {
-        return { success: false, error: "DOC_MCP_DRIVE_FOLDER_ID is not configured." };
+        return {
+            success: false,
+            error: "DOC_MCP_DRIVE_FOLDER_ID is not configured.",
+        };
     }
     try {
         // Auto-sync folder state before searching
@@ -30,7 +39,18 @@ export async function searchKnowledge(query, topK = 3) {
         }
         return {
             success: true,
-            results: results.map((r) => r.text).join("\n\n---\n\n"),
+            results: results.map((r) => {
+                let title = "Unknown Source";
+                if (r.metadata) {
+                    try {
+                        const metaObj = JSON.parse(r.metadata);
+                        if (metaObj.title)
+                            title = metaObj.title;
+                    }
+                    catch (e) { }
+                }
+                return `[File: ${title} | File ID: ${r.file_id || 'N/A'}]\n${r.text}`;
+            }).join("\n\n---\n\n"),
         };
     }
     catch (err) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@khoinguyen2002/doc-mcp",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "type": "module",
   "main": "dist/index.js",
   "bin": {

package/src/db/vector.ts

@@ -70,7 +70,7 @@ export async function embedText(text: string): Promise<number[]> {
   return json.data[0].embedding;
 }
 
-export async function upsertProjectDocument(projectId: string, text: string, metadata: Record<string, any> = {}): Promise<void> {
+export async function upsertProjectDocument(folderId: string, text: string, metadata: Record<string, any> = {}): Promise<void> {
   await initVectorDB();
   if (!client) throw new Error("Qdrant not initialized");
 
@@ -83,7 +83,7 @@ export async function upsertProjectDocument(projectId: string, text: string, met
         id: uuidv4(),
         vector: vector,
         payload: {
-          projectId,
+          folderId,
           text,
           source: metadata.source || "user",
           file_id: metadata.file_id || null,
@@ -95,10 +95,10 @@ export async function upsertProjectDocument(projectId: string, text: string, met
     ]
   });
 
-  console.error(`Upserted document chunk for project ${projectId}`);
+  console.error(`Upserted document chunk for folder ${folderId}`);
 }
 
-export async function searchProjectMemory(projectId: string, query: string, topK: number = 3): Promise<any[]> {
+export async function searchProjectMemory(folderId: string, query: string, topK: number = 3): Promise<any[]> {
   await initVectorDB();
   if (!client) throw new Error("Qdrant not initialized");
 
@@ -111,9 +111,9 @@ export async function searchProjectMemory(projectId: string, query: string, topK
     filter: {
       must: [
         {
-          key: "projectId",
+          key: "folderId",
           match: {
-            value: projectId
+            value: folderId
           }
         }
       ]
@@ -128,29 +128,29 @@ export async function searchProjectMemory(projectId: string, query: string, topK
   }));
 }
 
-export async function deleteProjectDocument(projectId: string, fileId: string): Promise<void> {
+export async function deleteProjectDocument(folderId: string, fileId: string): Promise<void> {
   await initVectorDB();
   if (!client) return;
 
   await client.delete(COLLECTION_NAME, {
     filter: {
       must: [
-        { key: "projectId", match: { value: projectId } },
+        { key: "folderId", match: { value: folderId } },
         { key: "file_id", match: { value: fileId } }
       ]
     }
   });
-  console.error(`Deleted old chunks from Qdrant for ${projectId} / ${fileId}`);
+  console.error(`Deleted old chunks from Qdrant for ${folderId} / ${fileId}`);
 }
 
-export async function checkProjectDocumentExists(projectId: string, fileId: string): Promise<boolean> {
+export async function checkProjectDocumentExists(folderId: string, fileId: string): Promise<boolean> {
   await initVectorDB();
   if (!client) return false;
 
   const res = await client.count(COLLECTION_NAME, {
     filter: {
       must: [
-        { key: "projectId", match: { value: projectId } },
+        { key: "folderId", match: { value: folderId } },
         { key: "file_id", match: { value: fileId } }
       ]
     }
@@ -158,14 +158,14 @@ export async function checkProjectDocumentExists(projectId: string, fileId: stri
   return res.count > 0;
 }
 
-export async function getProjectDocumentMetadata(projectId: string): Promise<Record<string, string>> {
+export async function getProjectDocumentMetadata(folderId: string): Promise<Record<string, string>> {
   await initVectorDB();
   if (!client) return {};
 
   const res = await client.scroll(COLLECTION_NAME, {
     filter: {
       must: [
-        { key: "projectId", match: { value: projectId } },
+        { key: "folderId", match: { value: folderId } },
         { key: "source", match: { value: "google_drive" } }
       ]
     },

package/src/mcp-server.ts

@@ -27,16 +27,20 @@ const server = new McpServer({
 server.registerTool(
   "list_drive_files",
   {
-    description: "List and search for Google Drive documents in the configured folder.",
+    description: "List and search for Google Drive documents and subfolders in a specific folder.",
     inputSchema: {
       keyword: z
         .string()
         .optional()
         .describe("Optional keyword to search for in document titles"),
+      targetFolderId: z
+        .string()
+        .optional()
+        .describe("Optional Google Drive folder ID to list contents from. Defaults to the root knowledge folder."),
     },
   },
-  async ({ keyword }) => {
-    const res = await listDriveFiles(keyword);
+  async ({ keyword, targetFolderId }) => {
+    const res = await listDriveFiles(keyword, targetFolderId);
     if (!res.success) {
       return {
         content: [{ type: "text", text: `Error: ${res.error}` }],

package/src/tools/driveTools.ts

@@ -31,8 +31,8 @@ function getDriveClient() {
   return google.drive({ version: "v3", auth });
 }
 
-export async function listDriveFiles(keyword?: string) {
-  const folderId = process.env.DOC_MCP_DRIVE_FOLDER_ID;
+export async function listDriveFiles(keyword?: string, targetFolderId?: string) {
+  const folderId = targetFolderId || config.DOC_MCP_DRIVE_FOLDER_ID;
   if (!folderId) {
     return {
       success: false,
@@ -42,7 +42,7 @@ export async function listDriveFiles(keyword?: string) {
 
   try {
     const drive = getDriveClient();
-    let q = "mimeType = 'application/vnd.google-apps.document'";
+    let q = "(mimeType = 'application/vnd.google-apps.document' or mimeType = 'application/vnd.google-apps.folder') and trashed = false";
     q = `'${folderId}' in parents and ${q}`;
 
     if (keyword) {
@@ -51,7 +51,7 @@ export async function listDriveFiles(keyword?: string) {
 
     const res = await drive.files.list({
       q,
-      fields: "files(id, name, description)",
+      fields: "files(id, name, description, mimeType)",
       spaces: "drive",
       pageSize: 50,
       supportsAllDrives: true,
@@ -112,12 +112,12 @@ export async function syncSingleDocument(fileId: string, folderId: string) {
     }
     return { synced: true, content, driveModifiedTime };
   }
-  
+
   return { synced: false, driveModifiedTime };
 }
 
 export async function readDriveDocument(fileId: string) {
-  const folderId = process.env.DOC_MCP_DRIVE_FOLDER_ID;
+  const folderId = config.DOC_MCP_DRIVE_FOLDER_ID;
   if (!folderId) {
     return {
       success: false,
@@ -127,7 +127,7 @@ export async function readDriveDocument(fileId: string) {
 
   try {
     const result = await syncSingleDocument(fileId, folderId);
-    
+
     // If not synced just now, we need to fetch content to return to the user
     let content = result.content;
     if (!content) {
@@ -159,19 +159,33 @@ export async function readDriveDocument(fileId: string) {
 export async function syncFolderState(folderId: string) {
   try {
     const drive = getDriveClient();
-    let q = "mimeType = 'application/vnd.google-apps.document'";
-    q = `'${folderId}' in parents and ${q}`;
 
-    const res = await drive.files.list({
-      q,
-      fields: "files(id, name, modifiedTime)",
-      spaces: "drive",
-      pageSize: 100,
-      supportsAllDrives: true,
-      includeItemsFromAllDrives: true,
-    });
+    async function getAllDocumentsFlat(): Promise<any[]> {
+      let allDocs: any[] = [];
+      let pageToken: string | undefined = undefined;
+
+      do {
+        const docsRes: any = await drive.files.list({
+          // Chú ý: Đéo check parentId nữa, gom sạch sành sanh mọi file .doc mà Service Account nhìn thấy
+          q: `mimeType = 'application/vnd.google-apps.document' and trashed = false`,
+          fields: "nextPageToken, files(id, name, modifiedTime)",
+          spaces: "drive",
+          pageSize: 100, // Google API limit mỗi page, tự động nhảy trang nếu nhiều hơn
+          pageToken,
+          supportsAllDrives: true,
+          includeItemsFromAllDrives: true,
+        });
+        
+        if (docsRes.data.files) {
+          allDocs = allDocs.concat(docsRes.data.files);
+        }
+        pageToken = docsRes.data.nextPageToken || undefined;
+      } while (pageToken);
+
+      return allDocs;
+    }
 
-    const driveFiles = res.data.files || [];
+    const driveFiles = await getAllDocumentsFlat();
     const dbMetaMap = await getProjectDocumentMetadata(folderId);
 
     // Sync updated or new files
@@ -185,7 +199,7 @@ export async function syncFolderState(folderId: string) {
 
     // Delete removed files from DB
     for (const dbFileId of Object.keys(dbMetaMap)) {
-      if (!driveFiles.find(f => f.id === dbFileId)) {
+      if (!driveFiles.find((f) => f.id === dbFileId)) {
         await deleteProjectDocument(folderId, dbFileId);
       }
     }

package/src/tools/knowledgeTools.ts

@@ -1,27 +1,35 @@
+import { config } from "../config.js";
 import { upsertProjectDocument, searchProjectMemory } from "../db/vector.js";
 import { syncFolderState } from "./driveTools.js";
 
 export async function saveAgentNote(content: string) {
-  const folderId = process.env.DOC_MCP_DRIVE_FOLDER_ID;
+  const folderId = config.DOC_MCP_DRIVE_FOLDER_ID;
   if (!folderId) {
-    return { success: false, error: "DOC_MCP_DRIVE_FOLDER_ID is not configured." };
+    return {
+      success: false,
+      error: "DOC_MCP_DRIVE_FOLDER_ID is not configured.",
+    };
   }
-
   try {
-    // We use folderId as the "projectId" parameter for vector-db namespace
     await upsertProjectDocument(folderId, content, {
       source: "agent",
     });
-    return { success: true, message: "Successfully stored note in vector memory." };
+    return {
+      success: true,
+      message: "Successfully stored note in vector memory.",
+    };
   } catch (err: any) {
     return { success: false, error: `Failed to store note: ${err.message}` };
   }
 }
 
 export async function searchKnowledge(query: string, topK: number = 3) {
-  const folderId = process.env.DOC_MCP_DRIVE_FOLDER_ID;
+  const folderId = config.DOC_MCP_DRIVE_FOLDER_ID;
   if (!folderId) {
-    return { success: false, error: "DOC_MCP_DRIVE_FOLDER_ID is not configured." };
+    return {
+      success: false,
+      error: "DOC_MCP_DRIVE_FOLDER_ID is not configured.",
+    };
   }
 
   try {
@@ -36,7 +44,16 @@ export async function searchKnowledge(query: string, topK: number = 3) {
 
     return {
       success: true,
-      results: results.map((r: any) => r.text).join("\n\n---\n\n"),
+      results: results.map((r: any) => {
+        let title = "Unknown Source";
+        if (r.metadata) {
+          try {
+            const metaObj = JSON.parse(r.metadata);
+            if (metaObj.title) title = metaObj.title;
+          } catch (e) {}
+        }
+        return `[File: ${title} | File ID: ${r.file_id || 'N/A'}]\n${r.text}`;
+      }).join("\n\n---\n\n"),
     };
   } catch (err: any) {
     return { success: false, error: `Failed to search: ${err.message}` };

package/src/hooks/driveSync.ts

@@ -1,108 +0,0 @@
-import {
-  getProjectDocumentMetadata,
-  deleteProjectDocument,
-  upsertProjectDocument,
-} from "../db/vector.js";
-import { RecursiveCharacterTextSplitter } from "@langchain/textsplitters";
-// Remove childLogger dependency to fully decouple from core
-// import { childLogger } from "@workspace/core";
-import { config } from "../config.js";
-
-const log = {
-  info: (obj: any, msg: string) => console.log(`[driveSync] ${msg}`, obj),
-  error: (obj: any, msg: string) => console.error(`[driveSync] ${msg}`, obj)
-};
-
-export async function syncProjectDriveFiles(
-  projectId: string,
-  onSyncMessage?: (msg: string) => void,
-): Promise<void> {
-  const dbMeta = await getProjectDocumentMetadata(projectId);
-  const fileIds = Object.keys(dbMeta);
-
-  if (fileIds.length === 0) return;
-
-  log.info({ projectId, fileCount: fileIds.length }, "Checking Drive files for updates...");
-
-  const { google } = await import("googleapis");
-
-  const clientEmail = config.DOC_MCP_GOOGLE_CLIENT_EMAIL;
-  let privateKey = config.DOC_MCP_GOOGLE_PRIVATE_KEY;
-  if (!clientEmail || !privateKey) return;
-  if (privateKey.startsWith('"') && privateKey.endsWith('"')) {
-    privateKey = privateKey.slice(1, -1);
-  }
-  privateKey = privateKey.replace(/\\n/g, "\n");
-
-  const auth = new google.auth.JWT({
-    email: clientEmail,
-    key: privateKey,
-    scopes: ["https://www.googleapis.com/auth/drive.readonly"],
-  });
-
-  const drive = google.drive({ version: "v3", auth });
-
-  let updatedCount = 0;
-  for (const fileId of fileIds) {
-    try {
-      const fileInfo = await drive.files.get({
-        fileId: fileId,
-        fields: "id, name, modifiedTime, trashed",
-        supportsAllDrives: true,
-      });
-
-      if (fileInfo.data.trashed) {
-        log.info({ fileId, projectId }, "File trashed on Drive, deleting from VectorDB...");
-        await deleteProjectDocument(projectId, fileId);
-        continue;
-      }
-
-      const driveModifiedTime = fileInfo.data.modifiedTime || "";
-      const dbModifiedTime = dbMeta[fileId];
-
-      if (driveModifiedTime !== dbModifiedTime) {
-        if (onSyncMessage) {
-          onSyncMessage(`🔄 Syncing updated file: ${fileInfo.data.name}...`);
-        }
-        
-        log.info({ fileId, projectId }, "File updated on Drive, syncing...");
-        await deleteProjectDocument(projectId, fileId);
-        
-        const res = await drive.files.export({
-          fileId: fileId,
-          mimeType: "text/plain",
-        });
-        
-        const content = res.data;
-        if (typeof content === "string" && content.trim() !== "") {
-          const splitter = new RecursiveCharacterTextSplitter({
-            chunkSize: config.CHUNK_SIZE,
-            chunkOverlap: config.CHUNK_OVERLAP,
-          });
-          const chunks = await splitter.splitText(content);
-          
-          for (const chunk of chunks) {
-            await upsertProjectDocument(projectId, chunk, {
-              title: fileInfo.data.name,
-              file_id: fileId,
-              source: "google_drive",
-              modified_time: driveModifiedTime,
-            });
-          }
-          updatedCount++;
-        }
-      }
-    } catch (err: any) {
-      if (err.code === 404) {
-        log.info({ fileId, projectId }, "File not found on Drive, deleting from VectorDB...");
-        await deleteProjectDocument(projectId, fileId);
-      } else {
-        log.error({ fileId, err: err.message }, "Error syncing drive file");
-      }
-    }
-  }
-
-  if (updatedCount > 0 && onSyncMessage) {
-    onSyncMessage(`✅ Synced ${updatedCount} files from Google Drive.`);
-  }
-}