@exaudeus/workrail 1.5.4 → 1.6.1

package/dist/manifest.json

@@ -686,16 +686,16 @@
       "bytes": 2789
     },
     "mcp/handlers/v2-execution/start.js": {
-      "sha256": "6d6763c4cae68d573d228826af6826e830ce70d016772c74d4780207807b1928",
-      "bytes": 15078
+      "sha256": "5eab7a7a5fdf021f7e4bc656de1dbfe791e6610ab9e1bacb15dad4f6423f1d58",
+      "bytes": 15248
     },
     "mcp/handlers/v2-resume.d.ts": {
       "sha256": "d88f6c35bcaf946666c837b72fda3702a2ebab5e478eb90f7b4b672a0e5fa24f",
       "bytes": 471
     },
     "mcp/handlers/v2-resume.js": {
-      "sha256": "b4ce6f9cc3dffef91ca7da2e428d659c4ff79ceaf984c87d672290215936db97",
-      "bytes": 2771
+      "sha256": "a8f51a680eec031e23148bd64f6d647d765c79eec738cbd6182650533248f16f",
+      "bytes": 2929
     },
     "mcp/handlers/v2-state-conversion.d.ts": {
       "sha256": "94bd06904ef58dd210ff17ffb75c2492beea8937eb06d99749e5d860c0e0d96b",
@@ -750,8 +750,8 @@
       "bytes": 168
     },
     "mcp/server.js": {
-      "sha256": "37ab570be76b379a4076e78e5bc82a06b7ee1fc73d3d5010ad48b5650e54365a",
-      "bytes": 10631
+      "sha256": "fd9e264c759db2d5446d88eafce14ec27ce9b73756e62ef10351b2fca52c7c7c",
+      "bytes": 11892
     },
     "mcp/tool-description-provider.d.ts": {
       "sha256": "1d46abc3112e11b68e57197e846f5708293ec9b2281fa71a9124ee2aad71e41b",
@@ -786,8 +786,8 @@
       "bytes": 8360
     },
     "mcp/types.d.ts": {
-      "sha256": "ce700982b88189f08c466384b4357b3345ec0ccb6842ff97a0831c2281185c6c",
-      "bytes": 4284
+      "sha256": "2ef7f9c1bffbe365b06469dfad43b45c4ae3b5292dd269b33423bf1c8c9c7c43",
+      "bytes": 4355
     },
     "mcp/types.js": {
       "sha256": "d10c4070e4c3454d80f0fa9cdc0e978c69c53c13fd09baa8710fcd802fed8926",
@@ -905,6 +905,14 @@
       "sha256": "08870a96c6f32c09e617f43f6eb83593c15007fe39241d498bd9bbbd420101fc",
       "bytes": 635
     },
+    "mcp/workspace-roots-manager.d.ts": {
+      "sha256": "32f342eb4cf2c1a94982f8077dea4185b74f118c1587b40a370200cf2aa06f9c",
+      "bytes": 363
+    },
+    "mcp/workspace-roots-manager.js": {
+      "sha256": "1037fa12885161c91af1beed0dc0a0cb05a5636b4a63d1756e1ed3cab76e1d32",
+      "bytes": 419
+    },
     "mcp/zod-to-json-schema.d.ts": {
       "sha256": "c80cfa6980a88c823ac47d04706b2223a351de3f4c76ee778f81ad3dda2c766b",
       "bytes": 456
@@ -1806,8 +1814,8 @@
       "bytes": 1069
     },
     "v2/infra/local/session-summary-provider/index.js": {
-      "sha256": "5dadd21632795d624d0be4be1957329242df6c105ccf507e024e397ab3e63e7f",
-      "bytes": 5728
+      "sha256": "736989809f7a78b3b6b0584609d8cce8424fde20e4eb8e7c513d7b012d45d9d2",
+      "bytes": 6463
     },
     "v2/infra/local/sha256/index.d.ts": {
       "sha256": "8a727b7e54a38275ca6f9f1b8730f97cfc0a212df035df1bdc58e716e6824230",
@@ -1842,12 +1850,12 @@
       "bytes": 304
     },
     "v2/infra/local/workspace-anchor/index.d.ts": {
-      "sha256": "b56193c3da9edb1012d2bbdd8b8dd33e7d62e23a08ecb82c328705d5bca14765",
-      "bytes": 431
+      "sha256": "90515ae92ac0230a279a28b82e26bc8be8f62c4cd6f6999630dc07f9a64e3296",
+      "bytes": 705
     },
     "v2/infra/local/workspace-anchor/index.js": {
-      "sha256": "1d49c3a0115603a7ee995a47f3b80bbd0fc80ca285ce9768e00036dc700b05a9",
-      "bytes": 1477
+      "sha256": "69541a7f2a6837ff0790c99740000a71521a60671b4fd7c4a2245a48e56bb9b7",
+      "bytes": 2102
     },
     "v2/ports/base32.port.d.ts": {
       "sha256": "64aa2f2003a552917cbf71474472fc5e4afffaa29577204bbcbe5ffa989ceb82",
@@ -1986,8 +1994,8 @@
       "bytes": 77
     },
     "v2/ports/workspace-anchor.port.d.ts": {
-      "sha256": "3c93de8b4e971c70c16adc0de85c8e849bc57c48a18525111648da45d044384c",
-      "bytes": 522
+      "sha256": "228cce9bff3cbaa430bf38be3a1bc326885ec07c7952af2824735bc7adbf9af3",
+      "bytes": 759
     },
     "v2/ports/workspace-anchor.port.js": {
       "sha256": "d43aa81f5bc89faa359e0f97c814ba25155591ff078fbb9bfd40f8c7c9683230",

package/dist/mcp/handlers/v2-execution/start.js

@@ -229,9 +229,12 @@ function executeStartWorkflow(input, ctx) {
         pinnedStore,
     })
         .andThen(({ workflow, firstStep, workflowHash, pinnedWorkflow }) => {
-        const workspaceAnchor = ctx.v2?.workspaceAnchor;
-        const anchorsRA = workspaceAnchor
-            ? workspaceAnchor.resolveAnchors()
+        const workspaceResolver = ctx.v2.workspaceResolver;
+        const primaryRootUri = ctx.v2.resolvedRootUris?.[0];
+        const anchorsRA = workspaceResolver
+            ? (primaryRootUri
+                ? workspaceResolver.resolveFromUri(primaryRootUri)
+                : workspaceResolver.resolveFromCwd())
                 .map((anchors) => (0, observation_builder_js_1.anchorsToObservations)(anchors))
                 .orElse(() => (0, neverthrow_1.okAsync)([]))
             : (0, neverthrow_1.okAsync)([]);

package/dist/mcp/handlers/v2-resume.js

@@ -19,8 +19,11 @@ async function handleV2ResumeSession(input, ctx) {
         return (0, types_js_1.errNotRetryable)('INTERNAL_ERROR', 'resume_session requires sessionSummaryProvider port');
     }
     let anchors = [];
-    if (v2.workspaceAnchor) {
-        const anchorRes = await v2.workspaceAnchor.resolveAnchors();
+    if (v2.workspaceResolver) {
+        const primaryRootUri = v2.resolvedRootUris?.[0];
+        const anchorRes = await (primaryRootUri
+            ? v2.workspaceResolver.resolveFromUri(primaryRootUri)
+            : v2.workspaceResolver.resolveFromCwd());
         if (anchorRes.isOk()) {
             anchors = anchorRes.value;
         }

package/dist/mcp/server.js

@@ -41,6 +41,7 @@ const tokens_js_1 = require("../di/tokens.js");
 const assert_never_js_1 = require("../runtime/assert-never.js");
 const token_codec_ports_js_1 = require("../v2/durable-core/tokens/token-codec-ports.js");
 const index_js_1 = require("../v2/infra/local/workspace-anchor/index.js");
+const workspace_roots_manager_js_1 = require("./workspace-roots-manager.js");
 const index_js_2 = require("../v2/infra/local/directory-listing/index.js");
 const index_js_3 = require("../v2/infra/local/session-summary-provider/index.js");
 const tool_factory_js_1 = require("./tool-factory.js");
@@ -101,7 +102,8 @@ async function createToolContext() {
                 crypto,
                 idFactory,
                 tokenCodecPorts,
-                workspaceAnchor: new index_js_1.LocalWorkspaceAnchorV2(process.cwd()),
+                resolvedRootUris: [],
+                workspaceResolver: new index_js_1.LocalWorkspaceAnchorV2(process.cwd()),
                 dataDir,
                 directoryListing,
                 sessionSummaryProvider: new index_js_3.LocalSessionSummaryProviderV2({
@@ -148,9 +150,10 @@ async function startServer() {
         default:
             (0, assert_never_js_1.assertNever)(workflowEdition);
     }
+    const rootsManager = new workspace_roots_manager_js_1.WorkspaceRootsManager();
     const { Server } = await Promise.resolve().then(() => __importStar(require('@modelcontextprotocol/sdk/server/index.js')));
     const { StdioServerTransport } = await Promise.resolve().then(() => __importStar(require('@modelcontextprotocol/sdk/server/stdio.js')));
-    const { CallToolRequestSchema, ListToolsRequestSchema, } = await Promise.resolve().then(() => __importStar(require('@modelcontextprotocol/sdk/types.js')));
+    const { CallToolRequestSchema, ListToolsRequestSchema, RootsListChangedNotificationSchema, } = await Promise.resolve().then(() => __importStar(require('@modelcontextprotocol/sdk/types.js')));
     const server = new Server({
         name: 'workrail-server',
         version: '0.1.0',
@@ -182,10 +185,31 @@ async function startServer() {
                 isError: true,
             };
         }
-        return handler(args ?? {}, ctx);
+        const requestCtx = ctx.v2
+            ? { ...ctx, v2: { ...ctx.v2, resolvedRootUris: rootsManager.getCurrentRootUris() } }
+            : ctx;
+        return handler(args ?? {}, requestCtx);
+    });
+    server.setNotificationHandler(RootsListChangedNotificationSchema, async () => {
+        try {
+            const result = await server.listRoots();
+            rootsManager.updateRootUris(result.roots.map((r) => r.uri));
+            console.error(`[Roots] Updated workspace roots: ${result.roots.map((r) => r.uri).join(', ') || '(none)'}`);
+        }
+        catch {
+            console.error('[Roots] Failed to fetch updated roots after change notification');
+        }
     });
     const transport = new StdioServerTransport();
     await server.connect(transport);
+    try {
+        const result = await server.listRoots();
+        rootsManager.updateRootUris(result.roots.map((r) => r.uri));
+        console.error(`[Roots] Initial workspace roots: ${result.roots.map((r) => r.uri).join(', ') || '(none)'}`);
+    }
+    catch {
+        console.error('[Roots] Client does not support roots/list; workspace context will use server CWD fallback');
+    }
     console.error('WorkRail MCP Server running on stdio');
     const shutdownEvents = container_js_1.container.resolve(tokens_js_1.DI.Runtime.ShutdownEvents);
     const processSignals = container_js_1.container.resolve(tokens_js_1.DI.Runtime.ProcessSignals);

package/dist/mcp/types.d.ts

@@ -12,7 +12,7 @@ import type { CryptoPortV2 } from '../v2/durable-core/canonical/hashing.js';
 import type { IdFactoryV2 } from '../v2/infra/local/id-factory/index.js';
 import type { JsonValue } from './output-schemas.js';
 import type { TokenCodecPorts } from '../v2/durable-core/tokens/token-codec-ports.js';
-import type { WorkspaceAnchorPortV2 } from '../v2/ports/workspace-anchor.port.js';
+import type { WorkspaceContextResolverPortV2 } from '../v2/ports/workspace-anchor.port.js';
 import type { DataDirPortV2 } from '../v2/ports/data-dir.port.js';
 import type { DirectoryListingPortV2 } from '../v2/ports/directory-listing.port.js';
 import type { SessionSummaryProviderPortV2 } from '../v2/ports/session-summary-provider.port.js';
@@ -55,7 +55,8 @@ export interface V2Dependencies {
     readonly crypto: CryptoPortV2;
     readonly idFactory: IdFactoryV2;
     readonly tokenCodecPorts: TokenCodecPorts;
-    readonly workspaceAnchor?: WorkspaceAnchorPortV2;
+    readonly resolvedRootUris?: readonly string[];
+    readonly workspaceResolver?: WorkspaceContextResolverPortV2;
     readonly dataDir?: DataDirPortV2;
     readonly directoryListing?: DirectoryListingPortV2;
     readonly sessionSummaryProvider?: SessionSummaryProviderPortV2;

package/dist/mcp/workspace-roots-manager.d.ts

@@ -0,0 +1,11 @@
+export interface RootsReader {
+    getCurrentRootUris(): readonly string[];
+}
+export interface RootsWriter {
+    updateRootUris(uris: readonly string[]): void;
+}
+export declare class WorkspaceRootsManager implements RootsReader, RootsWriter {
+    private rootUris;
+    updateRootUris(uris: readonly string[]): void;
+    getCurrentRootUris(): readonly string[];
+}

package/dist/mcp/workspace-roots-manager.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkspaceRootsManager = void 0;
+class WorkspaceRootsManager {
+    constructor() {
+        this.rootUris = Object.freeze([]);
+    }
+    updateRootUris(uris) {
+        this.rootUris = Object.freeze([...uris]);
+    }
+    getCurrentRootUris() {
+        return this.rootUris;
+    }
+}
+exports.WorkspaceRootsManager = WorkspaceRootsManager;

package/dist/v2/infra/local/session-summary-provider/index.js

@@ -68,7 +68,7 @@ class LocalSessionSummaryProviderV2 {
             const run = bestRun.run;
             const outputsRes = (0, node_outputs_js_1.projectNodeOutputsV2)(truth.events);
             const recapSnippet = outputsRes.isOk() && run.preferredTipNodeId
-                ? extractRecapFromOutputs(outputsRes.value, run.preferredTipNodeId)
+                ? extractAggregateRecap(outputsRes.value, run, run.preferredTipNodeId)
                 : null;
             const observations = extractObservations(truth.events);
             const workflow = extractWorkflowIdentity(truth.events, run.runId);
@@ -88,7 +88,19 @@ class LocalSessionSummaryProviderV2 {
     }
 }
 exports.LocalSessionSummaryProviderV2 = LocalSessionSummaryProviderV2;
-function extractRecapFromOutputs(outputs, nodeId) {
+const MAX_RECAP_ANCESTOR_DEPTH = 100;
+function collectAncestorNodeIds(nodesById, nodeId, remainingDepth) {
+    if (remainingDepth === 0)
+        return [nodeId];
+    const node = nodesById[nodeId];
+    if (!node)
+        return [nodeId];
+    const parentIds = node.parentNodeId
+        ? collectAncestorNodeIds(nodesById, node.parentNodeId, remainingDepth - 1)
+        : [];
+    return [nodeId, ...parentIds];
+}
+function extractNodeRecapMarkdown(outputs, nodeId) {
     const nodeView = outputs.nodesById[nodeId];
     if (!nodeView)
         return null;
@@ -99,9 +111,15 @@ function extractRecapFromOutputs(outputs, nodeId) {
     if (latest.payload.payloadKind !== 'notes')
         return null;
     const markdown = latest.payload.notesMarkdown;
-    if (!markdown || typeof markdown !== 'string')
+    return (markdown && typeof markdown === 'string') ? markdown : null;
+}
+function extractAggregateRecap(outputs, run, tipNodeId) {
+    const parts = collectAncestorNodeIds(run.nodesById, tipNodeId, MAX_RECAP_ANCESTOR_DEPTH)
+        .map((nodeId) => extractNodeRecapMarkdown(outputs, nodeId))
+        .filter((md) => md !== null);
+    if (parts.length === 0)
         return null;
-    return (0, resume_ranking_js_1.asRecapSnippet)(markdown);
+    return (0, resume_ranking_js_1.asRecapSnippet)(parts.join('\n\n'));
 }
 function extractObservations(events) {
     let gitHeadSha = null;

package/dist/v2/infra/local/workspace-anchor/index.d.ts

@@ -1,9 +1,12 @@
 import { ResultAsync as RA } from 'neverthrow';
-import type { WorkspaceAnchorPortV2, WorkspaceAnchor, WorkspaceAnchorError } from '../../../ports/workspace-anchor.port.js';
-export declare class LocalWorkspaceAnchorV2 implements WorkspaceAnchorPortV2 {
-    private readonly cwd;
-    constructor(cwd: string);
+import type { WorkspaceAnchorPortV2, WorkspaceContextResolverPortV2, WorkspaceAnchor, WorkspaceAnchorError } from '../../../ports/workspace-anchor.port.js';
+export declare class LocalWorkspaceAnchorV2 implements WorkspaceAnchorPortV2, WorkspaceContextResolverPortV2 {
+    private readonly defaultCwd;
+    constructor(defaultCwd: string);
     resolveAnchors(): RA<readonly WorkspaceAnchor[], WorkspaceAnchorError>;
+    resolveFromUri(rootUri: string): RA<readonly WorkspaceAnchor[], WorkspaceAnchorError>;
+    resolveFromCwd(): RA<readonly WorkspaceAnchor[], WorkspaceAnchorError>;
+    private resolveFromPath;
     private resolve;
     private gitCommand;
 }

package/dist/v2/infra/local/workspace-anchor/index.js

@@ -4,33 +4,46 @@ exports.LocalWorkspaceAnchorV2 = void 0;
 const neverthrow_1 = require("neverthrow");
 const child_process_1 = require("child_process");
 const util_1 = require("util");
+const url_1 = require("url");
 const execAsync = (0, util_1.promisify)(child_process_1.exec);
 class LocalWorkspaceAnchorV2 {
-    constructor(cwd) {
-        this.cwd = cwd;
+    constructor(defaultCwd) {
+        this.defaultCwd = defaultCwd;
     }
     resolveAnchors() {
-        return neverthrow_1.ResultAsync.fromPromise(this.resolve(), (cause) => ({
+        return this.resolveFromPath(this.defaultCwd);
+    }
+    resolveFromUri(rootUri) {
+        const fsPath = uriToFsPath(rootUri);
+        if (fsPath === null)
+            return (0, neverthrow_1.okAsync)([]);
+        return this.resolveFromPath(fsPath);
+    }
+    resolveFromCwd() {
+        return this.resolveFromPath(this.defaultCwd);
+    }
+    resolveFromPath(cwd) {
+        return neverthrow_1.ResultAsync.fromPromise(this.resolve(cwd), (cause) => ({
             code: 'ANCHOR_RESOLVE_FAILED',
             message: `Failed to resolve workspace anchors: ${String(cause)}`,
         }));
     }
-    async resolve() {
+    async resolve(cwd) {
         const anchors = [];
-        const branch = await this.gitCommand('git rev-parse --abbrev-ref HEAD');
+        const branch = await this.gitCommand('git rev-parse --abbrev-ref HEAD', cwd);
         if (branch && branch !== 'HEAD') {
             anchors.push({ key: 'git_branch', value: branch });
         }
-        const sha = await this.gitCommand('git rev-parse HEAD');
+        const sha = await this.gitCommand('git rev-parse HEAD', cwd);
         if (sha && /^[0-9a-f]{40}$/.test(sha)) {
             anchors.push({ key: 'git_head_sha', value: sha });
         }
         return anchors;
     }
-    async gitCommand(cmd) {
+    async gitCommand(cmd, cwd) {
         try {
             const { stdout } = await execAsync(cmd, {
-                cwd: this.cwd,
+                cwd,
                 timeout: 5000,
                 encoding: 'utf8',
             });
@@ -42,3 +55,13 @@ class LocalWorkspaceAnchorV2 {
     }
 }
 exports.LocalWorkspaceAnchorV2 = LocalWorkspaceAnchorV2;
+function uriToFsPath(uri) {
+    if (!uri.startsWith('file://'))
+        return null;
+    try {
+        return (0, url_1.fileURLToPath)(uri);
+    }
+    catch {
+        return null;
+    }
+}

package/dist/v2/ports/workspace-anchor.port.d.ts

@@ -16,3 +16,7 @@ export type WorkspaceAnchorError = {
 export interface WorkspaceAnchorPortV2 {
     resolveAnchors(): ResultAsync<readonly WorkspaceAnchor[], WorkspaceAnchorError>;
 }
+export interface WorkspaceContextResolverPortV2 {
+    resolveFromUri(rootUri: string): ResultAsync<readonly WorkspaceAnchor[], WorkspaceAnchorError>;
+    resolveFromCwd(): ResultAsync<readonly WorkspaceAnchor[], WorkspaceAnchorError>;
+}

package/package.json

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

package/workflows/test-session-persistence.json

@@ -0,0 +1,33 @@
+{
+  "id": "test-session-persistence",
+  "name": "Session Persistence Test",
+  "description": "Purpose-built workflow for testing v2 session persistence, cross-chat resume, and checkpoint mechanics. Has 5 simple steps that agents can complete quickly with unique marker outputs.",
+  "version": "1.0.0",
+  "steps": [
+    {
+      "id": "step-1-alpha",
+      "title": "Step 1: Alpha Marker",
+      "prompt": "You are testing WorkRail v2 session persistence.\n\nYour task is simple: provide output containing the unique marker **ALPHA**.\n\nInclude in your output:\n- The marker string exactly as shown above\n- A brief note that you completed this step\n\nThis marker will be used in a future test to verify cross-chat session resumption works correctly."
+    },
+    {
+      "id": "step-2-beta",
+      "title": "Step 2: Beta Marker",
+      "prompt": "You are testing WorkRail v2 session persistence.\n\nYour task is simple: provide output containing the unique marker **BETA**.\n\nInclude in your output:\n- The marker string exactly as shown above\n- A brief note that you completed this step\n\nThis marker will be used in a future test to verify cross-chat session resumption works correctly."
+    },
+    {
+      "id": "step-3-gamma",
+      "title": "Step 3: Gamma Marker",
+      "prompt": "You are testing WorkRail v2 session persistence.\n\nYour task is simple: provide output containing the unique marker **GAMMA**.\n\nInclude in your output:\n- The marker string exactly as shown above\n- A brief note that you completed this step\n\nThis marker will be used in a future test to verify cross-chat session resumption works correctly."
+    },
+    {
+      "id": "step-4-delta",
+      "title": "Step 4: Delta Marker",
+      "prompt": "You are testing WorkRail v2 session persistence.\n\nYour task is simple: provide output containing the unique marker **DELTA**.\n\nInclude in your output:\n- The marker string exactly as shown above\n- A brief note that you completed this step\n\nThis is the penultimate step."
+    },
+    {
+      "id": "step-5-final",
+      "title": "Step 5: Final Step",
+      "prompt": "You are testing WorkRail v2 session persistence.\n\nThis is the **final step**. Provide a summary:\n- How many unique markers did you write? (Should be 4: ALPHA, BETA, GAMMA, DELTA)\n- Did the workflow progress smoothly through all steps?\n- What CHAT_ID were you using?\n\nAfter you provide this output, the workflow will complete."
+    }
+  ]
+}