@axiom-lattice/core 1.0.35 → 1.0.39

package/dist/index.js

@@ -412,6 +412,7 @@ var import_zod = __toESM(require("zod"));
 
 // src/tool_lattice/ToolLatticeManager.ts
 var import_tools = require("@langchain/core/tools");
+var import_langgraph = require("@langchain/langgraph");
 var ToolLatticeManager = class _ToolLatticeManager extends BaseLatticeManager {
   /**
    * 获取ToolLatticeManager单例实例
@@ -439,6 +440,24 @@ var ToolLatticeManager = class _ToolLatticeManager extends BaseLatticeManager {
       const result = await executor(input, exe_config);
       return result;
     };
+    if (config.needUserApprove) {
+      toolExecutor = async (input, exe_config) => {
+        const contents = [
+          "```confirm",
+          JSON.stringify({
+            message: "try to " + config.name + ",please confirm the action"
+          }),
+          "```"
+        ].join("\n");
+        const feedback = await (0, import_langgraph.interrupt)(contents);
+        if (feedback.data.action === "yes") {
+          const result = await executor(input, exe_config);
+          return result;
+        } else {
+          return "user denied the action";
+        }
+      };
+    }
     const toolLattice = {
       key,
       config,
@@ -610,7 +629,7 @@ var import_protocols = require("@axiom-lattice/protocols");
 var import_prebuilt = require("@langchain/langgraph/prebuilt");
 
 // src/memory_lattice/DefaultMemorySaver.ts
-var import_langgraph = require("@langchain/langgraph");
+var import_langgraph2 = require("@langchain/langgraph");
 
 // src/memory_lattice/MemoryLatticeManager.ts
 var import_protocols2 = require("@axiom-lattice/protocols");
@@ -686,20 +705,20 @@ var getCheckpointSaver = (key) => MemoryLatticeManager.getInstance().getCheckpoi
 var registerCheckpointSaver = (key, saver) => MemoryLatticeManager.getInstance().registerCheckpointSaver(key, saver);
 
 // src/memory_lattice/DefaultMemorySaver.ts
-var memory = new import_langgraph.MemorySaver();
+var memory = new import_langgraph2.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_langgraph3 = require("@langchain/langgraph");
 var import_zod4 = require("zod");
-var ReActAgentState = import_langgraph2.MessagesZodState.extend({
+var ReActAgentState = import_langgraph3.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;
+  return schema ? import_langgraph3.MessagesZodState.extend(schema.shape) : void 0;
 };
 
 // src/agent_lattice/builders/ReActAgentGraphBuilder.ts
@@ -731,14 +750,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_langgraph4 = require("@langchain/langgraph");
+var import_langgraph5 = require("@langchain/langgraph");
 var import_prebuilt2 = require("@langchain/langgraph/prebuilt");
 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_langgraph3 = require("@langchain/langgraph");
+var import_langgraph4 = require("@langchain/langgraph");
 var import_zod5 = require("zod");
 
 // src/deep_agent/prompts.ts
@@ -1004,7 +1023,7 @@ var genUIMarkdown = (type, data) => {
 // src/deep_agent/tools.ts
 var writeTodos = (0, import_tools2.tool)(
   (input, config) => {
-    return new import_langgraph3.Command({
+    return new import_langgraph4.Command({
       update: {
         todos: input.todos,
         messages: [
@@ -1031,7 +1050,7 @@ var writeTodos = (0, import_tools2.tool)(
 );
 var ls = (0, import_tools2.tool)(
   () => {
-    const state = (0, import_langgraph3.getCurrentTaskInput)();
+    const state = (0, import_langgraph4.getCurrentTaskInput)();
     const files = state.files || {};
     return Object.keys(files);
   },
@@ -1043,7 +1062,7 @@ var ls = (0, import_tools2.tool)(
 );
 var readFile = (0, import_tools2.tool)(
   (input) => {
-    const state = (0, import_langgraph3.getCurrentTaskInput)();
+    const state = (0, import_langgraph4.getCurrentTaskInput)();
     const mockFilesystem = state.files || {};
     const { file_path, offset = 0, limit = 2e3 } = input;
     if (!(file_path in mockFilesystem)) {
@@ -1082,10 +1101,10 @@ var readFile = (0, import_tools2.tool)(
 );
 var writeFile = (0, import_tools2.tool)(
   (input, config) => {
-    const state = (0, import_langgraph3.getCurrentTaskInput)();
+    const state = (0, import_langgraph4.getCurrentTaskInput)();
     const files = { ...state.files || {} };
     files[input.file_path] = input.content;
-    return new import_langgraph3.Command({
+    return new import_langgraph4.Command({
       update: {
         files,
         messages: [
@@ -1108,7 +1127,7 @@ var writeFile = (0, import_tools2.tool)(
 );
 var editFile = (0, import_tools2.tool)(
   (input, config) => {
-    const state = (0, import_langgraph3.getCurrentTaskInput)();
+    const state = (0, import_langgraph4.getCurrentTaskInput)();
     const mockFilesystem = { ...state.files || {} };
     const { file_path, old_string, new_string, replace_all = false } = input;
     if (!(file_path in mockFilesystem)) {
@@ -1144,7 +1163,7 @@ var editFile = (0, import_tools2.tool)(
       newContent = content.replace(old_string, new_string);
     }
     mockFilesystem[file_path] = newContent;
-    return new import_langgraph3.Command({
+    return new import_langgraph4.Command({
       update: {
         files: mockFilesystem,
         messages: [
@@ -1223,7 +1242,7 @@ function createTaskTool(inputs) {
         ).join(", ")}`;
       }
       try {
-        const currentState = (0, import_langgraph4.getCurrentTaskInput)();
+        const currentState = (0, import_langgraph5.getCurrentTaskInput)();
         const modifiedState = {
           ...currentState,
           messages: [
@@ -1234,7 +1253,7 @@ function createTaskTool(inputs) {
           ]
         };
         const result = await reactAgent.invoke(modifiedState, config);
-        return new import_langgraph4.Command({
+        return new import_langgraph5.Command({
           update: {
             files: result.files || {},
             messages: [
@@ -1246,11 +1265,11 @@ function createTaskTool(inputs) {
           }
         });
       } catch (error) {
-        if (error instanceof import_langgraph4.GraphInterrupt) {
+        if (error instanceof import_langgraph5.GraphInterrupt) {
           throw error;
         }
         const errorMessage = error instanceof Error ? error.message : String(error);
-        return new import_langgraph4.Command({
+        return new import_langgraph5.Command({
           update: {
             messages: [
               new import_messages2.ToolMessage({
@@ -1280,7 +1299,7 @@ function createTaskTool(inputs) {
 
 // src/deep_agent/state.ts
 var import_zod7 = require("@langchain/langgraph/zod");
-var import_langgraph5 = require("@langchain/langgraph");
+var import_langgraph6 = require("@langchain/langgraph");
 var import_zod8 = require("@langchain/langgraph/zod");
 var import_zod9 = require("zod");
 function fileReducer(left, right) {
@@ -1298,7 +1317,7 @@ function todoReducer(left, right) {
   }
   return left || [];
 }
-var DeepAgentState = import_langgraph5.MessagesZodState.extend({
+var DeepAgentState = import_langgraph6.MessagesZodState.extend({
   todos: (0, import_zod8.withLangGraph)(import_zod9.z.custom(), {
     reducer: {
       schema: import_zod9.z.custom(),
@@ -1399,7 +1418,7 @@ var DeepAgentGraphBuilder = class {
 };
 
 // src/createPlanExecuteAgent.ts
-var import_langgraph7 = require("@langchain/langgraph");
+var import_langgraph8 = require("@langchain/langgraph");
 var import_messages6 = require("@langchain/core/messages");
 var import_zod10 = require("zod");
 
@@ -1599,10 +1618,10 @@ var getLastHumanMessageData = (messages) => {
 var import_tools7 = require("@langchain/core/tools");
 
 // src/util/PGMemory.ts
-var import_langgraph6 = require("@langchain/langgraph");
+var import_langgraph7 = 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 memory2 = new import_langgraph6.MemorySaver();
+var memory2 = new import_langgraph7.MemorySaver();
 var _MemoryManager = class _MemoryManager {
   static getInstance() {
     return _MemoryManager.instance;
@@ -1613,32 +1632,32 @@ var MemoryManager = _MemoryManager;
 
 // src/createPlanExecuteAgent.ts
 var import_prebuilt4 = require("@langchain/langgraph/prebuilt");
-var PlanExecuteState = import_langgraph7.Annotation.Root({
+var PlanExecuteState = import_langgraph8.Annotation.Root({
   // 输入
-  input: (0, import_langgraph7.Annotation)({
+  input: (0, import_langgraph8.Annotation)({
     reducer: (x, y) => y ?? x ?? ""
   }),
   // 计划步骤列表
-  plan: (0, import_langgraph7.Annotation)({
+  plan: (0, import_langgraph8.Annotation)({
     reducer: (x, y) => y ?? x ?? []
   }),
   // 已执行的步骤 [步骤名称, 执行结果]
-  pastSteps: (0, import_langgraph7.Annotation)({
+  pastSteps: (0, import_langgraph8.Annotation)({
     reducer: (x, y) => x.concat(y),
     default: () => []
   }),
   // 最终响应
-  response: (0, import_langgraph7.Annotation)({
+  response: (0, import_langgraph8.Annotation)({
     reducer: (x, y) => y ?? x
   }),
   // 错误信息
-  error: (0, import_langgraph7.Annotation)({
+  error: (0, import_langgraph8.Annotation)({
     reducer: (x, y) => y ?? x
   }),
   // 继承基础状态
-  "x-tenant-id": (0, import_langgraph7.Annotation)(),
-  messages: (0, import_langgraph7.Annotation)({
-    reducer: import_langgraph7.messagesStateReducer,
+  "x-tenant-id": (0, import_langgraph8.Annotation)(),
+  messages: (0, import_langgraph8.Annotation)({
+    reducer: import_langgraph8.messagesStateReducer,
     default: () => []
   })
 });
@@ -1738,7 +1757,7 @@ ${executorTools.map((tool5) => tool5.name).join("\n")}
         error: void 0
       };
     } catch (error) {
-      if (error instanceof import_langgraph7.GraphInterrupt) {
+      if (error instanceof import_langgraph8.GraphInterrupt) {
         throw error;
       }
       const errorMsg = `\u4EFB\u52A1\u6267\u884C\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`;
@@ -1831,7 +1850,7 @@ Only add steps to the plan that still NEED to be done. Do not return previously
         };
       }
     } catch (error) {
-      if (error instanceof import_langgraph7.GraphInterrupt) {
+      if (error instanceof import_langgraph8.GraphInterrupt) {
         throw error;
       }
       const errorMsg = `\u91CD\u65B0\u89C4\u5212\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`;
@@ -1863,8 +1882,8 @@ Only add steps to the plan that still NEED to be done. Do not return previously
     }
     return "continue";
   }
-  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,
+  const workflow = new import_langgraph8.StateGraph(PlanExecuteState).addNode("planner", planStep).addNode("executor", executeStep).addNode("replanner", replanStep).addEdge(import_langgraph8.START, "planner").addEdge("planner", "executor").addEdge("executor", "replanner").addConditionalEdges("replanner", shouldEnd, {
+    end: import_langgraph8.END,
     continue: "executor"
   });
   const compiledGraph = workflow.compile({