zeitlich 0.2.5 → 0.2.7

package/dist/index.d.cts

@@ -1,12 +1,12 @@
-import { A as AgentResponse, a as ActivityToolHandler, b as AskUserQuestionArgs, G as GlobArgs, F as FileEditArgs, B as BashArgs } from './workflow-Dg5JMeOC.cjs';
-export { c as AGENT_HANDLER_NAMES, d as AgentFile, e as AgentState, f as AgentStateManager, g as AgentStatus, h as AppendToolResultFn, i as BaseAgentState, j as BaseThreadManager, k as FileReadArgs, l as FileWriteArgs, m as GrepArgs, I as InferToolResults, J as JsonPrimitive, n as JsonSerializable, o as JsonValue, P as ParsedToolCall, p as ParsedToolCallUnion, q as PostToolUseFailureHook, r as PostToolUseFailureHookContext, s as PostToolUseFailureHookResult, t as PostToolUseHook, u as PostToolUseHookContext, v as PreToolUseHook, w as PreToolUseHookContext, x as PreToolUseHookResult, y as ProcessToolCallsContext, R as RawToolCall, z as RunAgentActivity, C as RunAgentConfig, S as SessionEndHook, D as SessionEndHookContext, E as SessionExitReason, H as SessionLifecycleHooks, K as SessionStartHook, L as SessionStartHookContext, M as SubagentArgs, N as SubagentConfig, O as SubagentHooks, Q as SubagentInput, T as TaskCreateArgs, U as TaskGetArgs, V as TaskListArgs, W as TaskStatus, X as TaskUpdateArgs, Y as ThreadManager, Z as ThreadManagerConfig, _ as ThreadOps, $ as ToolArgs, a0 as ToolCallResult, a1 as ToolCallResultUnion, a2 as ToolDefinition, a3 as ToolHandler, a4 as ToolHandlerContext, a5 as ToolHandlerResponse, a6 as ToolHooks, a7 as ToolMap, a8 as ToolMessageContent, a9 as ToolNames, aa as ToolResult, ab as ToolResultConfig, ac as ToolRouter, ad as ToolWithHandler, ae as WorkflowTask, af as ZeitlichAgentConfig, ag as ZeitlichSession, ah as ZeitlichSharedActivities, ai as askUserQuestionTool, aj as bashTool, ak as createAgentStateManager, al as createBashToolDescription, am as createSession, an as createSharedActivities, ao as createSubagentTool, ap as createTaskCreateHandler, aq as createTaskGetHandler, ar as createTaskListHandler, as as createTaskUpdateHandler, at as createThreadManager, au as createToolRouter, av as defineSubagent, aw as defineTool, ax as editTool, ay as getStateQuery, az as globTool, aA as grepTool, aB as hasNoOtherToolCalls, aC as isTerminalStatus, aD as proxyDefaultThreadOps, aE as readTool, aF as taskCreateTool, aG as taskGetTool, aH as taskListTool, aI as taskUpdateTool, aJ as withAutoAppend, aK as writeTool } from './workflow-Dg5JMeOC.cjs';
+import { A as AgentResponse, a as ActivityToolHandler, G as GlobArgs, F as FileEditArgs, B as BashArgs } from './workflow-CyYHDbrr.cjs';
+export { b as AGENT_HANDLER_NAMES, c as AgentConfig, d as AgentFile, e as AgentState, f as AgentStateManager, g as AgentStatus, h as AppendToolResultFn, i as AskUserQuestionArgs, j as BaseAgentState, k as BaseThreadManager, l as FileReadArgs, m as FileWriteArgs, n as GrepArgs, I as InferToolResults, J as JsonPrimitive, o as JsonSerializable, p as JsonValue, P as ParsedToolCall, q as ParsedToolCallUnion, r as PostToolUseFailureHook, s as PostToolUseFailureHookContext, t as PostToolUseFailureHookResult, u as PostToolUseHook, v as PostToolUseHookContext, w as PreToolUseHook, x as PreToolUseHookContext, y as PreToolUseHookResult, z as ProcessToolCallsContext, R as RawToolCall, C as RunAgentActivity, D as RunAgentConfig, S as SessionEndHook, E as SessionEndHookContext, H as SessionExitReason, K as SessionLifecycleHooks, L as SessionStartHook, M as SessionStartHookContext, N as SubagentArgs, O as SubagentConfig, Q as SubagentHooks, T as SubagentInput, U as TaskCreateArgs, V as TaskGetArgs, W as TaskListArgs, X as TaskStatus, Y as TaskUpdateArgs, Z as ThreadManager, _ as ThreadManagerConfig, $ as ThreadOps, a0 as ToolArgs, a1 as ToolCallResult, a2 as ToolCallResultUnion, a3 as ToolDefinition, a4 as ToolHandler, a5 as ToolHandlerContext, a6 as ToolHandlerResponse, a7 as ToolHooks, a8 as ToolMap, a9 as ToolMessageContent, aa as ToolNames, ab as ToolResult, ac as ToolResultConfig, ad as ToolRouter, ae as ToolWithHandler, af as WorkflowTask, ag as ZeitlichSession, ah as ZeitlichSharedActivities, ai as askUserQuestionTool, aj as bashTool, ak as createAgentStateManager, al as createAskUserQuestionHandler, am as createBashToolDescription, an as createSession, ao as createSharedActivities, ap as createSubagentTool, aq as createTaskCreateHandler, ar as createTaskGetHandler, as as createTaskListHandler, at as createTaskUpdateHandler, au as createThreadManager, av as createToolRouter, aw as defineSubagent, ax as defineTool, ay as editTool, az as globTool, aA as grepTool, aB as hasNoOtherToolCalls, aC as isTerminalStatus, aD as proxyDefaultThreadOps, aE as readTool, aF as taskCreateTool, aG as taskGetTool, aH as taskListTool, aI as taskUpdateTool, aJ as withAutoAppend, aK as writeTool } from './workflow-CyYHDbrr.cjs';
 import { SimplePlugin } from '@temporalio/plugin';
 import Redis from 'ioredis';
 import { WorkflowClient } from '@temporalio/client';
 import { BaseChatModel, BaseChatModelCallOptions, BindToolsInput } from '@langchain/core/language_models/chat_models';
-import { StoredMessage } from '@langchain/core/messages';
 import { IFileSystem, BashOptions } from 'just-bash';
 import '@temporalio/workflow';
+import '@langchain/core/messages';
 import 'zod';
 import '@temporalio/common';
 
@@ -55,13 +55,6 @@ declare function invokeModel({ redis, model, client, config: { threadId, agentNa
     }>;
 }): Promise<AgentResponse>;
 
-/**
- * Creates handler for user interaction tool - creates AI messages for display.
- */
-declare const createAskUserQuestionHandler: () => ActivityToolHandler<AskUserQuestionArgs, {
-    chatMessages: StoredMessage[];
-}>;
-
 /**
  * Result of a glob operation
  */
@@ -109,4 +102,4 @@ declare const toTree: (fs: IFileSystem, opts?: {
     sort?: boolean;
 }) => Promise<string>;
 
-export { ActivityToolHandler, AgentResponse, AskUserQuestionArgs, BashArgs, FileEditArgs, GlobArgs, type InvokeModelConfig, ZeitlichPlugin, type ZeitlichPluginOptions, createAskUserQuestionHandler, createBashHandler, createEditHandler, createGlobHandler, invokeModel, toTree };
+export { ActivityToolHandler, AgentResponse, BashArgs, FileEditArgs, GlobArgs, type InvokeModelConfig, ZeitlichPlugin, type ZeitlichPluginOptions, createBashHandler, createEditHandler, createGlobHandler, invokeModel, toTree };

package/dist/index.d.ts

@@ -1,12 +1,12 @@
-import { A as AgentResponse, a as ActivityToolHandler, b as AskUserQuestionArgs, G as GlobArgs, F as FileEditArgs, B as BashArgs } from './workflow-Dg5JMeOC.js';
-export { c as AGENT_HANDLER_NAMES, d as AgentFile, e as AgentState, f as AgentStateManager, g as AgentStatus, h as AppendToolResultFn, i as BaseAgentState, j as BaseThreadManager, k as FileReadArgs, l as FileWriteArgs, m as GrepArgs, I as InferToolResults, J as JsonPrimitive, n as JsonSerializable, o as JsonValue, P as ParsedToolCall, p as ParsedToolCallUnion, q as PostToolUseFailureHook, r as PostToolUseFailureHookContext, s as PostToolUseFailureHookResult, t as PostToolUseHook, u as PostToolUseHookContext, v as PreToolUseHook, w as PreToolUseHookContext, x as PreToolUseHookResult, y as ProcessToolCallsContext, R as RawToolCall, z as RunAgentActivity, C as RunAgentConfig, S as SessionEndHook, D as SessionEndHookContext, E as SessionExitReason, H as SessionLifecycleHooks, K as SessionStartHook, L as SessionStartHookContext, M as SubagentArgs, N as SubagentConfig, O as SubagentHooks, Q as SubagentInput, T as TaskCreateArgs, U as TaskGetArgs, V as TaskListArgs, W as TaskStatus, X as TaskUpdateArgs, Y as ThreadManager, Z as ThreadManagerConfig, _ as ThreadOps, $ as ToolArgs, a0 as ToolCallResult, a1 as ToolCallResultUnion, a2 as ToolDefinition, a3 as ToolHandler, a4 as ToolHandlerContext, a5 as ToolHandlerResponse, a6 as ToolHooks, a7 as ToolMap, a8 as ToolMessageContent, a9 as ToolNames, aa as ToolResult, ab as ToolResultConfig, ac as ToolRouter, ad as ToolWithHandler, ae as WorkflowTask, af as ZeitlichAgentConfig, ag as ZeitlichSession, ah as ZeitlichSharedActivities, ai as askUserQuestionTool, aj as bashTool, ak as createAgentStateManager, al as createBashToolDescription, am as createSession, an as createSharedActivities, ao as createSubagentTool, ap as createTaskCreateHandler, aq as createTaskGetHandler, ar as createTaskListHandler, as as createTaskUpdateHandler, at as createThreadManager, au as createToolRouter, av as defineSubagent, aw as defineTool, ax as editTool, ay as getStateQuery, az as globTool, aA as grepTool, aB as hasNoOtherToolCalls, aC as isTerminalStatus, aD as proxyDefaultThreadOps, aE as readTool, aF as taskCreateTool, aG as taskGetTool, aH as taskListTool, aI as taskUpdateTool, aJ as withAutoAppend, aK as writeTool } from './workflow-Dg5JMeOC.js';
+import { A as AgentResponse, a as ActivityToolHandler, G as GlobArgs, F as FileEditArgs, B as BashArgs } from './workflow-CyYHDbrr.js';
+export { b as AGENT_HANDLER_NAMES, c as AgentConfig, d as AgentFile, e as AgentState, f as AgentStateManager, g as AgentStatus, h as AppendToolResultFn, i as AskUserQuestionArgs, j as BaseAgentState, k as BaseThreadManager, l as FileReadArgs, m as FileWriteArgs, n as GrepArgs, I as InferToolResults, J as JsonPrimitive, o as JsonSerializable, p as JsonValue, P as ParsedToolCall, q as ParsedToolCallUnion, r as PostToolUseFailureHook, s as PostToolUseFailureHookContext, t as PostToolUseFailureHookResult, u as PostToolUseHook, v as PostToolUseHookContext, w as PreToolUseHook, x as PreToolUseHookContext, y as PreToolUseHookResult, z as ProcessToolCallsContext, R as RawToolCall, C as RunAgentActivity, D as RunAgentConfig, S as SessionEndHook, E as SessionEndHookContext, H as SessionExitReason, K as SessionLifecycleHooks, L as SessionStartHook, M as SessionStartHookContext, N as SubagentArgs, O as SubagentConfig, Q as SubagentHooks, T as SubagentInput, U as TaskCreateArgs, V as TaskGetArgs, W as TaskListArgs, X as TaskStatus, Y as TaskUpdateArgs, Z as ThreadManager, _ as ThreadManagerConfig, $ as ThreadOps, a0 as ToolArgs, a1 as ToolCallResult, a2 as ToolCallResultUnion, a3 as ToolDefinition, a4 as ToolHandler, a5 as ToolHandlerContext, a6 as ToolHandlerResponse, a7 as ToolHooks, a8 as ToolMap, a9 as ToolMessageContent, aa as ToolNames, ab as ToolResult, ac as ToolResultConfig, ad as ToolRouter, ae as ToolWithHandler, af as WorkflowTask, ag as ZeitlichSession, ah as ZeitlichSharedActivities, ai as askUserQuestionTool, aj as bashTool, ak as createAgentStateManager, al as createAskUserQuestionHandler, am as createBashToolDescription, an as createSession, ao as createSharedActivities, ap as createSubagentTool, aq as createTaskCreateHandler, ar as createTaskGetHandler, as as createTaskListHandler, at as createTaskUpdateHandler, au as createThreadManager, av as createToolRouter, aw as defineSubagent, ax as defineTool, ay as editTool, az as globTool, aA as grepTool, aB as hasNoOtherToolCalls, aC as isTerminalStatus, aD as proxyDefaultThreadOps, aE as readTool, aF as taskCreateTool, aG as taskGetTool, aH as taskListTool, aI as taskUpdateTool, aJ as withAutoAppend, aK as writeTool } from './workflow-CyYHDbrr.js';
 import { SimplePlugin } from '@temporalio/plugin';
 import Redis from 'ioredis';
 import { WorkflowClient } from '@temporalio/client';
 import { BaseChatModel, BaseChatModelCallOptions, BindToolsInput } from '@langchain/core/language_models/chat_models';
-import { StoredMessage } from '@langchain/core/messages';
 import { IFileSystem, BashOptions } from 'just-bash';
 import '@temporalio/workflow';
+import '@langchain/core/messages';
 import 'zod';
 import '@temporalio/common';
 
@@ -55,13 +55,6 @@ declare function invokeModel({ redis, model, client, config: { threadId, agentNa
     }>;
 }): Promise<AgentResponse>;
 
-/**
- * Creates handler for user interaction tool - creates AI messages for display.
- */
-declare const createAskUserQuestionHandler: () => ActivityToolHandler<AskUserQuestionArgs, {
-    chatMessages: StoredMessage[];
-}>;
-
 /**
  * Result of a glob operation
  */
@@ -109,4 +102,4 @@ declare const toTree: (fs: IFileSystem, opts?: {
     sort?: boolean;
 }) => Promise<string>;
 
-export { ActivityToolHandler, AgentResponse, AskUserQuestionArgs, BashArgs, FileEditArgs, GlobArgs, type InvokeModelConfig, ZeitlichPlugin, type ZeitlichPluginOptions, createAskUserQuestionHandler, createBashHandler, createEditHandler, createGlobHandler, invokeModel, toTree };
+export { ActivityToolHandler, AgentResponse, BashArgs, FileEditArgs, GlobArgs, type InvokeModelConfig, ZeitlichPlugin, type ZeitlichPluginOptions, createBashHandler, createEditHandler, createGlobHandler, invokeModel, toTree };

package/dist/index.js

@@ -1,7 +1,7 @@
-import { defineQuery, proxyActivities, setHandler, uuid4, workflowInfo, executeChild } from '@temporalio/workflow';
-import z3, { z } from 'zod';
+import { setHandler, defineUpdate, condition, proxyActivities, defineQuery, uuid4, workflowInfo, executeChild } from '@temporalio/workflow';
+import z13, { z } from 'zod';
 import { SimplePlugin } from '@temporalio/plugin';
-import { mapStoredMessagesToChatMessages, AIMessage, ToolMessage, HumanMessage } from '@langchain/core/messages';
+import { mapStoredMessagesToChatMessages, ToolMessage, AIMessage, SystemMessage, HumanMessage } from '@langchain/core/messages';
 import crypto from 'crypto';
 import { Context } from '@temporalio/activity';
 import { Bash } from 'just-bash';
@@ -39,10 +39,10 @@ function createSubagentTool(subagents) {
   return {
     name: SUBAGENT_TOOL,
     description: buildSubagentDescription(subagents),
-    schema: z3.object({
-      subagent: z3.enum(names).describe("The type of subagent to launch"),
-      description: z3.string().describe("A short (3-5 word) description of the task"),
-      prompt: z3.string().describe("The task for the agent to perform")
+    schema: z13.object({
+      subagent: z13.enum(names).describe("The type of subagent to launch"),
+      description: z13.string().describe("A short (3-5 word) description of the task"),
+      prompt: z13.string().describe("The task for the agent to perform")
     })
   };
 }
@@ -414,7 +414,8 @@ var createSession = async ({
   processToolsInParallel = true,
   hooks = {},
   appendSystemPrompt = true,
-  systemPrompt
+  systemPrompt,
+  waitForInputTimeout = "48h"
 }) => {
   const {
     appendToolResult,
@@ -443,6 +444,25 @@ var createSession = async ({
   };
   return {
     runSession: async ({ stateManager }) => {
+      setHandler(
+        defineUpdate(`add${agentName}Message`),
+        async (message) => {
+          if (hooks.onPreHumanMessageAppend) {
+            await hooks.onPreHumanMessageAppend({
+              message,
+              threadId
+            });
+          }
+          await appendHumanMessage(threadId, message);
+          if (hooks.onPostHumanMessageAppend) {
+            await hooks.onPostHumanMessageAppend({
+              message,
+              threadId
+            });
+          }
+          stateManager.run();
+        }
+      );
       if (hooks.onSessionStart) {
         await hooks.onSessionStart({
           threadId,
@@ -490,8 +510,15 @@ var createSession = async ({
             turn: currentTurn
           });
           if (stateManager.getStatus() === "WAITING_FOR_INPUT") {
-            exitReason = "waiting_for_input";
-            break;
+            const conditionMet = await condition(
+              () => stateManager.getStatus() === "RUNNING",
+              waitForInputTimeout
+            );
+            if (!conditionMet) {
+              stateManager.cancel();
+              await condition(() => false, "2s");
+              break;
+            }
           }
         }
         if (stateManager.getTurns() >= maxTurns && stateManager.isRunning()) {
@@ -532,8 +559,10 @@ function proxyDefaultThreadOps(options) {
 function isTerminalStatus(status) {
   return status === "COMPLETED" || status === "FAILED" || status === "CANCELLED";
 }
-var getStateQuery = defineQuery("getState");
-function createAgentStateManager(initialState) {
+function createAgentStateManager({
+  initialState,
+  agentConfig
+}) {
   let status = initialState?.status ?? "RUNNING";
   let version = initialState?.version ?? 0;
   let turns = initialState?.turns ?? 0;
@@ -557,9 +586,21 @@ function createAgentStateManager(initialState) {
       ...customState
     };
   }
-  setHandler(getStateQuery, () => {
+  setHandler(defineQuery(`get${agentConfig.agentName}State`), () => {
     return buildState();
   });
+  setHandler(
+    defineUpdate(
+      `waitFor${agentConfig.agentName}StateChange`
+    ),
+    async (lastKnownVersion) => {
+      await condition(
+        () => version > lastKnownVersion || isTerminalStatus(status),
+        "55s"
+      );
+      return buildState();
+    }
+  );
   return {
     getStatus() {
       return status;
@@ -648,38 +689,6 @@ var AGENT_HANDLER_NAMES = {
   waitForStateChange: "waitForStateChange",
   addMessage: "addMessage"
 };
-var askUserQuestionTool = {
-  name: "AskUserQuestion",
-  description: `Use this tool when you need to ask the user questions during execution. This allows you to:
-
-1. Gather user preferences or requirements
-2. Clarify ambiguous instructions
-3. Get decisions on implementation choices as you work
-4. Offer choices to the user about what direction to take.
-
-Usage notes:
-
-* Users will always be able to select "Other" to provide custom text input
-* 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: z3.object({
-    questions: z3.array(
-      z3.object({
-        question: z3.string().describe("The full question text to display"),
-        header: z3.string().describe("Short label for the question (max 12 characters)"),
-        options: z3.array(
-          z3.object({
-            label: z3.string(),
-            description: z3.string()
-          })
-        ).min(0).max(4).describe("Array of 0-4 choices, each with label and description"),
-        multiSelect: z3.boolean().describe("If true, users can select multiple options")
-      })
-    )
-  }),
-  strict: true
-};
 var globTool = {
   name: "Glob",
   description: `Search for files matching a glob pattern within the available file system.
@@ -836,17 +845,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: z3.object({
-    subject: z3.string().describe(
+  schema: z13.object({
+    subject: z13.string().describe(
       'A brief, actionable title in imperative form (e.g., "Fix authentication bug in login flow")'
     ),
-    description: z3.string().describe(
+    description: z13.string().describe(
       "Detailed description of what needs to be done, including context and acceptance criteria"
     ),
-    activeForm: z3.string().describe(
+    activeForm: z13.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: z3.record(z3.string(), z3.string()).describe("Arbitrary key-value pairs for tracking")
+    metadata: z13.record(z13.string(), z13.string()).describe("Arbitrary key-value pairs for tracking")
   })
 };
 function createTaskCreateHandler(stateManager) {
@@ -871,8 +880,8 @@ function createTaskCreateHandler(stateManager) {
 var taskGetTool = {
   name: "TaskGet",
   description: `Retrieve full task details including dependencies.`,
-  schema: z3.object({
-    taskId: z3.string().describe("The ID of the task to get")
+  schema: z13.object({
+    taskId: z13.string().describe("The ID of the task to get")
   })
 };
 
@@ -895,7 +904,7 @@ function createTaskGetHandler(stateManager) {
 var taskListTool = {
   name: "TaskList",
   description: `List all tasks with current state.`,
-  schema: z3.object({})
+  schema: z13.object({})
 };
 
 // src/tools/task-list/handler.ts
@@ -911,11 +920,11 @@ function createTaskListHandler(stateManager) {
 var taskUpdateTool = {
   name: "TaskUpdate",
   description: `Update status, add blockers, modify details.`,
-  schema: z3.object({
-    taskId: z3.string().describe("The ID of the task to get"),
-    status: z3.enum(["pending", "in_progress", "completed"]).describe("The status of the task"),
-    addBlockedBy: z3.array(z3.string()).describe("The IDs of the tasks that are blocking this task"),
-    addBlocks: z3.array(z3.string()).describe("The IDs of the tasks that this task is blocking")
+  schema: z13.object({
+    taskId: z13.string().describe("The ID of the task to get"),
+    status: z13.enum(["pending", "in_progress", "completed"]).describe("The status of the task"),
+    addBlockedBy: z13.array(z13.string()).describe("The IDs of the tasks that are blocking this task"),
+    addBlocks: z13.array(z13.string()).describe("The IDs of the tasks that this task is blocking")
   })
 };
 
@@ -983,13 +992,53 @@ Use this tool to:
 - Execute scripts and chain commands with pipes (|) or logical operators (&&, ||)
 - Inspect files and directories
 `,
-  schema: z3.object({
-    command: z3.string().describe(
+  schema: z13.object({
+    command: z13.string().describe(
       "The bash command to execute. Can include pipes (|), redirects (>, >>), logical operators (&&, ||), and shell features like command substitution $(...)."
     )
   }),
   strict: true
 };
+var askUserQuestionTool = {
+  name: "AskUserQuestion",
+  description: `Use this tool when you need to ask the user questions during execution. This allows you to:
+
+1. Gather user preferences or requirements
+2. Clarify ambiguous instructions
+3. Get decisions on implementation choices as you work
+4. Offer choices to the user about what direction to take.
+
+Usage notes:
+
+* Users will always be able to select "Other" to provide custom text input
+* 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: z13.object({
+    questions: z13.array(
+      z13.object({
+        question: z13.string().describe("The full question text to display"),
+        header: z13.string().describe("Short label for the question (max 12 characters)"),
+        options: z13.array(
+          z13.object({
+            label: z13.string(),
+            description: z13.string()
+          })
+        ).min(0).max(4).describe("Array of 0-4 choices, each with label and description"),
+        multiSelect: z13.boolean().describe("If true, users can select multiple options")
+      })
+    )
+  }),
+  strict: true
+};
+
+// src/tools/ask-user-question/handler.ts
+var createAskUserQuestionHandler = () => (args) => {
+  return {
+    toolResponse: "Question submitted",
+    data: { questions: args.questions }
+  };
+};
 
 // node_modules/uuid/dist/esm-node/stringify.js
 var byteToHex = [];
@@ -1071,6 +1120,12 @@ function createThreadManager(config) {
         content
       }).toDict();
     },
+    createSystemMessage(content) {
+      return new SystemMessage({
+        id: v4_default(),
+        content
+      }).toDict();
+    },
     createAIMessage(content, kwargs) {
       return new AIMessage({
         id: v4_default(),
@@ -1099,6 +1154,10 @@ function createThreadManager(config) {
     async appendAIMessage(content) {
       const message = helpers.createAIMessage(content);
       await base.append([message]);
+    },
+    async appendSystemMessage(content) {
+      const message = helpers.createSystemMessage(content);
+      await base.append([message]);
     }
   };
   return Object.assign(base, helpers);
@@ -1152,7 +1211,7 @@ async function invokeModel({
   const parentWorkflowId = info.workflowExecution.workflowId;
   const parentRunId = info.workflowExecution.runId;
   const handle = client.getHandle(parentWorkflowId, parentRunId);
-  const { tools } = await handle.query(getStateQuery);
+  const { tools } = await handle.query(`get${agentName}State`);
   const messages = await thread.load();
   const response = await model.invoke(
     [...mapStoredMessagesToChatMessages(messages)],
@@ -1179,19 +1238,6 @@ async function invokeModel({
     }
   };
 }
-var createAskUserQuestionHandler = () => async (args) => {
-  const messages = args.questions.map(
-    ({ question, header, options, multiSelect }) => new AIMessage({
-      content: question,
-      additional_kwargs: {
-        header,
-        options,
-        multiSelect
-      }
-    }).toDict()
-  );
-  return { toolResponse: "Question submitted", data: { chatMessages: messages } };
-};
 function createGlobHandler(fs) {
   return async (_args) => {
     new Bash({ fs });
@@ -1361,6 +1407,6 @@ var toTree = async (fs, opts = {}) => {
   return base + subtree;
 };
 
-export { AGENT_HANDLER_NAMES, ZeitlichPlugin, askUserQuestionTool, bashTool, createAgentStateManager, createAskUserQuestionHandler, createBashHandler, createBashToolDescription, createEditHandler, createGlobHandler, createSession, createSharedActivities, createSubagentTool, createTaskCreateHandler, createTaskGetHandler, createTaskListHandler, createTaskUpdateHandler, createThreadManager, createToolRouter, defineSubagent, defineTool, editTool, getStateQuery, globTool, grepTool, hasNoOtherToolCalls, invokeModel, isTerminalStatus, proxyDefaultThreadOps, readTool, taskCreateTool, taskGetTool, taskListTool, taskUpdateTool, toTree, withAutoAppend, writeTool };
+export { AGENT_HANDLER_NAMES, ZeitlichPlugin, askUserQuestionTool, bashTool, createAgentStateManager, createAskUserQuestionHandler, createBashHandler, createBashToolDescription, createEditHandler, createGlobHandler, createSession, createSharedActivities, createSubagentTool, createTaskCreateHandler, createTaskGetHandler, createTaskListHandler, createTaskUpdateHandler, createThreadManager, createToolRouter, defineSubagent, defineTool, editTool, globTool, grepTool, hasNoOtherToolCalls, invokeModel, isTerminalStatus, proxyDefaultThreadOps, readTool, taskCreateTool, taskGetTool, taskListTool, taskUpdateTool, toTree, withAutoAppend, writeTool };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map