@axlsdk/axl 0.8.0 → 0.9.0

package/dist/index.d.cts

@@ -912,8 +912,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 +1102,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 +1117,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 +1135,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 +1157,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

@@ -912,8 +912,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 +1102,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 +1117,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 +1135,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 +1157,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);