fix: add support for auto-returning download URLs for specific endpoints helping agent not sending whole file to LLM (#142)

Author: eirikb

src/endpoints.json

@@ -207,7 +207,8 @@
     "pathPattern": "/drives/{drive-id}/items/{driveItem-id}/content",
     "method": "get",
     "toolName": "download-onedrive-file-content",
-    "scopes": ["Files.Read"]
+    "scopes": ["Files.Read"],
+    "returnDownloadUrl": true
   },
   {
     "pathPattern": "/drives/{drive-id}/items/{driveItem-id}",
@@ -278,7 +279,8 @@
     "pathPattern": "/me/onenote/pages/{onenotePage-id}/content",
     "method": "get",
     "toolName": "get-onenote-page-content",
-    "scopes": ["Notes.Read"]
+    "scopes": ["Notes.Read"],
+    "returnDownloadUrl": true
   },
   {
     "pathPattern": "/me/onenote/pages",

src/graph-tools.ts

@@ -16,6 +16,7 @@ interface EndpointConfig {
   toolName: string;
   scopes?: string[];
   workScopes?: string[];
+  returnDownloadUrl?: boolean;
 }
 
 const endpointsData = JSON.parse(
@@ -261,7 +262,12 @@ export function registerGraphTools(
             tool.errors?.some((error) => error.description === 'Retrieved media content') ||
             path.endsWith('/content');
 
-          if (isProbablyMediaContent) {
+          if (endpointConfig?.returnDownloadUrl && path.endsWith('/content')) {
+            path = path.replace(/\/content$/, '');
+            logger.info(
+              `Auto-returning download URL for ${tool.alias} (returnDownloadUrl=true in endpoints.json)`
+            );
+          } else if (isProbablyMediaContent) {
             options.rawResponse = true;
           }