@minded-ai/mindedjs 1.0.122-beta.3 → 1.0.122-beta.4

package/dist/nodes/addToolRunNode.js

@@ -42,7 +42,7 @@ const addToolRunNode = async ({ graph, tools, toolNode, attachedToNodeName, agen
         const toolCallMessage = await tool.invoke(toolCallObj.tool_calls[0]);
         const toolStateUpdate = (0, extractStateMemoryResponse_1.default)(toolCallMessage);
         // Push the toolCallMessage into the messages array from toolStateUpdate
-        const updatedMessages = [toolCallMessage, ...(toolStateUpdate.messages || [])];
+        const updatedMessages = [toolCallMessage, ...((toolStateUpdate).messages || [])];
         // Return the full state update from the tool with the updated messages
         return {
             goto: null,

package/dist/nodes/addToolRunNode.js.map

@@ -1 +1 @@
-{"version":3,"file":"addToolRunNode.js","sourceRoot":"","sources":["../../src/nodes/addToolRunNode.ts"],"names":[],"mappings":";;;;;;AAIA,sDAA+E;AAC/E,iDAA8D;AAG9D,qGAA2E;AAE3E,4CAAyC;AACzC,8CAAqD;AAY9C,MAAM,oBAAoB,GAAG,CAAC,QAAgB,EAAE,EAAE,CAAC,GAAG,QAAQ,GAAG,iCAAmB,CAAC,QAAQ,EAAE,CAAC;AAA1F,QAAA,oBAAoB,wBAAsE;AAEhG,MAAM,cAAc,GAAG,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,kBAAkB,EAAE,KAAK,EAAwB,EAAE,EAAE;IAClH,MAAM,QAAQ,GAAiB,KAAK,EAAE,KAAmC,EAAE,EAAE;QAC3E,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC1E,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,mBAAmB,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,cAAc,GAAG,KAAK,EAAE,KAAwC,EAAE,EAAE;YACxE,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;gBACpE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC3B,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,4BAA4B,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,GAAG,SAAS,EAAE,CAAC,CAAC;gBAClH,OAAO,QAAQ,IAAI,EAAE,CAAC;YACxB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,6BAA6B,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;gBACxF,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC,CAAC;QACF,MAAM,IAAI,GAAG,IAAA,YAAa,EAAC,cAAc,EAAE;YACzC,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,WAAW,EAAE,WAAW,CAAC,WAAW;YACpC,MAAM,EAAE,WAAW,CAAC,KAAK;SAC1B,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAQ,CAAC;QACrE,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;QACD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QACrE,MAAM,eAAe,GAAG,IAAA,oCAAwB,EAAC,eAA8B,CAAC,CAAC;QAEjF,wEAAwE;QACxE,MAAM,eAAe,GAAG,CAAC,eAAe,EAAE,GAAG,CAAE,eAAuB,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC;QAExF,uEAAuE;QAEvE,OAAO;YACL,IAAI,EAAE,IAAI;YACV,GAAG,eAAe;YAClB,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,IAAA,2BAAiB,EAAc,KAAK,CAAC,OAAO,EAAE;gBACrD,IAAI,EAAE,sBAAQ,CAAC,IAAI;gBACnB,MAAM,EAAE,QAAQ,CAAC,IAAI;gBACrB,eAAe,EAAE,QAAQ,CAAC,WAAW;gBACrC,GAAG,EAAE,eAAe;gBACpB,UAAU,EAAE,EAAE;aACf,CAAC;SACH,CAAC;IACJ,CAAC,CAAC;IAEF,KAAK,CAAC,OAAO,CAAC,IAAA,4BAAoB,EAAC,kBAAkB,CAAC,EAAE,QAAQ,CAAC,CAAC;IAClE,KAAK,CAAC,OAAO,CAAC,kBAAyB,EAAE,IAAA,4BAAoB,EAAC,kBAAkB,CAAQ,CAAC,CAAC;AAC5F,CAAC,CAAC;AArDW,QAAA,cAAc,kBAqDzB"}
+{"version":3,"file":"addToolRunNode.js","sourceRoot":"","sources":["../../src/nodes/addToolRunNode.ts"],"names":[],"mappings":";;;;;;AAIA,sDAA+E;AAC/E,iDAA8D;AAG9D,qGAA2E;AAE3E,4CAAyC;AACzC,8CAAqD;AAY9C,MAAM,oBAAoB,GAAG,CAAC,QAAgB,EAAE,EAAE,CAAC,GAAG,QAAQ,GAAG,iCAAmB,CAAC,QAAQ,EAAE,CAAC;AAA1F,QAAA,oBAAoB,wBAAsE;AAEhG,MAAM,cAAc,GAAG,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,kBAAkB,EAAE,KAAK,EAAwB,EAAE,EAAE;IAClH,MAAM,QAAQ,GAAiB,KAAK,EAAE,KAAmC,EAAE,EAAE;QAC3E,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC1E,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,mBAAmB,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,cAAc,GAAG,KAAK,EAAE,KAAwC,EAAE,EAAE;YACxE,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;gBACpE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC3B,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,4BAA4B,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,GAAG,SAAS,EAAE,CAAC,CAAC;gBAClH,OAAO,QAAQ,IAAI,EAAE,CAAC;YACxB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,6BAA6B,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;gBACxF,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC,CAAC;QACF,MAAM,IAAI,GAAG,IAAA,YAAa,EAAC,cAAc,EAAE;YACzC,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,WAAW,EAAE,WAAW,CAAC,WAAW;YACpC,MAAM,EAAE,WAAW,CAAC,KAAK;SAC1B,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAQ,CAAC;QACrE,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;QACD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QACrE,MAAM,eAAe,GAAG,IAAA,oCAAwB,EAAC,eAA8B,CAAC,CAAC;QAEjF,wEAAwE;QACxE,MAAM,eAAe,GAAG,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,eAAe,CAAC,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC;QAEjF,uEAAuE;QAEvE,OAAO;YACL,IAAI,EAAE,IAAI;YACV,GAAG,eAAe;YAClB,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,IAAA,2BAAiB,EAAc,KAAK,CAAC,OAAO,EAAE;gBACrD,IAAI,EAAE,sBAAQ,CAAC,IAAI;gBACnB,MAAM,EAAE,QAAQ,CAAC,IAAI;gBACrB,eAAe,EAAE,QAAQ,CAAC,WAAW;gBACrC,GAAG,EAAE,eAAe;gBACpB,UAAU,EAAE,EAAE;aACf,CAAC;SACH,CAAC;IACJ,CAAC,CAAC;IAEF,KAAK,CAAC,OAAO,CAAC,IAAA,4BAAoB,EAAC,kBAAkB,CAAC,EAAE,QAAQ,CAAC,CAAC;IAClE,KAAK,CAAC,OAAO,CAAC,kBAAyB,EAAE,IAAA,4BAAoB,EAAC,kBAAkB,CAAQ,CAAC,CAAC;AAC5F,CAAC,CAAC;AArDW,QAAA,cAAc,kBAqDzB"}

package/dist/utils/extractStateMemoryResponse.d.ts

@@ -1,4 +1,5 @@
 import { ToolMessage } from '@langchain/core/messages';
-declare const extractToolStateResponse: <State>(toolMessage: ToolMessage) => Partial<State>;
+import { State } from '../types/LangGraph.types';
+declare const extractToolStateResponse: (toolMessage: ToolMessage) => Partial<State>;
 export default extractToolStateResponse;
 //# sourceMappingURL=extractStateMemoryResponse.d.ts.map

package/dist/utils/extractStateMemoryResponse.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"extractStateMemoryResponse.d.ts","sourceRoot":"","sources":["../../src/utils/extractStateMemoryResponse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAGvD,QAAA,MAAM,wBAAwB,GAAI,KAAK,EAAE,aAAa,WAAW,KAAG,OAAO,CAAC,KAAK,CAUhF,CAAC;AAEF,eAAe,wBAAwB,CAAC"}
+{"version":3,"file":"extractStateMemoryResponse.d.ts","sourceRoot":"","sources":["../../src/utils/extractStateMemoryResponse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAe,WAAW,EAA0C,MAAM,0BAA0B,CAAC;AAE5G,OAAO,EAAE,KAAK,EAAE,MAAM,0BAA0B,CAAC;AAEjD,QAAA,MAAM,wBAAwB,GAAI,aAAa,WAAW,KAAG,OAAO,CAAC,KAAK,CAazE,CAAC;AAuGF,eAAe,wBAAwB,CAAC"}

package/dist/utils/extractStateMemoryResponse.js

@@ -1,10 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+const messages_1 = require("@langchain/core/messages");
 const logger_1 = require("./logger");
 const extractToolStateResponse = (toolMessage) => {
     try {
         const parsed = JSON.parse(toolMessage.content);
         if (typeof parsed === 'object' && parsed !== null && 'state' in parsed) {
+            if (parsed.state.messages) {
+                parsed.state.messages = constructMessagesFromToolMessage(parsed.state.messages);
+            }
             return parsed.state;
         }
     }
@@ -13,5 +17,101 @@ const extractToolStateResponse = (toolMessage) => {
     }
     return {};
 };
+const constructMessagesFromToolMessage = (messages) => {
+    if (!Array.isArray(messages)) {
+        logger_1.logger.error({ msg: 'Expected messages to be an array', messages });
+        return [];
+    }
+    return messages.map((messageObj) => {
+        try {
+            // Handle case where message is already a BaseMessage instance
+            if (messageObj instanceof messages_1.BaseMessage) {
+                return messageObj;
+            }
+            // Handle case where message is a plain object that needs reconstruction
+            const messageType = messageObj._type || messageObj.type || messageObj.__class__ || 'unknown';
+            const content = messageObj.content || '';
+            const id = messageObj.id;
+            const additionalKwargs = messageObj.additional_kwargs || {};
+            switch (messageType) {
+                case 'ai':
+                case 'AIMessage':
+                    return new messages_1.AIMessage({
+                        content,
+                        id,
+                        additional_kwargs: additionalKwargs,
+                        tool_calls: messageObj.tool_calls || [],
+                        invalid_tool_calls: messageObj.invalid_tool_calls || [],
+                    });
+                case 'human':
+                case 'HumanMessage':
+                    return new messages_1.HumanMessage({
+                        content,
+                        id,
+                        additional_kwargs: additionalKwargs,
+                    });
+                case 'system':
+                case 'SystemMessage':
+                    return new messages_1.SystemMessage({
+                        content,
+                        id,
+                        additional_kwargs: additionalKwargs,
+                    });
+                case 'tool':
+                case 'ToolMessage':
+                    return new messages_1.ToolMessage({
+                        content,
+                        id,
+                        additional_kwargs: additionalKwargs,
+                        tool_call_id: messageObj.tool_call_id || '',
+                    });
+                default:
+                    // Fallback: try to determine type from message structure
+                    if (messageObj.tool_call_id) {
+                        return new messages_1.ToolMessage({
+                            content,
+                            id,
+                            additional_kwargs: additionalKwargs,
+                            tool_call_id: messageObj.tool_call_id,
+                        });
+                    }
+                    else if (messageObj.tool_calls || messageObj.invalid_tool_calls) {
+                        return new messages_1.AIMessage({
+                            content,
+                            id,
+                            additional_kwargs: additionalKwargs,
+                            tool_calls: messageObj.tool_calls || [],
+                            invalid_tool_calls: messageObj.invalid_tool_calls || [],
+                        });
+                    }
+                    else {
+                        // Default to HumanMessage for unknown types
+                        logger_1.logger.warn({
+                            msg: 'Unknown message type, defaulting to HumanMessage',
+                            messageType,
+                            messageObj
+                        });
+                        return new messages_1.HumanMessage({
+                            content,
+                            id,
+                            additional_kwargs: additionalKwargs,
+                        });
+                    }
+            }
+        }
+        catch (error) {
+            logger_1.logger.error({
+                msg: 'Error reconstructing message',
+                error,
+                messageObj
+            });
+            // Return a fallback HumanMessage in case of error
+            return new messages_1.HumanMessage({
+                content: messageObj.content || 'Error reconstructing message',
+                id: messageObj.id,
+            });
+        }
+    });
+};
 exports.default = extractToolStateResponse;
 //# sourceMappingURL=extractStateMemoryResponse.js.map

package/dist/utils/extractStateMemoryResponse.js.map

@@ -1 +1 @@
-{"version":3,"file":"extractStateMemoryResponse.js","sourceRoot":"","sources":["../../src/utils/extractStateMemoryResponse.ts"],"names":[],"mappings":";;AACA,qCAAkC;AAElC,MAAM,wBAAwB,GAAG,CAAQ,WAAwB,EAAkB,EAAE;IACnF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAiB,CAAC,CAAC;QACzD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;YACvE,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,mCAAmC,EAAE,KAAK,EAAE,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC;AAEF,kBAAe,wBAAwB,CAAC"}
+{"version":3,"file":"extractStateMemoryResponse.js","sourceRoot":"","sources":["../../src/utils/extractStateMemoryResponse.ts"],"names":[],"mappings":";;AAAA,uDAA4G;AAC5G,qCAAkC;AAGlC,MAAM,wBAAwB,GAAG,CAAC,WAAwB,EAAkB,EAAE;IAC5E,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAiB,CAAC,CAAC;QACzD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;YACvE,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC1B,MAAM,CAAC,KAAK,CAAC,QAAQ,GAAG,gCAAgC,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAClF,CAAC;YACD,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,mCAAmC,EAAE,KAAK,EAAE,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC;AAEF,MAAM,gCAAgC,GAAG,CAAC,QAAe,EAAiB,EAAE;IAC1E,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC7B,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,kCAAkC,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpE,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;QACjC,IAAI,CAAC;YACH,8DAA8D;YAC9D,IAAI,UAAU,YAAY,sBAAW,EAAE,CAAC;gBACtC,OAAO,UAAU,CAAC;YACpB,CAAC;YAED,wEAAwE;YACxE,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,IAAI,UAAU,CAAC,IAAI,IAAI,UAAU,CAAC,SAAS,IAAI,SAAS,CAAC;YAC7F,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,IAAI,EAAE,CAAC;YACzC,MAAM,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;YACzB,MAAM,gBAAgB,GAAG,UAAU,CAAC,iBAAiB,IAAI,EAAE,CAAC;YAE5D,QAAQ,WAAW,EAAE,CAAC;gBACpB,KAAK,IAAI,CAAC;gBACV,KAAK,WAAW;oBACd,OAAO,IAAI,oBAAS,CAAC;wBACnB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;wBACnC,UAAU,EAAE,UAAU,CAAC,UAAU,IAAI,EAAE;wBACvC,kBAAkB,EAAE,UAAU,CAAC,kBAAkB,IAAI,EAAE;qBACxD,CAAC,CAAC;gBAEL,KAAK,OAAO,CAAC;gBACb,KAAK,cAAc;oBACjB,OAAO,IAAI,uBAAY,CAAC;wBACtB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;qBACpC,CAAC,CAAC;gBAEL,KAAK,QAAQ,CAAC;gBACd,KAAK,eAAe;oBAClB,OAAO,IAAI,wBAAa,CAAC;wBACvB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;qBACpC,CAAC,CAAC;gBAEL,KAAK,MAAM,CAAC;gBACZ,KAAK,aAAa;oBAChB,OAAO,IAAI,sBAAW,CAAC;wBACrB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;wBACnC,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,EAAE;qBAC5C,CAAC,CAAC;gBAEL;oBACE,yDAAyD;oBACzD,IAAI,UAAU,CAAC,YAAY,EAAE,CAAC;wBAC5B,OAAO,IAAI,sBAAW,CAAC;4BACrB,OAAO;4BACP,EAAE;4BACF,iBAAiB,EAAE,gBAAgB;4BACnC,YAAY,EAAE,UAAU,CAAC,YAAY;yBACtC,CAAC,CAAC;oBACL,CAAC;yBAAM,IAAI,UAAU,CAAC,UAAU,IAAI,UAAU,CAAC,kBAAkB,EAAE,CAAC;wBAClE,OAAO,IAAI,oBAAS,CAAC;4BACnB,OAAO;4BACP,EAAE;4BACF,iBAAiB,EAAE,gBAAgB;4BACnC,UAAU,EAAE,UAAU,CAAC,UAAU,IAAI,EAAE;4BACvC,kBAAkB,EAAE,UAAU,CAAC,kBAAkB,IAAI,EAAE;yBACxD,CAAC,CAAC;oBACL,CAAC;yBAAM,CAAC;wBACN,4CAA4C;wBAC5C,eAAM,CAAC,IAAI,CAAC;4BACV,GAAG,EAAE,kDAAkD;4BACvD,WAAW;4BACX,UAAU;yBACX,CAAC,CAAC;wBACH,OAAO,IAAI,uBAAY,CAAC;4BACtB,OAAO;4BACP,EAAE;4BACF,iBAAiB,EAAE,gBAAgB;yBACpC,CAAC,CAAC;oBACL,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,KAAK,CAAC;gBACX,GAAG,EAAE,8BAA8B;gBACnC,KAAK;gBACL,UAAU;aACX,CAAC,CAAC;YACH,kDAAkD;YAClD,OAAO,IAAI,uBAAY,CAAC;gBACtB,OAAO,EAAE,UAAU,CAAC,OAAO,IAAI,8BAA8B;gBAC7D,EAAE,EAAE,UAAU,CAAC,EAAE;aAClB,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,kBAAe,wBAAwB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@minded-ai/mindedjs",
-  "version": "1.0.122-beta.3",
+  "version": "1.0.122-beta.4",
   "description": "MindedJS is a TypeScript library for building agents.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/nodes/addToolRunNode.ts

@@ -56,7 +56,7 @@ export const addToolRunNode = async ({ graph, tools, toolNode, attachedToNodeNam
     const toolStateUpdate = extractToolStateResponse(toolCallMessage as ToolMessage);
 
     // Push the toolCallMessage into the messages array from toolStateUpdate
-    const updatedMessages = [toolCallMessage, ...((toolStateUpdate as any).messages || [])];
+    const updatedMessages = [toolCallMessage, ...((toolStateUpdate).messages || [])];
 
     // Return the full state update from the tool with the updated messages
 

package/src/utils/extractStateMemoryResponse.ts

@@ -1,10 +1,14 @@
-import { ToolMessage } from '@langchain/core/messages';
+import { BaseMessage, ToolMessage, AIMessage, HumanMessage, SystemMessage } from '@langchain/core/messages';
 import { logger } from './logger';
+import { State } from '../types/LangGraph.types';
 
-const extractToolStateResponse = <State>(toolMessage: ToolMessage): Partial<State> => {
+const extractToolStateResponse = (toolMessage: ToolMessage): Partial<State> => {
   try {
     const parsed = JSON.parse(toolMessage.content as string);
     if (typeof parsed === 'object' && parsed !== null && 'state' in parsed) {
+      if (parsed.state.messages) {
+        parsed.state.messages = constructMessagesFromToolMessage(parsed.state.messages);
+      }
       return parsed.state;
     }
   } catch (error) {
@@ -13,4 +17,105 @@ const extractToolStateResponse = <State>(toolMessage: ToolMessage): Partial<Stat
   return {};
 };
 
+const constructMessagesFromToolMessage = (messages: any[]): BaseMessage[] => {
+  if (!Array.isArray(messages)) {
+    logger.error({ msg: 'Expected messages to be an array', messages });
+    return [];
+  }
+
+  return messages.map((messageObj) => {
+    try {
+      // Handle case where message is already a BaseMessage instance
+      if (messageObj instanceof BaseMessage) {
+        return messageObj;
+      }
+
+      // Handle case where message is a plain object that needs reconstruction
+      const messageType = messageObj._type || messageObj.type || messageObj.__class__ || 'unknown';
+      const content = messageObj.content || '';
+      const id = messageObj.id;
+      const additionalKwargs = messageObj.additional_kwargs || {};
+
+      switch (messageType) {
+        case 'ai':
+        case 'AIMessage':
+          return new AIMessage({
+            content,
+            id,
+            additional_kwargs: additionalKwargs,
+            tool_calls: messageObj.tool_calls || [],
+            invalid_tool_calls: messageObj.invalid_tool_calls || [],
+          });
+
+        case 'human':
+        case 'HumanMessage':
+          return new HumanMessage({
+            content,
+            id,
+            additional_kwargs: additionalKwargs,
+          });
+
+        case 'system':
+        case 'SystemMessage':
+          return new SystemMessage({
+            content,
+            id,
+            additional_kwargs: additionalKwargs,
+          });
+
+        case 'tool':
+        case 'ToolMessage':
+          return new ToolMessage({
+            content,
+            id,
+            additional_kwargs: additionalKwargs,
+            tool_call_id: messageObj.tool_call_id || '',
+          });
+
+        default:
+          // Fallback: try to determine type from message structure
+          if (messageObj.tool_call_id) {
+            return new ToolMessage({
+              content,
+              id,
+              additional_kwargs: additionalKwargs,
+              tool_call_id: messageObj.tool_call_id,
+            });
+          } else if (messageObj.tool_calls || messageObj.invalid_tool_calls) {
+            return new AIMessage({
+              content,
+              id,
+              additional_kwargs: additionalKwargs,
+              tool_calls: messageObj.tool_calls || [],
+              invalid_tool_calls: messageObj.invalid_tool_calls || [],
+            });
+          } else {
+            // Default to HumanMessage for unknown types
+            logger.warn({
+              msg: 'Unknown message type, defaulting to HumanMessage',
+              messageType,
+              messageObj
+            });
+            return new HumanMessage({
+              content,
+              id,
+              additional_kwargs: additionalKwargs,
+            });
+          }
+      }
+    } catch (error) {
+      logger.error({
+        msg: 'Error reconstructing message',
+        error,
+        messageObj
+      });
+      // Return a fallback HumanMessage in case of error
+      return new HumanMessage({
+        content: messageObj.content || 'Error reconstructing message',
+        id: messageObj.id,
+      });
+    }
+  });
+};
+
 export default extractToolStateResponse;