zeitlich 0.2.7 → 0.2.9

package/dist/{workflow-CyYHDbrr.d.cts → workflow-C2ShwjC7.d.cts}

@@ -116,7 +116,7 @@ interface ToolWithHandler<TName extends string = string, TSchema extends z$1.Zod
     strict?: boolean;
     max_uses?: number;
     /** Whether this tool is available to the agent (default: true). Disabled tools are excluded from definitions and rejected at parse time. */
-    enabled?: boolean;
+    enabled?: () => boolean;
     /** Per-tool lifecycle hooks (run in addition to global hooks) */
     hooks?: ToolHooks<z$1.infer<TSchema>, TResult>;
 }
@@ -135,7 +135,7 @@ type ToolMap = Record<string, {
     handler: ToolHandler<any, any, any>;
     strict?: boolean;
     max_uses?: number;
-    enabled?: boolean;
+    enabled?: () => boolean;
     hooks?: ToolHooks<any, any>;
 }>;
 /**
@@ -187,6 +187,8 @@ interface ToolHandlerResponse<TResult = null> {
      * payloads through Temporal's activity payload limit.
      */
     resultAppended?: boolean;
+    /** Token usage from the tool execution (e.g. child agent invocations) */
+    usage?: TokenUsage;
 }
 /**
  * Context passed to tool handlers for additional data beyond tool args.
@@ -237,6 +239,7 @@ interface ToolCallResult<TName extends string = string, TResult = unknown> {
     toolCallId: string;
     name: TName;
     data: TResult;
+    usage?: TokenUsage;
 }
 /**
  * Options for creating a tool router.
@@ -465,6 +468,10 @@ interface BaseAgentState {
     turns: number;
     tasks: Map<string, WorkflowTask>;
     systemPrompt: string;
+    totalInputTokens: number;
+    totalOutputTokens: number;
+    cachedWriteTokens: number;
+    cachedReadtTokens: number;
 }
 /**
  * File representation for agent workflows
@@ -481,17 +488,20 @@ interface AgentFile {
     /** MIME type of the file */
     mimeType?: string;
 }
+interface TokenUsage {
+    inputTokens?: number;
+    outputTokens?: number;
+    cachedWriteTokens?: number;
+    cachedReadTokens?: number;
+    reasonTokens?: number;
+}
 /**
  * Agent response from LLM invocation
  */
 interface AgentResponse<M = StoredMessage> {
     message: M;
     rawToolCalls: RawToolCall[];
-    usage?: {
-        input_tokens?: number;
-        output_tokens?: number;
-        total_tokens?: number;
-    };
+    usage?: TokenUsage;
 }
 /**
  * Thread operations required by a session.
@@ -566,9 +576,10 @@ interface SerializableToolDefinition {
 /**
  * Configuration passed to runAgent activity
  */
-interface RunAgentConfig {
+interface RunAgentConfig extends AgentConfig {
+    /** The thread ID to use for the session */
     threadId: string;
-    agentName: string;
+    /** Metadata for the session */
     metadata?: Record<string, unknown>;
 }
 /**
@@ -597,7 +608,7 @@ interface SubagentConfig<TResult extends z$1.ZodType = z$1.ZodType> {
     /** Description shown to the parent agent explaining what this subagent does */
     description: string;
     /** Whether this subagent is available (default: true). Disabled subagents are excluded from the Subagent tool. */
-    enabled?: boolean;
+    enabled?: () => boolean;
     /** Temporal workflow function or type name (used with executeChild) */
     workflow: string | Workflow;
     /** Optional task queue - defaults to parent's queue if not specified */
@@ -927,6 +938,23 @@ interface AgentStateManager<TCustom extends JsonSerializable<TCustom>> {
     deleteTask(id: string): boolean;
     /** Set the tools (converts Zod schemas to JSON Schema for serialization) */
     setTools(newTools: ToolDefinition[]): void;
+    /** Update the usage */
+    updateUsage(usage: {
+        inputTokens?: number;
+        outputTokens?: number;
+        cachedWriteTokens?: number;
+        cachedReadTokens?: number;
+        reasonTokens?: number;
+    }): void;
+    /** Get the total usage */
+    getTotalUsage(): {
+        totalInputTokens: number;
+        totalOutputTokens: number;
+        totalCachedWriteTokens: number;
+        totalCachedReadTokens: number;
+        totalReasonTokens: number;
+        turns: number;
+    };
 }
 /**
  * Creates an agent state manager for tracking workflow state.
@@ -954,7 +982,11 @@ declare const AGENT_HANDLER_NAMES: {
 interface ZeitlichSession<M = unknown> {
     runSession<T extends JsonSerializable<T>>(args: {
         stateManager: AgentStateManager<T>;
-    }): Promise<M | null>;
+    }): Promise<{
+        finalMessage: M | null;
+        exitReason: SessionExitReason;
+        usage: ReturnType<AgentStateManager<T>["getTotalUsage"]>;
+    }>;
 }
 /**
  * Session-level hooks for lifecycle events
@@ -965,7 +997,7 @@ interface SessionLifecycleHooks {
     /** Called when session ends */
     onSessionEnd?: SessionEndHook;
 }
-declare const createSession: <T extends ToolMap, M = unknown>({ threadId, agentName, maxTurns, metadata, runAgent, threadOps, buildContextMessage, subagents, tools, processToolsInParallel, hooks, appendSystemPrompt, systemPrompt, waitForInputTimeout, }: SessionConfig<T, M> & AgentConfig) => Promise<ZeitlichSession<M>>;
+declare const createSession: <T extends ToolMap, M = unknown>({ threadId, agentName, description, maxTurns, metadata, runAgent, threadOps, buildContextMessage, subagents, tools, processToolsInParallel, hooks, appendSystemPrompt, systemPrompt, waitForInputTimeout, }: SessionConfig<T, M> & AgentConfig) => Promise<ZeitlichSession<M>>;
 /**
  * Proxy the default ZeitlichSharedActivities as ThreadOps<StoredMessage>.
  * Call this in workflow code for the standard LangChain/StoredMessage setup.
@@ -1042,7 +1074,7 @@ declare const grepTool: {
 };
 type GrepArgs = z$1.infer<typeof grepTool.schema>;
 
-declare const readTool: {
+declare const readFileTool: {
     name: "FileRead";
     description: string;
     schema: z$1.ZodObject<{
@@ -1052,9 +1084,9 @@ declare const readTool: {
     }, z$1.core.$strip>;
     strict: true;
 };
-type FileReadArgs = z$1.infer<typeof readTool.schema>;
+type FileReadArgs = z$1.infer<typeof readFileTool.schema>;
 
-declare const writeTool: {
+declare const writeFileTool: {
     name: "FileWrite";
     description: string;
     schema: z$1.ZodObject<{
@@ -1063,7 +1095,7 @@ declare const writeTool: {
     }, z$1.core.$strip>;
     strict: true;
 };
-type FileWriteArgs = z$1.infer<typeof writeTool.schema>;
+type FileWriteArgs = z$1.infer<typeof writeFileTool.schema>;
 
 declare const editTool: {
     name: "FileEdit";
@@ -1188,7 +1220,7 @@ type AskUserQuestionArgs = z.infer<typeof askUserQuestionTool.schema>;
 /**
  * Creates handler for user interaction tool - creates AI messages for display.
  */
-declare const createAskUserQuestionHandler: () => ToolHandler<AskUserQuestionArgs, {
+declare const createAskUserQuestionHandler: () => ActivityToolHandler<AskUserQuestionArgs, {
     questions: {
         question: string;
         header: string;
@@ -1200,4 +1232,4 @@ declare const createAskUserQuestionHandler: () => ToolHandler<AskUserQuestionArg
     }[];
 }>;
 
-export { type ThreadOps as $, type AgentResponse as A, type BashArgs as B, type RunAgentActivity as C, type RunAgentConfig as D, type SessionEndHookContext as E, type FileEditArgs as F, type GlobArgs as G, type SessionExitReason as H, type InferToolResults as I, type JsonPrimitive as J, type SessionLifecycleHooks as K, type SessionStartHook as L, type SessionStartHookContext as M, type SubagentArgs as N, type SubagentConfig as O, type ParsedToolCall as P, type SubagentHooks as Q, type RawToolCall as R, type SessionEndHook as S, type SubagentInput as T, type TaskCreateArgs as U, type TaskGetArgs as V, type TaskListArgs as W, type TaskStatus as X, type TaskUpdateArgs as Y, type ThreadManager as Z, type ThreadManagerConfig as _, type ActivityToolHandler as a, type ToolArgs as a0, type ToolCallResult as a1, type ToolCallResultUnion as a2, type ToolDefinition as a3, type ToolHandler as a4, type ToolHandlerContext as a5, type ToolHandlerResponse as a6, type ToolHooks as a7, type ToolMap as a8, type ToolMessageContent as a9, grepTool as aA, hasNoOtherToolCalls as aB, isTerminalStatus as aC, proxyDefaultThreadOps as aD, readTool as aE, taskCreateTool as aF, taskGetTool as aG, taskListTool as aH, taskUpdateTool as aI, withAutoAppend as aJ, writeTool as aK, type ToolNames as aa, type ToolResult as ab, type ToolResultConfig as ac, type ToolRouter as ad, type ToolWithHandler as ae, type WorkflowTask as af, type ZeitlichSession as ag, type ZeitlichSharedActivities as ah, askUserQuestionTool as ai, bashTool as aj, createAgentStateManager as ak, createAskUserQuestionHandler as al, createBashToolDescription as am, createSession as an, createSharedActivities as ao, createSubagentTool as ap, createTaskCreateHandler as aq, createTaskGetHandler as ar, createTaskListHandler as as, createTaskUpdateHandler as at, createThreadManager as au, createToolRouter as av, defineSubagent as aw, defineTool as ax, editTool as ay, globTool as az, AGENT_HANDLER_NAMES as b, type AgentConfig as c, type AgentFile as d, type AgentState as e, type AgentStateManager as f, type AgentStatus as g, type AppendToolResultFn as h, type AskUserQuestionArgs as i, type BaseAgentState as j, type BaseThreadManager as k, type FileReadArgs as l, type FileWriteArgs as m, type GrepArgs as n, type JsonSerializable as o, type JsonValue as p, type ParsedToolCallUnion as q, type PostToolUseFailureHook as r, type PostToolUseFailureHookContext as s, type PostToolUseFailureHookResult as t, type PostToolUseHook as u, type PostToolUseHookContext as v, type PreToolUseHook as w, type PreToolUseHookContext as x, type PreToolUseHookResult as y, type ProcessToolCallsContext as z };
+export { type ThreadOps as $, type AgentResponse as A, type BashArgs as B, type RunAgentActivity as C, type RunAgentConfig as D, type SessionEndHookContext as E, type FileEditArgs as F, type GlobArgs as G, type SessionExitReason as H, type InferToolResults as I, type JsonPrimitive as J, type SessionLifecycleHooks as K, type SessionStartHook as L, type SessionStartHookContext as M, type SubagentArgs as N, type SubagentConfig as O, type ParsedToolCall as P, type SubagentHooks as Q, type RawToolCall as R, type SessionEndHook as S, type SubagentInput as T, type TaskCreateArgs as U, type TaskGetArgs as V, type TaskListArgs as W, type TaskStatus as X, type TaskUpdateArgs as Y, type ThreadManager as Z, type ThreadManagerConfig as _, type ActivityToolHandler as a, type ToolArgs as a0, type ToolCallResult as a1, type ToolCallResultUnion as a2, type ToolDefinition as a3, type ToolHandler as a4, type ToolHandlerContext as a5, type ToolHandlerResponse as a6, type ToolHooks as a7, type ToolMap as a8, type ToolMessageContent as a9, grepTool as aA, hasNoOtherToolCalls as aB, isTerminalStatus as aC, proxyDefaultThreadOps as aD, readFileTool as aE, taskCreateTool as aF, taskGetTool as aG, taskListTool as aH, taskUpdateTool as aI, withAutoAppend as aJ, writeFileTool as aK, type ToolNames as aa, type ToolResult as ab, type ToolResultConfig as ac, type ToolRouter as ad, type ToolWithHandler as ae, type WorkflowTask as af, type ZeitlichSession as ag, type ZeitlichSharedActivities as ah, askUserQuestionTool as ai, bashTool as aj, createAgentStateManager as ak, createAskUserQuestionHandler as al, createBashToolDescription as am, createSession as an, createSharedActivities as ao, createSubagentTool as ap, createTaskCreateHandler as aq, createTaskGetHandler as ar, createTaskListHandler as as, createTaskUpdateHandler as at, createThreadManager as au, createToolRouter as av, defineSubagent as aw, defineTool as ax, editTool as ay, globTool as az, AGENT_HANDLER_NAMES as b, type AgentConfig as c, type AgentFile as d, type AgentState as e, type AgentStateManager as f, type AgentStatus as g, type AppendToolResultFn as h, type AskUserQuestionArgs as i, type BaseAgentState as j, type BaseThreadManager as k, type FileReadArgs as l, type FileWriteArgs as m, type GrepArgs as n, type JsonSerializable as o, type JsonValue as p, type ParsedToolCallUnion as q, type PostToolUseFailureHook as r, type PostToolUseFailureHookContext as s, type PostToolUseFailureHookResult as t, type PostToolUseHook as u, type PostToolUseHookContext as v, type PreToolUseHook as w, type PreToolUseHookContext as x, type PreToolUseHookResult as y, type ProcessToolCallsContext as z };

package/dist/{workflow-CyYHDbrr.d.ts → workflow-C2ShwjC7.d.ts}

@@ -116,7 +116,7 @@ interface ToolWithHandler<TName extends string = string, TSchema extends z$1.Zod
     strict?: boolean;
     max_uses?: number;
     /** Whether this tool is available to the agent (default: true). Disabled tools are excluded from definitions and rejected at parse time. */
-    enabled?: boolean;
+    enabled?: () => boolean;
     /** Per-tool lifecycle hooks (run in addition to global hooks) */
     hooks?: ToolHooks<z$1.infer<TSchema>, TResult>;
 }
@@ -135,7 +135,7 @@ type ToolMap = Record<string, {
     handler: ToolHandler<any, any, any>;
     strict?: boolean;
     max_uses?: number;
-    enabled?: boolean;
+    enabled?: () => boolean;
     hooks?: ToolHooks<any, any>;
 }>;
 /**
@@ -187,6 +187,8 @@ interface ToolHandlerResponse<TResult = null> {
      * payloads through Temporal's activity payload limit.
      */
     resultAppended?: boolean;
+    /** Token usage from the tool execution (e.g. child agent invocations) */
+    usage?: TokenUsage;
 }
 /**
  * Context passed to tool handlers for additional data beyond tool args.
@@ -237,6 +239,7 @@ interface ToolCallResult<TName extends string = string, TResult = unknown> {
     toolCallId: string;
     name: TName;
     data: TResult;
+    usage?: TokenUsage;
 }
 /**
  * Options for creating a tool router.
@@ -465,6 +468,10 @@ interface BaseAgentState {
     turns: number;
     tasks: Map<string, WorkflowTask>;
     systemPrompt: string;
+    totalInputTokens: number;
+    totalOutputTokens: number;
+    cachedWriteTokens: number;
+    cachedReadtTokens: number;
 }
 /**
  * File representation for agent workflows
@@ -481,17 +488,20 @@ interface AgentFile {
     /** MIME type of the file */
     mimeType?: string;
 }
+interface TokenUsage {
+    inputTokens?: number;
+    outputTokens?: number;
+    cachedWriteTokens?: number;
+    cachedReadTokens?: number;
+    reasonTokens?: number;
+}
 /**
  * Agent response from LLM invocation
  */
 interface AgentResponse<M = StoredMessage> {
     message: M;
     rawToolCalls: RawToolCall[];
-    usage?: {
-        input_tokens?: number;
-        output_tokens?: number;
-        total_tokens?: number;
-    };
+    usage?: TokenUsage;
 }
 /**
  * Thread operations required by a session.
@@ -566,9 +576,10 @@ interface SerializableToolDefinition {
 /**
  * Configuration passed to runAgent activity
  */
-interface RunAgentConfig {
+interface RunAgentConfig extends AgentConfig {
+    /** The thread ID to use for the session */
     threadId: string;
-    agentName: string;
+    /** Metadata for the session */
     metadata?: Record<string, unknown>;
 }
 /**
@@ -597,7 +608,7 @@ interface SubagentConfig<TResult extends z$1.ZodType = z$1.ZodType> {
     /** Description shown to the parent agent explaining what this subagent does */
     description: string;
     /** Whether this subagent is available (default: true). Disabled subagents are excluded from the Subagent tool. */
-    enabled?: boolean;
+    enabled?: () => boolean;
     /** Temporal workflow function or type name (used with executeChild) */
     workflow: string | Workflow;
     /** Optional task queue - defaults to parent's queue if not specified */
@@ -927,6 +938,23 @@ interface AgentStateManager<TCustom extends JsonSerializable<TCustom>> {
     deleteTask(id: string): boolean;
     /** Set the tools (converts Zod schemas to JSON Schema for serialization) */
     setTools(newTools: ToolDefinition[]): void;
+    /** Update the usage */
+    updateUsage(usage: {
+        inputTokens?: number;
+        outputTokens?: number;
+        cachedWriteTokens?: number;
+        cachedReadTokens?: number;
+        reasonTokens?: number;
+    }): void;
+    /** Get the total usage */
+    getTotalUsage(): {
+        totalInputTokens: number;
+        totalOutputTokens: number;
+        totalCachedWriteTokens: number;
+        totalCachedReadTokens: number;
+        totalReasonTokens: number;
+        turns: number;
+    };
 }
 /**
  * Creates an agent state manager for tracking workflow state.
@@ -954,7 +982,11 @@ declare const AGENT_HANDLER_NAMES: {
 interface ZeitlichSession<M = unknown> {
     runSession<T extends JsonSerializable<T>>(args: {
         stateManager: AgentStateManager<T>;
-    }): Promise<M | null>;
+    }): Promise<{
+        finalMessage: M | null;
+        exitReason: SessionExitReason;
+        usage: ReturnType<AgentStateManager<T>["getTotalUsage"]>;
+    }>;
 }
 /**
  * Session-level hooks for lifecycle events
@@ -965,7 +997,7 @@ interface SessionLifecycleHooks {
     /** Called when session ends */
     onSessionEnd?: SessionEndHook;
 }
-declare const createSession: <T extends ToolMap, M = unknown>({ threadId, agentName, maxTurns, metadata, runAgent, threadOps, buildContextMessage, subagents, tools, processToolsInParallel, hooks, appendSystemPrompt, systemPrompt, waitForInputTimeout, }: SessionConfig<T, M> & AgentConfig) => Promise<ZeitlichSession<M>>;
+declare const createSession: <T extends ToolMap, M = unknown>({ threadId, agentName, description, maxTurns, metadata, runAgent, threadOps, buildContextMessage, subagents, tools, processToolsInParallel, hooks, appendSystemPrompt, systemPrompt, waitForInputTimeout, }: SessionConfig<T, M> & AgentConfig) => Promise<ZeitlichSession<M>>;
 /**
  * Proxy the default ZeitlichSharedActivities as ThreadOps<StoredMessage>.
  * Call this in workflow code for the standard LangChain/StoredMessage setup.
@@ -1042,7 +1074,7 @@ declare const grepTool: {
 };
 type GrepArgs = z$1.infer<typeof grepTool.schema>;
 
-declare const readTool: {
+declare const readFileTool: {
     name: "FileRead";
     description: string;
     schema: z$1.ZodObject<{
@@ -1052,9 +1084,9 @@ declare const readTool: {
     }, z$1.core.$strip>;
     strict: true;
 };
-type FileReadArgs = z$1.infer<typeof readTool.schema>;
+type FileReadArgs = z$1.infer<typeof readFileTool.schema>;
 
-declare const writeTool: {
+declare const writeFileTool: {
     name: "FileWrite";
     description: string;
     schema: z$1.ZodObject<{
@@ -1063,7 +1095,7 @@ declare const writeTool: {
     }, z$1.core.$strip>;
     strict: true;
 };
-type FileWriteArgs = z$1.infer<typeof writeTool.schema>;
+type FileWriteArgs = z$1.infer<typeof writeFileTool.schema>;
 
 declare const editTool: {
     name: "FileEdit";
@@ -1188,7 +1220,7 @@ type AskUserQuestionArgs = z.infer<typeof askUserQuestionTool.schema>;
 /**
  * Creates handler for user interaction tool - creates AI messages for display.
  */
-declare const createAskUserQuestionHandler: () => ToolHandler<AskUserQuestionArgs, {
+declare const createAskUserQuestionHandler: () => ActivityToolHandler<AskUserQuestionArgs, {
     questions: {
         question: string;
         header: string;
@@ -1200,4 +1232,4 @@ declare const createAskUserQuestionHandler: () => ToolHandler<AskUserQuestionArg
     }[];
 }>;
 
-export { type ThreadOps as $, type AgentResponse as A, type BashArgs as B, type RunAgentActivity as C, type RunAgentConfig as D, type SessionEndHookContext as E, type FileEditArgs as F, type GlobArgs as G, type SessionExitReason as H, type InferToolResults as I, type JsonPrimitive as J, type SessionLifecycleHooks as K, type SessionStartHook as L, type SessionStartHookContext as M, type SubagentArgs as N, type SubagentConfig as O, type ParsedToolCall as P, type SubagentHooks as Q, type RawToolCall as R, type SessionEndHook as S, type SubagentInput as T, type TaskCreateArgs as U, type TaskGetArgs as V, type TaskListArgs as W, type TaskStatus as X, type TaskUpdateArgs as Y, type ThreadManager as Z, type ThreadManagerConfig as _, type ActivityToolHandler as a, type ToolArgs as a0, type ToolCallResult as a1, type ToolCallResultUnion as a2, type ToolDefinition as a3, type ToolHandler as a4, type ToolHandlerContext as a5, type ToolHandlerResponse as a6, type ToolHooks as a7, type ToolMap as a8, type ToolMessageContent as a9, grepTool as aA, hasNoOtherToolCalls as aB, isTerminalStatus as aC, proxyDefaultThreadOps as aD, readTool as aE, taskCreateTool as aF, taskGetTool as aG, taskListTool as aH, taskUpdateTool as aI, withAutoAppend as aJ, writeTool as aK, type ToolNames as aa, type ToolResult as ab, type ToolResultConfig as ac, type ToolRouter as ad, type ToolWithHandler as ae, type WorkflowTask as af, type ZeitlichSession as ag, type ZeitlichSharedActivities as ah, askUserQuestionTool as ai, bashTool as aj, createAgentStateManager as ak, createAskUserQuestionHandler as al, createBashToolDescription as am, createSession as an, createSharedActivities as ao, createSubagentTool as ap, createTaskCreateHandler as aq, createTaskGetHandler as ar, createTaskListHandler as as, createTaskUpdateHandler as at, createThreadManager as au, createToolRouter as av, defineSubagent as aw, defineTool as ax, editTool as ay, globTool as az, AGENT_HANDLER_NAMES as b, type AgentConfig as c, type AgentFile as d, type AgentState as e, type AgentStateManager as f, type AgentStatus as g, type AppendToolResultFn as h, type AskUserQuestionArgs as i, type BaseAgentState as j, type BaseThreadManager as k, type FileReadArgs as l, type FileWriteArgs as m, type GrepArgs as n, type JsonSerializable as o, type JsonValue as p, type ParsedToolCallUnion as q, type PostToolUseFailureHook as r, type PostToolUseFailureHookContext as s, type PostToolUseFailureHookResult as t, type PostToolUseHook as u, type PostToolUseHookContext as v, type PreToolUseHook as w, type PreToolUseHookContext as x, type PreToolUseHookResult as y, type ProcessToolCallsContext as z };
+export { type ThreadOps as $, type AgentResponse as A, type BashArgs as B, type RunAgentActivity as C, type RunAgentConfig as D, type SessionEndHookContext as E, type FileEditArgs as F, type GlobArgs as G, type SessionExitReason as H, type InferToolResults as I, type JsonPrimitive as J, type SessionLifecycleHooks as K, type SessionStartHook as L, type SessionStartHookContext as M, type SubagentArgs as N, type SubagentConfig as O, type ParsedToolCall as P, type SubagentHooks as Q, type RawToolCall as R, type SessionEndHook as S, type SubagentInput as T, type TaskCreateArgs as U, type TaskGetArgs as V, type TaskListArgs as W, type TaskStatus as X, type TaskUpdateArgs as Y, type ThreadManager as Z, type ThreadManagerConfig as _, type ActivityToolHandler as a, type ToolArgs as a0, type ToolCallResult as a1, type ToolCallResultUnion as a2, type ToolDefinition as a3, type ToolHandler as a4, type ToolHandlerContext as a5, type ToolHandlerResponse as a6, type ToolHooks as a7, type ToolMap as a8, type ToolMessageContent as a9, grepTool as aA, hasNoOtherToolCalls as aB, isTerminalStatus as aC, proxyDefaultThreadOps as aD, readFileTool as aE, taskCreateTool as aF, taskGetTool as aG, taskListTool as aH, taskUpdateTool as aI, withAutoAppend as aJ, writeFileTool as aK, type ToolNames as aa, type ToolResult as ab, type ToolResultConfig as ac, type ToolRouter as ad, type ToolWithHandler as ae, type WorkflowTask as af, type ZeitlichSession as ag, type ZeitlichSharedActivities as ah, askUserQuestionTool as ai, bashTool as aj, createAgentStateManager as ak, createAskUserQuestionHandler as al, createBashToolDescription as am, createSession as an, createSharedActivities as ao, createSubagentTool as ap, createTaskCreateHandler as aq, createTaskGetHandler as ar, createTaskListHandler as as, createTaskUpdateHandler as at, createThreadManager as au, createToolRouter as av, defineSubagent as aw, defineTool as ax, editTool as ay, globTool as az, AGENT_HANDLER_NAMES as b, type AgentConfig as c, type AgentFile as d, type AgentState as e, type AgentStateManager as f, type AgentStatus as g, type AppendToolResultFn as h, type AskUserQuestionArgs as i, type BaseAgentState as j, type BaseThreadManager as k, type FileReadArgs as l, type FileWriteArgs as m, type GrepArgs as n, type JsonSerializable as o, type JsonValue as p, type ParsedToolCallUnion as q, type PostToolUseFailureHook as r, type PostToolUseFailureHookContext as s, type PostToolUseFailureHookResult as t, type PostToolUseHook as u, type PostToolUseHookContext as v, type PreToolUseHook as w, type PreToolUseHookContext as x, type PreToolUseHookResult as y, type ProcessToolCallsContext as z };

package/dist/workflow.cjs

@@ -8,18 +8,18 @@ function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 var z13__default = /*#__PURE__*/_interopDefault(z13);
 
 // src/lib/session.ts
-var SUBAGENT_TOOL = "Subagent";
+var SUBAGENT_TOOL_NAME = "Subagent";
 function buildSubagentDescription(subagents) {
   const subagentList = subagents.map((s) => `- **${s.agentName}**: ${s.description}`).join("\n");
   return `Launch a new agent to handle complex tasks autonomously.
 
-The ${SUBAGENT_TOOL} tool launches specialized agents (subprocesses) that autonomously handle complex tasks. Each agent type has specific capabilities and tools available to it.
+The ${SUBAGENT_TOOL_NAME} tool launches specialized agents (subprocesses) that autonomously handle complex tasks. Each agent type has specific capabilities and tools available to it.
 
 Available agent types:
 
 ${subagentList}
 
-When using the ${SUBAGENT_TOOL} tool, you must specify a subagent parameter to select which agent type to use.
+When using the ${SUBAGENT_TOOL_NAME} tool, you must specify a subagent parameter to select which agent type to use.
 
 Usage notes:
 
@@ -38,7 +38,7 @@ function createSubagentTool(subagents) {
   }
   const names = subagents.map((s) => s.agentName);
   return {
-    name: SUBAGENT_TOOL,
+    name: SUBAGENT_TOOL_NAME,
     description: buildSubagentDescription(subagents),
     schema: z13__default.default.object({
       subagent: z13__default.default.enum(names).describe("The type of subagent to launch"),
@@ -66,11 +66,12 @@ function createSubagentHandler(subagents) {
       args: [input],
       taskQueue: config.taskQueue ?? parentTaskQueue
     };
-    const { toolResponse, data } = typeof config.workflow === "string" ? await workflow.executeChild(config.workflow, childOpts) : await workflow.executeChild(config.workflow, childOpts);
+    const { toolResponse, data, usage } = typeof config.workflow === "string" ? await workflow.executeChild(config.workflow, childOpts) : await workflow.executeChild(config.workflow, childOpts);
     const validated = config.resultSchema ? config.resultSchema.parse(data) : null;
     return {
       toolResponse,
-      data: validated
+      data: validated,
+      ...usage && { usage }
     };
   };
 }
@@ -82,20 +83,17 @@ function createToolRouter(options) {
   for (const [_key, tool] of Object.entries(options.tools)) {
     toolMap.set(tool.name, tool);
   }
-  const isEnabled = (tool) => tool.enabled !== false;
+  const isEnabled = (tool) => tool.enabled?.() ?? true;
   if (options.subagents) {
-    const enabledSubagents = options.subagents.filter(
-      (s) => s.enabled !== false
-    );
-    if (enabledSubagents.length > 0) {
+    if (options.subagents.length > 0) {
       const subagentHooksMap = /* @__PURE__ */ new Map();
-      for (const s of enabledSubagents) {
+      for (const s of options.subagents) {
         if (s.hooks) subagentHooksMap.set(s.agentName, s.hooks);
       }
       const resolveSubagentName = (args) => args.subagent;
-      toolMap.set("Subagent", {
-        ...createSubagentTool(enabledSubagents),
-        handler: createSubagentHandler(enabledSubagents),
+      toolMap.set(SUBAGENT_TOOL_NAME, {
+        ...createSubagentTool(options.subagents),
+        handler: createSubagentHandler(options.subagents),
         ...subagentHooksMap.size > 0 && {
           hooks: {
             onPreToolUse: async (ctx) => {
@@ -286,13 +284,19 @@ function createToolRouter(options) {
       return Array.from(toolMap.entries()).filter(([, tool]) => isEnabled(tool)).map(([name]) => name);
     },
     getToolDefinitions() {
-      return Array.from(toolMap).filter(([, tool]) => isEnabled(tool)).map(([name, tool]) => ({
-        name,
-        description: tool.description,
-        schema: tool.schema,
-        strict: tool.strict,
-        max_uses: tool.max_uses
-      }));
+      const activeSubagents = options.subagents?.filter((subagent) => isEnabled(subagent)) ?? [];
+      return [
+        ...Array.from(toolMap).filter(
+          ([, tool]) => isEnabled(tool) && tool.name !== SUBAGENT_TOOL_NAME
+        ).map(([name, tool]) => ({
+          name,
+          description: tool.description,
+          schema: tool.schema,
+          strict: tool.strict,
+          max_uses: tool.max_uses
+        })),
+        ...activeSubagents.length > 0 ? [createSubagentTool(activeSubagents)] : []
+      ];
     },
     // --- Methods for processing tool calls ---
     async processToolCalls(toolCalls, context) {
@@ -386,6 +390,7 @@ function hasNoOtherToolCalls(toolCalls, excludeName) {
 var createSession = async ({
   threadId,
   agentName,
+  description,
   maxTurns = 50,
   metadata = {},
   runAgent,
@@ -425,7 +430,9 @@ var createSession = async ({
     }
   };
   return {
-    runSession: async ({ stateManager }) => {
+    runSession: async ({
+      stateManager
+    }) => {
       workflow.setHandler(
         workflow.defineUpdate(`add${agentName}Message`),
         async (message) => {
@@ -452,7 +459,6 @@ var createSession = async ({
           metadata
         });
       }
-      stateManager.setTools(toolRouter.getToolDefinitions());
       await initializeThread(threadId);
       if (appendSystemPrompt && systemPrompt && systemPrompt.trim() !== "") {
         await appendSystemMessage(threadId, systemPrompt);
@@ -463,15 +469,25 @@ var createSession = async ({
         while (stateManager.isRunning() && !stateManager.isTerminal() && stateManager.getTurns() < maxTurns) {
           stateManager.incrementTurns();
           const currentTurn = stateManager.getTurns();
-          const { message, rawToolCalls } = await runAgent({
+          stateManager.setTools(toolRouter.getToolDefinitions());
+          const { message, rawToolCalls, usage } = await runAgent({
             threadId,
             agentName,
-            metadata
+            metadata,
+            systemPrompt,
+            description
           });
+          if (usage) {
+            stateManager.updateUsage(usage);
+          }
           if (!toolRouter.hasTools() || rawToolCalls.length === 0) {
             stateManager.complete();
             exitReason = "completed";
-            return message;
+            return {
+              finalMessage: message,
+              exitReason,
+              usage: stateManager.getTotalUsage()
+            };
           }
           const parsedToolCalls = [];
           for (const tc of rawToolCalls) {
@@ -488,9 +504,17 @@ var createSession = async ({
               });
             }
           }
-          await toolRouter.processToolCalls(parsedToolCalls, {
-            turn: currentTurn
-          });
+          const toolCallResults = await toolRouter.processToolCalls(
+            parsedToolCalls,
+            {
+              turn: currentTurn
+            }
+          );
+          for (const result of toolCallResults) {
+            if (result.usage) {
+              stateManager.updateUsage(result.usage);
+            }
+          }
           if (stateManager.getStatus() === "WAITING_FOR_INPUT") {
             const conditionMet = await workflow.condition(
               () => stateManager.getStatus() === "RUNNING",
@@ -512,7 +536,11 @@ var createSession = async ({
       } finally {
         await callSessionEnd(exitReason, stateManager.getTurns());
       }
-      return null;
+      return {
+        finalMessage: null,
+        exitReason,
+        usage: stateManager.getTotalUsage()
+      };
     }
   };
 };
@@ -549,6 +577,11 @@ function createAgentStateManager({
   let version = initialState?.version ?? 0;
   let turns = initialState?.turns ?? 0;
   let tools = initialState?.tools ?? [];
+  let totalInputTokens = 0;
+  let totalOutputTokens = 0;
+  let totalCachedWriteTokens = 0;
+  let totalCachedReadTokens = 0;
+  let totalReasonTokens = 0;
   const tasks = new Map(initialState?.tasks);
   const {
     status: _,
@@ -663,6 +696,23 @@ function createAgentStateManager({
         version++;
       }
       return deleted;
+    },
+    updateUsage(usage) {
+      totalInputTokens += usage.inputTokens ?? 0;
+      totalOutputTokens += usage.outputTokens ?? 0;
+      totalCachedWriteTokens += usage.cachedWriteTokens ?? 0;
+      totalCachedReadTokens += usage.cachedReadTokens ?? 0;
+      totalReasonTokens += usage.reasonTokens ?? 0;
+    },
+    getTotalUsage() {
+      return {
+        totalInputTokens,
+        totalOutputTokens,
+        totalCachedWriteTokens,
+        totalCachedReadTokens,
+        totalReasonTokens,
+        turns
+      };
     }
   };
 }
@@ -715,7 +765,7 @@ Examples:
   }),
   strict: true
 };
-var readTool = {
+var readFileTool = {
   name: "FileRead",
   description: `Read file contents with optional pagination.
 
@@ -738,22 +788,21 @@ The tool returns the file content in an appropriate format:
   }),
   strict: true
 };
-var writeTool = {
+var writeFileTool = {
   name: "FileWrite",
   description: `Create or overwrite a file with new content.
 
 Usage:
-- Provide the absolute path to the file
 - The file will be created if it doesn't exist
 - If the file exists, it will be completely overwritten
 
 IMPORTANT:
 - You must read the file first (in this session) before writing to it
 - This is an atomic write operation - the entire file is replaced
-- Path must be absolute (e.g., "/docs/readme.md", not "docs/readme.md")
+- Path must be relative to the root of the file system (e.g., "docs/readme.md", not "/docs/readme.md")
 `,
   schema: z13.z.object({
-    file_path: z13.z.string().describe("The absolute path to the file to write"),
+    file_path: z13.z.string().describe("The path to the file to write"),
     content: z13.z.string().describe("The content to write to the file")
   }),
   strict: true
@@ -1015,7 +1064,7 @@ Usage notes:
 };
 
 // src/tools/ask-user-question/handler.ts
-var createAskUserQuestionHandler = () => (args) => {
+var createAskUserQuestionHandler = () => async (args) => {
   return {
     toolResponse: "Question submitted",
     data: { questions: args.questions }
@@ -1043,11 +1092,11 @@ exports.grepTool = grepTool;
 exports.hasNoOtherToolCalls = hasNoOtherToolCalls;
 exports.isTerminalStatus = isTerminalStatus;
 exports.proxyDefaultThreadOps = proxyDefaultThreadOps;
-exports.readTool = readTool;
+exports.readFileTool = readFileTool;
 exports.taskCreateTool = taskCreateTool;
 exports.taskGetTool = taskGetTool;
 exports.taskListTool = taskListTool;
 exports.taskUpdateTool = taskUpdateTool;
-exports.writeTool = writeTool;
+exports.writeFileTool = writeFileTool;
 //# sourceMappingURL=workflow.cjs.map
 //# sourceMappingURL=workflow.cjs.map