feat: script receive error message from chrome extension

Author: lynchee-owo

benchmark.py

@@ -62,7 +62,7 @@ def add_task_listener(driver, task_id, max_retries=3):
     var eventListener = function (e) {{
         if (e.detail.type == 'history') {{
             if (e.detail.status === 'success' || e.detail.status === 'error') {{
-                callback({{status: e.detail.status, type: 'history', data: e.detail.data}});
+                callback({{status: e.detail.status, type: 'history', data: e.detail.data, errorMessage: e.detail.errorMessage}});
                 document.removeEventListener('TaskUpdate', eventListener); // Optional: remove if you need continuous listening
             }}
             // Does not do anything when the status is 'running' or 'idle'. 
@@ -88,15 +88,10 @@ def handle_event(event_data):
             return
         if event_data['type'] == 'history':
             # Record history when task stops
-            result = event_data["status"]
-            # Determine the last action status
-            history = event_data['data']
-            last_action = history[-1]["action"]["operation"]["name"]
-            if last_action == "finish":
-                result = "success"
-            else:
-                result = "fail"
-            write_history(task_id, history)
+            if event_data['status'] == 'error':
+                logging.error(f"Task {task_id} error: {event_data['errorMessage']}")
+            result = event_data['status']
+            write_history(task_id, event_data['data'])
             attempts = 0
             return
         if event_data['type'] == 'screenshot':

src/state/currentTask.ts

@@ -42,6 +42,7 @@ export type CurrentTaskSlice = {
   history: TaskHistoryEntry[];
   status: "idle" | "running" | "success" | "error" | "interrupted";
   knowledgeInUse: Knowledge | null;
+  errorMessage: string;
   actionStatus:
     | "idle"
     | "attaching-debugger"
@@ -73,6 +74,7 @@ export const createCurrentTaskSlice: MyStateCreator<CurrentTaskSlice> = (
   status: "idle",
   actionStatus: "idle",
   knowledgeInUse: null,
+  errorMessage: "",
   actions: {
     runTask: async (onError) => {
       const voiceMode = get().settings.voiceMode;
@@ -132,6 +134,7 @@ export const createCurrentTaskSlice: MyStateCreator<CurrentTaskSlice> = (
             if (query == null) {
               set((state) => {
                 state.currentTask.status = "error";
+                state.currentTask.errorMessage = "Query is null.";
               });
               return false;
             }
@@ -239,6 +242,7 @@ export const createCurrentTaskSlice: MyStateCreator<CurrentTaskSlice> = (
             if (!pageDOM) {
               set((state) => {
                 state.currentTask.status = "error";
+                state.currentTask.errorMessage = "No page dom found";
               });
               break;
             }
@@ -278,6 +282,7 @@ export const createCurrentTaskSlice: MyStateCreator<CurrentTaskSlice> = (
         onError(e.message);
         set((state) => {
           state.currentTask.status = "error";
+          state.currentTask.errorMessage = e.message;
         });
       } finally {
         await detachAllDebuggers();

src/state/store.ts

@@ -1,7 +1,12 @@
 import { merge } from "lodash";
 import { create, StateCreator } from "zustand";
 import { immer } from "zustand/middleware/immer";
-import { createJSONStorage, devtools, persist , subscribeWithSelector } from "zustand/middleware";
+import {
+  createJSONStorage,
+  devtools,
+  persist,
+  subscribeWithSelector,
+} from "zustand/middleware";
 import { createCurrentTaskSlice, CurrentTaskSlice } from "./currentTask";
 import { createUiSlice, UiSlice } from "./ui";
 import { createSettingsSlice, SettingsSlice } from "./settings";
@@ -66,11 +71,18 @@ useAppState.subscribe(
   (state) => state.currentTask.status,
   // eslint-disable-next-line @typescript-eslint/no-unused-vars
   (status, previousStatus) => {
-    chrome.runtime.sendMessage({
+    const message = {
       action: "updateHistory",
       status,
       history: useAppState.getState().currentTask.history,
-    });
+      error: "",
+    };
+
+    if (status === "error") {
+      message.error = useAppState.getState().currentTask.errorMessage;
+    }
+
+    chrome.runtime.sendMessage(message);
   },
   { fireImmediately: true },
 );