@axiom-lattice/core 1.0.26 → 1.0.29

package/dist/index.mjs

@@ -483,7 +483,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(
@@ -550,6 +554,101 @@ import {
 
 // src/agent_lattice/builders/ReActAgentGraphBuilder.ts
 import { createReactAgent } from "@langchain/langgraph/prebuilt";
+
+// src/memory_lattice/DefaultMemorySaver.ts
+import { MemorySaver } from "@langchain/langgraph";
+
+// src/memory_lattice/MemoryLatticeManager.ts
+import { MemoryType } from "@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 MemorySaver();
+registerCheckpointSaver("default", memory);
+
+// src/agent_lattice/builders/state.ts
+import "@langchain/langgraph/zod";
+import { MessagesZodState } from "@langchain/langgraph";
+import { z as z3 } from "zod";
+var ReActAgentState = MessagesZodState.extend({
+  files: z3.object({
+    final_output: z3.record(z3.any())
+  })
+});
+var createReactAgentSchema = (schema) => {
+  return schema ? MessagesZodState.extend(schema.shape) : void 0;
+};
+
+// src/agent_lattice/builders/ReActAgentGraphBuilder.ts
 var ReActAgentGraphBuilder = class {
   /**
    * 构建ReAct Agent Graph
@@ -563,11 +662,14 @@ var ReActAgentGraphBuilder = class {
       const tool5 = getToolClient(t.key);
       return tool5;
     }).filter((tool5) => tool5 !== void 0);
+    const stateSchema = createReactAgentSchema(params.stateSchema);
     return createReactAgent({
       llm: params.model,
       tools,
       prompt: params.prompt,
-      name: agentLattice.config.name
+      name: agentLattice.config.name,
+      checkpointer: getCheckpointSaver("default"),
+      stateSchema
     });
   }
 };
@@ -581,13 +683,13 @@ import {
   GraphInterrupt
 } from "@langchain/langgraph";
 import { createReactAgent as createReactAgent2 } from "@langchain/langgraph/prebuilt";
-import { z as z4 } from "zod";
+import { z as z5 } from "zod";
 
 // src/deep_agent/tools.ts
 import { tool as tool2 } from "@langchain/core/tools";
 import { ToolMessage } from "@langchain/core/messages";
 import { Command, getCurrentTaskInput } from "@langchain/langgraph";
-import { z as z3 } from "zod";
+import { z as z4 } from "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.
@@ -867,11 +969,11 @@ var writeTodos = tool2(
   {
     name: "write_todos",
     description: WRITE_TODOS_DESCRIPTION,
-    schema: z3.object({
-      todos: z3.array(
-        z3.object({
-          content: z3.string().describe("Content of the todo item"),
-          status: z3.enum(["pending", "in_progress", "completed"]).describe("Status of the todo")
+    schema: z4.object({
+      todos: z4.array(
+        z4.object({
+          content: z4.string().describe("Content of the todo item"),
+          status: z4.enum(["pending", "in_progress", "completed"]).describe("Status of the todo")
         })
       ).describe("List of todo items to update")
     })
@@ -886,7 +988,7 @@ var ls = tool2(
   {
     name: "ls",
     description: "List all files in the mock filesystem",
-    schema: z3.object({})
+    schema: z4.object({})
   }
 );
 var readFile = tool2(
@@ -921,10 +1023,10 @@ var readFile = tool2(
   {
     name: "read_file",
     description: TOOL_DESCRIPTION,
-    schema: z3.object({
-      file_path: z3.string().describe("Absolute path to the file to read"),
-      offset: z3.number().optional().default(0).describe("Line offset to start reading from"),
-      limit: z3.number().optional().default(2e3).describe("Maximum number of lines to read")
+    schema: z4.object({
+      file_path: z4.string().describe("Absolute path to the file to read"),
+      offset: z4.number().optional().default(0).describe("Line offset to start reading from"),
+      limit: z4.number().optional().default(2e3).describe("Maximum number of lines to read")
     })
   }
 );
@@ -948,9 +1050,9 @@ var writeFile = tool2(
   {
     name: "write_file",
     description: "Write content to a file in the mock filesystem",
-    schema: z3.object({
-      file_path: z3.string().describe("Absolute path to the file to write"),
-      content: z3.string().describe("Content to write to the file")
+    schema: z4.object({
+      file_path: z4.string().describe("Absolute path to the file to write"),
+      content: z4.string().describe("Content to write to the file")
     })
   }
 );
@@ -1007,11 +1109,11 @@ var editFile = tool2(
   {
     name: "edit_file",
     description: EDIT_DESCRIPTION,
-    schema: z3.object({
-      file_path: z3.string().describe("Absolute path to the file to edit"),
-      old_string: z3.string().describe("String to be replaced (must match exactly)"),
-      new_string: z3.string().describe("String to replace with"),
-      replace_all: z3.boolean().optional().default(false).describe("Whether to replace all occurrences")
+    schema: z4.object({
+      file_path: z4.string().describe("Absolute path to the file to edit"),
+      old_string: z4.string().describe("String to be replaced (must match exactly)"),
+      new_string: z4.string().describe("String to replace with"),
+      replace_all: z4.boolean().optional().default(false).describe("Whether to replace all occurrences")
     })
   }
 );
@@ -1116,9 +1218,9 @@ function createTaskTool(inputs) {
         "{other_agents}",
         subagents.map((a) => `- ${a.name}: ${a.description}`).join("\n")
       ) + TASK_DESCRIPTION_SUFFIX,
-      schema: z4.object({
-        description: z4.string().describe("The task to execute with the selected agent"),
-        subagent_type: z4.string().describe(
+      schema: z5.object({
+        description: z5.string().describe("The task to execute with the selected agent"),
+        subagent_type: z5.string().describe(
           `Name of the agent to use. Available: ${subagents.map((a) => a.name).join(", ")}`
         )
       })
@@ -1128,9 +1230,9 @@ function createTaskTool(inputs) {
 
 // src/deep_agent/state.ts
 import "@langchain/langgraph/zod";
-import { MessagesZodState } from "@langchain/langgraph";
+import { MessagesZodState as MessagesZodState2 } from "@langchain/langgraph";
 import { withLangGraph } from "@langchain/langgraph/zod";
-import { z as z5 } from "zod";
+import { z as z6 } from "zod";
 function fileReducer(left, right) {
   if (left == null) {
     return right || {};
@@ -1146,16 +1248,16 @@ function todoReducer(left, right) {
   }
   return left || [];
 }
-var DeepAgentState = MessagesZodState.extend({
-  todos: withLangGraph(z5.custom(), {
+var DeepAgentState = MessagesZodState2.extend({
+  todos: withLangGraph(z6.custom(), {
     reducer: {
-      schema: z5.custom(),
+      schema: z6.custom(),
       fn: todoReducer
     }
   }),
-  files: withLangGraph(z5.custom(), {
+  files: withLangGraph(z6.custom(), {
     reducer: {
-      schema: z5.custom(),
+      schema: z6.custom(),
       fn: fileReducer
     }
   })
@@ -1163,81 +1265,6 @@ var DeepAgentState = MessagesZodState.extend({
 
 // src/deep_agent/graph.ts
 import { createReactAgent as createReactAgent3 } from "@langchain/langgraph/prebuilt";
-
-// src/memory_lattice/MemoryLatticeManager.ts
-import { MemoryType } from "@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\`
@@ -1333,7 +1360,7 @@ import {
 import {
   HumanMessage as HumanMessage2
 } from "@langchain/core/messages";
-import { z as z6 } from "zod";
+import { z as z7 } from "zod";
 
 // src/logger/Logger.ts
 import pino from "pino";
@@ -1533,16 +1560,16 @@ var getLastHumanMessageData = (messages) => {
 import { tool as tool4 } from "@langchain/core/tools";
 
 // src/util/PGMemory.ts
-import { MemorySaver } from "@langchain/langgraph";
+import { MemorySaver as MemorySaver2 } from "@langchain/langgraph";
 import { PostgresSaver } from "@langchain/langgraph-checkpoint-postgres";
 var globalMemory = PostgresSaver.fromConnString(process.env.DATABASE_URL);
-var memory = new MemorySaver();
+var memory2 = new MemorySaver2();
 var _MemoryManager = class _MemoryManager {
   static getInstance() {
     return _MemoryManager.instance;
   }
 };
-_MemoryManager.instance = memory;
+_MemoryManager.instance = memory2;
 var MemoryManager = _MemoryManager;
 
 // src/createPlanExecuteAgent.ts
@@ -1576,11 +1603,11 @@ var PlanExecuteState = Annotation.Root({
     default: () => []
   })
 });
-var planSchema = z6.object({
-  steps: z6.array(z6.string()).describe("\u9700\u8981\u6267\u884C\u7684\u6B65\u9AA4\u5217\u8868\uFF0C\u5E94\u6309\u7167\u6267\u884C\u987A\u5E8F\u6392\u5E8F")
+var planSchema = z7.object({
+  steps: z7.array(z7.string()).describe("\u9700\u8981\u6267\u884C\u7684\u6B65\u9AA4\u5217\u8868\uFF0C\u5E94\u6309\u7167\u6267\u884C\u987A\u5E8F\u6392\u5E8F")
 });
-var responseSchema = z6.object({
-  response: z6.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 = z7.object({
+  response: z7.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 {
@@ -1978,7 +2005,8 @@ var AgentParamsBuilder = class {
       tools,
       model,
       subAgents,
-      prompt: agentLattice.config.prompt
+      prompt: agentLattice.config.prompt,
+      stateSchema: agentLattice.config.schema
     };
   }
 };
@@ -2124,11 +2152,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
-import { MemorySaver as MemorySaver2 } from "@langchain/langgraph";
-var memory2 = new MemorySaver2();
-registerCheckpointSaver("default", memory2);
-
 // src/index.ts
 import * as Protocols from "@axiom-lattice/protocols";
 export {
@@ -2140,18 +2163,26 @@ export {
   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.mjs.map