Updated GDrive

modelcontextprotocol · Nov 20, 2024 · dc455ee · dc455ee
1 parent 0c5ec31
commit dc455ee
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/gdrive/README.md b/src/gdrive/README.md
@@ -15,7 +15,7 @@ This MCP server integrates with Google Drive to allow listing, reading, and sear
 
 The server provides access to Google Drive files:
 
-- **Files** (`gdrive://<file_id>`)
+- **Files** (`gdrive:///<file_id>`)
   - Supports all file types
   - Google Workspace files are automatically exported:
     - Docs → Markdown

diff --git a/src/gdrive/index.ts b/src/gdrive/index.ts
@@ -44,7 +44,7 @@ server.setRequestHandler(ListResourcesRequestSchema, async (request) => {
 
   return {
     resources: files.map((file) => ({
-      uri: `gdrive://${file.id}`,
+      uri: `gdrive:///${file.id}`,
       mimeType: file.mimeType,
       name: file.name,
     })),
@@ -53,7 +53,7 @@ server.setRequestHandler(ListResourcesRequestSchema, async (request) => {
 });
 
 server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
-  const fileId = request.params.uri.replace("gdrive://", "");
+  const fileId = request.params.uri.replace("gdrive:///", "");
 
   // First get file metadata to check mime type
   const file = await drive.files.get({
@@ -150,7 +150,7 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
 server.setRequestHandler(CallToolRequestSchema, async (request) => {
   if (request.params.name === "search") {
     const userQuery = request.params.arguments?.query as string;
-    const escapedQuery = userQuery.replace(/'/g, "\\'");
+    const escapedQuery = userQuery.replace(/\\/g, "\\\\").replace(/'/g, "\\'");
     const formattedQuery = `fullText contains '${escapedQuery}'`;
 
     const res = await drive.files.list({