zeitlich 0.2.43 → 0.2.44

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zeitlich",
-  "version": "0.2.43",
+  "version": "0.2.44",
   "description": "[EXPERIMENTAL] An opinionated AI agent implementation for Temporal",
   "main": "./dist/index.js",
   "module": "./dist/index.js",

package/src/lib/.env

@@ -0,0 +1 @@
+E2B_API_KEY=e2b_39af116424059782e2aee6942fd70237cc2126c9

package/src/lib/session/session.ts

@@ -383,46 +383,6 @@ export async function createSession<
         });
       }
 
-      // --- Virtual filesystem init (independent of sandbox) ----------------
-      if (virtualFsConfig) {
-        if (!virtualFsOps) {
-          throw ApplicationFailure.create({
-            message: "No virtualFsOps provided — cannot resolve file tree",
-            nonRetryable: true,
-          });
-        }
-        const result = await virtualFsOps.resolveFileTree(virtualFsConfig.ctx);
-        const skillFiles = skills ? collectSkillFiles(skills) : undefined;
-        const fileTree = skillFiles
-          ? [
-              ...result.fileTree,
-              ...Object.entries(skillFiles).map(([path, content]) => ({
-                id: `skill:${path}`,
-                path,
-                size: content.length,
-                mtime: new Date().toISOString(),
-                metadata: {},
-                // Carry the content directly on the entry so any handler that
-                // constructs a VirtualFileSystem from `fileTree` can read it
-                // without needing to also wire up `inlineFiles` from state.
-                inlineContent: content,
-              })),
-            ]
-          : result.fileTree;
-        stateManager.mergeUpdate({
-          fileTree,
-          virtualFsCtx: virtualFsConfig.ctx,
-          // `inlineFiles` is still the source of truth at read time:
-          // VirtualFileSystem checks the inlineFiles map first and only
-          // falls through to entry.inlineContent. Embedding the content on
-          // the entry is the migration target so that handlers building a
-          // VirtualFileSystem from `fileTree` alone (without forwarding
-          // `inlineFiles` from state) can read skill resources. Until a
-          // follow-up drops `inlineFiles`, both fields are populated.
-          ...(skillFiles && { inlineFiles: skillFiles }),
-        } as Partial<AgentState<TState>>);
-      }
-
       if (hooks.onSessionStart) {
         await hooks.onSessionStart({
           threadId,
@@ -474,6 +434,52 @@ export async function createSession<
           await initializeThread(threadId, threadKey);
         }
       }
+
+      // --- Virtual filesystem init (independent of sandbox) ----------------
+      // Runs AFTER thread rehydration so the freshly resolved tree is
+      // authoritative. Otherwise a stale `fileTree` carried in a persisted
+      // slice (from a run on older code that didn't strip it) could
+      // overwrite entries that no longer exist in the backing store and
+      // cause subsequent FileWrite calls to fail with "not_found".
+      if (virtualFsConfig) {
+        if (!virtualFsOps) {
+          throw ApplicationFailure.create({
+            message: "No virtualFsOps provided — cannot resolve file tree",
+            nonRetryable: true,
+          });
+        }
+        const result = await virtualFsOps.resolveFileTree(virtualFsConfig.ctx);
+        const skillFiles = skills ? collectSkillFiles(skills) : undefined;
+        const fileTree = skillFiles
+          ? [
+              ...result.fileTree,
+              ...Object.entries(skillFiles).map(([path, content]) => ({
+                id: `skill:${path}`,
+                path,
+                size: content.length,
+                mtime: new Date().toISOString(),
+                metadata: {},
+                // Carry the content directly on the entry so any handler that
+                // constructs a VirtualFileSystem from `fileTree` can read it
+                // without needing to also wire up `inlineFiles` from state.
+                inlineContent: content,
+              })),
+            ]
+          : result.fileTree;
+        stateManager.mergeUpdate({
+          fileTree,
+          virtualFsCtx: virtualFsConfig.ctx,
+          // `inlineFiles` is still the source of truth at read time:
+          // VirtualFileSystem checks the inlineFiles map first and only
+          // falls through to entry.inlineContent. Embedding the content on
+          // the entry is the migration target so that handlers building a
+          // VirtualFileSystem from `fileTree` alone (without forwarding
+          // `inlineFiles` from state) can read skill resources. Until a
+          // follow-up drops `inlineFiles`, both fields are populated.
+          ...(skillFiles && { inlineFiles: skillFiles }),
+        } as Partial<AgentState<TState>>);
+      }
+
       await appendHumanMessage(
         threadId,
         uuid4(),

package/src/lib/state/manager.integration.test.ts

@@ -385,6 +385,33 @@ describe("createAgentStateManager integration", () => {
     expect("tools" in slice.custom).toBe(false);
   });
 
+  it("getPersistedSlice strips runtime fields injected via mergeUpdate", () => {
+    const sm = createAgentStateManager<{ label: string }>({
+      initialState: { systemPrompt: "test", label: "original" },
+    });
+
+    sm.mergeUpdate({
+      label: "updated",
+      fileTree: [
+        {
+          id: "f1",
+          path: "/a.txt",
+          size: 1,
+          mtime: "2026-01-01T00:00:00Z",
+          metadata: {},
+        },
+      ],
+      virtualFsCtx: { workspaceBase: "/tmp" },
+      inlineFiles: { "/a.txt": "hi" },
+    } as Parameters<typeof sm.mergeUpdate>[0]);
+
+    const slice = sm.getPersistedSlice();
+    expect(slice.custom).toEqual({ label: "updated" });
+    expect("fileTree" in slice.custom).toBe(false);
+    expect("virtualFsCtx" in slice.custom).toBe(false);
+    expect("inlineFiles" in slice.custom).toBe(false);
+  });
+
   it("mergeUpdate replaces the task map when given a tasks field", () => {
     const sm = createAgentStateManager<{ label: string; extra?: string }>({
       initialState: {

package/src/lib/state/manager.ts

@@ -20,6 +20,32 @@ import type {
 } from "./types";
 import { z } from "zod";
 
+/**
+ * Fields that live on `AgentState` at runtime but must NOT be persisted into
+ * `PersistedThreadState`. They're either managed by the state manager itself
+ * (status/version/turns/tasks/tools/usage), kept in their own slot
+ * (systemPrompt), or rebuilt on each run (the virtual-fs trio).
+ *
+ * Centralizing the list keeps the constructor's destructure-omit and
+ * `getPersistedSlice` in lockstep.
+ */
+const RESERVED_STATE_KEYS = [
+  "status",
+  "version",
+  "turns",
+  "tasks",
+  "tools",
+  "systemPrompt",
+  "fileTree",
+  "inlineFiles",
+  "virtualFsCtx",
+  "totalInputTokens",
+  "totalOutputTokens",
+  "totalReasonTokens",
+  "cachedWriteTokens",
+  "cachedReadTokens",
+] as const;
+
 /**
  * Creates an agent state manager for tracking workflow state.
  * Automatically registers Temporal query and update handlers for the agent.
@@ -238,9 +264,21 @@ export function createAgentStateManager<
     },
 
     getPersistedSlice(): PersistedThreadState {
+      // `customState` can pick up reserved/runtime fields via `mergeUpdate`
+      // (e.g. `fileTree`, `virtualFsCtx`, `inlineFiles` written by the
+      // virtual-fs bootstrap on every run). Those are rebuilt per run and
+      // must never round-trip through the thread store, so strip them here
+      // rather than relying on callers to remember.
+      const source = customState as unknown as Record<string, JsonValue>;
+      const custom: Record<string, JsonValue> = {};
+      const reserved = new Set<string>(RESERVED_STATE_KEYS);
+      for (const [key, value] of Object.entries(source)) {
+        if (reserved.has(key)) continue;
+        custom[key] = value;
+      }
       return {
         tasks: Array.from(tasks.entries()),
-        custom: { ...(customState as unknown as Record<string, JsonValue>) },
+        custom,
       };
     },
 

package/src/tools/bash/.env

@@ -0,0 +1 @@
+E2B_API_KEY=e2b_39af116424059782e2aee6942fd70237cc2126c9