zeitlich 0.2.1 → 0.2.2

package/dist/workflow.cjs

@@ -1,11 +1,11 @@
 'use strict';
 
 var workflow = require('@temporalio/workflow');
-var z4 = require('zod');
+var z5 = require('zod');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
-var z4__default = /*#__PURE__*/_interopDefault(z4);
+var z5__default = /*#__PURE__*/_interopDefault(z5);
 
 // src/lib/session.ts
 var TASK_TOOL = "Task";
@@ -40,10 +40,10 @@ function createTaskTool(subagents) {
   return {
     name: TASK_TOOL,
     description: buildTaskDescription(subagents),
-    schema: z4__default.default.object({
-      subagent: z4__default.default.enum(names).describe("The type of subagent to launch"),
-      description: z4__default.default.string().describe("A short (3-5 word) description of the task"),
-      prompt: z4__default.default.string().describe("The task for the agent to perform")
+    schema: z5__default.default.object({
+      subagent: z5__default.default.enum(names).describe("The type of subagent to launch"),
+      description: z5__default.default.string().describe("A short (3-5 word) description of the task"),
+      prompt: z5__default.default.string().describe("The task for the agent to perform")
     })
   };
 }
@@ -57,16 +57,21 @@ function createTaskHandler(subagents) {
       );
     }
     const childWorkflowId = `${parentWorkflowId}-${args.subagent}-${workflow.uuid4()}`;
-    const childResult = await workflow.executeChild(config.workflowType, {
+    const input = {
+      prompt: args.prompt,
+      ...config.context && { context: config.context }
+    };
+    const childOpts = {
       workflowId: childWorkflowId,
-      args: [{ prompt: args.prompt }],
+      args: [input],
       taskQueue: config.taskQueue ?? parentTaskQueue
-    });
+    };
+    const childResult = typeof config.workflow === "string" ? await workflow.executeChild(config.workflow, childOpts) : await workflow.executeChild(config.workflow, childOpts);
     const validated = config.resultSchema ? config.resultSchema.parse(childResult) : childResult;
-    const content = typeof validated === "string" ? validated : JSON.stringify(validated, null, 2);
+    const toolResponse = typeof validated === "string" ? validated : JSON.stringify(validated, null, 2);
     return {
-      content,
-      result: {
+      toolResponse,
+      data: {
         result: validated,
         childWorkflowId
       }
@@ -93,8 +98,8 @@ Use this tool to:
 - Execute scripts and chain commands with pipes (|) or logical operators (&&, ||)
 - Inspect files and directories
 `,
-  schema: z4__default.default.object({
-    command: z4__default.default.string().describe(
+  schema: z5__default.default.object({
+    command: z5__default.default.string().describe(
       "The bash command to execute. Can include pipes (|), redirects (>, >>), logical operators (&&, ||), and shell features like command substitution $(...)."
     )
   }),
@@ -141,17 +146,17 @@ var taskCreateTool = {
   - Include enough detail in the description for another agent to understand and complete the task
   - After creating tasks, use TaskUpdate to set up dependencies (blocks/blockedBy) if needed
   - Check TaskList first to avoid creating duplicate tasks`,
-  schema: z4__default.default.object({
-    subject: z4__default.default.string().describe(
+  schema: z5__default.default.object({
+    subject: z5__default.default.string().describe(
       'A brief, actionable title in imperative form (e.g., "Fix authentication bug in login flow")'
     ),
-    description: z4__default.default.string().describe(
+    description: z5__default.default.string().describe(
       "Detailed description of what needs to be done, including context and acceptance criteria"
     ),
-    activeForm: z4__default.default.string().describe(
+    activeForm: z5__default.default.string().describe(
       'Present continuous form shown in spinner when task is in_progress (e.g., "Fixing authentication bug"). This is displayed to the user while you work on the task.'
     ),
-    metadata: z4__default.default.record(z4__default.default.string(), z4__default.default.string()).describe("Arbitrary key-value pairs for tracking")
+    metadata: z5__default.default.record(z5__default.default.string(), z5__default.default.string()).describe("Arbitrary key-value pairs for tracking")
   })
 };
 
@@ -175,9 +180,30 @@ function createToolRouter(options) {
     toolMap.set(tool.name, tool);
   }
   if (options.subagents) {
+    const subagentHooksMap = /* @__PURE__ */ new Map();
+    for (const s of options.subagents) {
+      if (s.hooks) subagentHooksMap.set(s.name, s.hooks);
+    }
+    const resolveSubagentName = (args) => args.subagent;
     toolMap.set("Task", {
       ...createTaskTool(options.subagents),
-      handler: createTaskHandler(options.subagents)
+      handler: createTaskHandler(options.subagents),
+      ...subagentHooksMap.size > 0 && {
+        hooks: {
+          onPreToolUse: async (ctx) => {
+            const hooks = subagentHooksMap.get(resolveSubagentName(ctx.args));
+            return hooks?.onPreExecution?.(ctx) ?? {};
+          },
+          onPostToolUse: async (ctx) => {
+            const hooks = subagentHooksMap.get(resolveSubagentName(ctx.args));
+            await hooks?.onPostExecution?.(ctx);
+          },
+          onPostToolUseFailure: async (ctx) => {
+            const hooks = subagentHooksMap.get(resolveSubagentName(ctx.args));
+            return hooks?.onExecutionFailure?.(ctx) ?? {};
+          }
+        }
+      }
     });
   }
   if (options.buildInTools) {
@@ -200,6 +226,8 @@ function createToolRouter(options) {
   }
   async function processToolCall(toolCall, turn, handlerContext) {
     const startTime = Date.now();
+    const tool = toolMap.get(toolCall.name);
+    const toolHooks = tool?.hooks;
     let effectiveArgs = toolCall.args;
     if (options.hooks?.onPreToolUse) {
       const preResult = await options.hooks.onPreToolUse({
@@ -222,7 +250,27 @@ function createToolRouter(options) {
         effectiveArgs = preResult.modifiedArgs;
       }
     }
-    const tool = toolMap.get(toolCall.name);
+    if (toolHooks?.onPreToolUse) {
+      const preResult = await toolHooks.onPreToolUse({
+        args: effectiveArgs,
+        threadId: options.threadId,
+        turn
+      });
+      if (preResult?.skip) {
+        await appendToolResult({
+          threadId: options.threadId,
+          toolCallId: toolCall.id,
+          content: JSON.stringify({
+            skipped: true,
+            reason: "Skipped by tool PreToolUse hook"
+          })
+        });
+        return null;
+      }
+      if (preResult?.modifiedArgs !== void 0) {
+        effectiveArgs = preResult.modifiedArgs;
+      }
+    }
     let result;
     let content;
     try {
@@ -231,30 +279,50 @@ function createToolRouter(options) {
           effectiveArgs,
           handlerContext ?? {}
         );
-        result = response.result;
-        content = response.content;
+        result = response.data;
+        content = response.toolResponse;
       } else {
         result = { error: `Unknown tool: ${toolCall.name}` };
         content = JSON.stringify(result, null, 2);
       }
     } catch (error) {
-      if (options.hooks?.onPostToolUseFailure) {
+      const err = error instanceof Error ? error : new Error(String(error));
+      let recovered = false;
+      if (toolHooks?.onPostToolUseFailure) {
+        const failureResult = await toolHooks.onPostToolUseFailure({
+          args: effectiveArgs,
+          error: err,
+          threadId: options.threadId,
+          turn
+        });
+        if (failureResult?.fallbackContent !== void 0) {
+          content = failureResult.fallbackContent;
+          result = { error: String(error), recovered: true };
+          recovered = true;
+        } else if (failureResult?.suppress) {
+          content = JSON.stringify({ error: String(error), suppressed: true });
+          result = { error: String(error), suppressed: true };
+          recovered = true;
+        }
+      }
+      if (!recovered && options.hooks?.onPostToolUseFailure) {
         const failureResult = await options.hooks.onPostToolUseFailure({
           toolCall,
-          error: error instanceof Error ? error : new Error(String(error)),
+          error: err,
           threadId: options.threadId,
           turn
         });
         if (failureResult?.fallbackContent !== void 0) {
           content = failureResult.fallbackContent;
           result = { error: String(error), recovered: true };
+          recovered = true;
         } else if (failureResult?.suppress) {
           content = JSON.stringify({ error: String(error), suppressed: true });
           result = { error: String(error), suppressed: true };
-        } else {
-          throw error;
+          recovered = true;
         }
-      } else {
+      }
+      if (!recovered) {
         throw error;
       }
     }
@@ -266,10 +334,19 @@ function createToolRouter(options) {
     const toolResult = {
       toolCallId: toolCall.id,
       name: toolCall.name,
-      result
+      data: result
     };
+    const durationMs = Date.now() - startTime;
+    if (toolHooks?.onPostToolUse) {
+      await toolHooks.onPostToolUse({
+        args: effectiveArgs,
+        result,
+        threadId: options.threadId,
+        turn,
+        durationMs
+      });
+    }
     if (options.hooks?.onPostToolUse) {
-      const durationMs = Date.now() - startTime;
       await options.hooks.onPostToolUse({
         toolCall,
         result: toolResult,
@@ -350,12 +427,12 @@ function createToolRouter(options) {
         await appendToolResult({
           threadId: options.threadId,
           toolCallId: toolCall.id,
-          content: response.content
+          content: response.toolResponse
         });
         return {
           toolCallId: toolCall.id,
           name: toolCall.name,
-          result: response.result
+          data: response.data ?? null
         };
       };
       if (options.parallel) {
@@ -381,6 +458,12 @@ function createToolRouter(options) {
     }
   };
 }
+function defineTool(tool) {
+  return tool;
+}
+function defineSubagent(config) {
+  return config;
+}
 function hasNoOtherToolCalls(toolCalls, excludeName) {
   return toolCalls.filter((tc) => tc.name !== excludeName).length === 0;
 }
@@ -485,17 +568,47 @@ var createSession = async ({
             return message;
           }
           const rawToolCalls = await parseToolCalls(message);
-          const parsedToolCalls = rawToolCalls.filter((tc) => tc.name !== "Task").map((tc) => toolRouter.parseToolCall(tc));
-          const taskToolCalls = subagents && subagents.length > 0 ? rawToolCalls.filter((tc) => tc.name === "Task").map((tc) => {
-            const parsedArgs = createTaskTool(subagents).schema.parse(
-              tc.args
-            );
-            return {
-              id: tc.id ?? "",
-              name: tc.name,
-              args: parsedArgs
-            };
-          }) : [];
+          const parsedToolCalls = [];
+          for (const tc of rawToolCalls.filter(
+            (tc2) => tc2.name !== "Task"
+          )) {
+            try {
+              parsedToolCalls.push(toolRouter.parseToolCall(tc));
+            } catch (error) {
+              await appendToolResult({
+                threadId,
+                toolCallId: tc.id ?? "",
+                content: JSON.stringify({
+                  error: `Invalid tool call for "${tc.name}": ${error instanceof Error ? error.message : String(error)}`
+                })
+              });
+            }
+          }
+          const taskToolCalls = [];
+          if (subagents && subagents.length > 0) {
+            for (const tc of rawToolCalls.filter(
+              (tc2) => tc2.name === "Task"
+            )) {
+              try {
+                const parsedArgs = createTaskTool(subagents).schema.parse(
+                  tc.args
+                );
+                taskToolCalls.push({
+                  id: tc.id ?? "",
+                  name: tc.name,
+                  args: parsedArgs
+                });
+              } catch (error) {
+                await appendToolResult({
+                  threadId,
+                  toolCallId: tc.id ?? "",
+                  content: JSON.stringify({
+                    error: `Invalid tool call for "Task": ${error instanceof Error ? error.message : String(error)}`
+                  })
+                });
+              }
+            }
+          }
           await toolRouter.processToolCalls(
             [...parsedToolCalls, ...taskToolCalls],
             {
@@ -525,8 +638,6 @@ var createSession = async ({
 function isTerminalStatus(status) {
   return status === "COMPLETED" || status === "FAILED" || status === "CANCELLED";
 }
-
-// src/lib/state-manager.ts
 var getStateQuery = workflow.defineQuery("getState");
 function createAgentStateManager(initialState) {
   let status = initialState?.status ?? "RUNNING";
@@ -621,7 +732,13 @@ function createAgentStateManager(initialState) {
       version++;
     },
     setTools(newTools) {
-      tools = newTools;
+      tools = newTools.map((tool) => ({
+        name: tool.name,
+        description: tool.description,
+        schema: z5.z.toJSONSchema(tool.schema),
+        strict: tool.strict,
+        max_uses: tool.max_uses
+      }));
     },
     deleteTask(id) {
       const deleted = tasks.delete(id);
@@ -652,18 +769,18 @@ Usage notes:
 * Use multiSelect: true to allow multiple answers to be selected for a question
 * If you recommend a specific option, make that the first option in the list and add "(Recommended)" at the end of the label
 `,
-  schema: z4__default.default.object({
-    questions: z4__default.default.array(
-      z4__default.default.object({
-        question: z4__default.default.string().describe("The full question text to display"),
-        header: z4__default.default.string().describe("Short label for the question (max 12 characters)"),
-        options: z4__default.default.array(
-          z4__default.default.object({
-            label: z4__default.default.string(),
-            description: z4__default.default.string()
+  schema: z5__default.default.object({
+    questions: z5__default.default.array(
+      z5__default.default.object({
+        question: z5__default.default.string().describe("The full question text to display"),
+        header: z5__default.default.string().describe("Short label for the question (max 12 characters)"),
+        options: z5__default.default.array(
+          z5__default.default.object({
+            label: z5__default.default.string(),
+            description: z5__default.default.string()
           })
         ).min(0).max(4).describe("Array of 0-4 choices, each with label and description"),
-        multiSelect: z4__default.default.boolean().describe("If true, users can select multiple options")
+        multiSelect: z5__default.default.boolean().describe("If true, users can select multiple options")
       })
     )
   }),
@@ -683,9 +800,9 @@ Examples:
 - "**/*.test.ts" - Find all test files recursively
 - "src/**/*.ts" - Find all TypeScript files in src directory
 `,
-  schema: z4.z.object({
-    pattern: z4.z.string().describe("Glob pattern to match files against"),
-    root: z4.z.string().optional().describe("Optional root directory to search from")
+  schema: z5.z.object({
+    pattern: z5.z.string().describe("Glob pattern to match files against"),
+    root: z5.z.string().optional().describe("Optional root directory to search from")
   }),
   strict: true
 };
@@ -703,13 +820,13 @@ Examples:
 - Search for function definitions with "function.*handleClick"
 - Search case-insensitively with ignoreCase: true
 `,
-  schema: z4.z.object({
-    pattern: z4.z.string().describe("Regex pattern to search for in file contents"),
-    ignoreCase: z4.z.boolean().optional().describe("Case-insensitive search (default: false)"),
-    maxMatches: z4.z.number().optional().describe("Maximum number of matches to return (default: 50)"),
-    includePatterns: z4.z.array(z4.z.string()).optional().describe("Glob patterns to include (e.g., ['*.ts', '*.js'])"),
-    excludePatterns: z4.z.array(z4.z.string()).optional().describe("Glob patterns to exclude (e.g., ['*.test.ts'])"),
-    contextLines: z4.z.number().optional().describe("Number of context lines to show around matches")
+  schema: z5.z.object({
+    pattern: z5.z.string().describe("Regex pattern to search for in file contents"),
+    ignoreCase: z5.z.boolean().optional().describe("Case-insensitive search (default: false)"),
+    maxMatches: z5.z.number().optional().describe("Maximum number of matches to return (default: 50)"),
+    includePatterns: z5.z.array(z5.z.string()).optional().describe("Glob patterns to include (e.g., ['*.ts', '*.js'])"),
+    excludePatterns: z5.z.array(z5.z.string()).optional().describe("Glob patterns to exclude (e.g., ['*.test.ts'])"),
+    contextLines: z5.z.number().optional().describe("Number of context lines to show around matches")
   }),
   strict: true
 };
@@ -727,12 +844,12 @@ The tool returns the file content in an appropriate format:
 - Images: Base64-encoded image data
 - PDFs: Extracted text content
 `,
-  schema: z4.z.object({
-    path: z4.z.string().describe("Virtual path to the file to read"),
-    offset: z4.z.number().optional().describe(
+  schema: z5.z.object({
+    path: z5.z.string().describe("Virtual path to the file to read"),
+    offset: z5.z.number().optional().describe(
       "Line number to start reading from (1-indexed, for text files)"
     ),
-    limit: z4.z.number().optional().describe("Maximum number of lines to read (for text files)")
+    limit: z5.z.number().optional().describe("Maximum number of lines to read (for text files)")
   }),
   strict: true
 };
@@ -750,9 +867,9 @@ IMPORTANT:
 - This is an atomic write operation - the entire file is replaced
 - Path must be absolute (e.g., "/docs/readme.md", not "docs/readme.md")
 `,
-  schema: z4.z.object({
-    file_path: z4.z.string().describe("The absolute virtual path to the file to write"),
-    content: z4.z.string().describe("The content to write to the file")
+  schema: z5.z.object({
+    file_path: z5.z.string().describe("The absolute virtual path to the file to write"),
+    content: z5.z.string().describe("The content to write to the file")
   }),
   strict: true
 };
@@ -772,27 +889,22 @@ IMPORTANT:
 - The operation fails if old_string is not found
 - old_string and new_string must be different
 `,
-  schema: z4.z.object({
-    file_path: z4.z.string().describe("The absolute virtual path to the file to modify"),
-    old_string: z4.z.string().describe("The exact text to replace"),
-    new_string: z4.z.string().describe(
+  schema: z5.z.object({
+    file_path: z5.z.string().describe("The absolute virtual path to the file to modify"),
+    old_string: z5.z.string().describe("The exact text to replace"),
+    new_string: z5.z.string().describe(
       "The text to replace it with (must be different from old_string)"
     ),
-    replace_all: z4.z.boolean().optional().describe(
+    replace_all: z5.z.boolean().optional().describe(
       "If true, replace all occurrences of old_string (default: false)"
     )
   }),
   strict: true
 };
-
-// src/tools/task-create/handler.ts
-function createTaskCreateHandler({
-  stateManager,
-  idGenerator
-}) {
+function createTaskCreateHandler(stateManager) {
   return (args) => {
     const task = {
-      id: idGenerator(),
+      id: workflow.uuid4(),
       subject: args.subject,
       description: args.description,
       activeForm: args.activeForm,
@@ -803,16 +915,16 @@ function createTaskCreateHandler({
     };
     stateManager.setTask(task);
     return {
-      content: JSON.stringify(task, null, 2),
-      result: task
+      toolResponse: JSON.stringify(task, null, 2),
+      data: task
     };
   };
 }
 var taskGetTool = {
   name: "TaskGet",
   description: `Retrieve full task details including dependencies.`,
-  schema: z4__default.default.object({
-    taskId: z4__default.default.string().describe("The ID of the task to get")
+  schema: z5__default.default.object({
+    taskId: z5__default.default.string().describe("The ID of the task to get")
   })
 };
 
@@ -822,20 +934,20 @@ function createTaskGetHandler(stateManager) {
     const task = stateManager.getTask(args.taskId) ?? null;
     if (!task) {
       return {
-        content: JSON.stringify({ error: `Task not found: ${args.taskId}` }),
-        result: null
+        toolResponse: JSON.stringify({ error: `Task not found: ${args.taskId}` }),
+        data: null
       };
     }
     return {
-      content: JSON.stringify(task, null, 2),
-      result: task
+      toolResponse: JSON.stringify(task, null, 2),
+      data: task
     };
   };
 }
 var taskListTool = {
   name: "TaskList",
   description: `List all tasks with current state.`,
-  schema: z4__default.default.object({})
+  schema: z5__default.default.object({})
 };
 
 // src/tools/task-list/handler.ts
@@ -843,19 +955,19 @@ function createTaskListHandler(stateManager) {
   return (_args) => {
     const taskList = stateManager.getTasks();
     return {
-      content: JSON.stringify(taskList, null, 2),
-      result: taskList
+      toolResponse: JSON.stringify(taskList, null, 2),
+      data: taskList
     };
   };
 }
 var taskUpdateTool = {
   name: "TaskUpdate",
   description: `Update status, add blockers, modify details.`,
-  schema: z4__default.default.object({
-    taskId: z4__default.default.string().describe("The ID of the task to get"),
-    status: z4__default.default.enum(["pending", "in_progress", "completed"]).describe("The status of the task"),
-    addBlockedBy: z4__default.default.array(z4__default.default.string()).describe("The IDs of the tasks that are blocking this task"),
-    addBlocks: z4__default.default.array(z4__default.default.string()).describe("The IDs of the tasks that this task is blocking")
+  schema: z5__default.default.object({
+    taskId: z5__default.default.string().describe("The ID of the task to get"),
+    status: z5__default.default.enum(["pending", "in_progress", "completed"]).describe("The status of the task"),
+    addBlockedBy: z5__default.default.array(z5__default.default.string()).describe("The IDs of the tasks that are blocking this task"),
+    addBlocks: z5__default.default.array(z5__default.default.string()).describe("The IDs of the tasks that this task is blocking")
   })
 };
 
@@ -865,8 +977,8 @@ function createTaskUpdateHandler(stateManager) {
     const task = stateManager.getTask(args.taskId);
     if (!task) {
       return {
-        content: JSON.stringify({ error: `Task not found: ${args.taskId}` }),
-        result: null
+        toolResponse: JSON.stringify({ error: `Task not found: ${args.taskId}` }),
+        data: null
       };
     }
     if (args.status) {
@@ -898,8 +1010,8 @@ function createTaskUpdateHandler(stateManager) {
     }
     stateManager.setTask(task);
     return {
-      content: JSON.stringify(task, null, 2),
-      result: task
+      toolResponse: JSON.stringify(task, null, 2),
+      data: task
     };
   };
 }
@@ -915,6 +1027,8 @@ exports.createTaskListHandler = createTaskListHandler;
 exports.createTaskTool = createTaskTool;
 exports.createTaskUpdateHandler = createTaskUpdateHandler;
 exports.createToolRouter = createToolRouter;
+exports.defineSubagent = defineSubagent;
+exports.defineTool = defineTool;
 exports.editTool = editTool;
 exports.globTool = globTool;
 exports.grepTool = grepTool;