@axiom-lattice/core 2.0.0 → 2.0.2

package/dist/index.d.mts

@@ -299,7 +299,7 @@ declare const validateToolInput: (key: string, input: any) => boolean;
 type AgentClient = CompiledStateGraph<any, any, any, any, any>;
 interface AgentLattice {
     config: AgentConfig;
-    client: AgentClient | undefined;
+    client?: AgentClient | undefined;
 }
 
 /**
@@ -363,6 +363,14 @@ declare class AgentLatticeManager extends BaseLatticeManager<AgentLattice> {
      * @returns 返回Agent构建参数
      */
     private buildAgentParams;
+    /**
+     * Create an AgentClient from AgentLattice config
+     *
+     * @param agentLattice Agent Lattice object
+     * @param options Build options
+     * @returns AgentClient instance
+     */
+    createAgentClientFromConfig(agentLattice: AgentLattice, options?: GraphBuildOptions): AgentClient;
     /**
      * 初始化Agent客户端
      *

package/dist/index.d.ts

@@ -299,7 +299,7 @@ declare const validateToolInput: (key: string, input: any) => boolean;
 type AgentClient = CompiledStateGraph<any, any, any, any, any>;
 interface AgentLattice {
     config: AgentConfig;
-    client: AgentClient | undefined;
+    client?: AgentClient | undefined;
 }
 
 /**
@@ -363,6 +363,14 @@ declare class AgentLatticeManager extends BaseLatticeManager<AgentLattice> {
      * @returns 返回Agent构建参数
      */
     private buildAgentParams;
+    /**
+     * Create an AgentClient from AgentLattice config
+     *
+     * @param agentLattice Agent Lattice object
+     * @param options Build options
+     * @returns AgentClient instance
+     */
+    createAgentClientFromConfig(agentLattice: AgentLattice, options?: GraphBuildOptions): AgentClient;
     /**
      * 初始化Agent客户端
      *

package/dist/index.js

@@ -44,7 +44,7 @@ __export(index_exports, {
   ThreadStatus: () => ThreadStatus,
   ToolLatticeManager: () => ToolLatticeManager,
   agentLatticeManager: () => agentLatticeManager,
-  getAgentClient: () => getAgentClient,
+  getAgentClient: () => getAgentClient2,
   getAgentConfig: () => getAgentConfig,
   getAgentLattice: () => getAgentLattice,
   getAllAgentConfigs: () => getAllAgentConfigs,
@@ -627,7 +627,7 @@ registerToolLattice(
   {
     name: "internet_search",
     description: "Run a web search",
-    needUserApprove: true,
+    needUserApprove: false,
     schema: import_zod2.default.object({
       query: import_zod2.default.string().describe("The search query"),
       maxResults: import_zod2.default.number().optional().default(5).describe("Maximum number of results to return"),
@@ -783,7 +783,6 @@ var ReActAgentGraphBuilder = class {
 var import_tools3 = require("@langchain/core/tools");
 var import_messages2 = require("@langchain/core/messages");
 var import_langgraph5 = require("@langchain/langgraph");
-var import_prebuilt2 = require("@langchain/langgraph/prebuilt");
 var import_zod6 = require("zod");
 
 // src/deep_agent/tools.ts
@@ -1215,6 +1214,7 @@ var editFile = (0, import_tools2.tool)(
 );
 
 // src/deep_agent/subAgent.ts
+var import_protocols3 = require("@axiom-lattice/protocols");
 var BUILTIN_TOOLS = {
   write_todos: writeTodos,
   read_file: readFile,
@@ -1236,30 +1236,28 @@ function createTaskTool(inputs) {
   const allTools = { ...BUILTIN_TOOLS, ...tools };
   const agentsMap = /* @__PURE__ */ new Map();
   for (const subagent of subagents) {
-    const subagentTools = [];
-    if (subagent.tools) {
-      for (const toolName of subagent.tools) {
-        const resolvedTool = allTools[toolName];
-        if (resolvedTool) {
-          subagentTools.push(resolvedTool);
-        } else {
-          console.warn(
-            `Warning: Tool '${toolName}' not found for agent '${subagent.name}'`
-          );
+    let reactAgent;
+    const agentLattice = getAgentLattice(subagent.key);
+    if (agentLattice) {
+      reactAgent = agentLattice.client;
+    } else {
+      const subagentTools = [];
+      if ((0, import_protocols3.hasTools)(subagent)) {
+        for (const toolName of (0, import_protocols3.getToolsFromConfig)(subagent)) {
+          const resolvedTool = allTools[toolName];
+          if (resolvedTool) {
+            subagentTools.push(resolvedTool);
+          } else {
+            console.warn(
+              `Warning: Tool '${toolName}' not found for agent '${subagent.name}'`
+            );
+          }
         }
+      } else {
+        subagentTools.push(...Object.values(allTools));
       }
-    } else {
-      subagentTools.push(...Object.values(allTools));
-    }
-    const reactAgent = (0, import_prebuilt2.createReactAgent)({
-      llm: model,
-      tools: subagentTools,
-      stateSchema,
-      messageModifier: subagent.prompt,
-      // checkpointer: false,
-      checkpointer: getCheckpointSaver("default"),
-      postModelHook
-    });
+      reactAgent = createAgentClientFromAgentLattice({ config: subagent });
+    }
     agentsMap.set(subagent.name, reactAgent);
   }
   return (0, import_tools3.tool)(
@@ -1363,7 +1361,7 @@ var DeepAgentState = import_langgraph6.MessagesZodState.extend({
 });
 
 // src/deep_agent/graph.ts
-var import_prebuilt3 = require("@langchain/langgraph/prebuilt");
+var import_prebuilt2 = require("@langchain/langgraph/prebuilt");
 var BASE_PROMPT = `You have access to a number of standard tools
 
 ## \`write_todos\`
@@ -1410,7 +1408,7 @@ function createDeepAgent(params = {}) {
     allTools.push(taskTool);
   }
   const finalInstructions = instructions ? instructions + BASE_PROMPT : BASE_PROMPT;
-  return (0, import_prebuilt3.createReactAgent)({
+  return (0, import_prebuilt2.createReactAgent)({
     llm: model,
     tools: allTools,
     stateSchema,
@@ -1436,13 +1434,9 @@ var DeepAgentGraphBuilder = class {
     return createDeepAgent({
       tools,
       model: params.model,
+      stateSchema: params.stateSchema,
       instructions: params.prompt,
-      subagents: params.subAgents.map((sa) => ({
-        name: sa.key,
-        description: sa.config.description,
-        prompt: sa.config.prompt,
-        tools: sa.config.tools || []
-      }))
+      subagents: params.subAgents.map((sa) => sa.config)
     });
   }
 };
@@ -1661,7 +1655,7 @@ _MemoryManager.instance = memory2;
 var MemoryManager = _MemoryManager;
 
 // src/createPlanExecuteAgent.ts
-var import_prebuilt4 = require("@langchain/langgraph/prebuilt");
+var import_prebuilt3 = require("@langchain/langgraph/prebuilt");
 var PlanExecuteState = import_langgraph8.Annotation.Root({
   // 输入
   input: (0, import_langgraph8.Annotation)({
@@ -1718,7 +1712,7 @@ function createPlanExecuteAgent(config) {
     maxSteps,
     toolsCount: executorTools.length
   });
-  const agentExecutor = (0, import_prebuilt4.createReactAgent)({
+  const agentExecutor = (0, import_prebuilt3.createReactAgent)({
     tools: executorTools,
     llm
   });
@@ -1820,7 +1814,7 @@ ${executorTools.map((tool5) => tool5.name).join("\n")}
       returnDirect: true
     }
   );
-  const replanAgent = (0, import_prebuilt4.createReactAgent)({
+  const replanAgent = (0, import_prebuilt3.createReactAgent)({
     tools: [responseTool, planTool],
     llm
   });
@@ -2043,6 +2037,7 @@ var AgentGraphBuilderFactory = class _AgentGraphBuilderFactory {
 };
 
 // src/agent_lattice/builders/AgentParamsBuilder.ts
+var import_protocols4 = require("@axiom-lattice/protocols");
 var AgentParamsBuilder = class {
   /**
    * constructor
@@ -2060,7 +2055,7 @@ var AgentParamsBuilder = class {
    * @returns Agent build parameters
    */
   buildParams(agentLattice, options) {
-    const toolKeys = options?.overrideTools || agentLattice.config.tools || [];
+    const toolKeys = options?.overrideTools || (0, import_protocols4.getToolsFromConfig)(agentLattice.config);
     const tools = toolKeys.map((toolKey) => {
       const toolLattice = toolLatticeManager.getToolLattice(toolKey);
       if (!toolLattice) {
@@ -2077,7 +2072,7 @@ var AgentParamsBuilder = class {
     if (!model) {
       throw new Error(`Model "${modelKey}" does not exist`);
     }
-    const subAgentKeys = agentLattice.config.subAgents || [];
+    const subAgentKeys = (0, import_protocols4.getSubAgentsFromConfig)(agentLattice.config);
     const subAgents = subAgentKeys.map((agentKey) => {
       const subAgentLattice = this.getAgentLatticeFunc(agentKey);
       if (!subAgentLattice) {
@@ -2089,10 +2084,17 @@ var AgentParamsBuilder = class {
         client: subAgentLattice.client
       };
     });
+    let internalSubAgents = [];
+    if ((0, import_protocols4.isDeepAgentConfig)(agentLattice.config)) {
+      internalSubAgents = agentLattice.config.internalSubAgents?.map((i) => ({
+        key: i.key,
+        config: i
+      })) || [];
+    }
     return {
       tools,
       model,
-      subAgents,
+      subAgents: [...subAgents, ...internalSubAgents],
       prompt: agentLattice.config.prompt,
       stateSchema: agentLattice.config.schema
     };
@@ -2195,11 +2197,25 @@ var AgentLatticeManager = class _AgentLatticeManager extends BaseLatticeManager
    * @returns 返回Agent构建参数
    */
   buildAgentParams(agentLattice, options) {
-    const paramsBuilder = new AgentParamsBuilder(
-      (key) => this.getAgentLattice(key)
-    );
+    const paramsBuilder = new AgentParamsBuilder((key) => {
+      this.initializeClient(key);
+      return this.getAgentLattice(key);
+    });
     return paramsBuilder.buildParams(agentLattice, options);
   }
+  /**
+   * Create an AgentClient from AgentLattice config
+   *
+   * @param agentLattice Agent Lattice object
+   * @param options Build options
+   * @returns AgentClient instance
+   */
+  createAgentClientFromConfig(agentLattice, options) {
+    const factory = AgentGraphBuilderFactory.getInstance();
+    const builder = factory.getBuilder(agentLattice.config.type);
+    const params = this.buildAgentParams(agentLattice, options);
+    return builder.build(agentLattice, params);
+  }
   /**
    * 初始化Agent客户端
    *
@@ -2217,10 +2233,7 @@ var AgentLatticeManager = class _AgentLatticeManager extends BaseLatticeManager
     if (agentLattice.client) {
       return agentLattice.client;
     }
-    const factory = AgentGraphBuilderFactory.getInstance();
-    const builder = factory.getBuilder(agentLattice.config.type);
-    const params = this.buildAgentParams(agentLattice, options);
-    const graph = builder.build(agentLattice, params);
+    const graph = this.createAgentClientFromConfig(agentLattice, options);
     agentLattice.client = graph;
     return graph;
   }
@@ -2238,7 +2251,8 @@ var getAgentLattice = (key) => agentLatticeManager.getAgentLattice(key);
 var getAgentConfig = (key) => agentLatticeManager.getAgentConfig(key);
 var getAllAgentConfigs = () => agentLatticeManager.getAllAgentConfigs();
 var validateAgentInput = (key, input) => agentLatticeManager.validateAgentInput(key, input);
-var getAgentClient = (key, options) => agentLatticeManager.initializeClient(key, options);
+var getAgentClient2 = (key, options) => agentLatticeManager.initializeClient(key, options);
+var createAgentClientFromAgentLattice = (agentLattice, options) => agentLatticeManager.createAgentClientFromConfig(agentLattice, options);
 
 // src/chunk_buffer_lattice/ChunkBuffer.ts
 var ChunkBuffer = class {