@credal/actions 0.2.61 → 0.2.64

package/dist/actions/providers/google-oauth/getDriveContentById.js

@@ -0,0 +1,161 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import pdf from "pdf-parse/lib/pdf-parse.js";
+import { axiosClient } from "../../util/axiosClient.js";
+import mammoth from "mammoth";
+import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
+const getDriveFileContentByID = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
+    if (!authParams.authToken) {
+        return { success: false, error: MISSING_AUTH_TOKEN };
+    }
+    const { fileId, limit } = params;
+    try {
+        // First, get file metadata to determine the file type
+        const metadataUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?fields=name,mimeType,size`;
+        const metadataRes = yield axiosClient.get(metadataUrl, {
+            headers: {
+                Authorization: `Bearer ${authParams.authToken}`,
+            },
+        });
+        const { name: fileName, mimeType, size } = metadataRes.data;
+        // Check if file is too large (50MB limit for safety)
+        if (size && parseInt(size) > 50 * 1024 * 1024) {
+            return {
+                success: false,
+                error: "File too large (>50MB)",
+            };
+        }
+        let content = "";
+        // Handle different file types - read content directly
+        if (mimeType === "application/vnd.google-apps.document") {
+            // Google Docs - download as plain text
+            const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media&format=txt`;
+            const downloadRes = yield axiosClient.get(downloadUrl, {
+                headers: {
+                    Authorization: `Bearer ${authParams.authToken}`,
+                },
+                responseType: 'text',
+            });
+            content = downloadRes.data;
+        }
+        else if (mimeType === "application/vnd.google-apps.spreadsheet") {
+            // Google Sheets - download as CSV
+            const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media&format=csv`;
+            const downloadRes = yield axiosClient.get(downloadUrl, {
+                headers: {
+                    Authorization: `Bearer ${authParams.authToken}`,
+                },
+                responseType: 'text',
+            });
+            content = downloadRes.data;
+        }
+        else if (mimeType === "application/vnd.google-apps.presentation") {
+            // Google Slides - download as plain text
+            const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media&format=txt`;
+            const downloadRes = yield axiosClient.get(downloadUrl, {
+                headers: {
+                    Authorization: `Bearer ${authParams.authToken}`,
+                },
+                responseType: 'text',
+            });
+            content = downloadRes.data;
+        }
+        else if (mimeType === "application/pdf") {
+            // PDF files - use pdf-parse
+            const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media`;
+            const downloadRes = yield axiosClient.get(downloadUrl, {
+                headers: {
+                    Authorization: `Bearer ${authParams.authToken}`,
+                },
+                responseType: 'arraybuffer',
+            });
+            try {
+                const pdfData = yield pdf(downloadRes.data);
+                content = pdfData.text;
+            }
+            catch (pdfError) {
+                return {
+                    success: false,
+                    error: `Failed to parse PDF: ${pdfError instanceof Error ? pdfError.message : 'Unknown PDF error'}`,
+                };
+            }
+        }
+        else if (mimeType === "application/vnd.openxmlformats-officedocument.wordprocessingml.document" ||
+            mimeType === "application/msword") {
+            // Word documents (.docx or .doc) - download and extract text using mammoth
+            const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media`;
+            const downloadRes = yield axiosClient.get(downloadUrl, {
+                headers: {
+                    Authorization: `Bearer ${authParams.authToken}`,
+                },
+                responseType: 'arraybuffer',
+            });
+            try {
+                // mammoth works with .docx files. It will ignore formatting and return raw text
+                const result = yield mammoth.extractRawText({ buffer: Buffer.from(downloadRes.data) });
+                content = result.value; // raw text
+            }
+            catch (wordError) {
+                return {
+                    success: false,
+                    error: `Failed to parse Word document: ${wordError instanceof Error ? wordError.message : 'Unknown Word error'}`,
+                };
+            }
+        }
+        else if (mimeType === "text/plain" ||
+            mimeType === "text/html" ||
+            mimeType === "application/rtf" ||
+            (mimeType === null || mimeType === void 0 ? void 0 : mimeType.startsWith("text/"))) {
+            // Text-based files
+            const downloadUrl = `https://www.googleapis.com/drive/v3/files/${encodeURIComponent(fileId)}?alt=media`;
+            const downloadRes = yield axiosClient.get(downloadUrl, {
+                headers: {
+                    Authorization: `Bearer ${authParams.authToken}`,
+                },
+                responseType: 'text',
+            });
+            content = downloadRes.data;
+        }
+        else if (mimeType === null || mimeType === void 0 ? void 0 : mimeType.startsWith("image/")) {
+            // Skip images
+            return {
+                success: false,
+                error: "Image files are not supported for text extraction",
+            };
+        }
+        else {
+            // Unsupported file type
+            return {
+                success: false,
+                error: `Unsupported file type: ${mimeType}`,
+            };
+        }
+        content = content.trim();
+        const originalLength = content.length;
+        // Naive way to truncate content
+        if (limit && content.length > limit) {
+            content = content.substring(0, limit);
+        }
+        return {
+            success: true,
+            content,
+            fileName,
+            fileLength: originalLength,
+        };
+    }
+    catch (error) {
+        console.error("Error getting Google Drive file content", error);
+        return {
+            success: false,
+            error: error instanceof Error ? error.message : "Unknown error",
+        };
+    }
+});
+export default getDriveFileContentByID;

package/dist/actions/providers/google-oauth/getDriveFileContentById.js

@@ -7,121 +7,48 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import { createAxiosClientWithTimeout, isAxiosTimeoutError } from "../../util/axiosClient.js";
+import { createAxiosClientWithTimeout } from "../../util/axiosClient.js";
 import mammoth from "mammoth";
 import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
 import { extractTextFromPdf } from "../../../utils/pdf.js";
-import { parseGoogleDocFromRawContentToPlainText, parseGoogleSheetsFromRawContentToPlainText, parseGoogleSlidesFromRawContentToPlainText, } from "../../../utils/google.js";
+import { getGoogleDocContent, getGoogleSheetContent, getGoogleSlidesContent } from "../../../utils/google.js";
 const getDriveFileContentById = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
     if (!authParams.authToken) {
         return { success: false, error: MISSING_AUTH_TOKEN };
     }
     const BASE_URL = "https://www.googleapis.com/drive/v3/files/";
     const { fileId, limit } = params;
-    const axiosClient = createAxiosClientWithTimeout(20000);
+    const timeoutLimit = params.timeoutLimit !== undefined && params.timeoutLimit > 0 ? params.timeoutLimit * 1000 : 15000; // timeoutLimit is in seconds, converted to ms`
+    const axiosClient = createAxiosClientWithTimeout(timeoutLimit); // timeoutLimit is in seconds, converted to ms
     try {
         // First, get file metadata to determine the file type and if it's in a shared drive
         const metadataUrl = `${BASE_URL}${encodeURIComponent(fileId)}?fields=name,mimeType,size,driveId,parents&supportsAllDrives=true`;
         const metadataRes = yield axiosClient.get(metadataUrl, {
-            headers: {
-                Authorization: `Bearer ${authParams.authToken}`,
-            },
+            headers: { Authorization: `Bearer ${authParams.authToken}` },
         });
         const { name: fileName, mimeType, size, driveId } = metadataRes.data;
         // Check if file is too large (50MB limit for safety)
-        const maxFileSize = 50 * 1024 * 1024;
+        const maxMegabytes = params.fileSizeLimit && params.fileSizeLimit > 0 ? params.fileSizeLimit : 50;
+        const maxFileSize = maxMegabytes * 1024 * 1024;
         if (size && parseInt(size) > maxFileSize) {
             return {
                 success: false,
-                error: "File too large (>50MB)",
+                error: "File too large (" + maxMegabytes + "MB)",
             };
         }
         let content = "";
         const sharedDriveParams = driveId ? "&supportsAllDrives=true" : "";
         // Google Docs - use Google Docs API instead of Drive export
         if (mimeType === "application/vnd.google-apps.document") {
-            try {
-                const docsUrl = `https://docs.googleapis.com/v1/documents/${fileId}`;
-                const docsRes = yield axiosClient.get(docsUrl, {
-                    headers: {
-                        Authorization: `Bearer ${authParams.authToken}`,
-                    },
-                });
-                content = parseGoogleDocFromRawContentToPlainText(docsRes.data);
-            }
-            catch (docsError) {
-                if (isAxiosTimeoutError(docsError)) {
-                    console.log("Request timed out using Google Docs API - dont retry");
-                }
-                else {
-                    console.log("Error using Google Docs API", docsError);
-                    // Fallback to Drive API export if Docs API fails
-                    const exportUrl = `${BASE_URL}${encodeURIComponent(fileId)}/export?mimeType=text/plain${sharedDriveParams}`;
-                    const exportRes = yield axiosClient.get(exportUrl, {
-                        headers: {
-                            Authorization: `Bearer ${authParams.authToken}`,
-                        },
-                        responseType: "text",
-                    });
-                    content = exportRes.data;
-                }
-            }
+            content = yield getGoogleDocContent(fileId, authParams.authToken, axiosClient, sharedDriveParams);
         }
         // Google Sheets - use Google Sheets API instead of Drive export
         else if (mimeType === "application/vnd.google-apps.spreadsheet") {
-            try {
-                const sheetsUrl = `https://sheets.googleapis.com/v4/spreadsheets/${fileId}?includeGridData=true`;
-                const sheetsRes = yield axiosClient.get(sheetsUrl, {
-                    headers: {
-                        Authorization: `Bearer ${authParams.authToken}`,
-                    },
-                });
-                content = parseGoogleSheetsFromRawContentToPlainText(sheetsRes.data);
-            }
-            catch (sheetsError) {
-                if (isAxiosTimeoutError(sheetsError)) {
-                    console.log("Request timed out using Google Sheets API - dont retry");
-                }
-                else {
-                    console.log("Error fetching file", sheetsError);
-                    const exportUrl = `${BASE_URL}${encodeURIComponent(fileId)}/export?mimeType=text/csv${sharedDriveParams}`;
-                    const exportRes = yield axiosClient.get(exportUrl, {
-                        headers: { Authorization: `Bearer ${authParams.authToken}` },
-                        responseType: "text",
-                    });
-                    content = exportRes.data
-                        .split("\n")
-                        .map((line) => line.replace(/,+$/, ""))
-                        .map((line) => line.replace(/,{2,}/g, ","))
-                        .join("\n");
-                }
-            }
+            content = yield getGoogleSheetContent(fileId, authParams.authToken, axiosClient, sharedDriveParams);
         }
         // Google Slides - use Google Slides API instead of Drive export
         else if (mimeType === "application/vnd.google-apps.presentation") {
-            try {
-                const slidesUrl = `https://slides.googleapis.com/v1/presentations/${fileId}`;
-                const slidesRes = yield axiosClient.get(slidesUrl, {
-                    headers: {
-                        Authorization: `Bearer ${authParams.authToken}`,
-                    },
-                });
-                content = parseGoogleSlidesFromRawContentToPlainText(slidesRes.data);
-            }
-            catch (slidesError) {
-                if (isAxiosTimeoutError(slidesError)) {
-                    console.log("Request timed out using Google Slides API - dont retry");
-                }
-                else {
-                    console.log("Error using Google Slides API", slidesError);
-                    const exportUrl = `${BASE_URL}${encodeURIComponent(fileId)}/export?mimeType=text/plain${sharedDriveParams}`;
-                    const exportRes = yield axiosClient.get(exportUrl, {
-                        headers: { Authorization: `Bearer ${authParams.authToken}` },
-                        responseType: "text",
-                    });
-                    content = exportRes.data;
-                }
-            }
+            content = yield getGoogleSlidesContent(fileId, authParams.authToken, axiosClient, sharedDriveParams);
         }
         // PDF files - download and extract text using pdfjs-dist
         else if (mimeType === "application/pdf") {
@@ -145,9 +72,7 @@ const getDriveFileContentById = (_a) => __awaiter(void 0, [_a], void 0, function
             mimeType === "application/msword") {
             const downloadUrl = `${BASE_URL}${encodeURIComponent(fileId)}?alt=media${sharedDriveParams}`;
             const downloadRes = yield axiosClient.get(downloadUrl, {
-                headers: {
-                    Authorization: `Bearer ${authParams.authToken}`,
-                },
+                headers: { Authorization: `Bearer ${authParams.authToken}` },
                 responseType: "arraybuffer",
             });
             try {
@@ -169,9 +94,7 @@ const getDriveFileContentById = (_a) => __awaiter(void 0, [_a], void 0, function
             // Text-based files
             const downloadUrl = `${BASE_URL}${encodeURIComponent(fileId)}?alt=media${sharedDriveParams}`;
             const downloadRes = yield axiosClient.get(downloadUrl, {
-                headers: {
-                    Authorization: `Bearer ${authParams.authToken}`,
-                },
+                headers: { Authorization: `Bearer ${authParams.authToken}` },
                 responseType: "text",
             });
             content = downloadRes.data;

package/dist/actions/providers/google-oauth/searchAndGetDriveContentByKeywords.d.ts

@@ -0,0 +1,3 @@
+import type { googleOauthSearchDriveByKeywordsFunction } from "../../autogen/types.js";
+declare const searchDriveByKeywords: googleOauthSearchDriveByKeywordsFunction;
+export default searchDriveByKeywords;

package/dist/actions/providers/google-oauth/searchAndGetDriveContentByKeywords.js

@@ -0,0 +1,47 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { axiosClient } from "../../util/axiosClient.js";
+import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
+const searchDriveByKeywords = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
+    var _b;
+    if (!authParams.authToken) {
+        return { success: false, error: MISSING_AUTH_TOKEN, files: [] };
+    }
+    const { keywords, limit } = params;
+    // Build the query: fullText contains 'keyword1' or fullText contains 'keyword2' ...
+    const query = keywords.map(kw => `fullText contains '${kw.replace(/'/g, "\\'")}'`).join(" or ");
+    const url = `https://www.googleapis.com/drive/v3/files?q=${encodeURIComponent(query)}&fields=files(id,name,mimeType,webViewLink)&supportsAllDrives=true&includeItemsFromAllDrives=true`;
+    // 1. Get the file metadata from google drive search
+    let files = [];
+    try {
+        const res = yield axiosClient.get(url, {
+            headers: {
+                Authorization: `Bearer ${authParams.authToken}`,
+            },
+        });
+        files =
+            ((_b = res.data.files) === null || _b === void 0 ? void 0 : _b.map((file) => ({
+                id: file.id || "",
+                name: file.name || "",
+                mimeType: file.mimeType || "",
+                url: file.webViewLink || "",
+            }))) || [];
+    }
+    catch (error) {
+        console.error("Error searching Google Drive", error);
+        return {
+            success: false,
+            error: error instanceof Error ? error.message : "Unknown error",
+            files: [],
+        };
+    }
+    files = limit ? files.splice(0, limit) : files;
+});
+export default searchDriveByKeywords;

package/dist/actions/providers/google-oauth/searchDriveAndGetContentByKeywords.d.ts

@@ -0,0 +1,3 @@
+import type { googleOauthSearchDriveAndGetContentByKeywordsFunction } from "../../autogen/types.js";
+declare const searchDriveAndGetContentByKeywords: googleOauthSearchDriveAndGetContentByKeywordsFunction;
+export default searchDriveAndGetContentByKeywords;

package/dist/actions/providers/google-oauth/searchDriveAndGetContentByKeywords.js

@@ -0,0 +1,110 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { axiosClient } from "../../util/axiosClient.js";
+import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
+import extractContentFromDriveFileId from "./utils/extractContentFromDriveFileId.js";
+import { normalizeText } from "../../../utils/string.js";
+/** Intelligently selects a section of text around the median occurrence of keywords */
+const intelligentSelectByMedianSection = (text, keywords, limit) => {
+    if (!text || text.length <= limit)
+        return text;
+    if (!(keywords === null || keywords === void 0 ? void 0 : keywords.length))
+        return text.substring(0, limit);
+    // Find all keyword positions (case-insensitive, limited to first 1000 matches)
+    const positions = [];
+    const normalizedText = normalizeText(text);
+    for (const keyword of keywords) {
+        if (keyword.length < 3)
+            continue; // Skip very short keywords
+        let pos = -1;
+        const normalizedKeyword = normalizeText(keyword);
+        while ((pos = normalizedText.indexOf(normalizedKeyword, pos + 1)) !== -1 && positions.length < 1000) {
+            positions.push(pos);
+        }
+    }
+    if (!positions.length)
+        return text.substring(0, limit);
+    // Find median position (using sort for simplicity, still fast for 1000 elements)
+    positions.sort((a, b) => a - b);
+    const medianPos = positions[Math.floor(positions.length / 2)];
+    // Return window around median
+    const half = Math.floor(limit / 2);
+    const start = Math.max(0, medianPos - half);
+    const end = Math.min(text.length, start + limit);
+    return text.substring(start, end);
+};
+const searchDriveAndGetContentByKeywords = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
+    var _b;
+    if (!authParams.authToken) {
+        return { success: false, error: MISSING_AUTH_TOKEN, files: [] };
+    }
+    const { keywords, fileLimit, fileSizeLimit } = params;
+    let files = [];
+    // 1. Search for files and get their metadata
+    // Build the query: fullText contains 'keyword1' or fullText contains 'keyword2' ...
+    const query = keywords.map(kw => `fullText contains '${kw.replace(/'/g, "\\'")}'`).join(" or ");
+    const url = `https://www.googleapis.com/drive/v3/files?q=${encodeURIComponent(query)}&fields=files(id,name,mimeType,webViewLink)&supportsAllDrives=true&includeItemsFromAllDrives=true`;
+    try {
+        const res = yield axiosClient.get(url, {
+            headers: {
+                Authorization: `Bearer ${authParams.authToken}`,
+            },
+        });
+        files =
+            ((_b = res.data.files) === null || _b === void 0 ? void 0 : _b.map((file) => ({
+                id: file.id,
+                name: file.name,
+                mimeType: file.mimeType,
+                url: file.webViewLink,
+            }))) || [];
+    }
+    catch (error) {
+        console.error("Error searching Google Drive", error);
+        return {
+            success: false,
+            error: error instanceof Error ? error.message : "Unknown error",
+            files: [],
+        };
+    }
+    files = fileLimit ? files.splice(0, fileLimit) : files;
+    // 2. Extract content from files and do some smart range selection
+    const processedFiles = yield Promise.all(files
+        .filter((file) => file.id && file.mimeType)
+        .map((file) => __awaiter(void 0, void 0, void 0, function* () {
+        const content = yield extractContentFromDriveFileId({
+            params: { fileId: file.id, mimeType: file.mimeType },
+            authParams,
+        });
+        if (content.success) {
+            let selectedContent = content.content;
+            if (fileSizeLimit && selectedContent && selectedContent.length > fileSizeLimit) {
+                selectedContent = intelligentSelectByMedianSection(selectedContent, keywords, fileSizeLimit);
+            }
+            return {
+                id: file.id || "",
+                name: file.name || "",
+                mimeType: file.mimeType || "",
+                url: file.url || "",
+                content: selectedContent,
+            };
+        }
+        else {
+            return {
+                id: file.id || "",
+                name: file.name || "",
+                mimeType: file.mimeType || "",
+                url: file.url || "",
+                error: content.error,
+            };
+        }
+    })));
+    return { success: true, files: processedFiles };
+});
+export default searchDriveAndGetContentByKeywords;

package/dist/actions/providers/google-oauth/searchDriveAndGetContentByQuery.d.ts

@@ -0,0 +1,3 @@
+import type { googleOauthSearchDriveAndGetContentByQueryFunction } from "../../autogen/types.js";
+declare const searchDriveAndGetContentByQuery: googleOauthSearchDriveAndGetContentByQueryFunction;
+export default searchDriveAndGetContentByQuery;

package/dist/actions/providers/google-oauth/searchDriveAndGetContentByQuery.js

@@ -0,0 +1,78 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { axiosClient } from "../../util/axiosClient.js";
+import { MISSING_AUTH_TOKEN } from "../../util/missingAuthConstants.js";
+import extractContentFromDriveFileId from "./utils/extractContentFromDriveFileId.js";
+const searchDriveAndGetContentByQuery = (_a) => __awaiter(void 0, [_a], void 0, function* ({ params, authParams, }) {
+    var _b;
+    if (!authParams.authToken) {
+        return { success: false, error: MISSING_AUTH_TOKEN, files: [] };
+    }
+    const { query, fileLimit, fileSizeLimit } = params;
+    let files = [];
+    // 1. Search for files and get their metadata
+    const url = `https://www.googleapis.com/drive/v3/files?q=${encodeURIComponent(query)}&fields=files(id,name,mimeType,webViewLink)&supportsAllDrives=true&includeItemsFromAllDrives=true&corpora=allDrives`;
+    try {
+        const res = yield axiosClient.get(url, {
+            headers: {
+                Authorization: `Bearer ${authParams.authToken}`,
+            },
+        });
+        files =
+            ((_b = res.data.files) === null || _b === void 0 ? void 0 : _b.map((file) => ({
+                id: file.id,
+                name: file.name,
+                mimeType: file.mimeType,
+                url: file.webViewLink,
+            }))) || [];
+    }
+    catch (error) {
+        console.error("Error searching Google Drive", error);
+        return {
+            success: false,
+            error: error instanceof Error ? error.message : "Unknown error",
+            files: [],
+        };
+    }
+    files = fileLimit ? files.splice(0, fileLimit) : files;
+    // 2. Extract content from files and do some smart range selection
+    const processedFiles = yield Promise.all(files
+        .filter((file) => file.id && file.mimeType)
+        .map((file) => __awaiter(void 0, void 0, void 0, function* () {
+        const content = yield extractContentFromDriveFileId({
+            params: { fileId: file.id, mimeType: file.mimeType },
+            authParams,
+        });
+        if (content.success) {
+            let selectedContent = content.content;
+            if (fileSizeLimit && selectedContent && selectedContent.length > fileSizeLimit) {
+                selectedContent = selectedContent.substring(0, fileSizeLimit);
+            }
+            return {
+                id: file.id || "",
+                name: file.name || "",
+                mimeType: file.mimeType || "",
+                url: file.url || "",
+                content: selectedContent,
+            };
+        }
+        else {
+            return {
+                id: file.id || "",
+                name: file.name || "",
+                mimeType: file.mimeType || "",
+                url: file.url || "",
+                error: content.error,
+            };
+        }
+    })));
+    return { success: true, files: processedFiles };
+});
+export default searchDriveAndGetContentByQuery;

package/dist/actions/providers/google-oauth/utils/extractContentFromDriveFileId.d.ts

@@ -0,0 +1,15 @@
+import type { AuthParamsType } from "../../../autogen/types.js";
+export type getDriveFileContentParams = {
+    fileId: string;
+    mimeType: string;
+};
+export type getDriveFileContentOutput = {
+    success: boolean;
+    content?: string;
+    error?: string;
+};
+declare const extractContentFromDriveFileId: ({ params, authParams, }: {
+    params: getDriveFileContentParams;
+    authParams: AuthParamsType;
+}) => Promise<getDriveFileContentOutput>;
+export default extractContentFromDriveFileId;