support zod v3/v4

Author: 404Wolf

src/helpers/beta/zod.ts

@@ -1,7 +1,7 @@
 import type { infer as zodInfer, ZodType } from 'zod/v4';
-import * as z from 'zod/v4';
 import type { BetaRunnableChatCompletionFunctionTool, Promisable } from '../../lib/beta/BetaRunnableTool';
 import type { ChatCompletionContentPart } from '../../resources';
+import { isZodV4, zodV3ToJsonSchema, zodV4ToJsonSchema } from '../zod';
 
 /**
  * Creates a tool using the provided Zod schema that can be passed
@@ -15,7 +15,10 @@ export function betaZodFunctionTool<InputSchema extends ZodType>(options: {
   description: string;
   run: (args: zodInfer<InputSchema>) => Promisable<string | ChatCompletionContentPart[]>;
 }): BetaRunnableChatCompletionFunctionTool<zodInfer<InputSchema>> {
-  const jsonSchema = z.toJSONSchema(options.parameters, { reused: 'ref' });
+  const { parameters, name } = options;
+
+  const jsonSchema =
+    isZodV4(parameters) ? zodV4ToJsonSchema(parameters) : zodV3ToJsonSchema(parameters, { name });
 
   return {
     type: 'function',

src/helpers/zod.ts

@@ -20,7 +20,7 @@ type InferZodType<T> =
   : T extends z3.ZodType ? z3.infer<T>
   : never;
 
-function zodV3ToJsonSchema(schema: z3.ZodType, options: { name: string }): Record<string, unknown> {
+export function zodV3ToJsonSchema(schema: z3.ZodType, options: { name: string }): Record<string, unknown> {
   return _zodToJsonSchema(schema, {
     openaiStrictMode: true,
     name: options.name,
@@ -30,15 +30,15 @@ function zodV3ToJsonSchema(schema: z3.ZodType, options: { name: string }): Recor
   });
 }
 
-function zodV4ToJsonSchema(schema: z4.ZodType): Record<string, unknown> {
+export function zodV4ToJsonSchema(schema: z4.ZodType): Record<string, unknown> {
   return toStrictJsonSchema(
     z4.toJSONSchema(schema, {
       target: 'draft-7',
     }) as JSONSchema,
   ) as Record<string, unknown>;
 }
 
-function isZodV4(zodObject: z3.ZodType | z4.ZodType): zodObject is z4.ZodType {
+export function isZodV4(zodObject: z3.ZodType | z4.ZodType): zodObject is z4.ZodType {
   return '_zod' in zodObject;
 }