@eko-ai/eko 4.0.1 → 4.0.3

package/dist/index.esm.js

@@ -34892,47 +34892,55 @@ class BaseBrowserLabelsAgent extends BaseBrowserAgent {
         return config$1.mode == "expert";
     }
     async handleMessages(agentContext, messages, tools) {
-        const pseudoHtmlDescription = "This is the environmental information after the operation, including the latest browser screenshot and page elements. Please perform the next operation based on the environmental information. Do not output the following elements and index information in your response.\n\nIndex and elements:\n";
-        let lastTool = this.lastToolResult(messages);
-        if (lastTool &&
-            lastTool.toolName !== "extract_page_content" &&
-            lastTool.toolName !== "get_all_tabs" &&
-            lastTool.toolName !== "variable_storage") {
-            await sleep(300);
-            const image_contents = [];
-            const result = await this.screenshot_and_html(agentContext);
-            if (await this.double_screenshots(agentContext, messages, tools)) {
-                const imageResult = result.double_screenshots
-                    ? result.double_screenshots
-                    : await this.screenshot_and_compress(agentContext, result.client_rect);
-                const image = toImage(imageResult.imageBase64);
-                image_contents.push({
-                    type: "file",
-                    data: image,
-                    mediaType: imageResult.imageType,
-                });
-            }
-            if (result.imageBase64) {
-                const image = toImage(result.imageBase64);
-                image_contents.push({
-                    type: "file",
-                    data: image,
-                    mediaType: result.imageType || "image/png",
+        try {
+            const pseudoHtmlDescription = "This is the environmental information after the operation, including the latest browser screenshot and page elements. Please perform the next operation based on the environmental information. Do not output the following elements and index information in your response.\n\nIndex and elements:\n";
+            let lastTool = this.lastToolResult(messages);
+            if (lastTool &&
+                lastTool.toolName !== "extract_page_content" &&
+                lastTool.toolName !== "get_all_tabs" &&
+                lastTool.toolName !== "variable_storage") {
+                await sleep(300);
+                const image_contents = [];
+                const result = await this.screenshot_and_html(agentContext);
+                if (await this.double_screenshots(agentContext, messages, tools)) {
+                    const imageResult = result.double_screenshots
+                        ? result.double_screenshots
+                        : await this.screenshot_and_compress(agentContext, result.client_rect);
+                    const image = toImage(imageResult.imageBase64);
+                    image_contents.push({
+                        type: "file",
+                        data: image,
+                        mediaType: imageResult.imageType,
+                    });
+                }
+                if (result.imageBase64) {
+                    const image = toImage(result.imageBase64);
+                    image_contents.push({
+                        type: "file",
+                        data: image,
+                        mediaType: result.imageType || "image/png",
+                    });
+                }
+                messages.push({
+                    role: "user",
+                    content: [
+                        ...image_contents,
+                        {
+                            type: "text",
+                            text: pseudoHtmlDescription +
+                                "```html\n" +
+                                result.pseudoHtml +
+                                "\n```",
+                        },
+                    ],
                 });
             }
-            messages.push({
-                role: "user",
-                content: [
-                    ...image_contents,
-                    {
-                        type: "text",
-                        text: pseudoHtmlDescription + "```html\n" + result.pseudoHtml + "\n```",
-                    },
-                ],
-            });
+            super.handleMessages(agentContext, messages, tools);
+            this.handlePseudoHtmlText(messages, pseudoHtmlDescription);
+        }
+        catch (error) {
+            Log.error("browser handleMessages error: ", error);
         }
-        super.handleMessages(agentContext, messages, tools);
-        this.handlePseudoHtmlText(messages, pseudoHtmlDescription);
     }
     handlePseudoHtmlText(messages, pseudoHtmlDescription) {
         for (let i = 0; i < messages.length; i++) {
@@ -37239,6 +37247,12 @@ class ChatAgent {
         tools.push(new TaskVariableStorageTool(this.chatContext, params));
         return tools;
     }
+    getMemory() {
+        return this.memory;
+    }
+    getTools() {
+        return this.tools;
+    }
     getChatContext() {
         return this.chatContext;
     }