zeitlich 0.2.47 → 0.2.49

package/dist/{proxy-CDh3Rsa7.d.cts → proxy-B7CWEV-T.d.cts}

@@ -1,5 +1,5 @@
 import { ActivityOptions, ActivityInterfaceFor } from '@temporalio/workflow';
-import { T as ThreadOps } from './types-DNEl5uxQ.cjs';
+import { T as ThreadOps } from './types-XUUFvrJ9.cjs';
 
 /**
  * Shared proxy helper for thread operations.

package/dist/{proxy-Du8ggERu.d.ts → proxy-ByFHMVRX.d.ts}

@@ -1,5 +1,5 @@
 import { ActivityOptions, ActivityInterfaceFor } from '@temporalio/workflow';
-import { T as ThreadOps } from './types-qQVZfhoT.js';
+import { T as ThreadOps } from './types-DeVNWqlb.js';
 
 /**
  * Shared proxy helper for thread operations.

package/dist/{thread-manager-DtHYws2F.d.ts → thread-manager-7AW4rhfu.d.ts}

@@ -1,7 +1,7 @@
 import Redis from 'ioredis';
 import Anthropic from '@anthropic-ai/sdk';
-import { J as JsonValue } from './types-qQVZfhoT.js';
-import { T as ThreadManagerHooks, P as ProviderThreadManager } from './types-CtdOquo3.js';
+import { J as JsonValue } from './types-DeVNWqlb.js';
+import { T as ThreadManagerHooks, P as ProviderThreadManager } from './types-DO4Tkwxo.js';
 
 /** SDK-native content type for Anthropic human messages */
 type AnthropicContent = string | Anthropic.Messages.ContentBlockParam[];

package/dist/{thread-manager-D8zKNFZ9.d.cts → thread-manager-B9rtMEVn.d.cts}

@@ -1,7 +1,7 @@
 import Redis from 'ioredis';
 import Anthropic from '@anthropic-ai/sdk';
-import { J as JsonValue } from './types-DNEl5uxQ.cjs';
-import { T as ThreadManagerHooks, P as ProviderThreadManager } from './types-BMJrsHo0.cjs';
+import { J as JsonValue } from './types-XUUFvrJ9.cjs';
+import { T as ThreadManagerHooks, P as ProviderThreadManager } from './types-BR-k7h0e.cjs';
 
 /** SDK-native content type for Anthropic human messages */
 type AnthropicContent = string | Anthropic.Messages.ContentBlockParam[];

package/dist/{thread-manager-BjoYYXgd.d.cts → thread-manager-Cibe0X5m.d.cts}

@@ -1,7 +1,7 @@
 import Redis from 'ioredis';
-import { J as JsonValue } from './types-DNEl5uxQ.cjs';
+import { J as JsonValue } from './types-XUUFvrJ9.cjs';
 import { MessageContent, StoredMessage, BaseMessage } from '@langchain/core/messages';
-import { T as ThreadManagerHooks, P as ProviderThreadManager } from './types-BMJrsHo0.cjs';
+import { T as ThreadManagerHooks, P as ProviderThreadManager } from './types-BR-k7h0e.cjs';
 
 /** SDK-native content type for LangChain human messages */
 type LangChainContent = string | MessageContent;

package/dist/{thread-manager-Dw96FKH1.d.ts → thread-manager-nK-WcFzM.d.ts}

@@ -1,7 +1,7 @@
 import Redis from 'ioredis';
-import { J as JsonValue } from './types-qQVZfhoT.js';
+import { J as JsonValue } from './types-DeVNWqlb.js';
 import { MessageContent, StoredMessage, BaseMessage } from '@langchain/core/messages';
-import { T as ThreadManagerHooks, P as ProviderThreadManager } from './types-CtdOquo3.js';
+import { T as ThreadManagerHooks, P as ProviderThreadManager } from './types-DO4Tkwxo.js';
 
 /** SDK-native content type for LangChain human messages */
 type LangChainContent = string | MessageContent;

package/dist/{types-BMJrsHo0.d.cts → types-BR-k7h0e.d.cts}

@@ -1,5 +1,5 @@
 import Redis from 'ioredis';
-import { P as PersistedThreadState, J as JsonValue } from './types-DNEl5uxQ.cjs';
+import { P as PersistedThreadState, J as JsonValue } from './types-XUUFvrJ9.cjs';
 
 interface ThreadManagerConfig<T> {
     redis: Redis;

package/dist/{types-CtdOquo3.d.ts → types-DO4Tkwxo.d.ts}

@@ -1,5 +1,5 @@
 import Redis from 'ioredis';
-import { P as PersistedThreadState, J as JsonValue } from './types-qQVZfhoT.js';
+import { P as PersistedThreadState, J as JsonValue } from './types-DeVNWqlb.js';
 
 interface ThreadManagerConfig<T> {
     redis: Redis;

package/dist/{types-qQVZfhoT.d.ts → types-DeVNWqlb.d.ts}

@@ -134,6 +134,32 @@ interface RouterContext {
     toolCallId: string;
     toolName: string;
     sandboxId?: string;
+    /**
+     * Id of the assistant message that issued this tool call (the message
+     * the session passed as `assistantMessageId` into `runAgent`). Present
+     * for any tool call processed through `processToolCalls` from a
+     * session; may be absent when the router is driven manually (e.g.
+     * tests, custom orchestrators).
+     *
+     * Subagent handlers that fork the parent's thread mid-call use this
+     * to truncate the orphan trailing assistant message from the forked
+     * thread so the child's first model call sees a well-formed history.
+     */
+    assistantMessageId?: string;
+    /**
+     * Persist the parent session's current `PersistedThreadState` slice
+     * (tasks + custom state) to the durable thread store. Wired up by
+     * the session — absent for manually-driven routers (tests, custom
+     * orchestrators).
+     *
+     * Subagent handlers invoke this before spawning a child that will
+     * read the parent's thread (`newThreadSource: "from-parent"` or an
+     * explicit parent threadId): the parent's slice otherwise only
+     * lands in storage at session-exit time, so the child would load a
+     * stale (or empty) snapshot. Best-effort — failures are logged by
+     * the session but never thrown.
+     */
+    persistThreadState?: () => Promise<void>;
 }
 /**
  * A handler function for a specific tool.
@@ -201,6 +227,23 @@ interface ProcessToolCallsContext {
     turn?: number;
     /** Active sandbox ID (when a sandbox is configured for this session) */
     sandboxId?: string;
+    /**
+     * Id of the assistant message that produced these tool calls. The
+     * router forwards it into every handler's {@link RouterContext} so
+     * handlers can reference the message they were issued from (e.g.
+     * subagent forks that need to truncate the orphan assistant message
+     * out of a parent-forked thread).
+     */
+    assistantMessageId?: string;
+    /**
+     * Optional callback that flushes the session's in-memory
+     * `PersistedThreadState` slice to the durable thread store. The
+     * router forwards it into every handler's {@link RouterContext}
+     * verbatim. The session uses this to let mid-loop tool handlers
+     * (notably subagents that fork or continue the parent's thread)
+     * persist the parent's slice before the child reads it.
+     */
+    persistThreadState?: () => Promise<void>;
 }
 /**
  * Signal that a tool handler requested a rewind. Attached to the
@@ -887,6 +930,14 @@ interface Hooks<T extends ToolMap, TResult = unknown, TContent = unknown> extend
  *   continue there. When the adapter has `onForkPrepareThread` and/or
  *   `onForkTransform` hooks configured, they are applied once to the forked
  *   thread before the session starts.
+ *
+ *   The optional `truncateAfterFork.fromMessageId` directs the session to
+ *   call `truncateThread` on the freshly forked thread immediately after
+ *   the fork, dropping that message and everything after. Used by
+ *   subagents that fork their parent's thread mid-tool-call to strip the
+ *   orphan assistant `tool_use` block (the one whose `tool_result` will
+ *   never arrive in the child's thread) so the first model call doesn't
+ *   reject on an unmatched tool-use/tool-result pair.
  */
 type ThreadInit = {
     mode: "new";
@@ -897,6 +948,9 @@ type ThreadInit = {
 } | {
     mode: "fork";
     threadId: string;
+    truncateAfterFork?: {
+        fromMessageId: string;
+    };
 };
 
 /**

package/dist/{types-DNEl5uxQ.d.cts → types-XUUFvrJ9.d.cts}

@@ -134,6 +134,32 @@ interface RouterContext {
     toolCallId: string;
     toolName: string;
     sandboxId?: string;
+    /**
+     * Id of the assistant message that issued this tool call (the message
+     * the session passed as `assistantMessageId` into `runAgent`). Present
+     * for any tool call processed through `processToolCalls` from a
+     * session; may be absent when the router is driven manually (e.g.
+     * tests, custom orchestrators).
+     *
+     * Subagent handlers that fork the parent's thread mid-call use this
+     * to truncate the orphan trailing assistant message from the forked
+     * thread so the child's first model call sees a well-formed history.
+     */
+    assistantMessageId?: string;
+    /**
+     * Persist the parent session's current `PersistedThreadState` slice
+     * (tasks + custom state) to the durable thread store. Wired up by
+     * the session — absent for manually-driven routers (tests, custom
+     * orchestrators).
+     *
+     * Subagent handlers invoke this before spawning a child that will
+     * read the parent's thread (`newThreadSource: "from-parent"` or an
+     * explicit parent threadId): the parent's slice otherwise only
+     * lands in storage at session-exit time, so the child would load a
+     * stale (or empty) snapshot. Best-effort — failures are logged by
+     * the session but never thrown.
+     */
+    persistThreadState?: () => Promise<void>;
 }
 /**
  * A handler function for a specific tool.
@@ -201,6 +227,23 @@ interface ProcessToolCallsContext {
     turn?: number;
     /** Active sandbox ID (when a sandbox is configured for this session) */
     sandboxId?: string;
+    /**
+     * Id of the assistant message that produced these tool calls. The
+     * router forwards it into every handler's {@link RouterContext} so
+     * handlers can reference the message they were issued from (e.g.
+     * subagent forks that need to truncate the orphan assistant message
+     * out of a parent-forked thread).
+     */
+    assistantMessageId?: string;
+    /**
+     * Optional callback that flushes the session's in-memory
+     * `PersistedThreadState` slice to the durable thread store. The
+     * router forwards it into every handler's {@link RouterContext}
+     * verbatim. The session uses this to let mid-loop tool handlers
+     * (notably subagents that fork or continue the parent's thread)
+     * persist the parent's slice before the child reads it.
+     */
+    persistThreadState?: () => Promise<void>;
 }
 /**
  * Signal that a tool handler requested a rewind. Attached to the
@@ -887,6 +930,14 @@ interface Hooks<T extends ToolMap, TResult = unknown, TContent = unknown> extend
  *   continue there. When the adapter has `onForkPrepareThread` and/or
  *   `onForkTransform` hooks configured, they are applied once to the forked
  *   thread before the session starts.
+ *
+ *   The optional `truncateAfterFork.fromMessageId` directs the session to
+ *   call `truncateThread` on the freshly forked thread immediately after
+ *   the fork, dropping that message and everything after. Used by
+ *   subagents that fork their parent's thread mid-tool-call to strip the
+ *   orphan assistant `tool_use` block (the one whose `tool_result` will
+ *   never arrive in the child's thread) so the first model call doesn't
+ *   reject on an unmatched tool-use/tool-result pair.
  */
 type ThreadInit = {
     mode: "new";
@@ -897,6 +948,9 @@ type ThreadInit = {
 } | {
     mode: "fork";
     threadId: string;
+    truncateAfterFork?: {
+        fromMessageId: string;
+    };
 };
 
 /**

package/dist/{workflow-BH9ImDGq.d.cts → workflow-KbGsxpfh.d.cts}

@@ -1,4 +1,4 @@
-import { au as ToolMap, _ as SandboxInit, ah as SubagentSandboxShutdown, a1 as SessionConfig, aF as ZeitlichSession, $ as SandboxShutdown, am as ThreadInit, u as JsonSerializable, B as BaseAgentState, p as AgentStateManager, aH as ToolRouterOptions, ax as ToolRouter, R as RouterContext, J as JsonValue, az as ToolWithHandler, w as ParsedToolCallUnion, av as ToolNames, ac as SubagentDefinition, af as SubagentHooks, ag as SubagentSandboxConfig, aa as SubagentConfig, ai as SubagentSessionInput, ad as SubagentFnResult, a7 as SessionStartHook, a2 as SessionEndHook, E as PostToolUseHook, z as PostToolUseFailureHook, a4 as SessionExitReason, an as TokenUsage, f as RunAgentConfig, A as AgentResponse, F as FileEntryMetadata, aB as VirtualFileTree, k as TreeMutation, s as FileEntry, aC as VirtualFsOps, h as SkillMetadata, i as Skill, c as ToolHandlerResponse, as as ToolHandler, aE as WorkflowTask, d as ActivityToolHandler } from './types-DNEl5uxQ.cjs';
+import { au as ToolMap, _ as SandboxInit, ah as SubagentSandboxShutdown, a1 as SessionConfig, aF as ZeitlichSession, $ as SandboxShutdown, am as ThreadInit, u as JsonSerializable, B as BaseAgentState, p as AgentStateManager, aH as ToolRouterOptions, ax as ToolRouter, R as RouterContext, J as JsonValue, az as ToolWithHandler, w as ParsedToolCallUnion, av as ToolNames, ac as SubagentDefinition, af as SubagentHooks, ag as SubagentSandboxConfig, aa as SubagentConfig, ai as SubagentSessionInput, ad as SubagentFnResult, a7 as SessionStartHook, a2 as SessionEndHook, E as PostToolUseHook, z as PostToolUseFailureHook, a4 as SessionExitReason, an as TokenUsage, f as RunAgentConfig, A as AgentResponse, F as FileEntryMetadata, aB as VirtualFileTree, k as TreeMutation, s as FileEntry, aC as VirtualFsOps, h as SkillMetadata, i as Skill, c as ToolHandlerResponse, as as ToolHandler, aE as WorkflowTask, d as ActivityToolHandler } from './types-XUUFvrJ9.cjs';
 import z$1, { z } from 'zod';
 import './types-CJ7tCdl6.cjs';
 import { Duration } from '@temporalio/common';

package/dist/{workflow-Cdw3-RNB.d.ts → workflow-uhOIj9D-.d.ts}

@@ -1,4 +1,4 @@
-import { au as ToolMap, _ as SandboxInit, ah as SubagentSandboxShutdown, a1 as SessionConfig, aF as ZeitlichSession, $ as SandboxShutdown, am as ThreadInit, u as JsonSerializable, B as BaseAgentState, p as AgentStateManager, aH as ToolRouterOptions, ax as ToolRouter, R as RouterContext, J as JsonValue, az as ToolWithHandler, w as ParsedToolCallUnion, av as ToolNames, ac as SubagentDefinition, af as SubagentHooks, ag as SubagentSandboxConfig, aa as SubagentConfig, ai as SubagentSessionInput, ad as SubagentFnResult, a7 as SessionStartHook, a2 as SessionEndHook, E as PostToolUseHook, z as PostToolUseFailureHook, a4 as SessionExitReason, an as TokenUsage, f as RunAgentConfig, A as AgentResponse, F as FileEntryMetadata, aB as VirtualFileTree, k as TreeMutation, s as FileEntry, aC as VirtualFsOps, h as SkillMetadata, i as Skill, c as ToolHandlerResponse, as as ToolHandler, aE as WorkflowTask, d as ActivityToolHandler } from './types-qQVZfhoT.js';
+import { au as ToolMap, _ as SandboxInit, ah as SubagentSandboxShutdown, a1 as SessionConfig, aF as ZeitlichSession, $ as SandboxShutdown, am as ThreadInit, u as JsonSerializable, B as BaseAgentState, p as AgentStateManager, aH as ToolRouterOptions, ax as ToolRouter, R as RouterContext, J as JsonValue, az as ToolWithHandler, w as ParsedToolCallUnion, av as ToolNames, ac as SubagentDefinition, af as SubagentHooks, ag as SubagentSandboxConfig, aa as SubagentConfig, ai as SubagentSessionInput, ad as SubagentFnResult, a7 as SessionStartHook, a2 as SessionEndHook, E as PostToolUseHook, z as PostToolUseFailureHook, a4 as SessionExitReason, an as TokenUsage, f as RunAgentConfig, A as AgentResponse, F as FileEntryMetadata, aB as VirtualFileTree, k as TreeMutation, s as FileEntry, aC as VirtualFsOps, h as SkillMetadata, i as Skill, c as ToolHandlerResponse, as as ToolHandler, aE as WorkflowTask, d as ActivityToolHandler } from './types-DeVNWqlb.js';
 import z$1, { z } from 'zod';
 import './types-CJ7tCdl6.js';
 import { Duration } from '@temporalio/common';

package/dist/workflow.cjs

@@ -123,7 +123,7 @@ function createToolRouter(options) {
       });
     }
   }
-  async function processToolCall(toolCall, turn, sandboxId, onRewindRequested) {
+  async function processToolCall(toolCall, turn, sandboxId, onRewindRequested, assistantMessageId, persistThreadState) {
     const startTime = Date.now();
     const tool = toolMap.get(toolCall.name);
     const preResult = await runPreHooks(toolCall, tool, turn);
@@ -158,7 +158,9 @@ function createToolRouter(options) {
           ...options.threadKey && { threadKey: options.threadKey },
           toolCallId: toolCall.id,
           toolName: toolCall.name,
-          ...sandboxId !== void 0 && { sandboxId }
+          ...sandboxId !== void 0 && { sandboxId },
+          ...assistantMessageId !== void 0 && { assistantMessageId },
+          ...persistThreadState !== void 0 && { persistThreadState }
         };
         const response = await tool.handler(
           effectiveArgs,
@@ -285,6 +287,8 @@ function createToolRouter(options) {
       }
       const turn = context?.turn ?? 0;
       const sandboxId = context?.sandboxId;
+      const assistantMessageId = context?.assistantMessageId;
+      const persistThreadState = context?.persistThreadState;
       let rewindSignal;
       if (options.parallel) {
         const scope = new workflow.CancellationScope({ cancellable: true });
@@ -297,7 +301,14 @@ function createToolRouter(options) {
         const outcomes = await scope.run(
           async () => Promise.allSettled(
             toolCalls.map(
-              (tc) => processToolCall(tc, turn, sandboxId, onRewindRequested)
+              (tc) => processToolCall(
+                tc,
+                turn,
+                sandboxId,
+                onRewindRequested,
+                assistantMessageId,
+                persistThreadState
+              )
             )
           )
         );
@@ -317,7 +328,14 @@ function createToolRouter(options) {
       }
       const results = [];
       for (const toolCall of toolCalls) {
-        const outcome = await processToolCall(toolCall, turn, sandboxId);
+        const outcome = await processToolCall(
+          toolCall,
+          turn,
+          sandboxId,
+          void 0,
+          assistantMessageId,
+          persistThreadState
+        );
         if (outcome.kind === "rewind") {
           rewindSignal = outcome.signal;
           break;
@@ -341,6 +359,12 @@ function createToolRouter(options) {
           toolName: toolCall.name,
           ...context?.sandboxId !== void 0 && {
             sandboxId: context.sandboxId
+          },
+          ...context?.assistantMessageId !== void 0 && {
+            assistantMessageId: context.assistantMessageId
+          },
+          ...context?.persistThreadState !== void 0 && {
+            persistThreadState: context.persistThreadState
           }
         };
         const response = await handler(
@@ -567,13 +591,26 @@ function createSubagentHandler(subagents) {
     const threadMode = config.thread ?? "new";
     const allowsContinuation = threadMode !== "new";
     const newThreadSource = config.newThreadSource ?? "new";
-    const continuationThreadId = !allowsContinuation ? void 0 : args.threadId ?? (newThreadSource === "from-parent" ? context.threadId : void 0);
+    const usingParentFallback = allowsContinuation && !args.threadId && newThreadSource === "from-parent";
+    const continuationThreadId = !allowsContinuation ? void 0 : args.threadId ?? (usingParentFallback ? context.threadId : void 0);
     let thread;
     if (continuationThreadId) {
-      thread = {
-        mode: threadMode,
-        threadId: continuationThreadId
-      };
+      if (threadMode === "fork") {
+        thread = {
+          mode: "fork",
+          threadId: continuationThreadId,
+          ...usingParentFallback && context.assistantMessageId ? {
+            truncateAfterFork: {
+              fromMessageId: context.assistantMessageId
+            }
+          } : {}
+        };
+      } else {
+        thread = {
+          mode: "continue",
+          threadId: continuationThreadId
+        };
+      }
     }
     let sandbox;
     let sandboxShutdownOverride;
@@ -685,6 +722,17 @@ function createSubagentHandler(subagents) {
     if (isSnapshotBaseCreator) {
       snapshotBaseCreatorAgent.set(childWorkflowId, config.agentName);
     }
+    if (continuationThreadId && continuationThreadId === context.threadId && context.persistThreadState) {
+      try {
+        await context.persistThreadState();
+      } catch (err) {
+        workflow.log.warn("failed to persist parent thread state for subagent", {
+          subagent: config.agentName,
+          childWorkflowId,
+          error: err instanceof Error ? err.message : String(err)
+        });
+      }
+    }
     workflow.log.info("subagent spawned", {
       subagent: config.agentName,
       childWorkflowId,
@@ -1034,6 +1082,7 @@ async function createSession(config) {
     appendSystemMessage,
     appendAgentMessage,
     forkThread,
+    truncateThread,
     loadThreadState,
     saveThreadState,
     hydrateThread,
@@ -1183,6 +1232,10 @@ async function createSession(config) {
       if (threadMode === "fork" && sourceThreadId) {
         await hydrateThread(sourceThreadId, threadKey);
         await forkThread(sourceThreadId, threadId, threadKey);
+        const truncate = threadInit.truncateAfterFork;
+        if (truncate?.fromMessageId) {
+          await truncateThread(threadId, truncate.fromMessageId, threadKey);
+        }
         const forkedSlice = await loadThreadState(threadId, threadKey);
         if (forkedSlice) rehydrateFromSlice(forkedSlice);
       } else if (threadMode === "continue") {
@@ -1312,7 +1365,20 @@ async function createSession(config) {
             parsedToolCalls,
             {
               turn: currentTurn,
-              ...sandboxId !== void 0 && { sandboxId }
+              ...sandboxId !== void 0 && { sandboxId },
+              ...assistantId !== void 0 && {
+                assistantMessageId: assistantId
+              },
+              // Hand handlers a way to persist the parent's slice
+              // mid-loop (subagents that fork or continue the parent's
+              // thread need this — otherwise the child loads a stale
+              // snapshot from the prior session, since `saveThreadState`
+              // would otherwise only run in the `finally` below).
+              persistThreadState: () => saveThreadState(
+                threadId,
+                stateManager.getPersistedSlice(),
+                threadKey
+              )
             }
           );
           for (const result of toolCallResults) {