feat: add tool getPageTextContent

mlc-ai · Oct 23, 2024 · ea06237 · ea06237
1 parent 723e8d5
commit ea06237
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 7 deletions.
diff --git a/src/action.ts b/src/action.ts
@@ -4,7 +4,7 @@ import { Tool } from "./tool";
 
 export const replaceSelectedText = (
   state: State,
-  parameters: { newText: string | string[] }
+  parameters: { newText: string | string[] },
 ): void => {
   const { newText } = parameters;
   const selection = state.currentSelection;
@@ -17,7 +17,7 @@ export const replaceSelectedText = (
     if (Array.isArray(newText)) {
       const fragment = document.createDocumentFragment();
       newText.forEach((text) =>
-        fragment.appendChild(document.createTextNode(text))
+        fragment.appendChild(document.createTextNode(text)),
       );
       range.insertNode(fragment);
     } else {
@@ -29,7 +29,7 @@ export const replaceSelectedText = (
 
 export const appendTextToDocument = (
   state: State,
-  parameters: { text: string }
+  parameters: { text: string },
 ): void => {
   if (window.location.hostname.includes("overleaf.com")) {
     const { text } = parameters;
@@ -59,7 +59,7 @@ export async function createGoogleCalendarEvent(
     startDateTime: string;
     endDateTime: string;
     timeZone?: string;
-  }
+  },
 ) {
   let { token } = parameters;
 
@@ -74,7 +74,7 @@ export async function createGoogleCalendarEvent(
     } catch (e) {
       throw new Error(
         "createGoogleCalendarEvent: `token` must be specified in parameters or `identity` permission must be added to the extension manifest.\n" +
-          e
+          e,
       );
     }
   }
@@ -108,7 +108,7 @@ export async function createGoogleCalendarEvent(
           "Content-Type": "application/json",
         },
         body: JSON.stringify(event),
-      }
+      },
     );
 
     if (!response.ok) {

diff --git a/src/retriever.ts b/src/retriever.ts
@@ -9,6 +9,13 @@ export const getSelectedText = (state: State, parameters: {}): string => {
   return state.currentSelection.toString();
 };
 
+export const getPageTextContent = (): string => {
+  if (document) {
+    return document.body.innerText || "";
+  }
+  return "";
+};
+
 export async function getCalendarEvents(
   state: State,
   parameters: { token?: string },
@@ -26,7 +33,7 @@ export async function getCalendarEvents(
     } catch (e) {
       throw new Error(
         "getCalendarEvents: `token` must be specified in parameters or `identity` permission must be added to the extension manifest.\n" +
-          e
+          e,
       );
     }
   }
@@ -87,6 +94,24 @@ export const retrievers: Record<string, Tool> = {
     caller: CallerType.Any,
     implementation: getSelectedText,
   },
+  getPageTextContent: {
+    name: "getPageTextContent",
+    displayName: "Get Page Content",
+    description: "Fetch the entire text content of the current webpage.",
+    schema: {
+      type: "function",
+      function: {
+        name: "getPageTextContent",
+        description:
+          "getPageTextContent() -> str - Fetches the entire text content of the current webpage.\n\n Returns:\n    str: The entire text content of the webpage.",
+        parameters: { type: "object", properties: {}, required: [] },
+      },
+    },
+    type: ToolType.Retriever,
+    scope: Scope.Any,
+    caller: CallerType.ContentScript,
+    implementation: getPageTextContent,
+  },
   getCalendarEvents: {
     name: "getGoogleCalendarEvents",
     displayName: "Get Google Calendar Events",