zeitlich 0.2.42 → 0.2.43

package/package.json

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

package/src/lib/session/session-edge-cases.integration.test.ts

@@ -176,9 +176,18 @@ describe("createSession edge cases", () => {
     idCounter = 0;
   });
 
-  // --- WAITING_FOR_INPUT flow (condition returns false = timeout) ---
-
-  it("cancels session when WAITING_FOR_INPUT times out (condition returns false)", async () => {
+  // NOTE: a previous test here ("cancels session when WAITING_FOR_INPUT
+  // times out") covered the condition-based wait + cancel-on-timeout flow
+  // that was removed in 7ab652b ("fix: stop halting workflow when
+  // waiting"). The session loop now exits cleanly the moment a tool puts
+  // the agent into WAITING_FOR_INPUT — there is no internal timeout to
+  // assert against — so the test was deleted rather than rewritten. The
+  // assertion below covers the surviving contract: the exit reason
+  // reflects the WAITING_FOR_INPUT state instead of the default
+  // "completed", so callers can distinguish a finished session from a
+  // parked one.
+
+  it("exits with 'waiting_for_input' when a tool parks the session", async () => {
     const { ops } = createMockThreadOps();
     let endReason: string | undefined;
     const capturedRef: {
@@ -191,10 +200,7 @@ describe("createSession edge cases", () => {
       schema: z.object({}),
       handler: async (_args: Record<string, never>, _ctx: RouterContext) => {
         capturedRef.stateManager?.waitForInput();
-        return {
-          toolResponse: "Please provide input.",
-          data: null,
-        };
+        return { toolResponse: "Please provide input.", data: null };
       },
     });
 
@@ -224,9 +230,9 @@ describe("createSession edge cases", () => {
 
     const result = await session.runSession({ stateManager });
 
-    expect(result.exitReason).toBe("cancelled");
-    expect(result.finalMessage).toBeNull();
-    expect(endReason).toBe("cancelled");
+    expect(result.exitReason).toBe("waiting_for_input");
+    expect(endReason).toBe("waiting_for_input");
+    expect(stateManager.getStatus()).toBe("WAITING_FOR_INPUT");
   });
 
   // --- All tool calls are invalid ---

package/src/lib/session/session.ts

@@ -596,6 +596,14 @@ export async function createSession<
             threadId,
             maxTurns,
           });
+        } else if (stateManager.getStatus() === "WAITING_FOR_INPUT") {
+          // A tool put the agent into WAITING_FOR_INPUT (e.g. an AskUser
+          // tool). The loop's `isRunning()` guard then exited cleanly; we
+          // report a dedicated exit reason rather than the misleading
+          // default `"completed"`, so callers can distinguish a finished
+          // session from one parked waiting on external input.
+          exitReason = "waiting_for_input";
+          log.info("session waiting for input", { agentName, threadId });
         }
       } catch (error) {
         exitReason = "failed";

package/src/lib/subagent/workflow.ts

@@ -168,9 +168,8 @@ export function defineSubagentWorkflow(
 
     // Auto-forward sandbox outputs captured from the session so user code
     // never has to thread them through manually. Explicit values on the fn
-    // result take precedence.
+    // result take precedence, so spread `result` LAST.
     return {
-      ...result,
       ...(capturedThreadId !== undefined && { threadId: capturedThreadId }),
       ...(capturedSandboxId !== undefined && { sandboxId: capturedSandboxId }),
       ...(capturedSnapshot !== undefined && { snapshot: capturedSnapshot }),
@@ -178,6 +177,7 @@ export function defineSubagentWorkflow(
         baseSnapshot: capturedBaseSnapshot,
       }),
       ...(capturedUsage !== undefined && { usage: capturedUsage }),
+      ...result,
     };
   };