@minded-ai/mindedjs 1.0.150 → 2.0.0-beta1

package/src/utils/extractStateMemoryResponse.ts

@@ -1,94 +0,0 @@
-import { BaseMessage, ToolMessage, AIMessage, HumanMessage, SystemMessage } from '@langchain/core/messages';
-import { logger } from './logger';
-import { State } from '../types/LangGraph.types';
-
-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 (err) {
-    logger.error({ message: 'Error parsing tool state response', err });
-  }
-  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.id[2];
-        const content = messageObj.kwargs.content || '';
-        const id = messageObj.kwargs.id;
-        const additionalKwargs = messageObj.additional_kwargs || {};
-
-        switch (messageType) {
-          case 'AIMessage':
-            return new AIMessage({
-              content,
-              id,
-              additional_kwargs: additionalKwargs,
-              tool_calls: messageObj.tool_calls || [],
-              invalid_tool_calls: messageObj.invalid_tool_calls || [],
-            });
-          case 'HumanMessage':
-            return new HumanMessage({
-              content,
-              id,
-              additional_kwargs: additionalKwargs,
-            });
-          case 'SystemMessage':
-            return new SystemMessage({
-              content,
-              id,
-              additional_kwargs: additionalKwargs,
-            });
-          case 'ToolMessage':
-            return new ToolMessage({
-              content,
-              id,
-              additional_kwargs: additionalKwargs,
-              tool_call_id: messageObj.tool_call_id || '',
-            });
-
-          default:
-            // Default to HumanMessage for unknown types
-            logger.warn({
-              msg: 'Unknown message type, defaulting to HumanMessage',
-              messageType,
-              messageObj,
-            });
-            return null;
-        }
-      } catch (err) {
-        logger.error({
-          msg: 'Error reconstructing message',
-          err,
-          messageObj,
-        });
-        // Return a fallback HumanMessage in case of error
-        return new HumanMessage({
-          content: messageObj.content || 'Error reconstructing message',
-          id: messageObj.id,
-        });
-      }
-    })
-    .filter(Boolean) as BaseMessage[];
-};
-
-export default extractToolStateResponse;