@axlsdk/axl 0.8.0 → 0.9.1

package/dist/index.d.cts

@@ -200,10 +200,12 @@ type VoteOptions<T> = {
 type VerifyRetry<T> = {
     /** Error message from the failed attempt (schema or validate). */
     error: string;
-    /** Raw return value from the previous fn call. */
+    /** Raw return value from the previous fn call. When fn() throws a ValidationError
+     *  or VerifyError, falls back to err.lastOutput so the retry has data to repair. */
     output: unknown;
     /** Schema-parsed object — only present when schema passed but validate failed.
-     *  Safe to modify and return as the next attempt. */
+     *  Also populated from ValidationError.lastOutput when fn() throws (e.g., inner
+     *  ctx.ask() exhausted its validate retries). Safe to modify and return. */
     parsed?: T;
 };
 /** Verify options */
@@ -912,8 +914,10 @@ declare class MemoryManager {
     close(): Promise<void>;
 }
 
-/** Convert a Zod schema to JSON Schema (subset). Exported for Studio tool introspection. */
-declare function zodToJsonSchema(schema: z.ZodTypeAny): unknown;
+/** Convert a Zod schema to JSON Schema. Exported for Studio tool introspection.
+ *  Wraps Zod v4's built-in `z.toJSONSchema()`, stripping the `$schema` key
+ *  since tool parameter schemas are embedded objects, not standalone documents. */
+declare function zodToJsonSchema(schema: z.ZodType): Record<string, unknown>;
 type WorkflowContextInit = {
     input: unknown;
     executionId: string;
@@ -1100,7 +1104,7 @@ type ToolHooks<TInput = unknown, TOutput = unknown> = {
     after?(output: TOutput, ctx: WorkflowContext): TOutput | Promise<TOutput>;
 };
 /** Tool configuration */
-type ToolConfig<TInput extends z.ZodTypeAny, TOutput = unknown> = {
+type ToolConfig<TInput extends z.ZodType, TOutput = unknown> = {
     name: string;
     description: string;
     input: TInput;
@@ -1115,7 +1119,7 @@ type ToolConfig<TInput extends z.ZodTypeAny, TOutput = unknown> = {
     hooks?: ToolHooks<z.infer<TInput>, TOutput>;
 };
 /** A defined tool instance */
-type Tool<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput = unknown> = {
+type Tool<TInput extends z.ZodType = z.ZodType, TOutput = unknown> = {
     readonly name: string;
     readonly description: string;
     readonly inputSchema: TInput;
@@ -1133,17 +1137,17 @@ type Tool<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput = unknown> = {
  * @param config - Tool configuration: name, description, input schema, handler, retry, and sensitivity options.
  * @returns A Tool instance that can be attached to agents and invoked via `tool.run()` or agent tool calling.
  */
-declare function tool<TInput extends z.ZodTypeAny, TOutput = unknown>(config: ToolConfig<TInput, TOutput>): Tool<TInput, TOutput>;
+declare function tool<TInput extends z.ZodType, TOutput = unknown>(config: ToolConfig<TInput, TOutput>): Tool<TInput, TOutput>;
 
 /** Workflow configuration */
-type WorkflowConfig<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput extends z.ZodTypeAny = z.ZodTypeAny> = {
+type WorkflowConfig<TInput extends z.ZodType = z.ZodType, TOutput extends z.ZodType = z.ZodType> = {
     name: string;
     input: TInput;
     output?: TOutput;
     handler: (ctx: WorkflowContext<z.infer<TInput>>) => Promise<z.infer<TOutput>>;
 };
 /** A defined workflow instance */
-type Workflow<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput extends z.ZodTypeAny = z.ZodTypeAny> = {
+type Workflow<TInput extends z.ZodType = z.ZodType, TOutput extends z.ZodType = z.ZodType> = {
     readonly name: string;
     readonly inputSchema: TInput;
     readonly outputSchema: TOutput | undefined;
@@ -1155,7 +1159,7 @@ type Workflow<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput extends z.ZodT
  * @param config - Workflow configuration: name, input schema, optional output schema, and async handler receiving a WorkflowContext.
  * @returns A Workflow instance ready to be registered with an AxlRuntime.
  */
-declare function workflow<TInput extends z.ZodTypeAny, TOutput extends z.ZodTypeAny = z.ZodTypeAny>(config: WorkflowConfig<TInput, TOutput>): Workflow<TInput, TOutput>;
+declare function workflow<TInput extends z.ZodType, TOutput extends z.ZodType = z.ZodType>(config: WorkflowConfig<TInput, TOutput>): Workflow<TInput, TOutput>;
 
 /**
  * A streamable workflow execution.

package/dist/index.d.ts

@@ -200,10 +200,12 @@ type VoteOptions<T> = {
 type VerifyRetry<T> = {
     /** Error message from the failed attempt (schema or validate). */
     error: string;
-    /** Raw return value from the previous fn call. */
+    /** Raw return value from the previous fn call. When fn() throws a ValidationError
+     *  or VerifyError, falls back to err.lastOutput so the retry has data to repair. */
     output: unknown;
     /** Schema-parsed object — only present when schema passed but validate failed.
-     *  Safe to modify and return as the next attempt. */
+     *  Also populated from ValidationError.lastOutput when fn() throws (e.g., inner
+     *  ctx.ask() exhausted its validate retries). Safe to modify and return. */
     parsed?: T;
 };
 /** Verify options */
@@ -912,8 +914,10 @@ declare class MemoryManager {
     close(): Promise<void>;
 }
 
-/** Convert a Zod schema to JSON Schema (subset). Exported for Studio tool introspection. */
-declare function zodToJsonSchema(schema: z.ZodTypeAny): unknown;
+/** Convert a Zod schema to JSON Schema. Exported for Studio tool introspection.
+ *  Wraps Zod v4's built-in `z.toJSONSchema()`, stripping the `$schema` key
+ *  since tool parameter schemas are embedded objects, not standalone documents. */
+declare function zodToJsonSchema(schema: z.ZodType): Record<string, unknown>;
 type WorkflowContextInit = {
     input: unknown;
     executionId: string;
@@ -1100,7 +1104,7 @@ type ToolHooks<TInput = unknown, TOutput = unknown> = {
     after?(output: TOutput, ctx: WorkflowContext): TOutput | Promise<TOutput>;
 };
 /** Tool configuration */
-type ToolConfig<TInput extends z.ZodTypeAny, TOutput = unknown> = {
+type ToolConfig<TInput extends z.ZodType, TOutput = unknown> = {
     name: string;
     description: string;
     input: TInput;
@@ -1115,7 +1119,7 @@ type ToolConfig<TInput extends z.ZodTypeAny, TOutput = unknown> = {
     hooks?: ToolHooks<z.infer<TInput>, TOutput>;
 };
 /** A defined tool instance */
-type Tool<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput = unknown> = {
+type Tool<TInput extends z.ZodType = z.ZodType, TOutput = unknown> = {
     readonly name: string;
     readonly description: string;
     readonly inputSchema: TInput;
@@ -1133,17 +1137,17 @@ type Tool<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput = unknown> = {
  * @param config - Tool configuration: name, description, input schema, handler, retry, and sensitivity options.
  * @returns A Tool instance that can be attached to agents and invoked via `tool.run()` or agent tool calling.
  */
-declare function tool<TInput extends z.ZodTypeAny, TOutput = unknown>(config: ToolConfig<TInput, TOutput>): Tool<TInput, TOutput>;
+declare function tool<TInput extends z.ZodType, TOutput = unknown>(config: ToolConfig<TInput, TOutput>): Tool<TInput, TOutput>;
 
 /** Workflow configuration */
-type WorkflowConfig<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput extends z.ZodTypeAny = z.ZodTypeAny> = {
+type WorkflowConfig<TInput extends z.ZodType = z.ZodType, TOutput extends z.ZodType = z.ZodType> = {
     name: string;
     input: TInput;
     output?: TOutput;
     handler: (ctx: WorkflowContext<z.infer<TInput>>) => Promise<z.infer<TOutput>>;
 };
 /** A defined workflow instance */
-type Workflow<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput extends z.ZodTypeAny = z.ZodTypeAny> = {
+type Workflow<TInput extends z.ZodType = z.ZodType, TOutput extends z.ZodType = z.ZodType> = {
     readonly name: string;
     readonly inputSchema: TInput;
     readonly outputSchema: TOutput | undefined;
@@ -1155,7 +1159,7 @@ type Workflow<TInput extends z.ZodTypeAny = z.ZodTypeAny, TOutput extends z.ZodT
  * @param config - Workflow configuration: name, input schema, optional output schema, and async handler receiving a WorkflowContext.
  * @returns A Workflow instance ready to be registered with an AxlRuntime.
  */
-declare function workflow<TInput extends z.ZodTypeAny, TOutput extends z.ZodTypeAny = z.ZodTypeAny>(config: WorkflowConfig<TInput, TOutput>): Workflow<TInput, TOutput>;
+declare function workflow<TInput extends z.ZodType, TOutput extends z.ZodType = z.ZodType>(config: WorkflowConfig<TInput, TOutput>): Workflow<TInput, TOutput>;
 
 /**
  * A streamable workflow execution.

package/dist/index.js

@@ -1922,7 +1922,7 @@ var defaultRegistry = new ProviderRegistry();
 
 // src/context.ts
 import { AsyncLocalStorage } from "async_hooks";
-import { ZodError } from "zod";
+import { z, ZodError } from "zod";
 
 // src/errors.ts
 var AxlError = class extends Error {
@@ -2102,48 +2102,9 @@ function resolveConfig(config) {
 // src/context.ts
 var signalStorage = new AsyncLocalStorage();
 function zodToJsonSchema(schema) {
-  const def = schema._def;
-  if (!def || !def.typeName) return {};
-  switch (def.typeName) {
-    case "ZodString":
-      return { type: "string" };
-    case "ZodNumber":
-      return { type: "number" };
-    case "ZodBoolean":
-      return { type: "boolean" };
-    case "ZodArray":
-      return { type: "array", items: zodToJsonSchema(def.type) };
-    case "ZodObject": {
-      const shape = def.shape?.() ?? {};
-      const properties = {};
-      const required = [];
-      for (const [key, value] of Object.entries(shape)) {
-        properties[key] = zodToJsonSchema(value);
-        const innerDef = value._def;
-        if (innerDef?.typeName !== "ZodOptional" && innerDef?.typeName !== "ZodDefault") {
-          required.push(key);
-        }
-      }
-      return { type: "object", properties, required: required.length > 0 ? required : void 0 };
-    }
-    case "ZodOptional":
-      return zodToJsonSchema(def.innerType);
-    case "ZodDefault":
-      return zodToJsonSchema(def.innerType);
-    case "ZodEnum":
-      return { type: "string", enum: def.values };
-    case "ZodLiteral": {
-      const v = def.value;
-      const t = v === null ? "null" : typeof v;
-      return { type: t, const: v };
-    }
-    case "ZodUnion":
-      return { oneOf: def.options.map((o) => zodToJsonSchema(o)) };
-    case "ZodNullable":
-      return { ...zodToJsonSchema(def.innerType), nullable: true };
-    default:
-      return {};
-  }
+  const result = z.toJSONSchema(schema, { unrepresentable: "any" });
+  delete result.$schema;
+  return result;
 }
 function estimateTokens(text) {
   return Math.ceil(text.length / 4);
@@ -3384,7 +3345,7 @@ ${summaryResponse.content}`
         if (err instanceof ValidationError) {
           lastRetry = {
             error: err.reason,
-            output: rawOutput,
+            output: rawOutput ?? err.lastOutput,
             parsed: err.lastOutput
           };
           if (attempt === maxRetries) {
@@ -3393,6 +3354,14 @@ ${summaryResponse.content}`
           }
           continue;
         }
+        if (err instanceof VerifyError) {
+          lastRetry = { error: err.message, output: rawOutput ?? err.lastOutput };
+          if (attempt === maxRetries) {
+            if (options?.fallback !== void 0) return options.fallback;
+            throw err;
+          }
+          continue;
+        }
         const errorMsg = err instanceof ZodError ? err.message : err instanceof Error ? err.message : String(err);
         lastRetry = { error: errorMsg, output: rawOutput };
         if (attempt === maxRetries) {