@exaudeus/workrail 1.7.2 → 1.7.3

package/dist/manifest.json

@@ -522,12 +522,12 @@
       "bytes": 3982
     },
     "mcp/handler-factory.d.ts": {
-      "sha256": "fad8fff32776b2fe7182708ce0e821c2434fdfce7191390bb9cfaf16a7d31a60",
-      "bytes": 796
+      "sha256": "27ee92b476ca59ed6daca56b228b27bc4c84a061106fa61f2554bba98703ba28",
+      "bytes": 838
     },
     "mcp/handler-factory.js": {
-      "sha256": "711bdf8bd71eaf2ad2a6b6b1bf84db004030e06223ce52075e01470a3bffd88d",
-      "bytes": 3805
+      "sha256": "8cc0d3e1a7871048b9033a90fb63b3dfd91148c16d22ac5336a767ed2d04a10b",
+      "bytes": 3907
     },
     "mcp/handlers/session.d.ts": {
       "sha256": "38926e69a0e4935d1dbcdc53848be9fff0e4d8f96827883da3216f217918fa82",
@@ -830,16 +830,16 @@
       "bytes": 408
     },
     "mcp/v2/tool-registry.js": {
-      "sha256": "4b192604ddafd986416865041db7e395d205da6fa1d398bec07646900f6ce28c",
-      "bytes": 3078
+      "sha256": "82539ddccffd58e3dee09b6ee859721fe69da024f55f25b3ee1113d48363d4e4",
+      "bytes": 3119
     },
     "mcp/v2/tools.d.ts": {
-      "sha256": "8d37b32d7a00d534b2ecbdc71018e3849714531ec3021862427494be85909eab",
-      "bytes": 5486
+      "sha256": "567938a5c153159b7792600cbe2326ce446a28f46d5f49e34dc1663cf826427e",
+      "bytes": 6737
     },
     "mcp/v2/tools.js": {
-      "sha256": "5af4e1a5a3a6c43df6db4ea40eb6d12fd0757af94f4e3e80d9f875ecf0181544",
-      "bytes": 7812
+      "sha256": "49d00fe392c526ce5f7e287ba98180cc82574b0c85ff0b550f3f01a216ac5dbf",
+      "bytes": 7938
     },
     "mcp/validation/bounded-json.d.ts": {
       "sha256": "82203ac6123d5c6989606c3b5405aaea99ab829c8958835f9ae3ba45b8bc8fd5",
@@ -858,12 +858,12 @@
       "bytes": 4664
     },
     "mcp/validation/schema-introspection.d.ts": {
-      "sha256": "7e0262e76234dd37079156027e95a30987b8949351f3e9ec0fd7b2be093a159d",
-      "bytes": 713
+      "sha256": "48b0f5c762bbfb82014b4424c65a5006e19fe871cc86a6ece4fe41c1110cc45f",
+      "bytes": 767
     },
     "mcp/validation/schema-introspection.js": {
-      "sha256": "850c09a3c01a5f22440ebc34236c393f3b428748210f5277258a9905cb847d71",
-      "bytes": 5293
+      "sha256": "03cc06529e37525264064a0d9c05dfa9cc17eb9043dcbf559c1b0a62fbd08081",
+      "bytes": 5448
     },
     "mcp/validation/string-similarity.d.ts": {
       "sha256": "4326210a768a526336b54d4ea20a128a939d92f53e8b2a5a33da06b5372d196a",
@@ -874,20 +874,20 @@
       "bytes": 2627
     },
     "mcp/validation/suggestion-config.d.ts": {
-      "sha256": "70b8395db74ec18bb1ef2309dd16516345b75839d2793bf9c5bfbd1e1d1baa0e",
-      "bytes": 388
+      "sha256": "ce63e2e63cfc316579d5f269210a60913beead0a131acb0cec8ad228ea81888f",
+      "bytes": 437
     },
     "mcp/validation/suggestion-config.js": {
-      "sha256": "efda43e48812979d0ddae2abe23809b6b5a3e5b955e74ca5a67716933e468db4",
-      "bytes": 592
+      "sha256": "907d37fdb1058491d27739f14b32a0371b79d892f5c193503d0c9e34833c444b",
+      "bytes": 667
     },
     "mcp/validation/suggestion-generator.d.ts": {
       "sha256": "159bde7596e7459f6025e152a24f1ab850d5cbd2bf55461036a05709dc5dbaa1",
       "bytes": 731
     },
     "mcp/validation/suggestion-generator.js": {
-      "sha256": "e4988260027f7c17398da69a7b1b8dcbb266f1b42100c63723a6d99426a0802e",
-      "bytes": 5250
+      "sha256": "e4e6740c4010b550de89ae97d50b9e66eebb8d34de13c33a62a6988c34264e79",
+      "bytes": 5284
     },
     "mcp/validation/suggestion-types.d.ts": {
       "sha256": "b93ae2e42f4b24789dcbe19db31a41af9534ad0dca85635339c2a10db42e298b",

package/dist/mcp/handler-factory.d.ts

@@ -3,5 +3,5 @@ import type { ToolContext, ToolResult } from './types.js';
 import type { PreValidateResult } from './validation/workflow-next-prevalidate.js';
 import type { WrappedToolHandler, McpCallToolResult } from './types/workflow-tool-edition.js';
 export declare function toMcpResult<T>(result: ToolResult<T>): McpCallToolResult;
-export declare function createHandler<TInput extends z.ZodType, TOutput>(schema: TInput, handler: (input: z.infer<TInput>, ctx: ToolContext) => Promise<ToolResult<TOutput>>): WrappedToolHandler;
+export declare function createHandler<TInput extends z.ZodType, TOutput>(schema: TInput, handler: (input: z.infer<TInput>, ctx: ToolContext) => Promise<ToolResult<TOutput>>, shapeSchema?: z.ZodObject<z.ZodRawShape>): WrappedToolHandler;
 export declare function createValidatingHandler<TInput extends z.ZodType, TOutput>(schema: TInput, preValidate: (args: unknown) => PreValidateResult, handler: (input: z.infer<TInput>, ctx: ToolContext) => Promise<ToolResult<TOutput>>): WrappedToolHandler;

package/dist/mcp/handler-factory.js

@@ -28,13 +28,14 @@ function toMcpResult(result) {
             };
     }
 }
-function createHandler(schema, handler) {
+function createHandler(schema, handler, shapeSchema) {
     return async (args, ctx) => {
         const parseResult = schema.safeParse(args);
         if (!parseResult.success) {
-            const suggestionResult = (0, index_js_1.generateSuggestions)(args, schema, index_js_1.DEFAULT_SUGGESTION_CONFIG);
+            const introspectionSchema = shapeSchema ?? schema;
+            const suggestionResult = (0, index_js_1.generateSuggestions)(args, introspectionSchema, index_js_1.DEFAULT_SUGGESTION_CONFIG);
             const suggestionDetails = (0, index_js_1.formatSuggestionDetails)(suggestionResult);
-            const patchedTemplate = (0, index_js_1.patchTemplateForFailedOptionals)(suggestionDetails.correctTemplate ?? null, args, parseResult.error.errors, schema, index_js_1.DEFAULT_SUGGESTION_CONFIG.maxTemplateDepth);
+            const patchedTemplate = (0, index_js_1.patchTemplateForFailedOptionals)(suggestionDetails.correctTemplate ?? null, args, parseResult.error.errors, introspectionSchema, index_js_1.DEFAULT_SUGGESTION_CONFIG.maxTemplateDepth);
             const patchedDetails = patchedTemplate !== suggestionDetails.correctTemplate
                 ? { ...suggestionDetails, correctTemplate: patchedTemplate }
                 : suggestionDetails;

package/dist/mcp/v2/tool-registry.js

@@ -50,7 +50,7 @@ function buildV2ToolRegistry(buildTool) {
         list_workflows: (0, handler_factory_js_1.createHandler)(tools_js_1.V2ListWorkflowsInput, v2_workflow_js_1.handleV2ListWorkflows),
         inspect_workflow: (0, handler_factory_js_1.createHandler)(tools_js_1.V2InspectWorkflowInput, v2_workflow_js_1.handleV2InspectWorkflow),
         start_workflow: (0, handler_factory_js_1.createHandler)(tools_js_1.V2StartWorkflowInput, v2_execution_js_1.handleV2StartWorkflow),
-        continue_workflow: (0, handler_factory_js_1.createHandler)(tools_js_1.V2ContinueWorkflowInput, v2_execution_js_1.handleV2ContinueWorkflow),
+        continue_workflow: (0, handler_factory_js_1.createHandler)(tools_js_1.V2ContinueWorkflowInput, v2_execution_js_1.handleV2ContinueWorkflow, tools_js_1.V2ContinueWorkflowInputShape),
         checkpoint_workflow: (0, handler_factory_js_1.createHandler)(tools_js_1.V2CheckpointWorkflowInput, v2_checkpoint_js_1.handleV2CheckpointWorkflow),
         resume_session: (0, handler_factory_js_1.createHandler)(tools_js_1.V2ResumeSessionInput, v2_resume_js_1.handleV2ResumeSession),
     };

package/dist/mcp/v2/tools.d.ts

@@ -27,6 +27,40 @@ export declare const V2StartWorkflowInput: z.ZodObject<{
     workspacePath?: string | undefined;
 }>;
 export type V2StartWorkflowInput = z.infer<typeof V2StartWorkflowInput>;
+export declare const V2ContinueWorkflowInputShape: z.ZodObject<{
+    intent: z.ZodOptional<z.ZodEnum<["advance", "rehydrate"]>>;
+    stateToken: z.ZodString;
+    ackToken: z.ZodOptional<z.ZodString>;
+    context: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    output: z.ZodOptional<z.ZodObject<{
+        notesMarkdown: z.ZodOptional<z.ZodString>;
+        artifacts: z.ZodOptional<z.ZodArray<z.ZodUnknown, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        notesMarkdown?: string | undefined;
+        artifacts?: unknown[] | undefined;
+    }, {
+        notesMarkdown?: string | undefined;
+        artifacts?: unknown[] | undefined;
+    }>>;
+}, "strict", z.ZodTypeAny, {
+    stateToken: string;
+    context?: Record<string, unknown> | undefined;
+    output?: {
+        notesMarkdown?: string | undefined;
+        artifacts?: unknown[] | undefined;
+    } | undefined;
+    intent?: "advance" | "rehydrate" | undefined;
+    ackToken?: string | undefined;
+}, {
+    stateToken: string;
+    context?: Record<string, unknown> | undefined;
+    output?: {
+        notesMarkdown?: string | undefined;
+        artifacts?: unknown[] | undefined;
+    } | undefined;
+    intent?: "advance" | "rehydrate" | undefined;
+    ackToken?: string | undefined;
+}>;
 export declare const V2ContinueWorkflowInput: z.ZodPipeline<z.ZodEffects<z.ZodObject<{
     intent: z.ZodOptional<z.ZodEnum<["advance", "rehydrate"]>>;
     stateToken: z.ZodString;

package/dist/mcp/v2/tools.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.V2_TOOL_ANNOTATIONS = exports.V2_TOOL_TITLES = exports.V2CheckpointWorkflowInput = exports.V2ResumeSessionInput = exports.V2ContinueWorkflowInput = exports.V2StartWorkflowInput = exports.V2InspectWorkflowInput = exports.V2ListWorkflowsInput = void 0;
+exports.V2_TOOL_ANNOTATIONS = exports.V2_TOOL_TITLES = exports.V2CheckpointWorkflowInput = exports.V2ResumeSessionInput = exports.V2ContinueWorkflowInput = exports.V2ContinueWorkflowInputShape = exports.V2StartWorkflowInput = exports.V2InspectWorkflowInput = exports.V2ListWorkflowsInput = void 0;
 const zod_1 = require("zod");
 exports.V2ListWorkflowsInput = zod_1.z.object({});
 exports.V2InspectWorkflowInput = zod_1.z.object({
@@ -15,7 +15,7 @@ exports.V2StartWorkflowInput = zod_1.z.object({
         .optional()
         .describe('Absolute path to your current workspace directory (e.g. the "Workspace:" value from your system parameters). Used to anchor this session to your workspace for future resume_session discovery. Pass this on every start_workflow call. If omitted, WorkRail uses the server process directory which may not match your workspace.'),
 });
-exports.V2ContinueWorkflowInput = zod_1.z.object({
+exports.V2ContinueWorkflowInputShape = zod_1.z.object({
     intent: zod_1.z.enum(['advance', 'rehydrate']).optional().describe('What you want to do. Auto-inferred from ackToken if omitted: ' +
         'ackToken present → "advance", ackToken absent → "rehydrate". ' +
         '"advance": I completed the current step — requires ackToken. ' +
@@ -30,10 +30,13 @@ exports.V2ContinueWorkflowInput = zod_1.z.object({
     })
         .optional()
         .describe('Durable output to attach to the current node. Only valid when intent is "advance".'),
-}).strict().transform((data) => {
+}).strict();
+exports.V2ContinueWorkflowInput = exports.V2ContinueWorkflowInputShape
+    .transform((data) => {
     const intent = data.intent ?? (data.ackToken ? 'advance' : 'rehydrate');
     return { ...data, intent };
-}).pipe(zod_1.z.custom().superRefine((data, ctx) => {
+})
+    .pipe(zod_1.z.custom().superRefine((data, ctx) => {
     if (data.intent === 'advance' && !data.ackToken) {
         ctx.addIssue({
             code: zod_1.z.ZodIssueCode.custom,

package/dist/mcp/validation/schema-introspection.d.ts

@@ -3,6 +3,6 @@ export declare function extractExpectedKeys(schema: z.ZodType): readonly string[
 export declare function extractRequiredKeys(schema: z.ZodType): readonly string[];
 export declare function findUnknownKeys(args: unknown, schema: z.ZodType): readonly string[];
 export declare function findMissingRequiredKeys(args: unknown, schema: z.ZodType): readonly string[];
-export declare function generateExampleValue(schema: z.ZodType, depth?: number, maxDepth?: number): unknown;
-export declare function generateTemplate(schema: z.ZodType, maxDepth?: number): Readonly<Record<string, unknown>> | null;
+export declare function generateExampleValue(schema: z.ZodType, depth?: number, maxDepth?: number, includeOptional?: boolean): unknown;
+export declare function generateTemplate(schema: z.ZodType, maxDepth?: number, includeOptional?: boolean): Readonly<Record<string, unknown>> | null;
 export declare function extractEnumValues(schema: z.ZodType, path: string): readonly string[];

package/dist/mcp/validation/schema-introspection.js

@@ -44,7 +44,7 @@ function findMissingRequiredKeys(args, schema) {
     const requiredKeys = extractRequiredKeys(schema);
     return requiredKeys.filter(key => !providedKeys.has(key));
 }
-function generateExampleValue(schema, depth = 0, maxDepth = 3) {
+function generateExampleValue(schema, depth = 0, maxDepth = 3, includeOptional = false) {
     if (depth > maxDepth) {
         return '...';
     }
@@ -52,23 +52,23 @@ function generateExampleValue(schema, depth = 0, maxDepth = 3) {
         return schema._def.defaultValue();
     }
     if (schema instanceof zod_1.z.ZodOptional) {
-        return generateExampleValue(schema._def.innerType, depth, maxDepth);
+        return generateExampleValue(schema._def.innerType, depth, maxDepth, includeOptional);
     }
     if (schema instanceof zod_1.z.ZodObject) {
         const shape = schema._def.shape();
         const result = {};
         for (const [key, value] of Object.entries(shape)) {
             const field = value;
-            if (field instanceof zod_1.z.ZodOptional)
+            if (!includeOptional && field instanceof zod_1.z.ZodOptional)
                 continue;
-            result[key] = generateExampleValue(field, depth + 1, maxDepth);
+            result[key] = generateExampleValue(field, depth + 1, maxDepth, includeOptional);
         }
         return result;
     }
     if (schema instanceof zod_1.z.ZodDiscriminatedUnion) {
         const options = schema._def.options;
         if (options.length > 0) {
-            return generateExampleValue(options[0], depth + 1, maxDepth);
+            return generateExampleValue(options[0], depth + 1, maxDepth, includeOptional);
         }
         return {};
     }
@@ -105,15 +105,15 @@ function generateExampleValue(schema, depth = 0, maxDepth = 3) {
         return '<any>';
     }
     if (schema instanceof zod_1.z.ZodEffects) {
-        return generateExampleValue(schema._def.schema, depth, maxDepth);
+        return generateExampleValue(schema._def.schema, depth, maxDepth, includeOptional);
     }
     return '<unknown>';
 }
-function generateTemplate(schema, maxDepth = 3) {
+function generateTemplate(schema, maxDepth = 3, includeOptional = false) {
     if (!(schema instanceof zod_1.z.ZodObject)) {
         return null;
     }
-    const example = generateExampleValue(schema, 0, maxDepth);
+    const example = generateExampleValue(schema, 0, maxDepth, includeOptional);
     if (typeof example === 'object' && example !== null) {
         return example;
     }

package/dist/mcp/validation/suggestion-config.d.ts

@@ -4,6 +4,7 @@ export interface SuggestionConfig {
     readonly maxSuggestions: number;
     readonly includeTemplate: boolean;
     readonly maxTemplateDepth: number;
+    readonly includeOptionalInTemplate: boolean;
 }
 export declare const DEFAULT_SUGGESTION_CONFIG: SuggestionConfig;
 export declare const MINIMAL_SUGGESTION_CONFIG: SuggestionConfig;

package/dist/mcp/validation/suggestion-config.js

@@ -7,10 +7,12 @@ exports.DEFAULT_SUGGESTION_CONFIG = {
     maxSuggestions: 3,
     includeTemplate: true,
     maxTemplateDepth: 3,
+    includeOptionalInTemplate: true,
 };
 exports.MINIMAL_SUGGESTION_CONFIG = {
     similarityThreshold: (0, suggestion_types_js_1.similarity)(0.7),
     maxSuggestions: 1,
     includeTemplate: false,
     maxTemplateDepth: 1,
+    includeOptionalInTemplate: false,
 };

package/dist/mcp/validation/suggestion-generator.js

@@ -56,7 +56,7 @@ function generateSuggestions(args, schema, config) {
         return suggestion_types_js_1.EMPTY_SUGGESTION_RESULT;
     }
     const correctTemplate = config.includeTemplate
-        ? (0, schema_introspection_js_1.generateTemplate)(schema, config.maxTemplateDepth)
+        ? (0, schema_introspection_js_1.generateTemplate)(schema, config.maxTemplateDepth, config.includeOptionalInTemplate)
         : null;
     return {
         suggestions,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exaudeus/workrail",
-  "version": "1.7.2",
+  "version": "1.7.3",
   "description": "Step-by-step workflow enforcement for AI agents via MCP",
   "license": "MIT",
   "repository": {