@khoinguyen2002/doc-mcp 1.0.1 → 1.0.3

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,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"}
+{"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,CAiC3G;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

@@ -24,7 +24,7 @@ export async function initVectorDB() {
                 },
             });
             await client.createPayloadIndex(COLLECTION_NAME, {
-                field_name: "projectId",
+                field_name: "folderId",
                 field_schema: "keyword",
             });
             await client.createPayloadIndex(COLLECTION_NAME, {
@@ -90,28 +90,34 @@ export async function searchProjectMemory(folderId, query, topK = 3) {
     await initVectorDB();
     if (!client)
         throw new Error("Qdrant not initialized");
-    const queryVector = await embedText(query);
-    const results = await client.search(COLLECTION_NAME, {
-        vector: queryVector,
-        limit: topK,
-        with_payload: true,
-        filter: {
-            must: [
-                {
-                    key: "folderId",
-                    match: {
-                        value: folderId
+    try {
+        const queryVector = await embedText(query);
+        const results = await client.search(COLLECTION_NAME, {
+            vector: queryVector,
+            limit: topK,
+            with_payload: true,
+            filter: {
+                must: [
+                    {
+                        key: "folderId",
+                        match: {
+                            value: folderId
+                        }
                     }
-                }
-            ]
-        }
-    });
-    // Map to match LanceDB format expected by other tools
-    return results.map(r => ({
-        id: r.id,
-        vector: r.vector,
-        ...r.payload
-    }));
+                ]
+            }
+        });
+        // Map to match LanceDB format expected by other tools
+        return results.map(r => ({
+            id: r.id,
+            vector: r.vector,
+            ...r.payload
+        }));
+    }
+    catch (err) {
+        console.error("Qdrant search error:", err.message);
+        return [];
+    }
 }
 export async function deleteProjectDocument(folderId, fileId) {
     await initVectorDB();

package/dist/mcp-server.js

@@ -43,9 +43,11 @@ server.registerTool("read_drive_document", {
     description: "Read the content of a specific Google Drive document. The document will also be automatically ingested into vector memory for future semantic search.",
     inputSchema: {
         fileId: z.string().describe("The Google Drive file ID to read"),
+        offset: z.number().optional().describe("Starting character index (default: 0)"),
+        limit: z.number().optional().describe("Maximum number of characters to return (default: 10000)"),
     },
-}, async ({ fileId }) => {
-    const res = await readDriveDocument(fileId);
+}, async ({ fileId, offset, limit }) => {
+    const res = await readDriveDocument(fileId, offset, limit);
     if (!res.success) {
         return {
             content: [{ type: "text", text: `Error: ${res.error}` }],
@@ -53,7 +55,7 @@ server.registerTool("read_drive_document", {
         };
     }
     return {
-        content: [{ type: "text", text: res.content || "No content found." }],
+        content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }],
     };
 });
 server.registerTool("save_agent_note", {

package/dist/tools/driveTools.d.ts

@@ -16,14 +16,23 @@ export declare function syncSingleDocument(fileId: string, folderId: string): Pr
     driveModifiedTime: string;
     content?: undefined;
 }>;
-export declare function readDriveDocument(fileId: string): Promise<{
+export declare function readDriveDocument(fileId: string, offset?: number, limit?: number): Promise<{
     success: boolean;
-    content: string;
+    data: {
+        content: string;
+        metadata: {
+            totalSize: number;
+            offset: number;
+            limit: number;
+            isTruncated: boolean;
+            warning: string | undefined;
+        };
+    };
     error?: undefined;
 } | {
     success: boolean;
     error: any;
-    content?: undefined;
+    data?: undefined;
 }>;
 export declare function syncFolderState(folderId: string): Promise<{
     success: boolean;

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,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"}
+{"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,EAAE,MAAM,GAAE,MAAU,EAAE,KAAK,GAAE,MAAc;;;;;;;;;;;;;;;;;GAsDhG;AAED,wBAAsB,eAAe,CAAC,QAAQ,EAAE,MAAM;;;;;;GAqDrD"}

package/dist/tools/driveTools.js

@@ -91,7 +91,7 @@ export async function syncSingleDocument(fileId, folderId) {
     }
     return { synced: false, driveModifiedTime };
 }
-export async function readDriveDocument(fileId) {
+export async function readDriveDocument(fileId, offset = 0, limit = 10000) {
     const folderId = config.DOC_MCP_DRIVE_FOLDER_ID;
     if (!folderId) {
         return {
@@ -112,15 +112,28 @@ export async function readDriveDocument(fileId) {
             content = typeof res.data === "string" ? res.data : "";
         }
         let finalContent = content;
-        const MAX_CHARS = 10000;
-        if (finalContent && finalContent.length > MAX_CHARS) {
-            finalContent =
-                finalContent.substring(0, MAX_CHARS) +
-                    "\n\n... [Content truncated due to length. The full document has been automatically ingested into Vector Memory. Use search_knowledge to query specific details.]";
+        const totalSize = finalContent ? finalContent.length : 0;
+        if (finalContent) {
+            finalContent = finalContent.substring(offset, offset + limit);
+        }
+        const isTruncated = offset + (finalContent?.length || 0) < totalSize;
+        let warning = undefined;
+        if (isTruncated) {
+            warning = `[WARNING]: This is not the entire document. Content has been truncated from character ${offset} to ${offset + finalContent.length} out of ${totalSize} total characters. Please use 'offset' and 'limit' parameters to read the rest of the document, or use search_knowledge to query specific details.`;
+            finalContent += `\n\n${warning}`;
         }
         return {
             success: true,
-            content: finalContent || "Empty file",
+            data: {
+                content: finalContent || "Empty file",
+                metadata: {
+                    totalSize,
+                    offset,
+                    limit,
+                    isTruncated,
+                    warning,
+                },
+            },
         };
     }
     catch (err) {

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

@@ -1 +1 @@
-{"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"}
+{"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;;;;;;;;GAqCpE"}

package/dist/tools/knowledgeTools.js

@@ -39,7 +39,8 @@ export async function searchKnowledge(query, topK = 3) {
         }
         return {
             success: true,
-            results: results.map((r) => {
+            results: results
+                .map((r) => {
                 let title = "Unknown Source";
                 if (r.metadata) {
                     try {
@@ -49,8 +50,9 @@ export async function searchKnowledge(query, topK = 3) {
                     }
                     catch (e) { }
                 }
-                return `[File: ${title} | File ID: ${r.file_id || 'N/A'}]\n${r.text}`;
-            }).join("\n\n---\n\n"),
+                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.1",
+  "version": "1.0.3",
   "type": "module",
   "main": "dist/index.js",
   "bin": {

package/src/db/vector.ts

@@ -28,7 +28,7 @@ export async function initVectorDB() {
         },
       });
       await client.createPayloadIndex(COLLECTION_NAME, {
-        field_name: "projectId",
+        field_name: "folderId",
         field_schema: "keyword",
       });
       await client.createPayloadIndex(COLLECTION_NAME, {
@@ -102,30 +102,35 @@ export async function searchProjectMemory(folderId: string, query: string, topK:
   await initVectorDB();
   if (!client) throw new Error("Qdrant not initialized");
 
-  const queryVector = await embedText(query);
-
-  const results = await client.search(COLLECTION_NAME, {
-    vector: queryVector,
-    limit: topK,
-    with_payload: true,
-    filter: {
-      must: [
-        {
-          key: "folderId",
-          match: {
-            value: folderId
+  try {
+    const queryVector = await embedText(query);
+
+    const results = await client.search(COLLECTION_NAME, {
+      vector: queryVector,
+      limit: topK,
+      with_payload: true,
+      filter: {
+        must: [
+          {
+            key: "folderId",
+            match: {
+              value: folderId
+            }
           }
-        }
-      ]
-    }
-  });
+        ]
+      }
+    });
 
-  // Map to match LanceDB format expected by other tools
-  return results.map(r => ({
-    id: r.id,
-    vector: r.vector,
-    ...r.payload
-  }));
+    // Map to match LanceDB format expected by other tools
+    return results.map(r => ({
+      id: r.id,
+      vector: r.vector,
+      ...r.payload
+    }));
+  } catch (err: any) {
+    console.error("Qdrant search error:", err.message);
+    return [];
+  }
 }
 
 export async function deleteProjectDocument(folderId: string, fileId: string): Promise<void> {

package/src/mcp-server.ts

@@ -60,10 +60,12 @@ server.registerTool(
       "Read the content of a specific Google Drive document. The document will also be automatically ingested into vector memory for future semantic search.",
     inputSchema: {
       fileId: z.string().describe("The Google Drive file ID to read"),
+      offset: z.number().optional().describe("Starting character index (default: 0)"),
+      limit: z.number().optional().describe("Maximum number of characters to return (default: 10000)"),
     },
   },
-  async ({ fileId }) => {
-    const res = await readDriveDocument(fileId);
+  async ({ fileId, offset, limit }) => {
+    const res = await readDriveDocument(fileId, offset, limit);
     if (!res.success) {
       return {
         content: [{ type: "text", text: `Error: ${res.error}` }],
@@ -71,7 +73,7 @@ server.registerTool(
       };
     }
     return {
-      content: [{ type: "text", text: res.content || "No content found." }],
+      content: [{ type: "text", text: JSON.stringify(res.data, null, 2) }],
     };
   },
 );

package/src/tools/driveTools.ts

@@ -116,7 +116,7 @@ export async function syncSingleDocument(fileId: string, folderId: string) {
   return { synced: false, driveModifiedTime };
 }
 
-export async function readDriveDocument(fileId: string) {
+export async function readDriveDocument(fileId: string, offset: number = 0, limit: number = 10000) {
   const folderId = config.DOC_MCP_DRIVE_FOLDER_ID;
   if (!folderId) {
     return {
@@ -140,16 +140,32 @@ export async function readDriveDocument(fileId: string) {
     }
 
     let finalContent = content;
-    const MAX_CHARS = 10000;
-    if (finalContent && finalContent.length > MAX_CHARS) {
-      finalContent =
-        finalContent.substring(0, MAX_CHARS) +
-        "\n\n... [Content truncated due to length. The full document has been automatically ingested into Vector Memory. Use search_knowledge to query specific details.]";
+    const totalSize = finalContent ? finalContent.length : 0;
+
+    if (finalContent) {
+      finalContent = finalContent.substring(offset, offset + limit);
+    }
+
+    const isTruncated = offset + (finalContent?.length || 0) < totalSize;
+    let warning = undefined;
+
+    if (isTruncated) {
+      warning = `[WARNING]: This is not the entire document. Content has been truncated from character ${offset} to ${offset + finalContent!.length} out of ${totalSize} total characters. Please use 'offset' and 'limit' parameters to read the rest of the document, or use search_knowledge to query specific details.`;
+      finalContent += `\n\n${warning}`;
     }
 
     return {
       success: true,
-      content: finalContent || "Empty file",
+      data: {
+        content: finalContent || "Empty file",
+        metadata: {
+          totalSize,
+          offset,
+          limit,
+          isTruncated,
+          warning,
+        },
+      },
     };
   } catch (err: any) {
     return { success: false, error: err.message };

package/src/tools/knowledgeTools.ts

@@ -44,16 +44,18 @@ export async function searchKnowledge(query: string, topK: number = 3) {
 
     return {
       success: true,
-      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"),
+      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}` };