zeitlich 0.2.33 → 0.2.34

package/dist/workflow.cjs

@@ -392,23 +392,50 @@ function createSubagentTool(subagents) {
   };
 }
 var childResultSignal = workflow.defineSignal("childResult");
+var childSandboxReadySignal = workflow.defineSignal("childSandboxReady");
 var destroySandboxSignal = workflow.defineSignal("destroySandbox");
 
 // src/lib/subagent/handler.ts
 function resolveSandboxConfig(config) {
-  if (!config || config === "none") return { source: "none" };
-  if (config === "inherit") return { source: "inherit" };
-  if (config === "own") return { source: "own" };
-  return { source: "own", shutdown: config.shutdown };
+  if (!config || config === "none") {
+    return { source: "none", init: "per-call", continuation: "fork" };
+  }
+  if (config.source === "inherit") {
+    return {
+      source: "inherit",
+      init: "per-call",
+      continuation: config.continuation,
+      shutdown: config.shutdown
+    };
+  }
+  return {
+    source: "own",
+    init: config.init ?? "per-call",
+    continuation: config.continuation,
+    shutdown: config.shutdown
+  };
 }
 function createSubagentHandler(subagents) {
   const { taskQueue: parentTaskQueue } = workflow.workflowInfo();
   const childResults = /* @__PURE__ */ new Map();
   const pendingDestroys = /* @__PURE__ */ new Map();
   const threadSandboxes = /* @__PURE__ */ new Map();
+  const persistentSandboxes = /* @__PURE__ */ new Map();
+  const persistentSandboxCreating = /* @__PURE__ */ new Set();
+  const lazyCreatorAgent = /* @__PURE__ */ new Map();
   workflow.setHandler(childResultSignal, ({ childWorkflowId, result }) => {
     childResults.set(childWorkflowId, result);
   });
+  workflow.setHandler(
+    childSandboxReadySignal,
+    ({ childWorkflowId, sandboxId }) => {
+      const agentName = lazyCreatorAgent.get(childWorkflowId);
+      if (agentName && !persistentSandboxes.has(agentName)) {
+        persistentSandboxes.set(agentName, sandboxId);
+        lazyCreatorAgent.delete(childWorkflowId);
+      }
+    }
+  );
   const handler = async (args, context) => {
     const config = subagents.find((s) => s.agentName === args.subagent);
     if (!config) {
@@ -435,21 +462,48 @@ function createSubagentHandler(subagents) {
       };
     }
     let sandbox;
+    let sandboxShutdownOverride;
+    let isLazyCreator = false;
     if (sandboxCfg.source === "inherit" && parentSandboxId) {
-      sandbox = {
-        mode: "inherit",
-        sandboxId: parentSandboxId
-      };
+      if (sandboxCfg.continuation === "fork") {
+        sandbox = { mode: "fork", sandboxId: parentSandboxId };
+      } else {
+        sandbox = { mode: "inherit", sandboxId: parentSandboxId };
+      }
     } else if (sandboxCfg.source === "own") {
-      const prevSbId = continuationThreadId ? threadSandboxes.get(continuationThreadId) : void 0;
-      if (prevSbId) {
-        sandbox = { mode: "fork", sandboxId: prevSbId };
+      const isLazy = sandboxCfg.init === "once";
+      let baseSandboxId;
+      if (isLazy) {
+        baseSandboxId = persistentSandboxes.get(config.agentName);
+        if (!baseSandboxId) {
+          if (persistentSandboxCreating.has(config.agentName)) {
+            await workflow.condition(() => persistentSandboxes.has(config.agentName));
+            baseSandboxId = persistentSandboxes.get(config.agentName);
+          } else {
+            persistentSandboxCreating.add(config.agentName);
+            isLazyCreator = true;
+          }
+        }
+      } else if (continuationThreadId) {
+        baseSandboxId = threadSandboxes.get(continuationThreadId);
+      }
+      if (baseSandboxId) {
+        sandbox = {
+          mode: sandboxCfg.continuation === "continue" ? "continue" : "fork",
+          sandboxId: baseSandboxId
+        };
+      }
+      const userShutdown = sandboxCfg.shutdown;
+      const alreadySurvives = userShutdown === "pause-until-parent-close" || userShutdown === "keep-until-parent-close" || userShutdown === "pause" || userShutdown === "keep";
+      const mustSurvive = isLazyCreator || sandboxCfg.continuation === "continue" || isLazy && sandboxCfg.continuation === "fork";
+      if (mustSurvive && !alreadySurvives) {
+        sandboxShutdownOverride = isLazyCreator ? "pause-until-parent-close" : "pause";
       }
     }
     const workflowInput = {
       ...thread && { thread },
       ...sandbox && { sandbox },
-      ...sandboxCfg.shutdown && { sandboxShutdown: sandboxCfg.shutdown }
+      sandboxShutdown: sandboxShutdownOverride ?? sandboxCfg.shutdown ?? void 0
     };
     const resolvedContext = config.context === void 0 ? void 0 : typeof config.context === "function" ? config.context() : config.context;
     const childOpts = {
@@ -457,6 +511,9 @@ function createSubagentHandler(subagents) {
       args: resolvedContext === void 0 ? [args.prompt, workflowInput] : [args.prompt, workflowInput, resolvedContext],
       taskQueue: config.taskQueue ?? parentTaskQueue
     };
+    if (isLazyCreator) {
+      lazyCreatorAgent.set(childWorkflowId, config.agentName);
+    }
     workflow.log.info("subagent spawned", {
       subagent: config.agentName,
       childWorkflowId,
@@ -464,10 +521,10 @@ function createSubagentHandler(subagents) {
       sandboxSource: sandboxCfg.source
     });
     const childHandle = await workflow.startChild(config.workflow, childOpts);
-    const effectiveShutdown = sandboxCfg.shutdown ?? "destroy";
-    const shouldDeferDestroy = effectiveShutdown === "pause-until-parent-close" && (sandboxCfg.source === "own" || allowsContinuation && sandboxCfg.source !== "inherit");
-    if (shouldDeferDestroy) {
-      pendingDestroys.set(childWorkflowId, childHandle);
+    const effectiveShutdown = sandboxShutdownOverride ?? sandboxCfg.shutdown ?? "destroy";
+    if (effectiveShutdown === "pause-until-parent-close" || effectiveShutdown === "keep-until-parent-close") {
+      const key = isLazyCreator ? `persistent:${config.agentName}` : childWorkflowId;
+      pendingDestroys.set(key, childHandle);
     }
     await Promise.race([
       workflow.condition(() => childResults.has(childWorkflowId)),
@@ -501,8 +558,16 @@ function createSubagentHandler(subagents) {
       sandboxId: childSandboxId,
       metadata
     } = childResult;
-    if (allowsContinuation && childSandboxId && childThreadId) {
-      threadSandboxes.set(childThreadId, childSandboxId);
+    if (childSandboxId) {
+      if (sandboxCfg.source === "own" && sandboxCfg.init === "once" && !persistentSandboxes.has(config.agentName)) {
+        persistentSandboxes.set(config.agentName, childSandboxId);
+      } else if (allowsContinuation && childThreadId) {
+        threadSandboxes.set(childThreadId, childSandboxId);
+      }
+    }
+    if (isLazyCreator) {
+      persistentSandboxCreating.delete(config.agentName);
+      lazyCreatorAgent.delete(childWorkflowId);
     }
     if (!toolResponse) {
       return {
@@ -708,6 +773,7 @@ async function createSession({
   thread: threadInit,
   sandbox: sandboxInit,
   sandboxShutdown = "destroy",
+  onSandboxReady,
   virtualFs: virtualFsConfig,
   virtualFsOps
 }) {
@@ -810,6 +876,9 @@ async function createSession({
           });
         }
         sandboxId = sandboxInit.sandboxId;
+        if (sandboxShutdown === "pause-until-parent-close") {
+          await sandboxOps.resumeSandbox(sandboxId);
+        }
         sandboxOwned = true;
       } else if (sandboxMode === "fork") {
         if (!sandboxOps) {
@@ -832,6 +901,9 @@ async function createSession({
           sandboxOwned = true;
         }
       }
+      if (sandboxId && onSandboxReady) {
+        onSandboxReady(sandboxId);
+      }
       if (virtualFsConfig) {
         if (!virtualFsOps) {
           throw workflow.ApplicationFailure.create({
@@ -1223,39 +1295,45 @@ function defineSubagent(definition, overrides) {
 function defineSubagentWorkflow(config, fn) {
   const workflow$1 = async (prompt, workflowInput, context) => {
     const effectiveShutdown = workflowInput.sandboxShutdown ?? config.sandboxShutdown ?? "destroy";
+    const { parent } = workflow.workflowInfo();
+    if (!parent) {
+      throw workflow.ApplicationFailure.create({
+        message: "Subagent workflow called without a parent workflow",
+        nonRetryable: true
+      });
+    }
+    const parentHandle = workflow.getExternalWorkflowHandle(parent.workflowId);
     const sessionInput = {
       agentName: config.name,
       sandboxShutdown: effectiveShutdown,
       ...workflowInput.thread && { thread: workflowInput.thread },
-      ...workflowInput.sandbox && { sandbox: workflowInput.sandbox }
+      ...workflowInput.sandbox && { sandbox: workflowInput.sandbox },
+      onSandboxReady: (sandboxId) => {
+        void parentHandle.signal(childSandboxReadySignal, {
+          childWorkflowId: workflow.workflowInfo().workflowId,
+          sandboxId
+        });
+      }
     };
     const { destroySandbox, ...result } = await fn(
       prompt,
       sessionInput,
       context ?? {}
     );
-    if (effectiveShutdown === "pause-until-parent-close") {
+    if (effectiveShutdown === "pause-until-parent-close" || effectiveShutdown === "keep-until-parent-close") {
       if (!destroySandbox) {
         throw workflow.ApplicationFailure.create({
-          message: `Subagent "${config.name}" has sandboxShutdown="pause-until-parent-close" but fn did not return a destroySandbox callback`,
+          message: `Subagent "${config.name}" has sandboxShutdown="${effectiveShutdown}" but fn did not return a destroySandbox callback`,
           nonRetryable: true
         });
       }
       if (!result.sandboxId) {
         throw workflow.ApplicationFailure.create({
-          message: `Subagent "${config.name}" has sandboxShutdown="pause-until-parent-close" but fn did not return a sandboxId`,
+          message: `Subagent "${config.name}" has sandboxShutdown="${effectiveShutdown}" but fn did not return a sandboxId`,
           nonRetryable: true
         });
       }
     }
-    const { parent } = workflow.workflowInfo();
-    if (!parent) {
-      throw workflow.ApplicationFailure.create({
-        message: "Subagent workflow called without a parent workflow",
-        nonRetryable: true
-      });
-    }
-    const parentHandle = workflow.getExternalWorkflowHandle(parent.workflowId);
     await parentHandle.signal(childResultSignal, {
       childWorkflowId: workflow.workflowInfo().workflowId,
       result