@axiom-lattice/core 1.0.26 → 1.0.27

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,88 @@ 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/ReActAgentGraphBuilder.ts
 var ReActAgentGraphBuilder = class {
   /**
    * 构建ReAct Agent Graph
@@ -613,7 +707,8 @@ var ReActAgentGraphBuilder = class {
       llm: params.model,
       tools,
       prompt: params.prompt,
-      name: agentLattice.config.name
+      name: agentLattice.config.name,
+      checkpointer: getCheckpointSaver("default")
     });
   }
 };
@@ -621,14 +716,14 @@ 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_langgraph3 = require("@langchain/langgraph");
 var import_prebuilt2 = require("@langchain/langgraph/prebuilt");
 var import_zod4 = 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_langgraph2 = require("@langchain/langgraph");
 var import_zod3 = require("zod");
 
 // src/deep_agent/prompts.ts
@@ -894,7 +989,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_langgraph2.Command({
       update: {
         todos: input.todos,
         messages: [
@@ -921,7 +1016,7 @@ var writeTodos = (0, import_tools2.tool)(
 );
 var ls = (0, import_tools2.tool)(
   () => {
-    const state = (0, import_langgraph.getCurrentTaskInput)();
+    const state = (0, import_langgraph2.getCurrentTaskInput)();
     const files = state.files || {};
     return Object.keys(files);
   },
@@ -933,7 +1028,7 @@ var ls = (0, import_tools2.tool)(
 );
 var readFile = (0, import_tools2.tool)(
   (input) => {
-    const state = (0, import_langgraph.getCurrentTaskInput)();
+    const state = (0, import_langgraph2.getCurrentTaskInput)();
     const mockFilesystem = state.files || {};
     const { file_path, offset = 0, limit = 2e3 } = input;
     if (!(file_path in mockFilesystem)) {
@@ -972,10 +1067,10 @@ var readFile = (0, import_tools2.tool)(
 );
 var writeFile = (0, import_tools2.tool)(
   (input, config) => {
-    const state = (0, import_langgraph.getCurrentTaskInput)();
+    const state = (0, import_langgraph2.getCurrentTaskInput)();
     const files = { ...state.files || {} };
     files[input.file_path] = input.content;
-    return new import_langgraph.Command({
+    return new import_langgraph2.Command({
       update: {
         files,
         messages: [
@@ -998,7 +1093,7 @@ var writeFile = (0, import_tools2.tool)(
 );
 var editFile = (0, import_tools2.tool)(
   (input, config) => {
-    const state = (0, import_langgraph.getCurrentTaskInput)();
+    const state = (0, import_langgraph2.getCurrentTaskInput)();
     const mockFilesystem = { ...state.files || {} };
     const { file_path, old_string, new_string, replace_all = false } = input;
     if (!(file_path in mockFilesystem)) {
@@ -1034,7 +1129,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_langgraph2.Command({
       update: {
         files: mockFilesystem,
         messages: [
@@ -1113,7 +1208,7 @@ function createTaskTool(inputs) {
         ).join(", ")}`;
       }
       try {
-        const currentState = (0, import_langgraph2.getCurrentTaskInput)();
+        const currentState = (0, import_langgraph3.getCurrentTaskInput)();
         const modifiedState = {
           ...currentState,
           messages: [
@@ -1124,7 +1219,7 @@ function createTaskTool(inputs) {
           ]
         };
         const result = await reactAgent.invoke(modifiedState, config);
-        return new import_langgraph2.Command({
+        return new import_langgraph3.Command({
           update: {
             files: result.files || {},
             messages: [
@@ -1136,11 +1231,11 @@ function createTaskTool(inputs) {
           }
         });
       } catch (error) {
-        if (error instanceof import_langgraph2.GraphInterrupt) {
+        if (error instanceof import_langgraph3.GraphInterrupt) {
           throw error;
         }
         const errorMessage = error instanceof Error ? error.message : String(error);
-        return new import_langgraph2.Command({
+        return new import_langgraph3.Command({
           update: {
             messages: [
               new import_messages2.ToolMessage({
@@ -1170,7 +1265,7 @@ function createTaskTool(inputs) {
 
 // src/deep_agent/state.ts
 var import_zod5 = require("@langchain/langgraph/zod");
-var import_langgraph3 = require("@langchain/langgraph");
+var import_langgraph4 = require("@langchain/langgraph");
 var import_zod6 = require("@langchain/langgraph/zod");
 var import_zod7 = require("zod");
 function fileReducer(left, right) {
@@ -1188,7 +1283,7 @@ function todoReducer(left, right) {
   }
   return left || [];
 }
-var DeepAgentState = import_langgraph3.MessagesZodState.extend({
+var DeepAgentState = import_langgraph4.MessagesZodState.extend({
   todos: (0, import_zod6.withLangGraph)(import_zod7.z.custom(), {
     reducer: {
       schema: import_zod7.z.custom(),
@@ -1205,81 +1300,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,7 +1384,7 @@ var DeepAgentGraphBuilder = class {
 };
 
 // src/createPlanExecuteAgent.ts
-var import_langgraph5 = require("@langchain/langgraph");
+var import_langgraph6 = require("@langchain/langgraph");
 var import_messages6 = require("@langchain/core/messages");
 var import_zod8 = require("zod");
 
@@ -1564,46 +1584,46 @@ var getLastHumanMessageData = (messages) => {
 var import_tools7 = require("@langchain/core/tools");
 
 // src/util/PGMemory.ts
-var import_langgraph4 = require("@langchain/langgraph");
+var import_langgraph5 = 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_langgraph5.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_langgraph6.Annotation.Root({
   // 输入
-  input: (0, import_langgraph5.Annotation)({
+  input: (0, import_langgraph6.Annotation)({
     reducer: (x, y) => y ?? x ?? ""
   }),
   // 计划步骤列表
-  plan: (0, import_langgraph5.Annotation)({
+  plan: (0, import_langgraph6.Annotation)({
     reducer: (x, y) => y ?? x ?? []
   }),
   // 已执行的步骤 [步骤名称, 执行结果]
-  pastSteps: (0, import_langgraph5.Annotation)({
+  pastSteps: (0, import_langgraph6.Annotation)({
     reducer: (x, y) => x.concat(y),
     default: () => []
   }),
   // 最终响应
-  response: (0, import_langgraph5.Annotation)({
+  response: (0, import_langgraph6.Annotation)({
     reducer: (x, y) => y ?? x
   }),
   // 错误信息
-  error: (0, import_langgraph5.Annotation)({
+  error: (0, import_langgraph6.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_langgraph6.Annotation)(),
+  messages: (0, import_langgraph6.Annotation)({
+    reducer: import_langgraph6.messagesStateReducer,
     default: () => []
   })
 });
@@ -1703,7 +1723,7 @@ ${executorTools.map((tool5) => tool5.name).join("\n")}
         error: void 0
       };
     } catch (error) {
-      if (error instanceof import_langgraph5.GraphInterrupt) {
+      if (error instanceof import_langgraph6.GraphInterrupt) {
         throw error;
       }
       const errorMsg = `\u4EFB\u52A1\u6267\u884C\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`;
@@ -1796,7 +1816,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_langgraph6.GraphInterrupt) {
         throw error;
       }
       const errorMsg = `\u91CD\u65B0\u89C4\u5212\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`;
@@ -1828,8 +1848,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_langgraph6.StateGraph(PlanExecuteState).addNode("planner", planStep).addNode("executor", executeStep).addNode("replanner", replanStep).addEdge(import_langgraph6.START, "planner").addEdge("planner", "executor").addEdge("executor", "replanner").addConditionalEdges("replanner", shouldEnd, {
+    end: import_langgraph6.END,
     continue: "executor"
   });
   const compiledGraph = workflow.compile({
@@ -2155,11 +2175,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 +2187,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