@minded-ai/mindedjs 2.0.13 → 2.0.14-beta-1

package/dist/utils/extractStateMemoryResponse.js

@@ -0,0 +1,91 @@
+"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;
+        }
+    }
+    catch (err) {
+        logger_1.logger.error({ message: 'Error parsing tool state response', err });
+    }
+    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.id[2];
+            const content = messageObj.kwargs.content || '';
+            const id = messageObj.kwargs.id;
+            const additionalKwargs = messageObj.additional_kwargs || {};
+            switch (messageType) {
+                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 'HumanMessage':
+                    return new messages_1.HumanMessage({
+                        content,
+                        id,
+                        additional_kwargs: additionalKwargs,
+                    });
+                case 'SystemMessage':
+                    return new messages_1.SystemMessage({
+                        content,
+                        id,
+                        additional_kwargs: additionalKwargs,
+                    });
+                case 'ToolMessage':
+                    return new messages_1.ToolMessage({
+                        content,
+                        id,
+                        additional_kwargs: additionalKwargs,
+                        tool_call_id: messageObj.tool_call_id || '',
+                    });
+                default:
+                    // Default to HumanMessage for unknown types
+                    logger_1.logger.warn({
+                        msg: 'Unknown message type, defaulting to HumanMessage',
+                        messageType,
+                        messageObj,
+                    });
+                    return null;
+            }
+        }
+        catch (err) {
+            logger_1.logger.error({
+                msg: 'Error reconstructing message',
+                err,
+                messageObj,
+            });
+            // Return a fallback HumanMessage in case of error
+            return new messages_1.HumanMessage({
+                content: messageObj.content || 'Error reconstructing message',
+                id: messageObj.id,
+            });
+        }
+    })
+        .filter(Boolean);
+};
+exports.default = extractToolStateResponse;
+//# sourceMappingURL=extractStateMemoryResponse.js.map

package/dist/utils/extractStateMemoryResponse.js.map

@@ -0,0 +1 @@
+{"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,GAAG,EAAE,CAAC;QACb,eAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC;IACtE,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;SACZ,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;QAClB,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,EAAE,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;YAChD,MAAM,EAAE,GAAG,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAChC,MAAM,gBAAgB,GAAG,UAAU,CAAC,iBAAiB,IAAI,EAAE,CAAC;YAE5D,QAAQ,WAAW,EAAE,CAAC;gBACpB,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;gBACL,KAAK,cAAc;oBACjB,OAAO,IAAI,uBAAY,CAAC;wBACtB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;qBACpC,CAAC,CAAC;gBACL,KAAK,eAAe;oBAClB,OAAO,IAAI,wBAAa,CAAC;wBACvB,OAAO;wBACP,EAAE;wBACF,iBAAiB,EAAE,gBAAgB;qBACpC,CAAC,CAAC;gBACL,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,4CAA4C;oBAC5C,eAAM,CAAC,IAAI,CAAC;wBACV,GAAG,EAAE,kDAAkD;wBACvD,WAAW;wBACX,UAAU;qBACX,CAAC,CAAC;oBACH,OAAO,IAAI,CAAC;YAChB,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,eAAM,CAAC,KAAK,CAAC;gBACX,GAAG,EAAE,8BAA8B;gBACnC,GAAG;gBACH,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;SACD,MAAM,CAAC,OAAO,CAAkB,CAAC;AACtC,CAAC,CAAC;AAEF,kBAAe,wBAAwB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@minded-ai/mindedjs",
-  "version": "2.0.13",
+  "version": "2.0.14-beta-1",
   "description": "MindedJS is a TypeScript library for building agents.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -70,4 +70,4 @@
   "peerDependencies": {
     "playwright": "^1.55.0"
   }
-}
+}

package/src/toolsLibrary/classifier.ts

@@ -35,6 +35,34 @@ const DEFAULT_CONFIG: Partial<ClassifierConfig> = {
   defaultReason: 'Unable to determine classification',
 };
 
+/**
+ * Validates the classification result
+ * @param result The classification result to validate
+ * @param config The classifier configuration
+ * @returns True if the result is valid, false otherwise
+ */
+function validateClassificationResult(result: ClassificationResult, config: ClassifierConfig): boolean {
+  // Check if result exists and has a class property
+  if (!result || !result.class) {
+    return false;
+  }
+
+  // Check if the class is one of the configured classes
+  const validClasses = config.classes.map((c) => c.name);
+  if (!validClasses.includes(result.class)) {
+    return false;
+  }
+
+  // If we expect JSON format with confidence, validate it
+  if (config.outputFormat === 'json' && result.confidence !== undefined) {
+    if (typeof result.confidence !== 'number' || result.confidence < 0 || result.confidence > 1) {
+      return false;
+    }
+  }
+
+  return true;
+}
+
 /**
  * Generic classifier utility that can be used standalone
  * @param content The content to classify
@@ -44,48 +72,98 @@ const DEFAULT_CONFIG: Partial<ClassifierConfig> = {
  */
 export async function classify(content: string, config: ClassifierConfig, llm: BaseLanguageModel): Promise<ClassificationResult> {
   const mergedConfig = { ...DEFAULT_CONFIG, ...config };
+  const MAX_RETRIES = 3;
+  let lastError: Error | null = null;
+  let lastInvalidResult: ClassificationResult | null = null;
 
-  try {
-    // Build the classification prompt
-    const classesDescription = mergedConfig.classes.map((c) => `${c.name}: ${c.description}`).join('\n');
+  for (let attempt = 1; attempt <= MAX_RETRIES; attempt++) {
+    try {
+      // Build the classification prompt
+      const classesDescription = mergedConfig.classes.map((c) => `${c.name}: ${c.description}`).join('\n');
+      const validClassNames = mergedConfig.classes.map((c) => c.name);
+
+      const basePrompt =
+        mergedConfig.systemPrompt ||
+        'You are a classifier. Your task is to classify the given content into one of the following categories:';
+
+      let prompt = `${basePrompt}\n\n${classesDescription}\n\n`;
+
+      // Add retry feedback if this is not the first attempt
+      if (attempt > 1 && lastInvalidResult) {
+        prompt += `\n⚠️ IMPORTANT: Your previous classification attempt was INVALID. `;
+        if (lastInvalidResult.class && !validClassNames.includes(lastInvalidResult.class)) {
+          prompt += `You selected "${lastInvalidResult.class}" which is NOT in the list of valid classes. `;
+        } else if (!lastInvalidResult.class) {
+          prompt += `You did not provide a class value. `;
+        }
+        prompt += `You MUST select ONLY from these valid classes:\n${validClassNames.join(', ')}\n\n`;
+      }
+
+      if (mergedConfig.outputFormat === 'json') {
+        prompt += `You should output the result in the following JSON format: {
+          "class": "<selected class name>",
+          ${mergedConfig.includeReason ? '"reason": "<explanation for the classification>",' : ''}
+          "confidence": <confidence score between 0 and 1>
+        }.\nReturn JSON and nothing more.`;
+      } else {
+        prompt += 'Return only the class name.';
+      }
+
+      prompt += `\n\nContent to classify:\n${content}`;
+
+      // Make the classification request
+      let result: ClassificationResult;
+      if (mergedConfig.outputFormat === 'json') {
+        const parser = new JsonOutputParser();
+        const parsedResult = await llm.pipe(parser).invoke([new SystemMessage(prompt)]);
+        result = parsedResult as ClassificationResult;
+      } else {
+        const llmResult = await llm.invoke([new SystemMessage(prompt)]);
+        const classText = typeof llmResult.content === 'string' ? llmResult.content.trim() : '';
+        result = { class: classText };
+      }
+
+      // Validate the result
+      if (!validateClassificationResult(result, mergedConfig)) {
+        lastInvalidResult = result;
+        throw new Error(`Invalid classification result: ${JSON.stringify(result)}`);
+      }
+
+      // Success - return the valid result
+      return result;
+    } catch (err) {
+      lastError = err as Error;
+      logger.warn({
+        message: `Classification attempt ${attempt} failed`,
+        attempt,
+        maxRetries: MAX_RETRIES,
+        error: lastError.message,
+        invalidResult: lastInvalidResult,
+      });
 
-    const basePrompt =
-      mergedConfig.systemPrompt || 'You are a classifier. Your task is to classify the given content into one of the following categories:';
-
-    let prompt = `${basePrompt}\n\n${classesDescription}\n\n`;
-
-    if (mergedConfig.outputFormat === 'json') {
-      prompt += `You should output the result in the following JSON format: {
-        "class": "<selected class name>",
-        ${mergedConfig.includeReason ? '"reason": "<explanation for the classification>",' : ''}
-        "confidence": <confidence score between 0 and 1>
-      }.\nReturn JSON and nothing more.`;
-    } else {
-      prompt += 'Return only the class name.';
+      // If this is not the last attempt, continue to retry
+      if (attempt < MAX_RETRIES) {
+        continue;
+      }
     }
+  }
 
-    prompt += `\n\nContent to classify:\n${content}`;
-
-    // Make the classification request
-    if (mergedConfig.outputFormat === 'json') {
-      const parser = new JsonOutputParser();
-      const result = await llm.pipe(parser).invoke([new SystemMessage(prompt)]);
-      return result as ClassificationResult;
-    } else {
-      const result = await llm.invoke([new SystemMessage(prompt)]);
-      const classText = typeof result.content === 'string' ? result.content.trim() : '';
-      return { class: classText };
-    }
-  } catch (err) {
-    logger.error({ message: 'Classification failed, using default', err });
+  // All retries exhausted - fallback to default
+  logger.warn({
+    message: 'Classification failed after max retries, using default class',
+    maxRetries: MAX_RETRIES,
+    lastError: lastError?.message,
+    lastInvalidResult,
+    defaultClass: mergedConfig.defaultClass,
+    defaultReason: mergedConfig.defaultReason,
+  });
 
-    // Return default classification on error
-    return {
-      class: mergedConfig.defaultClass || 'unknown',
-      reason: mergedConfig.defaultReason || (err as Error).message,
-      confidence: 0,
-    };
-  }
+  // Return default classification
+  return {
+    class: mergedConfig.defaultClass || 'unknown',
+    reason: mergedConfig.defaultReason || `Classification failed after ${MAX_RETRIES} attempts: ${lastError?.message}`,
+    confidence: 0,
+  };
 }
 
 /**