@axiom-lattice/core 1.0.26 → 1.0.29

package/dist/index.js

@@ -38,19 +38,27 @@ __export(index_exports, {
   MemoryType: () => import_protocols2.MemoryType,
   ModelLatticeManager: () => ModelLatticeManager,
   Protocols: () => Protocols,
+  ToolLatticeManager: () => ToolLatticeManager,
   agentLatticeManager: () => agentLatticeManager,
   getAgentClient: () => getAgentClient,
   getAgentConfig: () => getAgentConfig,
   getAgentLattice: () => getAgentLattice,
   getAllAgentConfigs: () => getAllAgentConfigs,
+  getAllToolDefinitions: () => getAllToolDefinitions,
   getCheckpointSaver: () => getCheckpointSaver,
   getModelLattice: () => getModelLattice,
+  getToolClient: () => getToolClient,
+  getToolDefinition: () => getToolDefinition,
+  getToolLattice: () => getToolLattice,
   modelLatticeManager: () => modelLatticeManager,
   registerAgentLattice: () => registerAgentLattice,
   registerAgentLattices: () => registerAgentLattices,
   registerCheckpointSaver: () => registerCheckpointSaver,
   registerModelLattice: () => registerModelLattice,
-  validateAgentInput: () => validateAgentInput
+  registerToolLattice: () => registerToolLattice,
+  toolLatticeManager: () => toolLatticeManager,
+  validateAgentInput: () => validateAgentInput,
+  validateToolInput: () => validateToolInput
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -533,7 +541,11 @@ var ToolLatticeManager = class _ToolLatticeManager extends BaseLatticeManager {
 };
 var toolLatticeManager = ToolLatticeManager.getInstance();
 var registerToolLattice = (key, config, executor) => toolLatticeManager.registerLattice(key, config, executor);
+var getToolLattice = (key) => toolLatticeManager.getToolLattice(key);
+var getToolDefinition = (key) => toolLatticeManager.getToolDefinition(key);
 var getToolClient = (key) => toolLatticeManager.getToolClient(key);
+var getAllToolDefinitions = () => toolLatticeManager.getAllToolDefinitions();
+var validateToolInput = (key, input) => toolLatticeManager.validateToolInput(key, input);
 
 // src/tool_lattice/get_current_date_time/index.ts
 registerToolLattice(
@@ -596,6 +608,101 @@ var import_protocols = require("@axiom-lattice/protocols");
 
 // src/agent_lattice/builders/ReActAgentGraphBuilder.ts
 var import_prebuilt = require("@langchain/langgraph/prebuilt");
+
+// src/memory_lattice/DefaultMemorySaver.ts
+var import_langgraph = require("@langchain/langgraph");
+
+// src/memory_lattice/MemoryLatticeManager.ts
+var import_protocols2 = require("@axiom-lattice/protocols");
+var _MemoryLatticeManager = class _MemoryLatticeManager extends BaseLatticeManager {
+  /**
+   * 私有构造函数，防止外部直接实例化
+   */
+  constructor() {
+    super();
+  }
+  /**
+   * 获取单例实例
+   */
+  static getInstance() {
+    if (!_MemoryLatticeManager.instance) {
+      _MemoryLatticeManager.instance = new _MemoryLatticeManager();
+    }
+    return _MemoryLatticeManager.instance;
+  }
+  /**
+   * 获取Lattice类型
+   */
+  getLatticeType() {
+    return "memory";
+  }
+  /**
+   * 注册检查点保存器
+   * @param key 保存器键名
+   * @param saver 检查点保存器实例
+   */
+  registerCheckpointSaver(key, saver) {
+    if (_MemoryLatticeManager.checkpointSavers.has(key)) {
+      throw new Error(`\u68C0\u67E5\u70B9\u4FDD\u5B58\u5668 "${key}" \u5DF2\u7ECF\u5B58\u5728\uFF0C\u65E0\u6CD5\u91CD\u590D\u6CE8\u518C`);
+    }
+    _MemoryLatticeManager.checkpointSavers.set(key, saver);
+  }
+  /**
+   * 获取检查点保存器
+   * @param key 保存器键名
+   */
+  getCheckpointSaver(key) {
+    const saver = _MemoryLatticeManager.checkpointSavers.get(key);
+    if (!saver) {
+      throw new Error(`\u68C0\u67E5\u70B9\u4FDD\u5B58\u5668 "${key}" \u4E0D\u5B58\u5728`);
+    }
+    return saver;
+  }
+  /**
+   * 获取所有已注册的检查点保存器键名
+   */
+  getCheckpointSaverKeys() {
+    return Array.from(_MemoryLatticeManager.checkpointSavers.keys());
+  }
+  /**
+   * 检查检查点保存器是否存在
+   * @param key 保存器键名
+   */
+  hasCheckpointSaver(key) {
+    return _MemoryLatticeManager.checkpointSavers.has(key);
+  }
+  /**
+   * 移除检查点保存器
+   * @param key 保存器键名
+   */
+  removeCheckpointSaver(key) {
+    return _MemoryLatticeManager.checkpointSavers.delete(key);
+  }
+};
+// 检查点保存器注册表
+_MemoryLatticeManager.checkpointSavers = /* @__PURE__ */ new Map();
+var MemoryLatticeManager = _MemoryLatticeManager;
+var getCheckpointSaver = (key) => MemoryLatticeManager.getInstance().getCheckpointSaver(key);
+var registerCheckpointSaver = (key, saver) => MemoryLatticeManager.getInstance().registerCheckpointSaver(key, saver);
+
+// src/memory_lattice/DefaultMemorySaver.ts
+var memory = new import_langgraph.MemorySaver();
+registerCheckpointSaver("default", memory);
+
+// src/agent_lattice/builders/state.ts
+var import_zod3 = require("@langchain/langgraph/zod");
+var import_langgraph2 = require("@langchain/langgraph");
+var import_zod4 = require("zod");
+var ReActAgentState = import_langgraph2.MessagesZodState.extend({
+  files: import_zod4.z.object({
+    final_output: import_zod4.z.record(import_zod4.z.any())
+  })
+});
+var createReactAgentSchema = (schema) => {
+  return schema ? import_langgraph2.MessagesZodState.extend(schema.shape) : void 0;
+};
+
+// src/agent_lattice/builders/ReActAgentGraphBuilder.ts
 var ReActAgentGraphBuilder = class {
   /**
    * 构建ReAct Agent Graph
@@ -609,11 +716,14 @@ var ReActAgentGraphBuilder = class {
       const tool5 = getToolClient(t.key);
       return tool5;
     }).filter((tool5) => tool5 !== void 0);
+    const stateSchema = createReactAgentSchema(params.stateSchema);
     return (0, import_prebuilt.createReactAgent)({
       llm: params.model,
       tools,
       prompt: params.prompt,
-      name: agentLattice.config.name
+      name: agentLattice.config.name,
+      checkpointer: getCheckpointSaver("default"),
+      stateSchema
     });
   }
 };
@@ -621,15 +731,15 @@ var ReActAgentGraphBuilder = class {
 // src/deep_agent/subAgent.ts
 var import_tools3 = require("@langchain/core/tools");
 var import_messages2 = require("@langchain/core/messages");
-var import_langgraph2 = require("@langchain/langgraph");
+var import_langgraph4 = require("@langchain/langgraph");
 var import_prebuilt2 = require("@langchain/langgraph/prebuilt");
-var import_zod4 = require("zod");
+var import_zod6 = require("zod");
 
 // src/deep_agent/tools.ts
 var import_tools2 = require("@langchain/core/tools");
 var import_messages = require("@langchain/core/messages");
-var import_langgraph = require("@langchain/langgraph");
-var import_zod3 = require("zod");
+var import_langgraph3 = require("@langchain/langgraph");
+var import_zod5 = require("zod");
 
 // src/deep_agent/prompts.ts
 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.
@@ -894,7 +1004,7 @@ var genUIMarkdown = (type, data) => {
 // src/deep_agent/tools.ts
 var writeTodos = (0, import_tools2.tool)(
   (input, config) => {
-    return new import_langgraph.Command({
+    return new import_langgraph3.Command({
       update: {
         todos: input.todos,
         messages: [
@@ -909,11 +1019,11 @@ var writeTodos = (0, import_tools2.tool)(
   {
     name: "write_todos",
     description: WRITE_TODOS_DESCRIPTION,
-    schema: import_zod3.z.object({
-      todos: import_zod3.z.array(
-        import_zod3.z.object({
-          content: import_zod3.z.string().describe("Content of the todo item"),
-          status: import_zod3.z.enum(["pending", "in_progress", "completed"]).describe("Status of the todo")
+    schema: import_zod5.z.object({
+      todos: import_zod5.z.array(
+        import_zod5.z.object({
+          content: import_zod5.z.string().describe("Content of the todo item"),
+          status: import_zod5.z.enum(["pending", "in_progress", "completed"]).describe("Status of the todo")
         })
       ).describe("List of todo items to update")
     })
@@ -921,19 +1031,19 @@ var writeTodos = (0, import_tools2.tool)(
 );
 var ls = (0, import_tools2.tool)(
   () => {
-    const state = (0, import_langgraph.getCurrentTaskInput)();
+    const state = (0, import_langgraph3.getCurrentTaskInput)();
     const files = state.files || {};
     return Object.keys(files);
   },
   {
     name: "ls",
     description: "List all files in the mock filesystem",
-    schema: import_zod3.z.object({})
+    schema: import_zod5.z.object({})
   }
 );
 var readFile = (0, import_tools2.tool)(
   (input) => {
-    const state = (0, import_langgraph.getCurrentTaskInput)();
+    const state = (0, import_langgraph3.getCurrentTaskInput)();
     const mockFilesystem = state.files || {};
     const { file_path, offset = 0, limit = 2e3 } = input;
     if (!(file_path in mockFilesystem)) {
@@ -963,19 +1073,19 @@ var readFile = (0, import_tools2.tool)(
   {
     name: "read_file",
     description: TOOL_DESCRIPTION,
-    schema: import_zod3.z.object({
-      file_path: import_zod3.z.string().describe("Absolute path to the file to read"),
-      offset: import_zod3.z.number().optional().default(0).describe("Line offset to start reading from"),
-      limit: import_zod3.z.number().optional().default(2e3).describe("Maximum number of lines to read")
+    schema: import_zod5.z.object({
+      file_path: import_zod5.z.string().describe("Absolute path to the file to read"),
+      offset: import_zod5.z.number().optional().default(0).describe("Line offset to start reading from"),
+      limit: import_zod5.z.number().optional().default(2e3).describe("Maximum number of lines to read")
     })
   }
 );
 var writeFile = (0, import_tools2.tool)(
   (input, config) => {
-    const state = (0, import_langgraph.getCurrentTaskInput)();
+    const state = (0, import_langgraph3.getCurrentTaskInput)();
     const files = { ...state.files || {} };
     files[input.file_path] = input.content;
-    return new import_langgraph.Command({
+    return new import_langgraph3.Command({
       update: {
         files,
         messages: [
@@ -990,15 +1100,15 @@ var writeFile = (0, import_tools2.tool)(
   {
     name: "write_file",
     description: "Write content to a file in the mock filesystem",
-    schema: import_zod3.z.object({
-      file_path: import_zod3.z.string().describe("Absolute path to the file to write"),
-      content: import_zod3.z.string().describe("Content to write to the file")
+    schema: import_zod5.z.object({
+      file_path: import_zod5.z.string().describe("Absolute path to the file to write"),
+      content: import_zod5.z.string().describe("Content to write to the file")
     })
   }
 );
 var editFile = (0, import_tools2.tool)(
   (input, config) => {
-    const state = (0, import_langgraph.getCurrentTaskInput)();
+    const state = (0, import_langgraph3.getCurrentTaskInput)();
     const mockFilesystem = { ...state.files || {} };
     const { file_path, old_string, new_string, replace_all = false } = input;
     if (!(file_path in mockFilesystem)) {
@@ -1034,7 +1144,7 @@ var editFile = (0, import_tools2.tool)(
       newContent = content.replace(old_string, new_string);
     }
     mockFilesystem[file_path] = newContent;
-    return new import_langgraph.Command({
+    return new import_langgraph3.Command({
       update: {
         files: mockFilesystem,
         messages: [
@@ -1049,11 +1159,11 @@ var editFile = (0, import_tools2.tool)(
   {
     name: "edit_file",
     description: EDIT_DESCRIPTION,
-    schema: import_zod3.z.object({
-      file_path: import_zod3.z.string().describe("Absolute path to the file to edit"),
-      old_string: import_zod3.z.string().describe("String to be replaced (must match exactly)"),
-      new_string: import_zod3.z.string().describe("String to replace with"),
-      replace_all: import_zod3.z.boolean().optional().default(false).describe("Whether to replace all occurrences")
+    schema: import_zod5.z.object({
+      file_path: import_zod5.z.string().describe("Absolute path to the file to edit"),
+      old_string: import_zod5.z.string().describe("String to be replaced (must match exactly)"),
+      new_string: import_zod5.z.string().describe("String to replace with"),
+      replace_all: import_zod5.z.boolean().optional().default(false).describe("Whether to replace all occurrences")
     })
   }
 );
@@ -1113,7 +1223,7 @@ function createTaskTool(inputs) {
         ).join(", ")}`;
       }
       try {
-        const currentState = (0, import_langgraph2.getCurrentTaskInput)();
+        const currentState = (0, import_langgraph4.getCurrentTaskInput)();
         const modifiedState = {
           ...currentState,
           messages: [
@@ -1124,7 +1234,7 @@ function createTaskTool(inputs) {
           ]
         };
         const result = await reactAgent.invoke(modifiedState, config);
-        return new import_langgraph2.Command({
+        return new import_langgraph4.Command({
           update: {
             files: result.files || {},
             messages: [
@@ -1136,11 +1246,11 @@ function createTaskTool(inputs) {
           }
         });
       } catch (error) {
-        if (error instanceof import_langgraph2.GraphInterrupt) {
+        if (error instanceof import_langgraph4.GraphInterrupt) {
           throw error;
         }
         const errorMessage = error instanceof Error ? error.message : String(error);
-        return new import_langgraph2.Command({
+        return new import_langgraph4.Command({
           update: {
             messages: [
               new import_messages2.ToolMessage({
@@ -1158,9 +1268,9 @@ function createTaskTool(inputs) {
         "{other_agents}",
         subagents.map((a) => `- ${a.name}: ${a.description}`).join("\n")
       ) + TASK_DESCRIPTION_SUFFIX,
-      schema: import_zod4.z.object({
-        description: import_zod4.z.string().describe("The task to execute with the selected agent"),
-        subagent_type: import_zod4.z.string().describe(
+      schema: import_zod6.z.object({
+        description: import_zod6.z.string().describe("The task to execute with the selected agent"),
+        subagent_type: import_zod6.z.string().describe(
           `Name of the agent to use. Available: ${subagents.map((a) => a.name).join(", ")}`
         )
       })
@@ -1169,10 +1279,10 @@ function createTaskTool(inputs) {
 }
 
 // src/deep_agent/state.ts
-var import_zod5 = require("@langchain/langgraph/zod");
-var import_langgraph3 = require("@langchain/langgraph");
-var import_zod6 = require("@langchain/langgraph/zod");
-var import_zod7 = require("zod");
+var import_zod7 = require("@langchain/langgraph/zod");
+var import_langgraph5 = require("@langchain/langgraph");
+var import_zod8 = require("@langchain/langgraph/zod");
+var import_zod9 = require("zod");
 function fileReducer(left, right) {
   if (left == null) {
     return right || {};
@@ -1188,16 +1298,16 @@ function todoReducer(left, right) {
   }
   return left || [];
 }
-var DeepAgentState = import_langgraph3.MessagesZodState.extend({
-  todos: (0, import_zod6.withLangGraph)(import_zod7.z.custom(), {
+var DeepAgentState = import_langgraph5.MessagesZodState.extend({
+  todos: (0, import_zod8.withLangGraph)(import_zod9.z.custom(), {
     reducer: {
-      schema: import_zod7.z.custom(),
+      schema: import_zod9.z.custom(),
       fn: todoReducer
     }
   }),
-  files: (0, import_zod6.withLangGraph)(import_zod7.z.custom(), {
+  files: (0, import_zod8.withLangGraph)(import_zod9.z.custom(), {
     reducer: {
-      schema: import_zod7.z.custom(),
+      schema: import_zod9.z.custom(),
       fn: fileReducer
     }
   })
@@ -1205,81 +1315,6 @@ var DeepAgentState = import_langgraph3.MessagesZodState.extend({
 
 // src/deep_agent/graph.ts
 var import_prebuilt3 = require("@langchain/langgraph/prebuilt");
-
-// src/memory_lattice/MemoryLatticeManager.ts
-var import_protocols2 = require("@axiom-lattice/protocols");
-var _MemoryLatticeManager = class _MemoryLatticeManager extends BaseLatticeManager {
-  /**
-   * 私有构造函数，防止外部直接实例化
-   */
-  constructor() {
-    super();
-  }
-  /**
-   * 获取单例实例
-   */
-  static getInstance() {
-    if (!_MemoryLatticeManager.instance) {
-      _MemoryLatticeManager.instance = new _MemoryLatticeManager();
-    }
-    return _MemoryLatticeManager.instance;
-  }
-  /**
-   * 获取Lattice类型
-   */
-  getLatticeType() {
-    return "memory";
-  }
-  /**
-   * 注册检查点保存器
-   * @param key 保存器键名
-   * @param saver 检查点保存器实例
-   */
-  registerCheckpointSaver(key, saver) {
-    if (_MemoryLatticeManager.checkpointSavers.has(key)) {
-      throw new Error(`\u68C0\u67E5\u70B9\u4FDD\u5B58\u5668 "${key}" \u5DF2\u7ECF\u5B58\u5728\uFF0C\u65E0\u6CD5\u91CD\u590D\u6CE8\u518C`);
-    }
-    _MemoryLatticeManager.checkpointSavers.set(key, saver);
-  }
-  /**
-   * 获取检查点保存器
-   * @param key 保存器键名
-   */
-  getCheckpointSaver(key) {
-    const saver = _MemoryLatticeManager.checkpointSavers.get(key);
-    if (!saver) {
-      throw new Error(`\u68C0\u67E5\u70B9\u4FDD\u5B58\u5668 "${key}" \u4E0D\u5B58\u5728`);
-    }
-    return saver;
-  }
-  /**
-   * 获取所有已注册的检查点保存器键名
-   */
-  getCheckpointSaverKeys() {
-    return Array.from(_MemoryLatticeManager.checkpointSavers.keys());
-  }
-  /**
-   * 检查检查点保存器是否存在
-   * @param key 保存器键名
-   */
-  hasCheckpointSaver(key) {
-    return _MemoryLatticeManager.checkpointSavers.has(key);
-  }
-  /**
-   * 移除检查点保存器
-   * @param key 保存器键名
-   */
-  removeCheckpointSaver(key) {
-    return _MemoryLatticeManager.checkpointSavers.delete(key);
-  }
-};
-// 检查点保存器注册表
-_MemoryLatticeManager.checkpointSavers = /* @__PURE__ */ new Map();
-var MemoryLatticeManager = _MemoryLatticeManager;
-var getCheckpointSaver = (key) => MemoryLatticeManager.getInstance().getCheckpointSaver(key);
-var registerCheckpointSaver = (key, saver) => MemoryLatticeManager.getInstance().registerCheckpointSaver(key, saver);
-
-// src/deep_agent/graph.ts
 var BASE_PROMPT = `You have access to a number of standard tools
 
 ## \`write_todos\`
@@ -1364,9 +1399,9 @@ var DeepAgentGraphBuilder = class {
 };
 
 // src/createPlanExecuteAgent.ts
-var import_langgraph5 = require("@langchain/langgraph");
+var import_langgraph7 = require("@langchain/langgraph");
 var import_messages6 = require("@langchain/core/messages");
-var import_zod8 = require("zod");
+var import_zod10 = require("zod");
 
 // src/logger/Logger.ts
 var import_pino = __toESM(require("pino"));
@@ -1564,54 +1599,54 @@ var getLastHumanMessageData = (messages) => {
 var import_tools7 = require("@langchain/core/tools");
 
 // src/util/PGMemory.ts
-var import_langgraph4 = require("@langchain/langgraph");
+var import_langgraph6 = require("@langchain/langgraph");
 var import_langgraph_checkpoint_postgres = require("@langchain/langgraph-checkpoint-postgres");
 var globalMemory = import_langgraph_checkpoint_postgres.PostgresSaver.fromConnString(process.env.DATABASE_URL);
-var memory = new import_langgraph4.MemorySaver();
+var memory2 = new import_langgraph6.MemorySaver();
 var _MemoryManager = class _MemoryManager {
   static getInstance() {
     return _MemoryManager.instance;
   }
 };
-_MemoryManager.instance = memory;
+_MemoryManager.instance = memory2;
 var MemoryManager = _MemoryManager;
 
 // src/createPlanExecuteAgent.ts
 var import_prebuilt4 = require("@langchain/langgraph/prebuilt");
-var PlanExecuteState = import_langgraph5.Annotation.Root({
+var PlanExecuteState = import_langgraph7.Annotation.Root({
   // 输入
-  input: (0, import_langgraph5.Annotation)({
+  input: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => y ?? x ?? ""
   }),
   // 计划步骤列表
-  plan: (0, import_langgraph5.Annotation)({
+  plan: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => y ?? x ?? []
   }),
   // 已执行的步骤 [步骤名称, 执行结果]
-  pastSteps: (0, import_langgraph5.Annotation)({
+  pastSteps: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => x.concat(y),
     default: () => []
   }),
   // 最终响应
-  response: (0, import_langgraph5.Annotation)({
+  response: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => y ?? x
   }),
   // 错误信息
-  error: (0, import_langgraph5.Annotation)({
+  error: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => y ?? x
   }),
   // 继承基础状态
-  "x-tenant-id": (0, import_langgraph5.Annotation)(),
-  messages: (0, import_langgraph5.Annotation)({
-    reducer: import_langgraph5.messagesStateReducer,
+  "x-tenant-id": (0, import_langgraph7.Annotation)(),
+  messages: (0, import_langgraph7.Annotation)({
+    reducer: import_langgraph7.messagesStateReducer,
     default: () => []
   })
 });
-var planSchema = import_zod8.z.object({
-  steps: import_zod8.z.array(import_zod8.z.string()).describe("\u9700\u8981\u6267\u884C\u7684\u6B65\u9AA4\u5217\u8868\uFF0C\u5E94\u6309\u7167\u6267\u884C\u987A\u5E8F\u6392\u5E8F")
+var planSchema = import_zod10.z.object({
+  steps: import_zod10.z.array(import_zod10.z.string()).describe("\u9700\u8981\u6267\u884C\u7684\u6B65\u9AA4\u5217\u8868\uFF0C\u5E94\u6309\u7167\u6267\u884C\u987A\u5E8F\u6392\u5E8F")
 });
-var responseSchema = import_zod8.z.object({
-  response: import_zod8.z.string().describe("\u7ED9\u7528\u6237\u7684\u6700\u7EC8\u54CD\u5E94\uFF0C\u5982\u679C\u4E0D\u9700\u8981\u6267\u884C\u6B65\u9AA4\uFF0C\u5219\u8FD4\u56DE\u8FD9\u4E2A\u5B57\u6BB5")
+var responseSchema = import_zod10.z.object({
+  response: import_zod10.z.string().describe("\u7ED9\u7528\u6237\u7684\u6700\u7EC8\u54CD\u5E94\uFF0C\u5982\u679C\u4E0D\u9700\u8981\u6267\u884C\u6B65\u9AA4\uFF0C\u5219\u8FD4\u56DE\u8FD9\u4E2A\u5B57\u6BB5")
 });
 function createPlanExecuteAgent(config) {
   const {
@@ -1703,7 +1738,7 @@ ${executorTools.map((tool5) => tool5.name).join("\n")}
         error: void 0
       };
     } catch (error) {
-      if (error instanceof import_langgraph5.GraphInterrupt) {
+      if (error instanceof import_langgraph7.GraphInterrupt) {
         throw error;
       }
       const errorMsg = `\u4EFB\u52A1\u6267\u884C\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`;
@@ -1796,7 +1831,7 @@ Only add steps to the plan that still NEED to be done. Do not return previously
         };
       }
     } catch (error) {
-      if (error instanceof import_langgraph5.GraphInterrupt) {
+      if (error instanceof import_langgraph7.GraphInterrupt) {
         throw error;
       }
       const errorMsg = `\u91CD\u65B0\u89C4\u5212\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`;
@@ -1828,8 +1863,8 @@ Only add steps to the plan that still NEED to be done. Do not return previously
     }
     return "continue";
   }
-  const workflow = new import_langgraph5.StateGraph(PlanExecuteState).addNode("planner", planStep).addNode("executor", executeStep).addNode("replanner", replanStep).addEdge(import_langgraph5.START, "planner").addEdge("planner", "executor").addEdge("executor", "replanner").addConditionalEdges("replanner", shouldEnd, {
-    end: import_langgraph5.END,
+  const workflow = new import_langgraph7.StateGraph(PlanExecuteState).addNode("planner", planStep).addNode("executor", executeStep).addNode("replanner", replanStep).addEdge(import_langgraph7.START, "planner").addEdge("planner", "executor").addEdge("executor", "replanner").addConditionalEdges("replanner", shouldEnd, {
+    end: import_langgraph7.END,
     continue: "executor"
   });
   const compiledGraph = workflow.compile({
@@ -2009,7 +2044,8 @@ var AgentParamsBuilder = class {
       tools,
       model,
       subAgents,
-      prompt: agentLattice.config.prompt
+      prompt: agentLattice.config.prompt,
+      stateSchema: agentLattice.config.schema
     };
   }
 };
@@ -2155,11 +2191,6 @@ var getAllAgentConfigs = () => agentLatticeManager.getAllAgentConfigs();
 var validateAgentInput = (key, input) => agentLatticeManager.validateAgentInput(key, input);
 var getAgentClient = (key, options) => agentLatticeManager.initializeClient(key, options);
 
-// src/memory_lattice/DefaultMemorySaver.ts
-var import_langgraph6 = require("@langchain/langgraph");
-var memory2 = new import_langgraph6.MemorySaver();
-registerCheckpointSaver("default", memory2);
-
 // src/index.ts
 var Protocols = __toESM(require("@axiom-lattice/protocols"));
 // Annotate the CommonJS export names for ESM import in node:
@@ -2172,18 +2203,26 @@ var Protocols = __toESM(require("@axiom-lattice/protocols"));
   MemoryType,
   ModelLatticeManager,
   Protocols,
+  ToolLatticeManager,
   agentLatticeManager,
   getAgentClient,
   getAgentConfig,
   getAgentLattice,
   getAllAgentConfigs,
+  getAllToolDefinitions,
   getCheckpointSaver,
   getModelLattice,
+  getToolClient,
+  getToolDefinition,
+  getToolLattice,
   modelLatticeManager,
   registerAgentLattice,
   registerAgentLattices,
   registerCheckpointSaver,
   registerModelLattice,
-  validateAgentInput
+  registerToolLattice,
+  toolLatticeManager,
+  validateAgentInput,
+  validateToolInput
 });
 //# sourceMappingURL=index.js.map