@axiom-lattice/core 1.0.27 → 1.0.30

package/dist/index.js

@@ -689,6 +689,19 @@ var registerCheckpointSaver = (key, saver) => MemoryLatticeManager.getInstance()
 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 {
   /**
@@ -703,12 +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,
-      checkpointer: getCheckpointSaver("default")
+      checkpointer: getCheckpointSaver("default"),
+      stateSchema
     });
   }
 };
@@ -716,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_langgraph3 = 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_langgraph2 = 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.
@@ -989,7 +1004,7 @@ var genUIMarkdown = (type, data) => {
 // src/deep_agent/tools.ts
 var writeTodos = (0, import_tools2.tool)(
   (input, config) => {
-    return new import_langgraph2.Command({
+    return new import_langgraph3.Command({
       update: {
         todos: input.todos,
         messages: [
@@ -1004,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")
     })
@@ -1016,19 +1031,19 @@ var writeTodos = (0, import_tools2.tool)(
 );
 var ls = (0, import_tools2.tool)(
   () => {
-    const state = (0, import_langgraph2.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_langgraph2.getCurrentTaskInput)();
+    const state = (0, import_langgraph3.getCurrentTaskInput)();
     const mockFilesystem = state.files || {};
     const { file_path, offset = 0, limit = 2e3 } = input;
     if (!(file_path in mockFilesystem)) {
@@ -1058,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_langgraph2.getCurrentTaskInput)();
+    const state = (0, import_langgraph3.getCurrentTaskInput)();
     const files = { ...state.files || {} };
     files[input.file_path] = input.content;
-    return new import_langgraph2.Command({
+    return new import_langgraph3.Command({
       update: {
         files,
         messages: [
@@ -1085,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_langgraph2.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)) {
@@ -1129,7 +1144,7 @@ var editFile = (0, import_tools2.tool)(
       newContent = content.replace(old_string, new_string);
     }
     mockFilesystem[file_path] = newContent;
-    return new import_langgraph2.Command({
+    return new import_langgraph3.Command({
       update: {
         files: mockFilesystem,
         messages: [
@@ -1144,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")
     })
   }
 );
@@ -1208,7 +1223,7 @@ function createTaskTool(inputs) {
         ).join(", ")}`;
       }
       try {
-        const currentState = (0, import_langgraph3.getCurrentTaskInput)();
+        const currentState = (0, import_langgraph4.getCurrentTaskInput)();
         const modifiedState = {
           ...currentState,
           messages: [
@@ -1219,7 +1234,7 @@ function createTaskTool(inputs) {
           ]
         };
         const result = await reactAgent.invoke(modifiedState, config);
-        return new import_langgraph3.Command({
+        return new import_langgraph4.Command({
           update: {
             files: result.files || {},
             messages: [
@@ -1231,11 +1246,11 @@ function createTaskTool(inputs) {
           }
         });
       } catch (error) {
-        if (error instanceof import_langgraph3.GraphInterrupt) {
+        if (error instanceof import_langgraph4.GraphInterrupt) {
           throw error;
         }
         const errorMessage = error instanceof Error ? error.message : String(error);
-        return new import_langgraph3.Command({
+        return new import_langgraph4.Command({
           update: {
             messages: [
               new import_messages2.ToolMessage({
@@ -1253,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(", ")}`
         )
       })
@@ -1264,10 +1279,10 @@ function createTaskTool(inputs) {
 }
 
 // src/deep_agent/state.ts
-var import_zod5 = require("@langchain/langgraph/zod");
-var import_langgraph4 = 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 || {};
@@ -1283,16 +1298,16 @@ function todoReducer(left, right) {
   }
   return left || [];
 }
-var DeepAgentState = import_langgraph4.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
     }
   })
@@ -1384,9 +1399,9 @@ var DeepAgentGraphBuilder = class {
 };
 
 // src/createPlanExecuteAgent.ts
-var import_langgraph6 = 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"));
@@ -1584,10 +1599,10 @@ var getLastHumanMessageData = (messages) => {
 var import_tools7 = require("@langchain/core/tools");
 
 // src/util/PGMemory.ts
-var import_langgraph5 = 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 memory2 = new import_langgraph5.MemorySaver();
+var memory2 = new import_langgraph6.MemorySaver();
 var _MemoryManager = class _MemoryManager {
   static getInstance() {
     return _MemoryManager.instance;
@@ -1598,40 +1613,40 @@ var MemoryManager = _MemoryManager;
 
 // src/createPlanExecuteAgent.ts
 var import_prebuilt4 = require("@langchain/langgraph/prebuilt");
-var PlanExecuteState = import_langgraph6.Annotation.Root({
+var PlanExecuteState = import_langgraph7.Annotation.Root({
   // 输入
-  input: (0, import_langgraph6.Annotation)({
+  input: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => y ?? x ?? ""
   }),
   // 计划步骤列表
-  plan: (0, import_langgraph6.Annotation)({
+  plan: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => y ?? x ?? []
   }),
   // 已执行的步骤 [步骤名称, 执行结果]
-  pastSteps: (0, import_langgraph6.Annotation)({
+  pastSteps: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => x.concat(y),
     default: () => []
   }),
   // 最终响应
-  response: (0, import_langgraph6.Annotation)({
+  response: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => y ?? x
   }),
   // 错误信息
-  error: (0, import_langgraph6.Annotation)({
+  error: (0, import_langgraph7.Annotation)({
     reducer: (x, y) => y ?? x
   }),
   // 继承基础状态
-  "x-tenant-id": (0, import_langgraph6.Annotation)(),
-  messages: (0, import_langgraph6.Annotation)({
-    reducer: import_langgraph6.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 {
@@ -1723,7 +1738,7 @@ ${executorTools.map((tool5) => tool5.name).join("\n")}
         error: void 0
       };
     } catch (error) {
-      if (error instanceof import_langgraph6.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"}`;
@@ -1816,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_langgraph6.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"}`;
@@ -1848,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_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,
+  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({
@@ -2029,7 +2044,8 @@ var AgentParamsBuilder = class {
       tools,
       model,
       subAgents,
-      prompt: agentLattice.config.prompt
+      prompt: agentLattice.config.prompt,
+      stateSchema: agentLattice.config.schema
     };
   }
 };