@stablyai/playwright-base 0.1.7-next.3 → 0.1.7-next.4

package/dist/ai/extract.d.ts

@@ -4,10 +4,9 @@ export interface ExtractSchema extends z4.$ZodType {
     safeParseAsync(data: unknown, params?: z4.ParseContext<z4.$ZodIssue>): Promise<z4.util.SafeParseResult<z4.output<this>>>;
 }
 export type SchemaOutput<T extends ExtractSchema> = z4.output<T>;
-type ExtractSubject = Page | Locator;
 type BaseExtractArgs = {
     prompt: string;
-    pageOrLocator: ExtractSubject;
+    pageOrLocator: Page | Locator;
 };
 type ExtractArgsWithSchema<T extends ExtractSchema> = BaseExtractArgs & {
     schema: T;

package/dist/ai/extract.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.extract = extract;
 const runtime_1 = require("../runtime");
+const is_object_1 = require("~/type-predicate/is-object");
 const EXTRACT_ENDPOINT = "https://api.stably.ai/internal/v2/extract";
 const zodV4 = (() => {
     try {
@@ -11,23 +12,17 @@ const zodV4 = (() => {
         return undefined;
     }
 })();
-const isObject = (value) => {
-    return typeof value === "object" && value !== null;
-};
 const isExtractionResponse = (value) => {
-    if (!isObject(value)) {
+    if (!(0, is_object_1.isObject)(value)) {
         return false;
     }
     if (value.success === true) {
         return "value" in value;
     }
-    if (value.success === false && typeof value.error === "string") {
-        return true;
-    }
-    return false;
+    return value.success === false && typeof value.error === "string";
 };
 const isErrorResponse = (value) => {
-    return isObject(value) && typeof value.error === "string";
+    return (0, is_object_1.isObject)(value) && typeof value.error === "string";
 };
 class ExtractValidationError extends Error {
     issues;
@@ -49,7 +44,7 @@ async function extract({ prompt, pageOrLocator, schema, }) {
         throw new Error("Schema support requires installing zod@4. Please add it to enable schemas.");
     }
     const jsonSchema = schema && zodV4
-        ? zodV4.toJSONSchema(schema)
+        ? zodV4?.toJSONSchema(schema)
         : undefined;
     const apiKey = (0, runtime_1.requireApiKey)();
     const form = new FormData();
@@ -83,8 +78,5 @@ async function extract({ prompt, pageOrLocator, schema, }) {
                 ? value
                 : JSON.stringify(value);
     }
-    if (isErrorResponse(raw)) {
-        throw new Error(raw.error);
-    }
-    throw new Error("Extract failed");
+    throw new Error(isErrorResponse(raw) ? raw.error : "Extract failed");
 }

package/dist/ai/verify-prompt.js

@@ -1,16 +1,32 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.verifyPrompt = verifyPrompt;
-const PROMPT_ASSERTION_ENDPOINT = "https://api.stably.ai/internal/v1/assert";
-const zod_1 = require("zod");
+const is_object_1 = require("~/type-predicate/is-object");
 const runtime_1 = require("../runtime");
-const zSuccess = zod_1.z.object({
-    success: zod_1.z.boolean(),
-    reason: zod_1.z.string().optional(),
-});
-const zError = zod_1.z.object({
-    error: zod_1.z.string(),
-});
+const PROMPT_ASSERTION_ENDPOINT = "https://api.stably.ai/internal/v1/assert";
+const parseSuccessResponse = (value) => {
+    if (!(0, is_object_1.isObject)(value)) {
+        throw new Error("Verify prompt returned unexpected response shape");
+    }
+    const { success, reason } = value;
+    if (typeof success !== "boolean") {
+        throw new Error("Verify prompt returned unexpected response shape");
+    }
+    if (reason !== undefined && typeof reason !== "string") {
+        throw new Error("Verify prompt returned unexpected response shape");
+    }
+    return {
+        success,
+        reason,
+    };
+};
+const parseErrorResponse = (value) => {
+    if (!(0, is_object_1.isObject)(value)) {
+        return undefined;
+    }
+    const { error } = value;
+    return typeof error !== "string" ? undefined : { error };
+};
 async function verifyPrompt({ prompt, screenshot, }) {
     const apiKey = (0, runtime_1.requireApiKey)();
     const form = new FormData();
@@ -27,12 +43,12 @@ async function verifyPrompt({ prompt, screenshot, }) {
     });
     const parsed = await response.json().catch(() => undefined);
     if (response.ok) {
-        const { success, reason } = zSuccess.parse(parsed);
+        const { success, reason } = parseSuccessResponse(parsed);
         return {
             pass: success,
             reason,
         };
     }
-    const err = zError.safeParse(parsed);
-    throw new Error(`Verify prompt failed (${response.status})${err.success ? `: ${err.data.error}` : ""}`);
+    const err = parseErrorResponse(parsed);
+    throw new Error(`Verify prompt failed (${response.status})${err ? `: ${err.error}` : ""}`);
 }

package/dist/internal/autoheal-patch.d.ts

@@ -0,0 +1 @@
+export declare function installAutohealZodPatch(): void;

package/dist/internal/autoheal-patch.js

@@ -0,0 +1,178 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.installAutohealZodPatch = installAutohealZodPatch;
+const module_1 = __importDefault(require("module"));
+const module_2 = require("module");
+const nodeRequire = (0, module_2.createRequire)(__filename);
+let patchInstalled = false;
+function installAutohealZodPatch() {
+    if (patchInstalled) {
+        return;
+    }
+    patchInstalled = true;
+    let targetModule;
+    try {
+        targetModule = nodeRequire.resolve("@stablyai/internal-playwright-core/lib/server/stably/ai-tools/stablyApiCaller");
+    }
+    catch {
+        return;
+    }
+    const moduleAny = module_1.default;
+    const originalLoad = moduleAny._load;
+    const originalResolve = moduleAny._resolveFilename;
+    if (typeof originalLoad !== "function" || typeof originalResolve !== "function") {
+        return;
+    }
+    moduleAny._load = function loadPatchedModule(request, parent, isMain, options) {
+        const args = arguments;
+        let resolved;
+        try {
+            resolved = originalResolve.call(this, request, parent, isMain, options);
+        }
+        catch {
+            return originalLoad.apply(this, args);
+        }
+        if (resolved === targetModule) {
+            const cached = nodeRequire.cache[targetModule];
+            if (cached?.loaded) {
+                return cached.exports;
+            }
+            const patchedExports = createPatchedAutohealModule();
+            const ModuleConstructor = moduleAny.Module ?? module_1.default.Module;
+            const syntheticModule = new ModuleConstructor(targetModule, parent ?? undefined);
+            syntheticModule.filename = targetModule;
+            syntheticModule.exports = patchedExports;
+            syntheticModule.loaded = true;
+            nodeRequire.cache[targetModule] = syntheticModule;
+            return patchedExports;
+        }
+        return originalLoad.apply(this, args);
+    };
+}
+function createPatchedAutohealModule() {
+    const { createMultipartFormData, makeHTTPRequest, } = nodeRequire("@stablyai/internal-playwright-core/lib/server/stably/ai-tools/http-request");
+    const { STABLY_API_HOSTNAME, STABLY_API_PORT } = nodeRequire("@stablyai/internal-playwright-core/lib/server/stably/constants");
+    const LOCATOR_AUTOHEAL_ENDPOINT = "/v1/internal/sdk/autohealLocator";
+    const SCREENSHOT_AUTOHEAL_ENDPOINT = "/v1/internal/sdk/autohealToHaveScreenshot";
+    function parseLocatorResponse(raw) {
+        if (!isObject(raw)) {
+            throw new Error("Invalid auto-heal response");
+        }
+        const { reasoning, success } = raw;
+        if (typeof reasoning !== "string") {
+            throw new Error("Invalid auto-heal response: missing reasoning");
+        }
+        if (typeof success !== "boolean") {
+            throw new Error("Invalid auto-heal response: missing success flag");
+        }
+        if (success === true) {
+            const { coordinates } = raw;
+            if (!Array.isArray(coordinates) ||
+                coordinates.length !== 2 ||
+                !coordinates.every((value) => typeof value === "number" && Number.isFinite(value))) {
+                throw new Error("Invalid auto-heal response: missing coordinates");
+            }
+            return {
+                reasoning,
+                success: true,
+                coordinates: [coordinates[0], coordinates[1]],
+            };
+        }
+        return {
+            reasoning,
+            success: false,
+        };
+    }
+    function parseScreenshotResponse(raw) {
+        if (!isObject(raw)) {
+            throw new Error("Invalid auto-heal response");
+        }
+        const { result, reason } = raw;
+        if (typeof result !== "boolean" || typeof reason !== "string") {
+            throw new Error("Invalid auto-heal response");
+        }
+        return { result, reason };
+    }
+    async function callAiToAutohealLocator({ description, screenshot, originalSelector, actionName, model = "claude-sonnet-4-5", }) {
+        const stablyApiKey = process.env.STABLY_API_KEY;
+        if (!stablyApiKey) {
+            throw new Error("STABLY_API_KEY environment variable is required for auto-healing");
+        }
+        const { formData, boundary } = createMultipartFormData({
+            fields: {
+                originalSelector,
+                description,
+                actionName,
+            },
+            files: [
+                {
+                    fieldName: "file",
+                    filename: "screenshot.png",
+                    data: screenshot,
+                    contentType: "image/png",
+                },
+            ],
+        });
+        const options = {
+            hostname: STABLY_API_HOSTNAME,
+            port: STABLY_API_PORT,
+            path: `${LOCATOR_AUTOHEAL_ENDPOINT}?model=${encodeURIComponent(model)}`,
+            method: "POST",
+            headers: {
+                "Content-Type": `multipart/form-data; boundary=${boundary}`,
+                Authorization: stablyApiKey,
+                "Content-Length": formData.length.toString(),
+            },
+        };
+        const responseData = await makeHTTPRequest(options, formData);
+        const parsed = parseLocatorResponse(JSON.parse(responseData));
+        return {
+            coordinate: parsed.success ? parsed.coordinates : undefined,
+            reasoning: parsed.reasoning,
+        };
+    }
+    async function callAiToAutohealScreenshot({ actualScreenshot, expectedScreenshot, model = "claude-sonnet-4-5", }) {
+        const stablyApiKey = process.env.STABLY_API_KEY;
+        if (!stablyApiKey) {
+            throw new Error("STABLY_API_KEY environment variable is required for auto-healing");
+        }
+        const files = [
+            {
+                fieldName: "actualScreenshot",
+                filename: "actual.png",
+                data: actualScreenshot,
+                contentType: "image/png",
+            },
+            {
+                fieldName: "expectedScreenshot",
+                filename: "expected.png",
+                data: expectedScreenshot,
+                contentType: "image/png",
+            },
+        ];
+        const { formData, boundary } = createMultipartFormData({ files });
+        const options = {
+            hostname: STABLY_API_HOSTNAME,
+            port: STABLY_API_PORT,
+            path: `${SCREENSHOT_AUTOHEAL_ENDPOINT}?model=${encodeURIComponent(model)}`,
+            method: "POST",
+            headers: {
+                "Content-Type": `multipart/form-data; boundary=${boundary}`,
+                Authorization: stablyApiKey,
+                "Content-Length": formData.length.toString(),
+            },
+        };
+        const responseData = await makeHTTPRequest(options, formData);
+        return parseScreenshotResponse(JSON.parse(responseData));
+    }
+    return {
+        callAiToAutohealLocator,
+        callAiToAutohealScreenshot,
+    };
+}
+function isObject(value) {
+    return typeof value === "object" && value !== null;
+}

package/dist/type-predicate/is-object.d.ts

@@ -0,0 +1 @@
+export declare const isObject: (value: unknown) => value is Record<string, unknown>;

package/dist/type-predicate/is-object.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isObject = void 0;
+const isObject = (value) => {
+    return typeof value === "object" && value !== null;
+};
+exports.isObject = isObject;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stablyai/playwright-base",
-  "version": "0.1.7-next.3",
+  "version": "0.1.7-next.4",
   "description": "Shared augmentation runtime for Stably Playwright wrappers",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",