@axiom-lattice/core 2.1.22 → 2.1.23

package/dist/index.js

@@ -5331,7 +5331,7 @@ function returnCommandWithStateUpdate(result, toolCallId) {
       ...stateUpdate,
       messages: [
         new import_langchain37.ToolMessage({
-          content: lastMessage?.content || "Task completed",
+          content: lastMessage?.content || "Task Failed to complete",
           tool_call_id: toolCallId,
           name: "task"
         })
@@ -5412,28 +5412,42 @@ function createTaskTool(options) {
   return (0, import_langchain37.tool)(
     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 = (0, import_langgraph6.getCurrentTaskInput)();
-      const subagentState = filterStateForSubagent(currentState);
-      subagentState.messages = [new import_messages.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 = (0, import_langgraph6.getCurrentTaskInput)();
+        const subagentState = filterStateForSubagent(currentState);
+        subagentState.messages = [new import_messages.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 import_langgraph6.Command({
+          update: {
+            messages: [
+              new import_langchain37.ToolMessage({
+                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",