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

package/dist/toolsLibrary/classifier.js

@@ -6,15 +6,25 @@ exports.createClassifier = createClassifier;
 exports.multiClassify = multiClassify;
 const zod_1 = require("zod");
 const logger_1 = require("../utils/logger");
-const output_parsers_1 = require("@langchain/core/output_parsers");
 const messages_1 = require("@langchain/core/messages");
-// Default configuration
-const DEFAULT_CONFIG = {
-    outputFormat: 'json',
-    includeReason: true,
-    defaultClass: 'unknown',
-    defaultReason: 'Unable to determine classification',
-};
+function supportsStructuredOutput(llm) {
+    return 'withStructuredOutput' in llm && typeof llm.withStructuredOutput === 'function';
+}
+/**
+ * Create a dynamic Zod schema for classification result based on valid classes
+ */
+function createClassificationSchema(config) {
+    const validClassNames = config.classes.map((c) => c.name);
+    // Create a union type for valid class names
+    const classEnum = zod_1.z.enum(validClassNames);
+    // Build the schema - always include all fields
+    const schemaShape = {
+        class: classEnum.describe('The selected classification category'),
+        reason: zod_1.z.string().describe('Explanation for the classification'),
+        confidence: zod_1.z.number().min(0).max(1).describe('Confidence score between 0 and 1'),
+    };
+    return zod_1.z.object(schemaShape);
+}
 /**
  * Generic classifier utility that can be used standalone
  * @param content The content to classify
@@ -23,49 +33,73 @@ const DEFAULT_CONFIG = {
  * @returns The classification result
  */
 async function classify(content, config, llm) {
-    const mergedConfig = { ...DEFAULT_CONFIG, ...config };
-    try {
-        // Build the classification prompt
-        const classesDescription = mergedConfig.classes.map((c) => `${c.name}: ${c.description}`).join('\n');
-        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.';
-        }
-        prompt += `\n\nContent to classify:\n${content}`;
-        // Make the classification request
-        if (mergedConfig.outputFormat === 'json') {
-            const parser = new output_parsers_1.JsonOutputParser();
-            const result = await llm.pipe(parser).invoke([new messages_1.SystemMessage(prompt)]);
+    const MAX_RETRIES = 3;
+    let lastError = null;
+    // Check if LLM supports structured output upfront
+    if (!supportsStructuredOutput(llm)) {
+        throw new Error('LLM does not support structured output, which is required for classification');
+    }
+    for (let attempt = 1; attempt <= MAX_RETRIES; attempt++) {
+        try {
+            // Build the classification prompt
+            const classesDescription = config.classes.map((c) => `${c.name}: ${c.description}`).join('\n');
+            const basePrompt = config.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) {
+                const validClassNames = config.classes.map((c) => c.name);
+                prompt += `\n⚠️ IMPORTANT: Your previous classification attempt failed. `;
+                prompt += `You MUST select ONLY from these valid classes:\n${validClassNames.join(', ')}\n\n`;
+            }
+            // Create dynamic schema based on valid classes
+            const classificationSchema = createClassificationSchema(config);
+            // Add content to classify to the prompt
+            prompt += `\n\nContent to classify:\n${content}`;
+            // Use structured output for guaranteed compliance
+            const structuredLLM = llm.withStructuredOutput(classificationSchema);
+            const messages = [new messages_1.SystemMessage(prompt)];
+            const result = (await structuredLLM.invoke(messages));
+            // The result is already validated by withStructuredOutput
+            logger_1.logger.debug({
+                message: 'Classification successful with structured output',
+                attempt,
+                class: result.class,
+            });
             return result;
         }
-        else {
-            const result = await llm.invoke([new messages_1.SystemMessage(prompt)]);
-            const classText = typeof result.content === 'string' ? result.content.trim() : '';
-            return { class: classText };
+        catch (err) {
+            lastError = err;
+            logger_1.logger.warn({
+                message: `Classification attempt ${attempt} failed`,
+                attempt,
+                maxRetries: MAX_RETRIES,
+                error: lastError.message,
+            });
+            // If this is not the last attempt, continue to retry
+            if (attempt < MAX_RETRIES) {
+                continue;
+            }
         }
     }
-    catch (err) {
-        logger_1.logger.error({ message: 'Classification failed, using default', err });
-        // Return default classification on error
-        return {
-            class: mergedConfig.defaultClass || 'unknown',
-            reason: mergedConfig.defaultReason || err.message,
-            confidence: 0,
-        };
-    }
+    // All retries exhausted - fallback to default
+    logger_1.logger.warn({
+        message: 'Classification failed after max retries, using default class',
+        maxRetries: MAX_RETRIES,
+        lastError: lastError === null || lastError === void 0 ? void 0 : lastError.message,
+        defaultClass: config.defaultClass,
+        defaultReason: config.defaultReason,
+    });
+    // Return default classification
+    return {
+        class: config.defaultClass,
+        reason: config.defaultReason,
+        confidence: 0,
+    };
 }
 /**
  * Create a classifier from a simple class list
  * @param classes Array of class names or [name, description] tuples
- * @param options Additional configuration options
+ * @param options Configuration options (includeReason, defaultClass, defaultReason are required)
  * @returns A configured classify function
  */
 function createClassifier(classes, options) {
@@ -96,19 +130,17 @@ exports.schema = zod_1.z.object({
         .optional()
         .describe('Classes to classify into. Can be strings, [name, description] tuples, or objects with name and description'),
     systemPrompt: zod_1.z.string().optional().describe('Custom system prompt for classification'),
-    includeReason: zod_1.z.boolean().optional().default(true).describe('Whether to include reasoning in the response'),
-    outputFormat: zod_1.z.enum(['json', 'text']).optional().default('json').describe('Output format for the classification'),
-    defaultClass: zod_1.z.string().optional().describe('Default class to use if classification fails'),
-    defaultReason: zod_1.z.string().optional().describe('Default reason to use if classification fails'),
+    defaultClass: zod_1.z.string().describe('Default class to use if classification fails'),
+    defaultReason: zod_1.z.string().describe('Default reason to use if classification fails'),
 });
 const classifierTool = {
     name: 'minded-classifier',
-    description: 'Classify content into predefined categories using AI. Supports custom classes, system prompts, and various output formats. Can be configured with default fallback values.',
+    description: 'Classify content into predefined categories using AI. Supports custom classes and system prompts. Uses structured output for guaranteed schema compliance. Requires default fallback values.',
     input: exports.schema,
     isGlobal: false,
     execute: async ({ input, state, agent }) => {
         var _a, _b;
-        const { content, classes, systemPrompt, includeReason, outputFormat, defaultClass, defaultReason } = input;
+        const { content, classes, systemPrompt, defaultClass, defaultReason } = input;
         logger_1.logger.info({
             message: 'Classifying content',
             sessionId: state.sessionId,
@@ -140,8 +172,6 @@ const classifierTool = {
         const config = {
             classes: classDefinitions,
             systemPrompt,
-            includeReason,
-            outputFormat,
             defaultClass,
             defaultReason,
         };
@@ -175,39 +205,41 @@ const classifierTool = {
 exports.default = classifierTool;
 // Export utility for multi-label classification
 async function multiClassify(content, config, llm) {
-    const mergedConfig = { ...DEFAULT_CONFIG, ...config };
-    const maxClasses = mergedConfig.maxClasses || 3;
+    const maxClasses = config.maxClasses || 3;
+    // Check if LLM supports structured output
+    if (!supportsStructuredOutput(llm)) {
+        throw new Error('LLM does not support structured output, which is required for multi-classification');
+    }
     try {
-        const classesDescription = mergedConfig.classes.map((c) => `${c.name}: ${c.description}`).join('\n');
-        const basePrompt = mergedConfig.systemPrompt || 'You are a multi-label classifier. Select all applicable categories for the given content:';
+        const classesDescription = config.classes.map((c) => `${c.name}: ${c.description}`).join('\n');
+        const validClassNames = config.classes.map((c) => c.name);
+        const basePrompt = config.systemPrompt || 'You are a multi-label classifier. Select all applicable categories for the given content:';
         const prompt = `${basePrompt}\n\n${classesDescription}\n\n
-You should output the result as a JSON array of up to ${maxClasses} classifications, ordered by relevance:
-[
-  {
-    "class": "<class name>",
-    ${mergedConfig.includeReason ? '"reason": "<explanation>",' : ''}
-    "confidence": <confidence score between 0 and 1>
-  },
-  ...
-]
-Return JSON and nothing more.
+Select up to ${maxClasses} classifications, ordered by relevance.
 
 Content to classify:
 ${content}`;
-        const parser = new output_parsers_1.JsonOutputParser();
-        const result = await llm.pipe(parser).invoke([new messages_1.SystemMessage(prompt)]);
-        if (Array.isArray(result)) {
-            return result;
-        }
-        // If single result returned, wrap in array
-        return [result];
+        // Create schema for array of classifications
+        const classEnum = zod_1.z.enum(validClassNames);
+        const multiClassSchema = zod_1.z
+            .array(zod_1.z.object({
+            class: classEnum.describe('The selected classification category'),
+            reason: zod_1.z.string().describe('Explanation for the classification'),
+            confidence: zod_1.z.number().min(0).max(1).describe('Confidence score between 0 and 1'),
+        }))
+            .min(1)
+            .max(maxClasses);
+        const structuredLLM = llm.withStructuredOutput(multiClassSchema);
+        const messages = [new messages_1.SystemMessage(prompt)];
+        const result = (await structuredLLM.invoke(messages));
+        return result;
     }
     catch (err) {
         logger_1.logger.error({ message: 'Multi-classification failed', err });
         return [
             {
-                class: mergedConfig.defaultClass || 'unknown',
-                reason: mergedConfig.defaultReason || err.message,
+                class: config.defaultClass,
+                reason: config.defaultReason,
                 confidence: 0,
             },
         ];

package/dist/toolsLibrary/classifier.js.map

@@ -1 +1 @@
-{"version":3,"file":"classifier.js","sourceRoot":"","sources":["../../src/toolsLibrary/classifier.ts"],"names":[],"mappings":";;;AA4CA,4BA4CC;AAQD,4CAcC;AA2GD,sCAiDC;AA1QD,6BAAwB;AAExB,4CAAyC;AACzC,mEAAkE;AAClE,uDAAyD;AAyBzD,wBAAwB;AACxB,MAAM,cAAc,GAA8B;IAChD,YAAY,EAAE,MAAM;IACpB,aAAa,EAAE,IAAI;IACnB,YAAY,EAAE,SAAS;IACvB,aAAa,EAAE,oCAAoC;CACpD,CAAC;AAEF;;;;;;GAMG;AACI,KAAK,UAAU,QAAQ,CAAC,OAAe,EAAE,MAAwB,EAAE,GAAsB;IAC9F,MAAM,YAAY,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IAEtD,IAAI,CAAC;QACH,kCAAkC;QAClC,MAAM,kBAAkB,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAErG,MAAM,UAAU,GACd,YAAY,CAAC,YAAY,IAAI,wGAAwG,CAAC;QAExI,IAAI,MAAM,GAAG,GAAG,UAAU,OAAO,kBAAkB,MAAM,CAAC;QAE1D,IAAI,YAAY,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;YACzC,MAAM,IAAI;;UAEN,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,mDAAmD,CAAC,CAAC,CAAC,EAAE;;wCAEvD,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,6BAA6B,CAAC;QAC1C,CAAC;QAED,MAAM,IAAI,6BAA6B,OAAO,EAAE,CAAC;QAEjD,kCAAkC;QAClC,IAAI,YAAY,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,IAAI,iCAAgB,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,wBAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC1E,OAAO,MAA8B,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,wBAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC7D,MAAM,SAAS,GAAG,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAClF,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAC9B,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,eAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,sCAAsC,EAAE,GAAG,EAAE,CAAC,CAAC;QAEvE,yCAAyC;QACzC,OAAO;YACL,KAAK,EAAE,YAAY,CAAC,YAAY,IAAI,SAAS;YAC7C,MAAM,EAAE,YAAY,CAAC,aAAa,IAAK,GAAa,CAAC,OAAO;YAC5D,UAAU,EAAE,CAAC;SACd,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,SAAgB,gBAAgB,CAAC,OAAsC,EAAE,OAAmC;IAC1G,MAAM,gBAAgB,GAAsB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC5D,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;QACtC,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAqB;QAC/B,GAAG,OAAO;QACV,OAAO,EAAE,gBAAgB;KAC1B,CAAC;IAEF,OAAO,CAAC,OAAe,EAAE,GAAsB,EAAE,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;AACrF,CAAC;AAED,iCAAiC;AACpB,QAAA,MAAM,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7B,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;IACvD,OAAO,EAAE,OAAC;SACP,KAAK,CACJ,OAAC,CAAC,KAAK,CAAC;QACN,OAAC,CAAC,MAAM,EAAE;QACV,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QACjC,OAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;YAChB,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;SACxB,CAAC;KACH,CAAC,CACH;SACA,QAAQ,EAAE;SACV,QAAQ,CAAC,4GAA4G,CAAC;IACzH,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;IACvF,aAAa,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,8CAA8C,CAAC;IAC5G,YAAY,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,sCAAsC,CAAC;IAClH,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8CAA8C,CAAC;IAC5F,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,+CAA+C,CAAC;CAC/F,CAAC,CAAC;AAEH,MAAM,cAAc,GAA6B;IAC/C,IAAI,EAAE,mBAAmB;IACzB,WAAW,EACT,4KAA4K;IAC9K,KAAK,EAAE,cAAM;IACb,QAAQ,EAAE,KAAK;IACf,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE;;QACzC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,YAAY,EAAE,YAAY,EAAE,aAAa,EAAE,GAAG,KAAK,CAAC;QAE3G,eAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,qBAAqB;YAC9B,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,aAAa,EAAE,OAAO,CAAC,MAAM;YAC7B,UAAU,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,MAAM;SAC5B,CAAC,CAAC;QAEH,kDAAkD;QAClD,IAAI,gBAAgB,GAAsB,EAAE,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,gBAAgB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACnC,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;oBAC1B,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;gBACtC,CAAC;qBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5B,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3C,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAoB,CAAC;gBAC9B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,uDAAuD;QACvD,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,KAAI,MAAA,MAAA,KAAK,CAAC,MAAM,0CAAE,gBAAgB,0CAAE,OAAO,CAAA,EAAE,CAAC;YAC7E,gBAAgB,GAAG,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC;QAC3D,CAAC;QAED,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,MAAM,GAAqB;YAC/B,OAAO,EAAE,gBAAgB;YACzB,YAAY;YACZ,aAAa;YACb,YAAY;YACZ,YAAY;YACZ,aAAa;SACd,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;YAE1D,eAAM,CAAC,IAAI,CAAC;gBACV,OAAO,EAAE,0BAA0B;gBACnC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,UAAU,EAAE,MAAM,CAAC,UAAU;aAC9B,CAAC,CAAC;YAEH,KAAK,CAAC,MAAM,CAAC,kBAAkB,GAAG;gBAChC,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxE,MAAM;gBACN,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC;YAEF,OAAO;gBACL,MAAM;aACP,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,eAAM,CAAC,KAAK,CAAC;gBACX,OAAO,EAAE,uBAAuB;gBAChC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,GAAG;aACJ,CAAC,CAAC;YAEH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF,CAAC;AAEF,kBAAe,cAAc,CAAC;AAE9B,gDAAgD;AACzC,KAAK,UAAU,aAAa,CACjC,OAAe,EACf,MAAkD,EAClD,GAAsB;IAEtB,MAAM,YAAY,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IACtD,MAAM,UAAU,GAAG,YAAY,CAAC,UAAU,IAAI,CAAC,CAAC;IAEhD,IAAI,CAAC;QACH,MAAM,kBAAkB,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAErG,MAAM,UAAU,GACd,YAAY,CAAC,YAAY,IAAI,2FAA2F,CAAC;QAE3H,MAAM,MAAM,GAAG,GAAG,UAAU,OAAO,kBAAkB;wDACD,UAAU;;;;MAI5D,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,EAAE;;;;;;;;EAQlE,OAAO,EAAE,CAAC;QAER,MAAM,MAAM,GAAG,IAAI,iCAAgB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,wBAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAE1E,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,OAAO,MAAgC,CAAC;QAC1C,CAAC;QAED,2CAA2C;QAC3C,OAAO,CAAC,MAA8B,CAAC,CAAC;IAC1C,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,eAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC;QAE9D,OAAO;YACL;gBACE,KAAK,EAAE,YAAY,CAAC,YAAY,IAAI,SAAS;gBAC7C,MAAM,EAAE,YAAY,CAAC,aAAa,IAAK,GAAa,CAAC,OAAO;gBAC5D,UAAU,EAAE,CAAC;aACd;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}
+{"version":3,"file":"classifier.js","sourceRoot":"","sources":["../../src/toolsLibrary/classifier.ts"],"names":[],"mappings":";;;AA8DA,4BA4EC;AAQD,4CAcC;AAuGD,sCAqDC;AA5TD,6BAAwB;AAExB,4CAAyC;AAEzC,uDAAyD;AAQzD,SAAS,wBAAwB,CAAC,GAAsB;IACtD,OAAO,sBAAsB,IAAI,GAAG,IAAI,OAAQ,GAAW,CAAC,oBAAoB,KAAK,UAAU,CAAC;AAClG,CAAC;AAsBD;;GAEG;AACH,SAAS,0BAA0B,CAAC,MAAwB;IAC1D,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAE1D,4CAA4C;IAC5C,MAAM,SAAS,GAAG,OAAC,CAAC,IAAI,CAAC,eAAwC,CAAC,CAAC;IAEnE,+CAA+C;IAC/C,MAAM,WAAW,GAAiC;QAChD,KAAK,EAAE,SAAS,CAAC,QAAQ,CAAC,sCAAsC,CAAC;QACjE,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oCAAoC,CAAC;QACjE,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,kCAAkC,CAAC;KAClF,CAAC;IAEF,OAAO,OAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;AAC/B,CAAC;AAED;;;;;;GAMG;AACI,KAAK,UAAU,QAAQ,CAAC,OAAe,EAAE,MAAwB,EAAE,GAAsB;IAC9F,MAAM,WAAW,GAAG,CAAC,CAAC;IACtB,IAAI,SAAS,GAAiB,IAAI,CAAC;IAEnC,kDAAkD;IAClD,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,8EAA8E,CAAC,CAAC;IAClG,CAAC;IAED,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;QACxD,IAAI,CAAC;YACH,kCAAkC;YAClC,MAAM,kBAAkB,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE/F,MAAM,UAAU,GACd,MAAM,CAAC,YAAY,IAAI,wGAAwG,CAAC;YAElI,IAAI,MAAM,GAAG,GAAG,UAAU,OAAO,kBAAkB,MAAM,CAAC;YAE1D,sDAAsD;YACtD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBAChB,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBAC1D,MAAM,IAAI,+DAA+D,CAAC;gBAC1E,MAAM,IAAI,mDAAmD,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;YAChG,CAAC;YAED,+CAA+C;YAC/C,MAAM,oBAAoB,GAAG,0BAA0B,CAAC,MAAM,CAAC,CAAC;YAEhE,wCAAwC;YACxC,MAAM,IAAI,6BAA6B,OAAO,EAAE,CAAC;YAEjD,kDAAkD;YAClD,MAAM,aAAa,GAAG,GAAG,CAAC,oBAAoB,CAAC,oBAAoB,CAAC,CAAC;YACrE,MAAM,QAAQ,GAAG,CAAC,IAAI,wBAAa,CAAC,MAAM,CAAC,CAAC,CAAC;YAC7C,MAAM,MAAM,GAAG,CAAC,MAAM,aAAa,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAyB,CAAC;YAE9E,0DAA0D;YAC1D,eAAM,CAAC,KAAK,CAAC;gBACX,OAAO,EAAE,kDAAkD;gBAC3D,OAAO;gBACP,KAAK,EAAE,MAAM,CAAC,KAAK;aACpB,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,GAAG,GAAY,CAAC;YACzB,eAAM,CAAC,IAAI,CAAC;gBACV,OAAO,EAAE,0BAA0B,OAAO,SAAS;gBACnD,OAAO;gBACP,UAAU,EAAE,WAAW;gBACvB,KAAK,EAAE,SAAS,CAAC,OAAO;aACzB,CAAC,CAAC;YAEH,qDAAqD;YACrD,IAAI,OAAO,GAAG,WAAW,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;QACH,CAAC;IACH,CAAC;IAED,8CAA8C;IAC9C,eAAM,CAAC,IAAI,CAAC;QACV,OAAO,EAAE,8DAA8D;QACvE,UAAU,EAAE,WAAW;QACvB,SAAS,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,OAAO;QAC7B,YAAY,EAAE,MAAM,CAAC,YAAY;QACjC,aAAa,EAAE,MAAM,CAAC,aAAa;KACpC,CAAC,CAAC;IAEH,gCAAgC;IAChC,OAAO;QACL,KAAK,EAAE,MAAM,CAAC,YAAY;QAC1B,MAAM,EAAE,MAAM,CAAC,aAAa;QAC5B,UAAU,EAAE,CAAC;KACd,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAgB,gBAAgB,CAAC,OAAsC,EAAE,OAA0C;IACjH,MAAM,gBAAgB,GAAsB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC5D,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;QACtC,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAqB;QAC/B,GAAG,OAAO;QACV,OAAO,EAAE,gBAAgB;KAC1B,CAAC;IAEF,OAAO,CAAC,OAAe,EAAE,GAAsB,EAAE,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;AACrF,CAAC;AAED,iCAAiC;AACpB,QAAA,MAAM,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7B,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;IACvD,OAAO,EAAE,OAAC;SACP,KAAK,CACJ,OAAC,CAAC,KAAK,CAAC;QACN,OAAC,CAAC,MAAM,EAAE;QACV,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QACjC,OAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;YAChB,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE;SACxB,CAAC;KACH,CAAC,CACH;SACA,QAAQ,EAAE;SACV,QAAQ,CAAC,4GAA4G,CAAC;IACzH,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;IACvF,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8CAA8C,CAAC;IACjF,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+CAA+C,CAAC;CACpF,CAAC,CAAC;AAEH,MAAM,cAAc,GAA6B;IAC/C,IAAI,EAAE,mBAAmB;IACzB,WAAW,EACT,8LAA8L;IAChM,KAAK,EAAE,cAAM;IACb,QAAQ,EAAE,KAAK;IACf,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE;;QACzC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,aAAa,EAAE,GAAG,KAAK,CAAC;QAE9E,eAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,qBAAqB;YAC9B,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,aAAa,EAAE,OAAO,CAAC,MAAM;YAC7B,UAAU,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,MAAM;SAC5B,CAAC,CAAC;QAEH,kDAAkD;QAClD,IAAI,gBAAgB,GAAsB,EAAE,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,gBAAgB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACnC,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;oBAC1B,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;gBACtC,CAAC;qBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC5B,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3C,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAoB,CAAC;gBAC9B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,uDAAuD;QACvD,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,KAAI,MAAA,MAAA,KAAK,CAAC,MAAM,0CAAE,gBAAgB,0CAAE,OAAO,CAAA,EAAE,CAAC;YAC7E,gBAAgB,GAAG,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC;QAC3D,CAAC;QAED,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,MAAM,GAAqB;YAC/B,OAAO,EAAE,gBAAgB;YACzB,YAAY;YACZ,YAAY;YACZ,aAAa;SACd,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;YAE1D,eAAM,CAAC,IAAI,CAAC;gBACV,OAAO,EAAE,0BAA0B;gBACnC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,UAAU,EAAE,MAAM,CAAC,UAAU;aAC9B,CAAC,CAAC;YAEH,KAAK,CAAC,MAAM,CAAC,kBAAkB,GAAG;gBAChC,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxE,MAAM;gBACN,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC;YAEF,OAAO;gBACL,MAAM;aACP,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,eAAM,CAAC,KAAK,CAAC;gBACX,OAAO,EAAE,uBAAuB;gBAChC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,GAAG;aACJ,CAAC,CAAC;YAEH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF,CAAC;AAEF,kBAAe,cAAc,CAAC;AAE9B,gDAAgD;AACzC,KAAK,UAAU,aAAa,CACjC,OAAe,EACf,MAAkD,EAClD,GAAsB;IAEtB,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC;IAE1C,0CAA0C;IAC1C,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,oFAAoF,CAAC,CAAC;IACxG,CAAC;IAED,IAAI,CAAC;QACH,MAAM,kBAAkB,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/F,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAE1D,MAAM,UAAU,GAAG,MAAM,CAAC,YAAY,IAAI,2FAA2F,CAAC;QAEtI,MAAM,MAAM,GAAG,GAAG,UAAU,OAAO,kBAAkB;eAC1C,UAAU;;;EAGvB,OAAO,EAAE,CAAC;QAER,6CAA6C;QAC7C,MAAM,SAAS,GAAG,OAAC,CAAC,IAAI,CAAC,eAAwC,CAAC,CAAC;QACnE,MAAM,gBAAgB,GAAG,OAAC;aACvB,KAAK,CACJ,OAAC,CAAC,MAAM,CAAC;YACP,KAAK,EAAE,SAAS,CAAC,QAAQ,CAAC,sCAAsC,CAAC;YACjE,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oCAAoC,CAAC;YACjE,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,kCAAkC,CAAC;SAClF,CAAC,CACH;aACA,GAAG,CAAC,CAAC,CAAC;aACN,GAAG,CAAC,UAAU,CAAC,CAAC;QAEnB,MAAM,aAAa,GAAG,GAAG,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QACjE,MAAM,QAAQ,GAAG,CAAC,IAAI,wBAAa,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,CAAC,MAAM,aAAa,CAAC,MAAM,CAAC,QAAQ,CAAC,CAA2B,CAAC;QAEhF,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,eAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC;QAE9D,OAAO;YACL;gBACE,KAAK,EAAE,MAAM,CAAC,YAAY;gBAC1B,MAAM,EAAE,MAAM,CAAC,aAAa;gBAC5B,UAAU,EAAE,CAAC;aACd;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/utils/extractStateMemoryResponse.d.ts

@@ -0,0 +1,5 @@
+import { ToolMessage } from '@langchain/core/messages';
+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

@@ -0,0 +1 @@
+{"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,gBAAiB,WAAW,KAAG,OAAO,CAAC,KAAK,CAazE,CAAC;AA4EF,eAAe,wBAAwB,CAAC"}

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.14",
+  "version": "2.0.15-beta-1",
   "description": "MindedJS is a TypeScript library for building agents.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/nodes/addRpaNode.ts

@@ -12,6 +12,8 @@ import { LLMProviders } from '../types/LLM.types';
 import { AIMessage, ToolMessage } from '@langchain/core/messages';
 import { v4 as uuidv4 } from 'uuid';
 import { executeRpaStep } from './rpaStepsExecutor';
+import { createBrowserSession } from '../browserTask/executeBrowserTask';
+import { getConfig } from '../platform/config';
 
 type AddRpaNodeParams = {
   graph: PreCompiledGraph;
@@ -29,13 +31,22 @@ export const addRpaNode = async ({ graph, node, agent, llm }: AddRpaNodeParams)
 
     let browser: Browser | null = null;
     let page: Page | null = null;
+    const { browserTaskMode } = getConfig();
 
+    const session = await createBrowserSession({
+      sessionId: state.sessionId,
+      browserTaskMode,
+    });
+
+    if (!session.sessionId || !session.cdpUrl) {
+      throw new Error('Failed to create browser session: missing session details');
+    }
     // Create tool call for RPA execution
     const toolCallId = uuidv4();
     const aiMessageId = uuidv4();
-
+    state.cdpUrl = session.cdpUrl;
     // Get CDP URL from state
-    const cdpUrl = state.cdpUrl;
+    const cdpUrl = session.cdpUrl;
 
     const toolCall = {
       id: toolCallId,