zeitlich 0.2.43 → 0.2.45

package/dist/index.js

@@ -1092,10 +1092,7 @@ async function createSession(config) {
           stateManager.run();
         }
       );
-      const lifecycle = resolveSessionLifecycle(
-        sandboxInit,
-        sandboxShutdown
-      );
+      const lifecycle = resolveSessionLifecycle(sandboxInit, sandboxShutdown);
       const sandboxMode = lifecycle.mode;
       const resolvedShutdown = lifecycle.shutdown;
       let sandboxId;
@@ -1170,6 +1167,34 @@ async function createSession(config) {
           ...baseSnapshot && { baseSnapshot }
         });
       }
+      const sessionStartMs = Date.now();
+      const systemPrompt = stateManager.getSystemPrompt();
+      const rehydrateFromSlice = (slice) => {
+        stateManager.mergeUpdate({
+          tasks: new Map(slice.tasks),
+          ...slice.custom
+        });
+      };
+      if (threadMode === "fork" && sourceThreadId) {
+        await forkThread(sourceThreadId, threadId, threadKey);
+        const forkedSlice = await loadThreadState(threadId, threadKey);
+        if (forkedSlice) rehydrateFromSlice(forkedSlice);
+      } else if (threadMode === "continue") {
+        const continuedSlice = await loadThreadState(threadId, threadKey);
+        if (continuedSlice) rehydrateFromSlice(continuedSlice);
+      } else {
+        if (appendSystemPrompt) {
+          if (systemPrompt == null || typeof systemPrompt === "string" && systemPrompt.trim() === "") {
+            throw ApplicationFailure.create({
+              message: "No system prompt in state",
+              nonRetryable: true
+            });
+          }
+          await appendSystemMessage(threadId, uuid4(), systemPrompt, threadKey);
+        } else {
+          await initializeThread(threadId, threadKey);
+        }
+      }
       if (virtualFsConfig) {
         if (!virtualFsOps) {
           throw ApplicationFailure.create({
@@ -1206,6 +1231,14 @@ async function createSession(config) {
           ...skillFiles && { inlineFiles: skillFiles }
         });
       }
+      await appendHumanMessage(
+        threadId,
+        uuid4(),
+        await buildContextMessage(),
+        threadKey
+      );
+      let exitReason = "completed";
+      let finalMessage = null;
       if (hooks.onSessionStart) {
         await hooks.onSessionStart({
           threadId,
@@ -1220,42 +1253,6 @@ async function createSession(config) {
         maxTurns,
         ...sandboxId && { sandboxId }
       });
-      const sessionStartMs = Date.now();
-      const systemPrompt = stateManager.getSystemPrompt();
-      const rehydrateFromSlice = (slice) => {
-        stateManager.mergeUpdate({
-          tasks: new Map(slice.tasks),
-          ...slice.custom
-        });
-      };
-      if (threadMode === "fork" && sourceThreadId) {
-        await forkThread(sourceThreadId, threadId, threadKey);
-        const forkedSlice = await loadThreadState(threadId, threadKey);
-        if (forkedSlice) rehydrateFromSlice(forkedSlice);
-      } else if (threadMode === "continue") {
-        const continuedSlice = await loadThreadState(threadId, threadKey);
-        if (continuedSlice) rehydrateFromSlice(continuedSlice);
-      } else {
-        if (appendSystemPrompt) {
-          if (systemPrompt == null || typeof systemPrompt === "string" && systemPrompt.trim() === "") {
-            throw ApplicationFailure.create({
-              message: "No system prompt in state",
-              nonRetryable: true
-            });
-          }
-          await appendSystemMessage(threadId, uuid4(), systemPrompt, threadKey);
-        } else {
-          await initializeThread(threadId, threadKey);
-        }
-      }
-      await appendHumanMessage(
-        threadId,
-        uuid4(),
-        await buildContextMessage(),
-        threadKey
-      );
-      let exitReason = "completed";
-      let finalMessage = null;
       try {
         let assistantId;
         while (stateManager.isRunning() && !stateManager.isTerminal() && stateManager.getTurns() < maxTurns) {
@@ -1452,6 +1449,22 @@ function getThreadStateKey(threadKey, threadId) {
 function isTerminalStatus(status) {
   return status === "COMPLETED" || status === "FAILED" || status === "CANCELLED";
 }
+var RESERVED_STATE_KEYS = [
+  "status",
+  "version",
+  "turns",
+  "tasks",
+  "tools",
+  "systemPrompt",
+  "fileTree",
+  "inlineFiles",
+  "virtualFsCtx",
+  "totalInputTokens",
+  "totalOutputTokens",
+  "totalReasonTokens",
+  "cachedWriteTokens",
+  "cachedReadTokens"
+];
 function createAgentStateManager({
   initialState
 }) {
@@ -1605,9 +1618,16 @@ function createAgentStateManager({
       return deleted;
     },
     getPersistedSlice() {
+      const source = customState;
+      const custom2 = {};
+      const reserved = new Set(RESERVED_STATE_KEYS);
+      for (const [key, value] of Object.entries(source)) {
+        if (reserved.has(key)) continue;
+        custom2[key] = value;
+      }
       return {
         tasks: Array.from(tasks.entries()),
-        custom: { ...customState }
+        custom: custom2
       };
     },
     updateUsage(usage) {