zeitlich 0.2.38 → 0.2.40

package/src/lib/tool-router/types.ts

@@ -2,6 +2,7 @@ import type { TokenUsage, ToolResultConfig } from "../types";
 import type { JsonValue } from "../state/types";
 import type { z } from "zod";
 import type { ActivityFunctionWithOptions } from "@temporalio/workflow";
+import type { SandboxSnapshot } from "../sandbox/types";
 
 // ============================================================================
 // Tool Definition Types
@@ -140,14 +141,16 @@ export interface ToolHandlerResponse<
    */
   resultAppended?: boolean;
   /**
-   * When true, the session will rewind: any in-flight parallel tool calls
-   * are cancelled, previously appended tool results and the triggering
-   * assistant message are removed from the thread, and the LLM call that
-   * produced the tool calls is retried.
+   * When true, the session will rewind: any in-flight parallel tool
+   * calls are cancelled and the LLM call is retried. The session reuses
+   * the same `assistantMessageId` for the retry; the next `runAgent`
+   * activity truncates the thread from that id on entry, wiping the
+   * triggering assistant message and any tool results already appended
+   * before re-invoking the LLM.
    *
    * The `toolResponse` for a rewinding tool call is ignored (never
-   * appended) since the session truncates the thread back to the
-   * pre-invocation state.
+   * appended) since the thread is rewound back to the pre-assistant
+   * state on the next invocation.
    */
   rewind?: boolean;
   /** Token usage from the tool execution (e.g. child agent invocations) */
@@ -156,6 +159,10 @@ export interface ToolHandlerResponse<
   threadId?: string;
   /** Sandbox ID created or used by the handler (e.g. child agent sandbox) */
   sandboxId?: string;
+  /** Snapshot captured on exit when `sandboxShutdown === "snapshot"`. */
+  snapshot?: SandboxSnapshot;
+  /** Snapshot captured immediately after sandbox seeding (before the agent loop starts) when `sandbox.mode === "new"` and `sandboxShutdown === "snapshot"`. Intended as a reusable "base" for new threads that want to skip re-seeding. */
+  baseSnapshot?: SandboxSnapshot;
   /** Unvalidated metadata passthrough from handler to hooks (e.g. infrastructure state) */
   metadata?: Record<string, unknown>;
 }
@@ -291,8 +298,9 @@ export interface ProcessToolCallsContext {
 
 /**
  * Signal that a tool handler requested a rewind. Attached to the
- * {@link ProcessToolCallsResult} so the session can roll the thread
- * back to the pre-invocation snapshot and retry the LLM call.
+ * {@link ProcessToolCallsResult} so the session can reuse the same
+ * `assistantMessageId` for the retry; the next `runAgent` activity
+ * then truncates the thread from that id on entry.
  */
 export interface RewindSignal {
   toolCallId: string;

package/src/lib/types.ts

@@ -92,6 +92,18 @@ export interface RunAgentConfig extends AgentConfig {
   threadKey?: string;
   /** Metadata for the session */
   metadata?: Record<string, unknown>;
+  /**
+   * The id under which the assistant message produced by this call will
+   * be appended. The activity truncates the thread from this id on
+   * entry (no-op on the first attempt) so that:
+   *
+   * - Rewind retries can reuse the same id and the previous (bad)
+   *   assistant + its tool results are wiped before the retry LLM call.
+   * - Resetting the Temporal workflow to this activity restores the
+   *   pre-call thread state: replay re-truncates, re-invokes, and
+   *   appends under the same id.
+   */
+  assistantMessageId: string;
 }
 
 /**

package/src/workflow.ts

@@ -40,6 +40,11 @@ export type {
 
 // Thread utilities
 export { getShortId } from "./lib/thread/id";
+export {
+  THREAD_TTL_SECONDS,
+  getThreadListKey,
+  getThreadMetaKey,
+} from "./lib/thread/keys";
 
 // State management
 export { createAgentStateManager } from "./lib/state";
@@ -49,6 +54,7 @@ export type {
   JsonSerializable,
   JsonValue,
   JsonPrimitive,
+  PersistedThreadState,
 } from "./lib/state";
 
 // Tool router (includes registry functionality)
@@ -57,7 +63,11 @@ export {
   hasNoOtherToolCalls,
   defineTool,
 } from "./lib/tool-router";
-export { defineSubagent, defineSubagentWorkflow } from "./lib/subagent";
+export {
+  defineSubagent,
+  defineSubagentWorkflow,
+  DEFAULT_SUBAGENT_WORKFLOW_RUN_TIMEOUT,
+} from "./lib/subagent";
 export type {
   // Tool definition types
   ToolDefinition,
@@ -153,6 +163,7 @@ export { proxyRunAgent } from "./lib/model/proxy";
 // Subagent types
 export type {
   SubagentConfig,
+  SubagentChildWorkflowOptions,
   SubagentDefinition,
   SubagentFnResult,
   SubagentHooks,

package/tsup.config.ts

@@ -4,6 +4,7 @@ export default defineConfig({
   entry: {
     index: "src/index.ts",
     workflow: "src/workflow.ts",
+    "adapters/thread/index": "src/adapters/thread/index.ts",
     "adapters/thread/langchain/index": "src/adapters/thread/langchain/index.ts",
     "adapters/thread/langchain/workflow":
       "src/adapters/thread/langchain/proxy.ts",