@stablyai/playwright-base 0.1.6 → 0.1.7-next.2

package/dist/ai/extract.d.ts

@@ -1,7 +1,18 @@
 import type { Locator, Page } from "@stablyai/internal-playwright-test";
 import * as z4 from "zod/v4/core";
+import { z } from "zod";
 export type ExtractSchema = z4.$ZodType;
 export type SchemaOutput<T extends ExtractSchema> = z4.output<T>;
+export declare const zExtractionResponse: z.ZodDiscriminatedUnion<[z.ZodObject<{
+    success: z.ZodLiteral<true>;
+    value: z.ZodUnknown;
+}, z4.$strict>, z.ZodObject<{
+    success: z.ZodLiteral<false>;
+    error: z.ZodString;
+}, z4.$strict>], "success">;
+export declare const zError: z.ZodObject<{
+    error: z.ZodString;
+}, z4.$strict>;
 type ExtractSubject = Page | Locator;
 type BaseExtractArgs = {
     prompt: string;

package/dist/ai/extract.js

@@ -33,13 +33,17 @@ var __importStar = (this && this.__importStar) || (function () {
     };
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.zError = exports.zExtractionResponse = void 0;
 exports.extract = extract;
 const z4 = __importStar(require("zod/v4/core"));
 const zod_1 = require("zod");
 const runtime_1 = require("../runtime");
-const EXTRACT_ENDPOINT = "https://api.stably.ai/internal/v1/extract";
-const zSuccess = zod_1.z.object({ value: zod_1.z.unknown() });
-const zError = zod_1.z.object({ error: zod_1.z.string() });
+const EXTRACT_ENDPOINT = "https://api.stably.ai/internal/v2/extract";
+exports.zExtractionResponse = zod_1.z.discriminatedUnion("success", [
+    zod_1.z.object({ success: zod_1.z.literal(true), value: zod_1.z.unknown() }).strict(),
+    zod_1.z.object({ success: zod_1.z.literal(false), error: zod_1.z.string() }).strict(),
+]);
+exports.zError = zod_1.z.object({ error: zod_1.z.string() }).strict();
 class ExtractValidationError extends Error {
     issues;
     constructor(message, issues) {
@@ -74,15 +78,23 @@ async function extract({ prompt, pageOrLocator, schema, }) {
         },
         body: form,
     });
-    const parsed = await response.json().catch(() => undefined);
+    const raw = await response.json().catch(() => undefined);
     if (response.ok) {
-        const { value } = zSuccess.parse(parsed);
+        const parsed = exports.zExtractionResponse.safeParse(raw);
+        if (!parsed.success) {
+            throw new Error("Extract returned unexpected response shape");
+        }
+        // AI could not find requested content; return message for display
+        if (parsed.data.success === false) {
+            return parsed.data.error;
+        }
+        const { value } = parsed.data;
         return schema
             ? await validateWithSchema(schema, value)
             : typeof value === "string"
                 ? value
                 : JSON.stringify(value);
     }
-    const err = zError.safeParse(parsed);
-    throw new Error(`Extract failed (${response.status})${err.success ? `: ${err.data.error}` : ""}`);
+    const err = exports.zError.safeParse(raw);
+    throw new Error(err.success ? err.data.error : "Extract failed");
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stablyai/playwright-base",
-  "version": "0.1.6",
+  "version": "0.1.7-next.2",
   "description": "Shared augmentation runtime for Stably Playwright wrappers",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -11,7 +11,7 @@
     "node": ">=18"
   },
   "dependencies": {
-    "@stablyai/internal-playwright-test": "0.1.2",
+    "@stablyai/internal-playwright-test": "0.1.8",
     "jpeg-js": "^0.4.4",
     "pixelmatch": "^5.3.0",
     "pngjs": "^7.0.0"
@@ -19,6 +19,11 @@
   "peerDependencies": {
     "zod": "^3.25.0 || ^4.0.0"
   },
+  "peerDependenciesMeta": {
+    "zod": {
+      "optional": true
+    }
+  },
   "devDependencies": {
     "@types/jpeg-js": "^0.3.7",
     "@types/pngjs": "^6.0.5",