misoai-web 1.0.4 → 1.0.5

package/dist/lib/puppeteer.js

@@ -1706,7 +1706,7 @@ var import_js_yaml3 = __toESM(require("js-yaml"));
 var import_semver = __toESM(require("semver"));
 
 // package.json
-var version = "1.0.4";
+var version = "1.0.5";
 
 // src/common/task-cache.ts
 var debug3 = (0, import_logger3.getDebug)("cache");
@@ -1888,10 +1888,13 @@ var PageAgent = class {
         generateReport: true,
         autoPrintReportMsg: true,
         groupName: "Midscene Report",
-        groupDescription: ""
+        groupDescription: "",
+        enableCumulativeContext: true,
+        autoClearContext: false
       },
       opts || {}
     );
+    this.initializeContextStore();
     if (this.page.pageType === "puppeteer" || this.page.pageType === "playwright") {
       this.page.waitForNavigationTimeout = this.opts.waitForNavigationTimeout || import_constants2.DEFAULT_WAIT_FOR_NAVIGATION_TIMEOUT;
       this.page.waitForNetworkIdleTimeout = this.opts.waitForNetworkIdleTimeout || import_constants2.DEFAULT_WAIT_FOR_NETWORK_IDLE_TIMEOUT;
@@ -1918,6 +1921,69 @@ var PageAgent = class {
       opts?.testId || this.page.pageType || "web"
     );
   }
+  /**
+   * Initialize context store for cumulative context functionality
+   */
+  async initializeContextStore() {
+    if (!this.opts.enableCumulativeContext) {
+      debug4("Cumulative context disabled via options");
+      return;
+    }
+    try {
+      const aiModel = await import("misoai-core/ai-model");
+      this.contextStore = aiModel.getContextStore();
+      debug4("Context store initialized successfully", {
+        autoClearContext: this.opts.autoClearContext,
+        testId: this.opts.testId
+      });
+      if (this.opts.autoClearContext) {
+        this.contextStore.clear();
+        debug4("Context store cleared due to autoClearContext option");
+      } else {
+        const existingData = this.contextStore.getAllData();
+        const existingSteps = this.contextStore.getRecentSteps(100).length;
+        debug4("Context store preserving existing data", {
+          existingDataKeys: Object.keys(existingData),
+          existingStepsCount: existingSteps
+        });
+      }
+    } catch (error) {
+      debug4("Failed to initialize context store:", error);
+      console.warn("⚠️ Could not initialize context store:", error);
+    }
+  }
+  /**
+   * Get the context store instance
+   */
+  getContextStore() {
+    return this.contextStore;
+  }
+  /**
+   * Clear the context store
+   */
+  clearContext() {
+    if (this.contextStore) {
+      this.contextStore.clear();
+    }
+  }
+  /**
+   * Get all stored data from context store
+   */
+  getStoredData() {
+    if (this.contextStore) {
+      return this.contextStore.getAllData();
+    }
+    return {};
+  }
+  /**
+   * Get step summary from context store
+   */
+  getStepSummary() {
+    if (this.contextStore) {
+      return this.contextStore.getStepSummary();
+    }
+    return "";
+  }
   async getUIContext(action) {
     if (action && (action === "extract" || action === "assert" || action === "captcha")) {
       return await parseContextFromWebPage(this.page, {
@@ -2153,18 +2219,30 @@ var PageAgent = class {
     };
   }
   async aiAction(taskPrompt, opt) {
-    try {
-      const aiModel = await import("misoai-core/ai-model");
-      const contextStore = aiModel.getContextStore();
-      const processedPrompt = contextStore.replaceAllReferences(taskPrompt, "action");
-      contextStore.addStep({
-        type: "action",
-        summary: `Action: ${processedPrompt}`,
-        prompt: processedPrompt
-      });
-      taskPrompt = processedPrompt;
-    } catch (error) {
-      debug4("Context store not available:", error);
+    if (this.opts.enableCumulativeContext && this.contextStore) {
+      try {
+        const originalPrompt = taskPrompt;
+        const processedPrompt = this.contextStore.replaceAllReferences(taskPrompt, "action");
+        if (originalPrompt !== processedPrompt) {
+          debug4("Context replacement in aiAction:", {
+            original: originalPrompt,
+            processed: processedPrompt,
+            storedData: this.contextStore.getAllData()
+          });
+        }
+        this.contextStore.addStep({
+          type: "action",
+          summary: `Action: ${processedPrompt}`,
+          prompt: processedPrompt
+        });
+        debug4("Added action step to context store:", {
+          stepNumber: this.contextStore.getRecentSteps(1)[0]?.stepNumber,
+          totalSteps: this.contextStore.getRecentSteps(100).length
+        });
+        taskPrompt = processedPrompt;
+      } catch (error) {
+        debug4("Context store operation failed:", error);
+      }
     }
     const cacheable = opt?.cacheable;
     const isVlmUiTars = (0, import_env2.vlLocateMode)() === "vlm-ui-tars";
@@ -2235,38 +2313,50 @@ var PageAgent = class {
       debug4("Context store not available:", error);
     }
     const { output, executor } = await this.taskExecutor.query(processedDemand);
-    if (storageKey && output) {
-      try {
-        const aiModel = await import("misoai-core/ai-model");
-        const contextStore = aiModel.getContextStore();
-        const pendingAliases = contextStore._pendingAliases;
-        if (pendingAliases) {
-          contextStore.storeDataWithAliases(storageKey, output, pendingAliases, typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand));
-          delete contextStore._pendingAliases;
-        } else {
-          contextStore.storeData(storageKey, output);
+    if (this.opts.enableCumulativeContext && this.contextStore) {
+      if (storageKey && output) {
+        try {
+          const pendingAliases = this.contextStore._pendingAliases;
+          if (pendingAliases) {
+            this.contextStore.storeDataWithAliases(storageKey, output, pendingAliases, typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand));
+            delete this.contextStore._pendingAliases;
+            debug4("Stored query result with aliases:", {
+              key: storageKey,
+              value: output,
+              aliases: pendingAliases
+            });
+          } else {
+            this.contextStore.storeData(storageKey, output);
+            debug4("Stored query result:", {
+              key: storageKey,
+              value: output
+            });
+          }
+          this.contextStore.addStep({
+            type: "query",
+            summary: `Query: ${typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand)} (stored as ${storageKey})`,
+            data: output,
+            prompt: typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand)
+          });
+          debug4("Added query step to context store:", {
+            storageKey,
+            totalStoredItems: Object.keys(this.contextStore.getAllData()).length,
+            totalSteps: this.contextStore.getRecentSteps(100).length
+          });
+        } catch (error) {
+          debug4("Failed to store query result:", error);
+        }
+      } else {
+        try {
+          this.contextStore.addStep({
+            type: "query",
+            summary: `Query: ${typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand)}`,
+            data: output,
+            prompt: typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand)
+          });
+        } catch (error) {
+          debug4("Failed to add query step:", error);
         }
-        contextStore.addStep({
-          type: "query",
-          summary: `Query: ${typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand)} (stored as ${storageKey})`,
-          data: output,
-          prompt: typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand)
-        });
-      } catch (error) {
-        debug4("Failed to store query result:", error);
-      }
-    } else {
-      try {
-        const aiModel = await import("misoai-core/ai-model");
-        const contextStore = aiModel.getContextStore();
-        contextStore.addStep({
-          type: "query",
-          summary: `Query: ${typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand)}`,
-          data: output,
-          prompt: typeof processedDemand === "string" ? processedDemand : JSON.stringify(processedDemand)
-        });
-      } catch (error) {
-        debug4("Failed to add query step:", error);
       }
     }
     const metadata = this.afterTaskRunning(executor);
@@ -2379,17 +2469,29 @@ var PageAgent = class {
   }
   async aiAssert(assertion, msg, opt) {
     let processedAssertion = assertion;
-    try {
-      const aiModel = await import("misoai-core/ai-model");
-      const contextStore = aiModel.getContextStore();
-      processedAssertion = contextStore.replaceAllReferences(assertion, "assertion");
-      contextStore.addStep({
-        type: "assertion",
-        summary: `Assertion: ${processedAssertion}`,
-        prompt: processedAssertion
-      });
-    } catch (error) {
-      debug4("Context store not available:", error);
+    if (this.opts.enableCumulativeContext && this.contextStore) {
+      try {
+        const originalAssertion = assertion;
+        processedAssertion = this.contextStore.replaceAllReferences(assertion, "assertion");
+        if (originalAssertion !== processedAssertion) {
+          debug4("Context replacement in aiAssert:", {
+            original: originalAssertion,
+            processed: processedAssertion,
+            context: "assertion",
+            storedData: this.contextStore.getAllData()
+          });
+        }
+        this.contextStore.addStep({
+          type: "assertion",
+          summary: `Assertion: ${processedAssertion}`,
+          prompt: processedAssertion
+        });
+        debug4("Added assertion step to context store:", {
+          totalSteps: this.contextStore.getRecentSteps(100).length
+        });
+      } catch (error) {
+        debug4("Context store operation failed:", error);
+      }
     }
     let currentUrl = "";
     if (this.page.url) {