@axiom-lattice/core 2.1.45 → 2.1.46

package/dist/index.mjs

@@ -648,11 +648,11 @@ var ToolLatticeManager = class _ToolLatticeManager extends BaseLatticeManager {
    * @param key Lattice键名
    * @param tool 已有的StructuredTool实例
    */
-  registerExistingTool(key, tool50) {
+  registerExistingTool(key, tool51) {
     const config = {
-      name: tool50.name,
-      description: tool50.description,
-      schema: tool50.schema,
+      name: tool51.name,
+      description: tool51.description,
+      schema: tool51.schema,
       // StructuredTool的schema已经是Zod兼容的
       needUserApprove: false
       // MCP工具默认不需要用户批准
@@ -660,7 +660,7 @@ var ToolLatticeManager = class _ToolLatticeManager extends BaseLatticeManager {
     const toolLattice = {
       key,
       config,
-      client: tool50
+      client: tool51
     };
     this.register(key, toolLattice);
   }
@@ -686,7 +686,7 @@ var ToolLatticeManager = class _ToolLatticeManager extends BaseLatticeManager {
 };
 var toolLatticeManager = ToolLatticeManager.getInstance();
 var registerToolLattice = (key, config, executor) => toolLatticeManager.registerLattice(key, config, executor);
-var registerExistingTool = (key, tool50) => toolLatticeManager.registerExistingTool(key, tool50);
+var registerExistingTool = (key, tool51) => toolLatticeManager.registerExistingTool(key, tool51);
 var getToolLattice = (key) => toolLatticeManager.getToolLattice(key);
 var getToolDefinition = (key) => toolLatticeManager.getToolDefinition(key);
 var getToolClient = (key) => toolLatticeManager.getToolClient(key);
@@ -9239,7 +9239,7 @@ function createShowWidgetTool() {
     },
     {
       name: "show_widget",
-      description: "Render an interactive HTML widget or SVG diagram visible to the user. Use for: charts, dashboards, calculators, forms, diagrams, timers, games, visualizations. The widget appears in a panel next to the chat. Users can interact with it and send data back via window.sendToAgent(data). IMPORTANT: Always call load_guidelines before your first show_widget.",
+      description: "Render an interactive HTML widget or SVG diagram visible to the user. Use for: charts, dashboards, calculators, forms, diagrams, timers, games, visualizations. The widget appears in a panel next to the chat. Users can interact with it and send data back via window.sendPrompt(data). IMPORTANT: Always call load_guidelines before your first show_widget.",
       schema: ShowWidgetInputSchema
     }
   );
@@ -9770,9 +9770,198 @@ ${bootstrapSections.join("\n\n")}
   });
 }
 
+// src/middlewares/unknownToolHandlerMiddleware.ts
+import { createMiddleware as createMiddleware11 } from "langchain";
+import { AIMessage as AIMessage2, ToolMessage as ToolMessage3 } from "@langchain/core/messages";
+function createUnknownToolHandlerMiddleware(config = {}) {
+  const {
+    strategy = "error",
+    errorMessageTemplate = (name, tools) => `Error: Tool "${name}" does not exist.
+
+Available tools:
+` + tools.map((t) => `  - ${t}`).join("\n") + `
+
+Please select a valid tool from the list above.`
+  } = config;
+  return createMiddleware11({
+    name: "UnknownToolHandlerMiddleware",
+    /**
+     * wrapModelCall hook
+     *
+     * The only place to access request.tools (all available tools).
+     * Identifies unknown tools and stores error info in metadata.
+     *
+     * Key: Preserve all tool_calls in original AIMessage, including unknown ones.
+     * This allows the model to see what it selected in the next round.
+     */
+    wrapModelCall: async (request, handler) => {
+      const availableTools = request.tools || [];
+      const availableToolNames = availableTools.map((t) => t.name);
+      const availableToolSet = new Set(availableToolNames);
+      const aiResponse = await handler(request);
+      if (!aiResponse.tool_calls || aiResponse.tool_calls.length === 0) {
+        return aiResponse;
+      }
+      const unknownToolCalls = [];
+      for (const toolCall of aiResponse.tool_calls) {
+        if (!availableToolSet.has(toolCall.name)) {
+          unknownToolCalls.push(toolCall);
+        }
+      }
+      if (unknownToolCalls.length === 0) {
+        return aiResponse;
+      }
+      if (strategy === "strict") {
+        const unknownNames = unknownToolCalls.map((t) => t.name).join(", ");
+        throw new Error(
+          `Unknown tools detected: ${unknownNames}. Available tools: ${availableToolNames.join(", ")}`
+        );
+      }
+      const unknownToolErrors = unknownToolCalls.map((toolCall) => ({
+        toolName: toolCall.name,
+        toolCallId: toolCall.id,
+        errorMessage: errorMessageTemplate(toolCall.name, availableToolNames)
+      }));
+      const modifiedResponse = new AIMessage2({
+        content: aiResponse.content,
+        tool_calls: aiResponse.tool_calls,
+        // Key: preserve all tool_calls, don't delete unknown
+        response_metadata: {
+          ...aiResponse.response_metadata,
+          _unknownToolErrors: unknownToolErrors
+        }
+      });
+      if (aiResponse.id) {
+        modifiedResponse.id = aiResponse.id;
+      }
+      return modifiedResponse;
+    },
+    /**
+     * afterModel hook
+     *
+     * Processes unknown tool errors stored in metadata.
+     * Generates error ToolMessages and jumps back to model.
+     *
+     * Key: Uses jumpTo: "model" to form agent loop.
+     * Without jumpTo, the graph would end after ToolNode without returning to model.
+     */
+    afterModel: {
+      canJumpTo: ["model"],
+      hook: async (state) => {
+        const { messages } = state;
+        if (!messages || messages.length === 0) {
+          return;
+        }
+        const lastMessage = messages[messages.length - 1];
+        if (!AIMessage2.isInstance(lastMessage)) {
+          return;
+        }
+        const unknownToolErrors = lastMessage.response_metadata?._unknownToolErrors;
+        if (!unknownToolErrors || !Array.isArray(unknownToolErrors) || unknownToolErrors.length === 0) {
+          return;
+        }
+        const errorToolMessages = unknownToolErrors.map(
+          (error) => new ToolMessage3({
+            content: error.errorMessage,
+            name: error.toolName,
+            tool_call_id: error.toolCallId,
+            status: "error"
+          })
+        );
+        delete lastMessage.response_metadata._unknownToolErrors;
+        return {
+          messages: errorToolMessages,
+          jumpTo: "model"
+        };
+      }
+    }
+  });
+}
+
+// src/deep_agent_new/middleware/date.ts
+import { createMiddleware as createMiddleware12, tool as tool46 } from "langchain";
+import { z as z47 } from "zod";
+function formatCurrentDate(timezone = "UTC") {
+  const now = /* @__PURE__ */ new Date();
+  let validTimezone = timezone;
+  try {
+    Intl.DateTimeFormat(void 0, { timeZone: timezone });
+  } catch {
+    validTimezone = "UTC";
+  }
+  const dateFormatOptions = {
+    timeZone: validTimezone,
+    year: "numeric",
+    month: "2-digit",
+    day: "2-digit"
+  };
+  return new Intl.DateTimeFormat("en-US", dateFormatOptions).format(now);
+}
+function generateDateContext(timezone = "UTC") {
+  const formattedDate = formatCurrentDate(timezone);
+  if (timezone === "UTC") {
+    return `Current Date: ${formattedDate}`;
+  }
+  const utcDate = formatCurrentDate("UTC");
+  return `Current Date: ${formattedDate} (UTC: ${utcDate})`;
+}
+function createDateMiddleware(options = {}) {
+  const timezone = options.timezone || "UTC";
+  const dateContext = generateDateContext(timezone);
+  return createMiddleware12({
+    name: "DateMiddleware",
+    tools: [
+      tool46(
+        async () => {
+          const now = /* @__PURE__ */ new Date();
+          let validTimezone = timezone;
+          try {
+            Intl.DateTimeFormat(void 0, { timeZone: timezone });
+          } catch {
+            validTimezone = "UTC";
+          }
+          const localTimeOptions = {
+            timeZone: validTimezone,
+            year: "numeric",
+            month: "2-digit",
+            day: "2-digit",
+            hour: "2-digit",
+            minute: "2-digit",
+            second: "2-digit",
+            hour12: false
+          };
+          const localTime = new Intl.DateTimeFormat("en-US", localTimeOptions).format(now);
+          return JSON.stringify({
+            local_time: `${localTime} (${validTimezone})`,
+            iso_date: now.toISOString().split("T")[0],
+            timestamp: now.toISOString(),
+            timezone: validTimezone
+          });
+        },
+        {
+          name: "get_current_date_time",
+          description: "Get the exact current date and time at the moment of invocation. Use this when the user asks about the current time (e.g., 'what time is it', '\u51E0\u70B9\u4E86', '\u73B0\u5728\u51E0\u70B9'), or when you need to know the precise time for scheduling, deadlines, or time-sensitive operations.",
+          schema: z47.object({})
+        }
+      )
+    ],
+    wrapModelCall: async (request, handler) => {
+      const currentSystemPrompt = request.systemPrompt || "";
+      const newSystemPrompt = currentSystemPrompt ? `${dateContext}
+
+${currentSystemPrompt}` : dateContext;
+      return handler({
+        ...request,
+        systemPrompt: newSystemPrompt
+      });
+    }
+  });
+}
+
 // src/agent_lattice/builders/commonMiddleware.ts
 async function createCommonMiddlewares(middlewareConfigs, filesystemBackend) {
   const middlewares = [];
+  middlewares.push(createUnknownToolHandlerMiddleware());
   middlewares.push(createModelSelectorMiddleware());
   const filesystemConfig = middlewareConfigs.find((m) => m.type === "filesystem");
   if (filesystemConfig?.enabled && filesystemBackend) {
@@ -9833,6 +10022,9 @@ async function createCommonMiddlewares(middlewareConfigs, filesystemBackend) {
           }));
         }
         break;
+      case "date":
+        middlewares.push(createDateMiddleware(config.config));
+        break;
     }
   }
   return middlewares;
@@ -9886,9 +10078,9 @@ var ReActAgentGraphBuilder = class {
    */
   async build(agentLattice, params) {
     const tools = params.tools.map((t) => {
-      const tool50 = getToolClient(t.key);
-      return tool50;
-    }).filter((tool50) => tool50 !== void 0);
+      const tool51 = getToolClient(t.key);
+      return tool51;
+    }).filter((tool51) => tool51 !== void 0);
     const stateSchema2 = createReactAgentSchema(params.stateSchema);
     const middlewareConfigs = params.middleware || [];
     const filesystemBackend = this.createFilesystemBackendFactory(middlewareConfigs, agentLattice);
@@ -9914,12 +10106,12 @@ import {
 } from "langchain";
 
 // src/deep_agent_new/middleware/subagents.ts
-import { z as z47 } from "zod/v3";
+import { z as z48 } from "zod/v3";
 import {
-  createMiddleware as createMiddleware11,
+  createMiddleware as createMiddleware13,
   createAgent as createAgent2,
-  tool as tool46,
-  ToolMessage as ToolMessage3,
+  tool as tool47,
+  ToolMessage as ToolMessage4,
   humanInTheLoopMiddleware
 } from "langchain";
 import { Command as Command2, getCurrentTaskInput as getCurrentTaskInput2, GraphInterrupt as GraphInterrupt3 } from "@langchain/langgraph";
@@ -10490,7 +10682,7 @@ function returnCommandWithStateUpdate(result, toolCallId) {
     update: {
       ...stateUpdate,
       messages: [
-        new ToolMessage3({
+        new ToolMessage4({
           content: lastMessage?.content || "Task Failed to complete",
           tool_call_id: toolCallId,
           name: "task"
@@ -10572,7 +10764,7 @@ function createTaskTool(options) {
     generalPurposeAgent
   });
   const finalTaskDescription = taskDescription ? taskDescription : getTaskToolDescription(subagentDescriptions);
-  return tool46(
+  return tool47(
     async (input, config) => {
       const { description, subagent_type } = input;
       let assistant_id = subagent_type;
@@ -10626,7 +10818,7 @@ function createTaskTool(options) {
         return new Command2({
           update: {
             messages: [
-              new ToolMessage3({
+              new ToolMessage4({
                 content: error instanceof Error ? error.message : "Task Failed to complete",
                 tool_call_id: config.toolCall.id,
                 name: "task"
@@ -10639,9 +10831,9 @@ function createTaskTool(options) {
     {
       name: "task",
       description: finalTaskDescription,
-      schema: z47.object({
-        description: z47.string().describe("The task to execute with the selected agent"),
-        subagent_type: z47.string().describe(
+      schema: z48.object({
+        description: z48.string().describe("The task to execute with the selected agent"),
+        subagent_type: z48.string().describe(
           `Name of the agent to use. Available: ${Object.keys(
             subagentGraphs
           ).join(", ")}`
@@ -10670,7 +10862,7 @@ function createSubAgentMiddleware(options) {
     generalPurposeAgent,
     taskDescription
   });
-  return createMiddleware11({
+  return createMiddleware13({
     name: "subAgentMiddleware",
     tools: [taskTool],
     wrapModelCall: async (request, handler) => {
@@ -10691,14 +10883,14 @@ ${systemPrompt}` : systemPrompt;
 
 // src/deep_agent_new/middleware/patch_tool_calls.ts
 import {
-  createMiddleware as createMiddleware12,
-  ToolMessage as ToolMessage4,
-  AIMessage as AIMessage2
+  createMiddleware as createMiddleware14,
+  ToolMessage as ToolMessage5,
+  AIMessage as AIMessage3
 } from "langchain";
 import { RemoveMessage } from "@langchain/core/messages";
 import { REMOVE_ALL_MESSAGES } from "@langchain/langgraph";
 function createPatchToolCallsMiddleware() {
-  return createMiddleware12({
+  return createMiddleware14({
     name: "patchToolCallsMiddleware",
     beforeAgent: async (state) => {
       const messages = state.messages;
@@ -10709,15 +10901,15 @@ function createPatchToolCallsMiddleware() {
       for (let i = 0; i < messages.length; i++) {
         const msg = messages[i];
         patchedMessages.push(msg);
-        if (AIMessage2.isInstance(msg) && msg.tool_calls != null) {
+        if (AIMessage3.isInstance(msg) && msg.tool_calls != null) {
           for (const toolCall of msg.tool_calls) {
             const correspondingToolMsg = messages.slice(i).find(
-              (m) => ToolMessage4.isInstance(m) && m.tool_call_id === toolCall.id
+              (m) => ToolMessage5.isInstance(m) && m.tool_call_id === toolCall.id
             );
             if (!correspondingToolMsg) {
               const toolMsg = `Tool call ${toolCall.name} with id ${toolCall.id} was cancelled - another message came in before it could be completed.`;
               patchedMessages.push(
-                new ToolMessage4({
+                new ToolMessage5({
                   content: toolMsg,
                   name: toolCall.name,
                   tool_call_id: toolCall.id
@@ -11834,8 +12026,8 @@ var MemoryBackend = class {
 
 // src/deep_agent_new/middleware/todos.ts
 import { Command as Command3 } from "@langchain/langgraph";
-import { z as z48 } from "zod";
-import { createMiddleware as createMiddleware13, tool as tool47, ToolMessage as ToolMessage5 } from "langchain";
+import { z as z49 } from "zod";
+import { createMiddleware as createMiddleware15, tool as tool48, ToolMessage as ToolMessage6 } from "langchain";
 var WRITE_TODOS_DESCRIPTION = `Use this tool to create and manage a structured task list for your current work session. This helps you track progress, organize complex tasks, and demonstrate thoroughness to the user.
 It also helps the user understand the progress of the task and overall progress of their requests.
 Only use this tool if you think it will be helpful in staying organized. If the user's request is trivial and takes less than 3 steps, it is better to NOT use this tool and just do the taks directly.
@@ -12062,20 +12254,20 @@ Writing todos takes time and tokens, use it when it is helpful for managing comp
 ## Important To-Do List Usage Notes to Remember
 - The \`write_todos\` tool should never be called multiple times in parallel.
 - Don't be afraid to revise the To-Do list as you go. New information may reveal new tasks that need to be done, or old tasks that are irrelevant.`;
-var TodoStatus = z48.enum(["pending", "in_progress", "completed"]).describe("Status of the todo");
-var TodoSchema = z48.object({
-  content: z48.string().describe("Content of the todo item"),
+var TodoStatus = z49.enum(["pending", "in_progress", "completed"]).describe("Status of the todo");
+var TodoSchema = z49.object({
+  content: z49.string().describe("Content of the todo item"),
   status: TodoStatus
 });
-var stateSchema = z48.object({ todos: z48.array(TodoSchema).default([]) });
+var stateSchema = z49.object({ todos: z49.array(TodoSchema).default([]) });
 function todoListMiddleware(options) {
-  const writeTodos = tool47(
+  const writeTodos = tool48(
     ({ todos }, config) => {
       return new Command3({
         update: {
           todos,
           messages: [
-            new ToolMessage5({
+            new ToolMessage6({
               content: genUIMarkdown("todo_list", todos),
               tool_call_id: config.toolCall?.id
             })
@@ -12086,12 +12278,12 @@ function todoListMiddleware(options) {
     {
       name: "write_todos",
       description: options?.toolDescription ?? WRITE_TODOS_DESCRIPTION,
-      schema: z48.object({
-        todos: z48.array(TodoSchema).describe("List of todo items to update")
+      schema: z49.object({
+        todos: z49.array(TodoSchema).describe("List of todo items to update")
       })
     }
   );
-  return createMiddleware13({
+  return createMiddleware15({
     name: "todoListMiddleware",
     stateSchema,
     tools: [writeTodos],
@@ -12245,7 +12437,7 @@ var DeepAgentGraphBuilder = class {
     const tools = params.tools.map((t) => {
       const toolClient = getToolClient(t.key);
       return toolClient;
-    }).filter((tool50) => tool50 !== void 0);
+    }).filter((tool51) => tool51 !== void 0);
     const subagents = await Promise.all(params.subAgents.map(async (sa) => {
       if (sa.client) {
         return {
@@ -12285,7 +12477,7 @@ var DeepAgentGraphBuilder = class {
 };
 
 // src/agent_team/agent_team.ts
-import { z as z51 } from "zod/v3";
+import { z as z52 } from "zod/v3";
 import { createAgent as createAgent5 } from "langchain";
 
 // src/agent_team/types.ts
@@ -12721,14 +12913,14 @@ var InMemoryMailboxStore = class {
 };
 
 // src/agent_team/middleware/team.ts
-import { z as z50 } from "zod/v3";
-import { createMiddleware as createMiddleware14, createAgent as createAgent4, tool as tool49, ToolMessage as ToolMessage7 } from "langchain";
+import { z as z51 } from "zod/v3";
+import { createMiddleware as createMiddleware16, createAgent as createAgent4, tool as tool50, ToolMessage as ToolMessage8 } from "langchain";
 import { Command as Command5, getCurrentTaskInput as getCurrentTaskInput3 } from "@langchain/langgraph";
 import { v4 as uuidv4 } from "uuid";
 
 // src/agent_team/middleware/teammate_tools.ts
-import { z as z49 } from "zod/v3";
-import { tool as tool48, ToolMessage as ToolMessage6 } from "langchain";
+import { z as z50 } from "zod/v3";
+import { tool as tool49, ToolMessage as ToolMessage7 } from "langchain";
 import { Command as Command4 } from "@langchain/langgraph";
 
 // src/agent_team/middleware/formatMessages.ts
@@ -12753,7 +12945,7 @@ ${meta}${body}`;
 // src/agent_team/middleware/teammate_tools.ts
 function createTeammateTools(options) {
   const { teamId, agentId, taskListStore, mailboxStore } = options;
-  const claimTaskTool = tool48(
+  const claimTaskTool = tool49(
     async (input) => {
       const task = await taskListStore.claimTaskById(
         teamId,
@@ -12778,12 +12970,12 @@ function createTeammateTools(options) {
     {
       name: "claim_task",
       description: "Pick a task to work on by task_id. Use check_tasks first to see all tasks; then call this with the task_id you choose. The task's assignee is set to you and you should focus on that task until you complete_task or fail_task it.",
-      schema: z49.object({
-        task_id: z49.string().describe("ID of the task to claim (e.g. task-01). Use check_tasks to see IDs.")
+      schema: z50.object({
+        task_id: z50.string().describe("ID of the task to claim (e.g. task-01). Use check_tasks to see IDs.")
       })
     }
   );
-  const completeTaskTool = tool48(
+  const completeTaskTool = tool49(
     async (input) => {
       const task = await taskListStore.completeTask(
         teamId,
@@ -12804,13 +12996,13 @@ function createTeammateTools(options) {
     {
       name: "complete_task",
       description: "Mark a claimed task as completed with a result summary. Call this after you have finished working on a task.",
-      schema: z49.object({
-        task_id: z49.string().describe("ID of the task to complete"),
-        result: z49.string().describe("Summary of the task result")
+      schema: z50.object({
+        task_id: z50.string().describe("ID of the task to complete"),
+        result: z50.string().describe("Summary of the task result")
       })
     }
   );
-  const failTaskTool = tool48(
+  const failTaskTool = tool49(
     async (input) => {
       const task = await taskListStore.failTask(
         teamId,
@@ -12831,13 +13023,13 @@ function createTeammateTools(options) {
     {
       name: "fail_task",
       description: "Mark a claimed task as failed with an error description. Call this if you cannot complete the task.",
-      schema: z49.object({
-        task_id: z49.string().describe("ID of the task to fail"),
-        error: z49.string().describe("Description of why the task failed")
+      schema: z50.object({
+        task_id: z50.string().describe("ID of the task to fail"),
+        error: z50.string().describe("Description of why the task failed")
       })
     }
   );
-  const sendMessageTool = tool48(
+  const sendMessageTool = tool49(
     async (input) => {
       await mailboxStore.sendMessage(
         teamId,
@@ -12851,11 +13043,11 @@ function createTeammateTools(options) {
     {
       name: "send_message",
       description: 'Send a message to the team lead or another teammate via the mailbox. Use "team_lead" to message the team lead. Use this to report discoveries, request guidance, or suggest new tasks.',
-      schema: z49.object({
-        to: z49.string().describe(
+      schema: z50.object({
+        to: z50.string().describe(
           'Recipient agent name (e.g. "team_lead" or a teammate name)'
         ),
-        content: z49.string().describe("Message content")
+        content: z50.string().describe("Message content")
       })
     }
   );
@@ -12875,7 +13067,7 @@ function createTeammateTools(options) {
       read: msg.read
     }));
   };
-  const readMessagesTool = tool48(
+  const readMessagesTool = tool49(
     async (input, config) => {
       const formatAndMarkAsRead = async (msgs2) => {
         for (const msg of msgs2) {
@@ -12887,7 +13079,7 @@ function createTeammateTools(options) {
       if (msgs.length > 0) {
         const formatted2 = await formatAndMarkAsRead(msgs);
         const relevantMsgs2 = await getRelevantMessagesForState();
-        const toolMessage2 = new ToolMessage6({
+        const toolMessage2 = new ToolMessage7({
           content: formatted2,
           tool_call_id: config.toolCall?.id,
           name: "read_messages"
@@ -12912,7 +13104,7 @@ function createTeammateTools(options) {
       });
       const relevantMsgs = await getRelevantMessagesForState();
       if (msgs.length === 0) {
-        const toolMessage2 = new ToolMessage6({
+        const toolMessage2 = new ToolMessage7({
           content: "No unread messages.",
           tool_call_id: config.toolCall?.id,
           name: "read_messages"
@@ -12922,7 +13114,7 @@ function createTeammateTools(options) {
         });
       }
       const formatted = await formatAndMarkAsRead(msgs);
-      const toolMessage = new ToolMessage6({
+      const toolMessage = new ToolMessage7({
         content: formatted,
         tool_call_id: config.toolCall?.id,
         name: "read_messages"
@@ -12934,10 +13126,10 @@ function createTeammateTools(options) {
     {
       name: "read_messages",
       description: "Read unread messages from the mailbox. Returns immediately if messages exist, otherwise waits for up to 3 minutes for new messages.",
-      schema: z49.object({})
+      schema: z50.object({})
     }
   );
-  const checkTasksTool = tool48(
+  const checkTasksTool = tool49(
     async () => {
       const tasks = await taskListStore.getAllTasks(teamId);
       return formatTaskSummary(tasks);
@@ -12945,10 +13137,10 @@ function createTeammateTools(options) {
     {
       name: "check_tasks",
       description: "Use this tool to get the current status of all tasks in a team. This is your primary way to monitor task progress.",
-      schema: z49.object({})
+      schema: z50.object({})
     }
   );
-  const broadcastMessageTool = tool48(
+  const broadcastMessageTool = tool49(
     async (input) => {
       const allAgents = await mailboxStore.getRegisteredAgents(teamId);
       const recipients = allAgents.filter((a) => a !== agentId);
@@ -12967,8 +13159,8 @@ function createTeammateTools(options) {
     {
       name: "broadcast_message",
       description: "Send a message to everyone in the team except yourself. Use this to share updates or information with all teammates and the team lead at once.",
-      schema: z49.object({
-        content: z49.string().describe("Message content to broadcast to others")
+      schema: z50.object({
+        content: z50.string().describe("Message content to broadcast to others")
       })
     }
   );
@@ -13202,12 +13394,12 @@ async function spawnTeammate(options) {
 function createTeamMiddleware(options) {
   const { teamConfig, taskListStore, mailboxStore, tenantId } = options;
   const defaultModel = teamConfig.model ?? "claude-sonnet-4-5-20250929";
-  const createTeamTool = tool49(
+  const createTeamTool = tool50(
     async (input, config) => {
       const state = getCurrentTaskInput3();
       if (state?.team?.teamId) {
         const existingId = state.team.teamId;
-        const msg = new ToolMessage7({
+        const msg = new ToolMessage8({
           content: `A team is already active (id: ${existingId}). Use this team_id for \`check_tasks\`, \`read_messages\`, \`add_tasks\`, \`send_message\`, \`assign_task\`, \`set_task_status\`, and \`set_task_dependencies\`. Do not call \`create_team\` again unless you need a fresh team for a new objective.`,
           tool_call_id: config.toolCall?.id,
           name: "create_team"
@@ -13296,7 +13488,7 @@ Teammates are now working in the background. Keep calling \`check_tasks\` and \`
 \`\`\`json
 ${teamJson}
 \`\`\``;
-      const toolMessage = new ToolMessage7({
+      const toolMessage = new ToolMessage8({
         content: summary,
         tool_call_id: config.toolCall?.id,
         name: "create_team"
@@ -13357,20 +13549,20 @@ After calling create_team, you MUST:
 2. When messages indicate task changes, call check_tasks to get full task status
 3. Continue until all tasks show "completed" or "failed"
 4. Do NOT assume tasks are done - always verify with check_tasks`,
-      schema: z50.object({
-        tasks: z50.array(
-          z50.object({
-            id: z50.string().describe("Task ID in format task-01, task-02, etc."),
-            title: z50.string().describe("Short task title"),
-            description: z50.string().describe("Detailed task description - what exactly needs to be done"),
-            dependencies: z50.array(z50.string()).optional().default([]).describe('Array of task IDs that must complete before this task (e.g. ["task-01"])')
+      schema: z51.object({
+        tasks: z51.array(
+          z51.object({
+            id: z51.string().describe("Task ID in format task-01, task-02, etc."),
+            title: z51.string().describe("Short task title"),
+            description: z51.string().describe("Detailed task description - what exactly needs to be done"),
+            dependencies: z51.array(z51.string()).optional().default([]).describe('Array of task IDs that must complete before this task (e.g. ["task-01"])')
           })
         ).describe("List of tasks for teammates to work on. Each task needs unique ID (task-01, task-02, etc.)."),
-        teammates: z50.array(
-          z50.object({
-            name: z50.string().describe("Teammate name (must match a pre-configured teammate type)"),
-            role: z50.string().describe("Role category (e.g. researcher, writer, coder, reviewer)"),
-            description: z50.string().describe("What this teammate will focus on - specific instructions for their work")
+        teammates: z51.array(
+          z51.object({
+            name: z51.string().describe("Teammate name (must match a pre-configured teammate type)"),
+            role: z51.string().describe("Role category (e.g. researcher, writer, coder, reviewer)"),
+            description: z51.string().describe("What this teammate will focus on - specific instructions for their work")
           })
         ).describe("Teammate agents to create. Each should have a clear role and focus.")
       })
@@ -13381,7 +13573,7 @@ After calling create_team, you MUST:
     if (state?.team?.teamId) return state.team.teamId;
     throw new Error("No team_id provided and no team in state. Call create_team first.");
   };
-  const addTasksTool = tool49(
+  const addTasksTool = tool50(
     async (input, config) => {
       const teamId = resolveTeamId();
       const created = await taskListStore.addTasks(
@@ -13395,7 +13587,7 @@ After calling create_team, you MUST:
         }))
       );
       const summary = created.map((t) => `- ${t.id}: "${t.title}"`).join("\n");
-      return new ToolMessage7({
+      return new ToolMessage8({
         content: `Added ${created.length} task(s) to team ${teamId}:
 ${summary}
 Sleeping teammates will wake up and claim these.`,
@@ -13433,33 +13625,33 @@ IMPORTANT: Dependencies
 
 IMPORTANT: Assigning to a specific teammate
 - When you need a particular teammate to do the work, set assignee to that teammate's name (e.g. assignee: "researcher"). They can then claim or see the task as assigned to them.`,
-      schema: z50.object({
-        tasks: z50.array(
-          z50.object({
-            id: z50.string().describe("Task ID in format task-01, task-02, etc. Must be unique."),
-            title: z50.string().describe("Short task title"),
-            description: z50.string().describe("Detailed task description - what needs to be done"),
-            assignee: z50.string().optional().describe("Teammate name to assign this task to (use when you need that person to do the work)"),
-            dependencies: z50.array(z50.string()).optional().default([]).describe("Array of task IDs that must complete before this task")
+      schema: z51.object({
+        tasks: z51.array(
+          z51.object({
+            id: z51.string().describe("Task ID in format task-01, task-02, etc. Must be unique."),
+            title: z51.string().describe("Short task title"),
+            description: z51.string().describe("Detailed task description - what needs to be done"),
+            assignee: z51.string().optional().describe("Teammate name to assign this task to (use when you need that person to do the work)"),
+            dependencies: z51.array(z51.string()).optional().default([]).describe("Array of task IDs that must complete before this task")
           })
         ).describe("New tasks to add to the team")
       })
     }
   );
-  const assignTaskTool = tool49(
+  const assignTaskTool = tool50(
     async (input, config) => {
       const teamId = resolveTeamId();
       const task = await taskListStore.updateTask(teamId, input.task_id, {
         assignee: input.assignee
       });
       if (!task) {
-        return new ToolMessage7({
+        return new ToolMessage8({
           content: `Task ${input.task_id} not found in team ${teamId}.`,
           tool_call_id: config.toolCall?.id,
           name: "assign_task"
         });
       }
-      return new ToolMessage7({
+      return new ToolMessage8({
         content: `Task "${task.title}" (${task.id}) assigned to ${input.assignee}.`,
         tool_call_id: config.toolCall?.id,
         name: "assign_task"
@@ -13468,26 +13660,26 @@ IMPORTANT: Assigning to a specific teammate
     {
       name: "assign_task",
       description: "Assign a task to a specific teammate. Use when you need to reassign work to a different teammate. Omit team_id to use the active team from state.",
-      schema: z50.object({
-        task_id: z50.string().describe("Task ID to assign"),
-        assignee: z50.string().describe("Teammate name to assign this task to")
+      schema: z51.object({
+        task_id: z51.string().describe("Task ID to assign"),
+        assignee: z51.string().describe("Teammate name to assign this task to")
       })
     }
   );
-  const setTaskStatusTool = tool49(
+  const setTaskStatusTool = tool50(
     async (input, config) => {
       const teamId = resolveTeamId();
       const task = await taskListStore.updateTask(teamId, input.task_id, {
         status: input.status
       });
       if (!task) {
-        return new ToolMessage7({
+        return new ToolMessage8({
           content: `Task ${input.task_id} not found in team ${teamId}.`,
           tool_call_id: config.toolCall?.id,
           name: "set_task_status"
         });
       }
-      return new ToolMessage7({
+      return new ToolMessage8({
         content: `Task "${task.title}" (${task.id}) status set to ${input.status}.`,
         tool_call_id: config.toolCall?.id,
         name: "set_task_status"
@@ -13496,26 +13688,26 @@ IMPORTANT: Assigning to a specific teammate
     {
       name: "set_task_status",
       description: "Set a task's status. Use to reopen a task (set to pending), mark as failed, or correct status. Values: pending, claimed, in_progress, completed, failed. Omit team_id to use the active team from state.",
-      schema: z50.object({
-        task_id: z50.string().describe("Task ID to update"),
-        status: z50.enum(["pending", "claimed", "in_progress", "completed", "failed"]).describe("New status for the task")
+      schema: z51.object({
+        task_id: z51.string().describe("Task ID to update"),
+        status: z51.enum(["pending", "claimed", "in_progress", "completed", "failed"]).describe("New status for the task")
       })
     }
   );
-  const setTaskDependenciesTool = tool49(
+  const setTaskDependenciesTool = tool50(
     async (input, config) => {
       const teamId = resolveTeamId();
       const task = await taskListStore.updateTask(teamId, input.task_id, {
         dependencies: input.dependencies
       });
       if (!task) {
-        return new ToolMessage7({
+        return new ToolMessage8({
           content: `Task ${input.task_id} not found in team ${teamId}.`,
           tool_call_id: config.toolCall?.id,
           name: "set_task_dependencies"
         });
       }
-      return new ToolMessage7({
+      return new ToolMessage8({
         content: `Task "${task.title}" (${task.id}) dependencies set to [${input.dependencies.join(", ")}].`,
         tool_call_id: config.toolCall?.id,
         name: "set_task_dependencies"
@@ -13524,13 +13716,13 @@ IMPORTANT: Assigning to a specific teammate
     {
       name: "set_task_dependencies",
       description: 'Set which task IDs must complete before this task can be claimed. Pass an array of task IDs (e.g. ["task-01", "task-02"]). Use to fix task order or add/remove dependencies. Omit team_id to use the active team from state.',
-      schema: z50.object({
-        task_id: z50.string().describe("Task ID to update"),
-        dependencies: z50.array(z50.string()).describe("Task IDs that must complete before this task can be claimed")
+      schema: z51.object({
+        task_id: z51.string().describe("Task ID to update"),
+        dependencies: z51.array(z51.string()).describe("Task IDs that must complete before this task can be claimed")
       })
     }
   );
-  const checkTasksTool = tool49(
+  const checkTasksTool = tool50(
     async (input, config) => {
       const teamId = resolveTeamId();
       const tasks = await taskListStore.getAllTasks(teamId);
@@ -13539,7 +13731,7 @@ IMPORTANT: Assigning to a specific teammate
         update: {
           tasks: tasksSnapshot,
           messages: [
-            new ToolMessage7({
+            new ToolMessage8({
               content: formatTaskSummary(tasks),
               tool_call_id: config.toolCall?.id,
               name: "check_tasks"
@@ -13570,12 +13762,12 @@ Task Status Values:
 - in_progress: Teammate is actively working on this task
 - completed: Task finished successfully
 - failed: Task encountered an error`,
-      schema: z50.object({
-        team_id: z50.string().optional().describe("Team ID (omit to use active team)")
+      schema: z51.object({
+        team_id: z51.string().optional().describe("Team ID (omit to use active team)")
       })
     }
   );
-  const sendMessageTool = tool49(
+  const sendMessageTool = tool50(
     async (input, config) => {
       const teamId = resolveTeamId();
       await mailboxStore.sendMessage(
@@ -13585,7 +13777,7 @@ Task Status Values:
         input.content,
         "direct_message" /* DIRECT_MESSAGE */
       );
-      return new ToolMessage7({
+      return new ToolMessage8({
         content: `Message sent to ${input.to}.`,
         tool_call_id: config.toolCall?.id,
         name: "send_message"
@@ -13594,13 +13786,13 @@ Task Status Values:
     {
       name: "send_message",
       description: "Send a message to a specific teammate in the team. Omit team_id to use the active team from state.",
-      schema: z50.object({
-        to: z50.string().describe("Recipient teammate name"),
-        content: z50.string().describe("Message content")
+      schema: z51.object({
+        to: z51.string().describe("Recipient teammate name"),
+        content: z51.string().describe("Message content")
       })
     }
   );
-  const readMessagesTool = tool49(
+  const readMessagesTool = tool50(
     async (input, config) => {
       const teamId = resolveTeamId();
       const formatAndMarkAsRead = async (msgs2) => {
@@ -13628,7 +13820,7 @@ Task Status Values:
       if (msgs.length > 0) {
         const formatted2 = await formatAndMarkAsRead(msgs);
         const allTeamMessages2 = await getAllTeamMessagesForState();
-        const toolMessage2 = new ToolMessage7({
+        const toolMessage2 = new ToolMessage8({
           content: formatted2,
           tool_call_id: config.toolCall?.id,
           name: "read_messages"
@@ -13660,7 +13852,7 @@ Task Status Values:
       );
       const allTeamMessages = await getAllTeamMessagesForState();
       if (msgs.length === 0) {
-        const toolMessage2 = new ToolMessage7({
+        const toolMessage2 = new ToolMessage8({
           content: "No unread messages from teammates.",
           tool_call_id: config.toolCall?.id,
           name: "read_messages"
@@ -13670,7 +13862,7 @@ Task Status Values:
         });
       }
       const formatted = await formatAndMarkAsRead(msgs);
-      const toolMessage = new ToolMessage7({
+      const toolMessage = new ToolMessage8({
         content: formatted,
         tool_call_id: config.toolCall?.id,
         name: "read_messages"
@@ -13682,12 +13874,12 @@ Task Status Values:
     {
       name: "read_messages",
       description: "Read unread messages from teammates. Returns immediately if messages exist, otherwise waits for up to 3 minutes for new messages.",
-      schema: z50.object({
-        team_id: z50.string().optional().describe("Team ID (omit to use active team)")
+      schema: z51.object({
+        team_id: z51.string().optional().describe("Team ID (omit to use active team)")
       })
     }
   );
-  const disbandTeamTool = tool49(
+  const disbandTeamTool = tool50(
     async (input, config) => {
       const teamId = resolveTeamId();
       await mailboxStore.broadcastMessage(
@@ -13697,7 +13889,7 @@ Task Status Values:
         "shutdown_request" /* SHUTDOWN_REQUEST */
       );
       await new Promise((r) => setTimeout(r, 2e3));
-      return new ToolMessage7({
+      return new ToolMessage8({
         content: `Team ${teamId} has been disbanded. All teammates notified and resources cleaned up.`,
         tool_call_id: config.toolCall?.id,
         name: "disband_team"
@@ -13708,7 +13900,7 @@ Task Status Values:
       description: "Disband a team when all work is done. Before calling: (1) Call check_tasks to verify no tasks are still pending/in_progress; (2) if any are, discuss with the team via read_messages and broadcast_message/send_message whether to continue or stop/cancel them; (3) only after alignment (all tasks completed/failed or explicitly stopped), then call this tool. This will: 1) Send a shutdown message to all teammates, 2) Wait briefly for them to clean up, 3) Clear all tasks and messages. Omit team_id to use the active team from state."
     }
   );
-  const broadcastMessageTool = tool49(
+  const broadcastMessageTool = tool50(
     async (input, config) => {
       const teamId = resolveTeamId();
       await mailboxStore.broadcastMessage(
@@ -13717,7 +13909,7 @@ Task Status Values:
         input.content,
         "broadcast" /* BROADCAST */
       );
-      return new ToolMessage7({
+      return new ToolMessage8({
         content: `Broadcast message sent to all teammates.`,
         tool_call_id: config.toolCall?.id,
         name: "broadcast_message"
@@ -13726,12 +13918,12 @@ Task Status Values:
     {
       name: "broadcast_message",
       description: "Send a message to all teammates at once. Use this to communicate with everyone in the team. Omit team_id to use the active team from state.",
-      schema: z50.object({
-        content: z50.string().describe("Message content to broadcast to all teammates")
+      schema: z51.object({
+        content: z51.string().describe("Message content to broadcast to all teammates")
       })
     }
   );
-  return createMiddleware14({
+  return createMiddleware16({
     name: "teamMiddleware",
     tools: [
       createTeamTool,
@@ -13759,37 +13951,37 @@ ${TEAM_SYSTEM_PROMPT}` : TEAM_SYSTEM_PROMPT;
 }
 
 // src/agent_team/agent_team.ts
-var TeammateInfoSchema = z51.object({
-  name: z51.string().describe("Teammate name"),
-  role: z51.string().describe("Role category (e.g. research, writing, review)"),
-  description: z51.string().describe("What this teammate focuses on")
+var TeammateInfoSchema = z52.object({
+  name: z52.string().describe("Teammate name"),
+  role: z52.string().describe("Role category (e.g. research, writing, review)"),
+  description: z52.string().describe("What this teammate focuses on")
 });
-var TeamTaskInfoSchema = z51.object({
-  id: z51.string(),
-  title: z51.string(),
-  description: z51.string(),
-  status: z51.string().optional()
+var TeamTaskInfoSchema = z52.object({
+  id: z52.string(),
+  title: z52.string(),
+  description: z52.string(),
+  status: z52.string().optional()
 });
-var MailboxMessageSchema = z51.object({
-  id: z51.string().describe("Unique message identifier"),
-  from: z51.string().describe("Sender agent name"),
-  to: z51.string().describe("Recipient agent name"),
-  content: z51.string().describe("Message content"),
-  timestamp: z51.string().describe("ISO timestamp when the message was sent"),
-  type: z51.nativeEnum(MessageType).describe("Message type"),
-  read: z51.boolean().describe("Whether the recipient has read this message")
+var MailboxMessageSchema = z52.object({
+  id: z52.string().describe("Unique message identifier"),
+  from: z52.string().describe("Sender agent name"),
+  to: z52.string().describe("Recipient agent name"),
+  content: z52.string().describe("Message content"),
+  timestamp: z52.string().describe("ISO timestamp when the message was sent"),
+  type: z52.nativeEnum(MessageType).describe("Message type"),
+  read: z52.boolean().describe("Whether the recipient has read this message")
 });
-var TeamInfoSchema = z51.object({
-  teamId: z51.string().describe("Unique team identifier"),
-  teamLeadId: z51.string().default("team_lead").describe("Team lead agent ID"),
-  teammates: z51.array(TeammateInfoSchema).describe("Active teammates in this team"),
-  tasks: z51.array(TeamTaskInfoSchema).optional().describe("Initial tasks snapshot"),
-  createdAt: z51.string().optional().describe("ISO timestamp when team was created")
+var TeamInfoSchema = z52.object({
+  teamId: z52.string().describe("Unique team identifier"),
+  teamLeadId: z52.string().default("team_lead").describe("Team lead agent ID"),
+  teammates: z52.array(TeammateInfoSchema).describe("Active teammates in this team"),
+  tasks: z52.array(TeamTaskInfoSchema).optional().describe("Initial tasks snapshot"),
+  createdAt: z52.string().optional().describe("ISO timestamp when team was created")
 });
-var TEAM_STATE_SCHEMA = z51.object({
+var TEAM_STATE_SCHEMA = z52.object({
   team: TeamInfoSchema.optional().describe("Team info: teamId, teamLeadId, teammates, tasks. Set when create_team succeeds."),
-  tasks: z51.array(TeamTaskInfoSchema).optional().describe("Current tasks snapshot from check_tasks. Updated on each check."),
-  team_mailbox: z51.array(MailboxMessageSchema).optional().describe("All team mailbox messages for display")
+  tasks: z52.array(TeamTaskInfoSchema).optional().describe("Current tasks snapshot from check_tasks. Updated on each check."),
+  team_mailbox: z52.array(MailboxMessageSchema).optional().describe("All team mailbox messages for display")
 });
 var TEAM_LEAD_BASE_PROMPT = `You are a team lead that coordinates a team of specialized agents. In order to complete the objective that the user asks of you, you will need to:
 
@@ -13870,7 +14062,7 @@ var TeamAgentGraphBuilder = class {
     const tools = params.tools.map((t) => {
       const toolClient = getToolClient(t.key);
       return toolClient;
-    }).filter((tool50) => tool50 !== void 0);
+    }).filter((tool51) => tool51 !== void 0);
     const teammates = params.subAgents.map((sa) => {
       const baseConfig = sa.config;
       return {
@@ -14603,6 +14795,9 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
     buffer2.expiresAt = Date.now() + this.config.ttl;
     buffer2.status = "active" /* ACTIVE */;
   }
+  async ensureThread(threadId) {
+    this.getOrCreateBuffer(threadId);
+  }
   async completeThread(threadId) {
     const buffer2 = this.getBufferIfValid(threadId);
     if (buffer2) {
@@ -16756,10 +16951,10 @@ var McpLatticeManager = class _McpLatticeManager extends BaseLatticeManager {
     }
     const tools = await this.getAllTools();
     console.log(`[MCP] Registering ${tools.length} tools to Tool Lattice...`);
-    for (const tool50 of tools) {
-      const toolKey = prefix ? `${prefix}_${tool50.name}` : tool50.name;
-      tool50.name = toolKey;
-      toolLatticeManager.registerExistingTool(toolKey, tool50);
+    for (const tool51 of tools) {
+      const toolKey = prefix ? `${prefix}_${tool51.name}` : tool51.name;
+      tool51.name = toolKey;
+      toolLatticeManager.registerExistingTool(toolKey, tool51);
       console.log(`[MCP] Registered tool: ${toolKey}`);
     }
     console.log(`[MCP] Successfully registered ${tools.length} tools to Tool Lattice`);
@@ -17366,6 +17561,7 @@ var Agent = class {
         id: messageId
       });
     }
+    await this.chunkBuffer.ensureThread(this.thread_id);
     this.startQueueProcessorIfNeeded().catch((err) => {
       console.error("Failed to start queue processor:", err);
     });
@@ -17499,12 +17695,16 @@ var Agent = class {
   /**
    * Abort the current agent execution
    * This will cancel any ongoing invoke or stream operations
+   * and clear all queued messages (pending + processing)
    */
-  abort() {
+  async abort() {
     if (this.abortController) {
       this.abortController.abort();
       this.abortController = null;
     }
+    await this.chunkBuffer.abortThread(this.thread_id);
+    const store = this.getQueueStore();
+    await store.clearMessages(this.thread_id);
   }
   /**
    * Check if the agent is currently being aborted
@@ -17812,6 +18012,7 @@ export {
   createQueryTablesListTool,
   createTeamMiddleware,
   createTeammateTools,
+  createUnknownToolHandlerMiddleware,
   createWidgetMiddleware,
   decrypt,
   describeCronExpression,