@stablyai/internal-playwright-core 0.1.0 → 0.1.2

package/lib/server/frames.js

@@ -923,7 +923,7 @@ class Frame extends import_instrumentation.SdkObject {
     try {
       return await this._retryWithProgressIfNotConnected(progress, selector, strict, performActionPreChecks, action);
     } catch (error) {
-      const isAutoHealEnabled = autoHeal ?? (0, import_healingService.isHealingEnabled)();
+      const isAutoHealEnabled = autoHeal ?? (0, import_healingService.isLocatorHealingEnabled)();
       if (!isAutoHealEnabled)
         throw error;
       const healedResult = await this._attemptHealingRetry(

package/lib/server/stably/autohealing/healingService.js

@@ -21,8 +21,11 @@ __export(healingService_exports, {
   attemptAutohealLocator: () => attemptAutohealLocator,
   attemptToHaveScreenshotHealing: () => attemptToHaveScreenshotHealing,
   getHealingConfig: () => getHealingConfig,
+  getLocatorHealingConfig: () => getLocatorHealingConfig,
+  getScreenshotHealingConfig: () => getScreenshotHealingConfig,
   healLocator: () => healLocator,
-  isHealingEnabled: () => isHealingEnabled,
+  isLocatorHealingEnabled: () => isLocatorHealingEnabled,
+  isScreenshotHealingEnabled: () => isScreenshotHealingEnabled,
   setHealingConfig: () => setHealingConfig
 });
 module.exports = __toCommonJS(healingService_exports);
@@ -44,8 +47,31 @@ function setHealingConfig(config) {
 function getHealingConfig() {
   return { ...globalHealingConfig };
 }
-function isHealingEnabled() {
-  return globalHealingConfig.enabled;
+function getLocatorHealingConfig() {
+  const globalConfig = getHealingConfig();
+  const locatorConfig = globalConfig.locators || {};
+  return {
+    enabled: locatorConfig.enabled ?? globalConfig.enabled,
+    model: locatorConfig.model ?? globalConfig.model,
+    timeout: locatorConfig.timeout ?? globalConfig.timeout
+  };
+}
+function getScreenshotHealingConfig() {
+  const globalConfig = getHealingConfig();
+  const screenshotConfig = globalConfig.screenshots || {};
+  return {
+    enabled: screenshotConfig.enabled ?? globalConfig.enabled,
+    model: screenshotConfig.model ?? globalConfig.model,
+    timeout: screenshotConfig.timeout ?? globalConfig.timeout
+  };
+}
+function isLocatorHealingEnabled() {
+  const config = getLocatorHealingConfig();
+  return config.enabled;
+}
+function isScreenshotHealingEnabled() {
+  const config = getScreenshotHealingConfig();
+  return config.enabled;
 }
 async function healLocator({
   frame,
@@ -59,7 +85,7 @@ async function healLocator({
       message: "No description provided for healing. Use .describe() to enable auto-healing."
     };
   }
-  const config = getHealingConfig();
+  const config = getLocatorHealingConfig();
   try {
     const screenshot = await (0, import_screenshotFrame.captureScreenshot)(frame);
     const aiResponse = await (0, import_stablyApiCaller.callAiToAutohealLocator)({
@@ -178,7 +204,7 @@ async function healToHaveScreenshot({
   actualScreenshot,
   expectedScreenshot
 }) {
-  const config = getHealingConfig();
+  const config = getScreenshotHealingConfig();
   try {
     return await (0, import_stablyApiCaller2.callAiToAutohealScreenshot)({
       actualScreenshot,
@@ -196,21 +222,15 @@ async function attemptToHaveScreenshotHealing({
   actualScreenshot,
   expectedScreenshot
 }) {
-  const { result, reason } = await healToHaveScreenshot({
+  const healingResult = await healToHaveScreenshot({
     actualScreenshot,
     expectedScreenshot
   });
-  await saveToHaveScreenshotAutohealResult({ result, reason });
-  return result;
+  await saveToHaveScreenshotAutohealResult(healingResult);
+  return healingResult.result;
 }
-async function saveToHaveScreenshotAutohealResult({
-  result,
-  reason
-}) {
-  const healingDataBody = JSON.stringify({
-    reason,
-    result
-  }, null, 2);
+async function saveToHaveScreenshotAutohealResult(result) {
+  const healingDataBody = JSON.stringify(result, null, 2);
   const stepInfoOrTestInfo = (0, import_stepContext.getCurrentStepInfo)() ?? (0, import_globals.currentTestInfo)();
   stepInfoOrTestInfo?.attach("toHaveScreenshot-autoheal-results", {
     body: healingDataBody,
@@ -222,7 +242,10 @@ async function saveToHaveScreenshotAutohealResult({
   attemptAutohealLocator,
   attemptToHaveScreenshotHealing,
   getHealingConfig,
+  getLocatorHealingConfig,
+  getScreenshotHealingConfig,
   healLocator,
-  isHealingEnabled,
+  isLocatorHealingEnabled,
+  isScreenshotHealingEnabled,
   setHealingConfig
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stablyai/internal-playwright-core",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "A high-level API to automate web browsers",
   "repository": {
     "type": "git",