add v4 schema support for zodFunction

Author: karpetrosyan

src/helpers/zod.ts

@@ -142,14 +142,32 @@ export function zodFunction<Parameters extends ZodType>(options: {
   arguments: Parameters;
   name: string;
   function: (args: zodInfer<Parameters>) => unknown;
-}> {
-  // @ts-expect-error TODO
+}>;
+export function zodFunction<Parameters extends ZodTypeV4>(options: {
+  name: string;
+  parameters: Parameters;
+  function?: ((args: zodInferV4<Parameters>) => unknown | Promise<unknown>) | undefined;
+  description?: string | undefined;
+}): AutoParseableTool<{
+  arguments: Parameters;
+  name: string;
+  function: (args: zodInferV4<Parameters>) => unknown;
+}>;
+export function zodFunction<Parameters extends ZodType | ZodTypeV4>(options: {
+  name: string;
+  parameters: Parameters;
+  function?: ((args: any) => unknown | Promise<unknown>) | undefined;
+  description?: string | undefined;
+}): unknown {
   return makeParseableTool<any>(
     {
       type: 'function',
       function: {
         name: options.name,
-        parameters: zodToJsonSchema(options.parameters, { name: options.name }),
+        parameters:
+          isZodV4(options.parameters) ?
+            toJSONSchema(options.parameters)
+          : zodToJsonSchema(options.parameters, { name: options.name }),
         strict: true,
         ...(options.description ? { description: options.description } : undefined),
       },

src/lib/transform.ts

@@ -98,10 +98,10 @@ function ensureStrictJsonSchema(
 
     const resolved = resolveRef(root, ref);
     if (typeof resolved === 'boolean') {
-      throw new ValueError(`Expected \`$ref: ${ref}\` to resolve to an object schema but got boolean`);
+      throw new Error(`Expected \`$ref: ${ref}\` to resolve to an object schema but got boolean`);
     }
     if (!isDict(resolved)) {
-      throw new ValueError(
+      throw new Error(
         `Expected \`$ref: ${ref}\` to resolve to a dictionary but got ${JSON.stringify(resolved)}`,
       );
     }
@@ -120,7 +120,7 @@ function ensureStrictJsonSchema(
 
 function resolveRef(root: JSONSchema, ref: string): JSONSchemaDefinition {
   if (!ref.startsWith('#/')) {
-    throw new ValueError(`Unexpected $ref format ${JSON.stringify(ref)}; Does not start with #/`);
+    throw new Error(`Unexpected $ref format ${JSON.stringify(ref)}; Does not start with #/`);
   }
 
   const pathParts = ref.slice(2).split('/');
@@ -156,10 +156,3 @@ function hasMoreThanNKeys(obj: Record<string, any>, n: number): boolean {
   }
   return false;
 }
-
-class ValueError extends Error {
-  constructor(message: string) {
-    super(message);
-    this.name = 'ValueError';
-  }
-}