update zod to schema

Author: tkattkat

packages/core/lib/v3/zodCompat.ts

@@ -4,6 +4,7 @@ import type {
   ZodRawShape as Zod4RawShape,
   ZodTypeAny as Zod4TypeAny,
 } from "zod";
+import zodToJsonSchema from "zod-to-json-schema";
 import type * as z3 from "zod/v3";
 export type StagehandZodSchema = Zod4TypeAny | z3.ZodTypeAny;
 
@@ -29,118 +30,12 @@ export const isZod3Schema = (
 
 export type JsonSchemaDocument = Record<string, unknown>;
 
-// Manual converter for zod/v3 schemas to JSON Schema
-function ZodToJsonSchema(schema: z3.ZodTypeAny): JsonSchemaDocument {
-  const _def = (schema as unknown as { _def?: Record<string, unknown> })._def;
 
-  if (!_def) {
-    return { type: "null" };
-  }
-
-  const typeName = _def.typeName;
-
-  switch (typeName) {
-    case "ZodObject": {
-      const shape =
-        typeof _def.shape === "function" ? _def.shape() : _def.shape;
-      const properties: Record<string, JsonSchemaDocument> = {};
-      const required: string[] = [];
-
-      for (const [key, value] of Object.entries(
-        shape as Record<string, z3.ZodTypeAny>,
-      )) {
-        properties[key] = ZodToJsonSchema(value);
-        // Check if field is not optional
-        const valueDef = (value as unknown as { _def?: { typeName?: string } })
-          ._def;
-        if (valueDef?.typeName !== "ZodOptional") {
-          required.push(key);
-        }
-      }
-
-      return {
-        type: "object",
-        properties,
-        required,
-        additionalProperties: _def.unknownKeys === "passthrough",
-      };
-    }
-
-    case "ZodArray": {
-      const itemType = _def.type as z3.ZodTypeAny;
-      return {
-        type: "array",
-        items: ZodToJsonSchema(itemType),
-      };
-    }
-
-    case "ZodString": {
-      const result: JsonSchemaDocument = { type: "string" };
-      // Check for URL validation
-      const checks = _def.checks as Array<{ kind?: string }> | undefined;
-      if (checks) {
-        for (const check of checks) {
-          if (check.kind === "url") {
-            result.format = "url";
-            break;
-          }
-        }
-      }
-      return result;
-    }
-
-    case "ZodNumber":
-      return { type: "number" };
-
-    case "ZodBoolean":
-      return { type: "boolean" };
-
-    case "ZodOptional":
-      return ZodToJsonSchema(_def.innerType as z3.ZodTypeAny);
-
-    case "ZodNullable": {
-      const innerSchema = ZodToJsonSchema(_def.innerType as z3.ZodTypeAny);
-      return {
-        ...innerSchema,
-        nullable: true,
-      };
-    }
-
-    case "ZodEnum":
-      return {
-        type: "string",
-        enum: _def.values,
-      };
-
-    case "ZodLiteral":
-      return {
-        type: typeof _def.value,
-        const: _def.value,
-      };
-
-    case "ZodUnion":
-      return {
-        anyOf: (_def.options as z3.ZodTypeAny[]).map((opt) =>
-          ZodToJsonSchema(opt),
-        ),
-      };
-
-    default:
-      console.warn(`Unknown Zod type: ${typeName}`);
-      return { type: "null" };
-  }
-}
 
 export function toJsonSchema(schema: StagehandZodSchema): JsonSchemaDocument {
-  // For v3 schemas, use manual converter
-  // Note: We can't use zod-to-json-schema for v3 schemas when zod v4 is installed
-  // because the library imports 'zod' (v4) and tries to access ZodFirstPartyTypeKind which doesn't exist in v4
+
   if (!isZod4Schema(schema)) {
-    const result = {
-      $schema: "http://json-schema.org/draft-07/schema#",
-      ...ZodToJsonSchema(schema),
-    };
-    return result;
+    return zodToJsonSchema(schema);
   }
 
   // For v4 schemas, use built-in z.toJSONSchema() method

packages/core/package.json

@@ -56,7 +56,8 @@
     "pino": "^9.6.0",
     "pino-pretty": "^13.0.0",
     "playwright": "^1.52.0",
-    "ws": "^8.18.0"
+    "ws": "^8.18.0",
+    "zod-to-json-schema": "^3.25.0"
   },
   "optionalDependencies": {
     "@ai-sdk/anthropic": "^2.0.34",