@axiom-lattice/core 1.0.26 → 1.0.27

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,88 @@ 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/ReActAgentGraphBuilder.ts
 var ReActAgentGraphBuilder = class {
   /**
    * 构建ReAct Agent Graph
@@ -567,7 +653,8 @@ var ReActAgentGraphBuilder = class {
       llm: params.model,
       tools,
       prompt: params.prompt,
-      name: agentLattice.config.name
+      name: agentLattice.config.name,
+      checkpointer: getCheckpointSaver("default")
     });
   }
 };
@@ -1163,81 +1250,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\`
@@ -1533,16 +1545,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
@@ -2124,11 +2136,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 +2147,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