@axiom-lattice/core 2.1.21 → 2.1.23

package/dist/index.mjs

@@ -5232,7 +5232,7 @@ function returnCommandWithStateUpdate(result, toolCallId) {
       ...stateUpdate,
       messages: [
         new ToolMessage2({
-          content: lastMessage?.content || "Task completed",
+          content: lastMessage?.content || "Task Failed to complete",
           tool_call_id: toolCallId,
           name: "task"
         })
@@ -5313,28 +5313,42 @@ function createTaskTool(options) {
   return tool34(
     async (input, config) => {
       const { description, subagent_type } = input;
-      if (!(subagent_type in subagentGraphs)) {
-        const allowedTypes = Object.keys(subagentGraphs).map((k) => `\`${k}\``).join(", ");
-        throw new Error(
-          `Error: invoked agent of type ${subagent_type}, the only allowed types are ${allowedTypes}`
-        );
-      }
-      const subagent = subagentGraphs[subagent_type];
-      const currentState = getCurrentTaskInput2();
-      const subagentState = filterStateForSubagent(currentState);
-      subagentState.messages = [new HumanMessage({ content: description })];
-      const subagent_thread_id = config.configurable?.thread_id + "____" + subagent_type + "_" + config.toolCall.id;
-      console.log(subagent_thread_id);
-      const workerResult = await agentWorkerGraph.invoke({
-        assistant_id: subagent_type,
-        thread_id: subagent_thread_id,
-        input: { ...subagentState, message: description }
-      });
-      const result = workerResult.finalState?.values;
-      if (!config.toolCall?.id) {
-        throw new Error("Tool call ID is required for subagent invocation");
+      try {
+        if (!(subagent_type in subagentGraphs)) {
+          const allowedTypes = Object.keys(subagentGraphs).map((k) => `\`${k}\``).join(", ");
+          throw new Error(
+            `Error: invoked agent of type ${subagent_type}, the only allowed types are ${allowedTypes}`
+          );
+        }
+        const subagent = subagentGraphs[subagent_type];
+        const currentState = getCurrentTaskInput2();
+        const subagentState = filterStateForSubagent(currentState);
+        subagentState.messages = [new HumanMessage({ content: description })];
+        const subagent_thread_id = config.configurable?.thread_id + "____" + subagent_type + "_" + config.toolCall.id;
+        console.log(subagent_thread_id);
+        const workerResult = await agentWorkerGraph.invoke({
+          assistant_id: subagent_type,
+          thread_id: subagent_thread_id,
+          input: { ...subagentState, message: description }
+        });
+        const result = workerResult.finalState?.values;
+        if (!config.toolCall?.id) {
+          throw new Error("Tool call ID is required for subagent invocation");
+        }
+        return returnCommandWithStateUpdate(result, config.toolCall.id);
+      } catch (error) {
+        return new Command2({
+          update: {
+            messages: [
+              new ToolMessage2({
+                content: error instanceof Error ? error.message : "Task Failed to complete",
+                tool_call_id: config.toolCall.id,
+                name: "task"
+              })
+            ]
+          }
+        });
       }
-      return returnCommandWithStateUpdate(result, config.toolCall.id);
     },
     {
       name: "task",
@@ -5999,6 +6013,26 @@ var AgentParamsBuilder = class {
   }
 };
 
+// src/agent_lattice/utils/mergeAgentConfig.ts
+var NON_INHERITABLE_FIELDS = /* @__PURE__ */ new Set([
+  "key",
+  "extendsAgent"
+]);
+function mergeAgentConfig(parent, child) {
+  const merged = { ...parent };
+  for (const [key, value] of Object.entries(child)) {
+    if (NON_INHERITABLE_FIELDS.has(key)) {
+      continue;
+    }
+    if (value !== void 0) {
+      merged[key] = value;
+    }
+  }
+  merged.key = child.key;
+  delete merged.extendsAgent;
+  return merged;
+}
+
 // src/agent_lattice/AgentLatticeManager.ts
 function assistantToConfig(assistant) {
   const graphDef = typeof assistant.graphDefinition === "object" && assistant.graphDefinition !== null ? assistant.graphDefinition : {};
@@ -6154,6 +6188,39 @@ var AgentLatticeManager = class _AgentLatticeManager extends BaseLatticeManager
       return false;
     }
   }
+  /**
+   * Resolve inherited config by walking the extendsAgent chain.
+   *
+   * If config.extendsAgent is set, fetches the parent config, recursively
+   * resolves the parent's own inheritance, then merges parent with child
+   * (child's explicit fields override parent's).
+   *
+   * @param config - The agent config to resolve
+   * @param visited - Set of visited keys to detect circular inheritance
+   * @returns A fully resolved AgentConfig with inheritance applied
+   */
+  resolveInheritedConfig(config, visited = /* @__PURE__ */ new Set()) {
+    if (!config.extendsAgent) {
+      return config;
+    }
+    if (visited.has(config.key)) {
+      throw new Error(
+        `Circular agent inheritance detected: "${config.key}" already visited in chain [${[...visited].join(" -> ")}]`
+      );
+    }
+    visited.add(config.key);
+    const parentLattice = this.getAgentLattice(config.extendsAgent);
+    if (!parentLattice) {
+      throw new Error(
+        `Parent agent "${config.extendsAgent}" not found when resolving inheritance for "${config.key}"`
+      );
+    }
+    const resolvedParent = this.resolveInheritedConfig(
+      parentLattice.config,
+      visited
+    );
+    return mergeAgentConfig(resolvedParent, config);
+  }
   /**
    * 构建Agent参数
    *
@@ -6176,10 +6243,15 @@ var AgentLatticeManager = class _AgentLatticeManager extends BaseLatticeManager
    * @returns AgentClient instance
    */
   createAgentClientFromConfig(agentLattice, options) {
+    const resolvedConfig = this.resolveInheritedConfig(agentLattice.config);
+    const resolvedLattice = {
+      ...agentLattice,
+      config: resolvedConfig
+    };
     const factory = AgentGraphBuilderFactory.getInstance();
-    const builder = factory.getBuilder(agentLattice.config.type);
-    const params = this.buildAgentParams(agentLattice, options);
-    return builder.build(agentLattice, params);
+    const builder = factory.getBuilder(resolvedConfig.type);
+    const params = this.buildAgentParams(resolvedLattice, options);
+    return builder.build(resolvedLattice, params);
   }
   /**
    * 初始化Agent客户端