@n8n/ai-workflow-builder 0.19.0 → 0.21.0

package/dist/tools/update-node-parameters.tool.d.ts

@@ -1,19 +1,22 @@
 import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
 import type { INodeTypeDescription, Logger } from 'n8n-workflow';
 import { z } from 'zod';
-export declare function createUpdateNodeParametersTool(nodeTypes: INodeTypeDescription[], llm: BaseChatModel, logger?: Logger, instanceUrl?: string): import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{
-    nodeId: z.ZodString;
-    changes: z.ZodArray<z.ZodString, "many">;
-}, "strip", z.ZodTypeAny, {
-    nodeId: string;
-    changes: string[];
-}, {
-    nodeId: string;
-    changes: string[];
-}>, {
-    nodeId: string;
-    changes: string[];
-}, {
-    nodeId: string;
-    changes: string[];
-}, import("@langchain/langgraph").Command<unknown>>;
+export declare function createUpdateNodeParametersTool(nodeTypes: INodeTypeDescription[], llm: BaseChatModel, logger?: Logger, instanceUrl?: string): {
+    tool: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{
+        nodeId: z.ZodString;
+        changes: z.ZodArray<z.ZodString, "many">;
+    }, "strip", z.ZodTypeAny, {
+        nodeId: string;
+        changes: string[];
+    }, {
+        nodeId: string;
+        changes: string[];
+    }>, {
+        nodeId: string;
+        changes: string[];
+    }, {
+        nodeId: string;
+        changes: string[];
+    }, import("@langchain/langgraph").Command<unknown>>;
+    displayTitle: string;
+};

package/dist/tools/update-node-parameters.tool.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.createUpdateNodeParametersTool = createUpdateNodeParametersTool;
 const tools_1 = require("@langchain/core/tools");
 const zod_1 = require("zod");
+const trim_workflow_context_1 = require("../utils/trim-workflow-context");
 const parameter_updater_1 = require("../chains/parameter-updater");
 const errors_1 = require("../errors");
 const progress_1 = require("./helpers/progress");
@@ -10,6 +11,7 @@ const response_1 = require("./helpers/response");
 const state_1 = require("./helpers/state");
 const validation_1 = require("./helpers/validation");
 const parameter_update_utils_1 = require("./utils/parameter-update.utils");
+const DISPLAY_TITLE = 'Updating node parameters';
 const updateNodeParametersSchema = zod_1.z.object({
     nodeId: zod_1.z.string().describe('The ID of the node to update'),
     changes: zod_1.z
@@ -32,7 +34,7 @@ async function processParameterUpdates(node, nodeType, nodeId, changes, state, l
         requestedChanges: changes,
     }, logger);
     const newParameters = (await parametersChain.invoke({
-        workflow_json: workflow,
+        workflow_json: (0, trim_workflow_context_1.trimWorkflowJSON)(workflow),
         execution_schema: state.workflowContext?.executionSchema ?? 'NO SCHEMA',
         execution_data: state.workflowContext?.executionData ?? 'NO EXECUTION DATA YET',
         node_id: nodeId,
@@ -57,15 +59,26 @@ async function processParameterUpdates(node, nodeType, nodeId, changes, state, l
     }
     return (0, parameter_update_utils_1.fixExpressionPrefixes)(newParameters.parameters);
 }
+function getCustomNodeTitle(input, nodes) {
+    if ('nodeId' in input && typeof input['nodeId'] === 'string') {
+        const targetNode = nodes?.find((node) => node.id === input.nodeId);
+        if (targetNode) {
+            return `Updating "${targetNode.name}" node parameters`;
+        }
+    }
+    return DISPLAY_TITLE;
+}
 function createUpdateNodeParametersTool(nodeTypes, llm, logger, instanceUrl) {
-    return (0, tools_1.tool)(async (input, config) => {
-        const reporter = (0, progress_1.createProgressReporter)(config, 'update_node_parameters');
+    const dynamicTool = (0, tools_1.tool)(async (input, config) => {
+        const reporter = (0, progress_1.createProgressReporter)(config, 'update_node_parameters', DISPLAY_TITLE);
         try {
             const validatedInput = updateNodeParametersSchema.parse(input);
             const { nodeId, changes } = validatedInput;
-            reporter.start(validatedInput);
             const state = (0, state_1.getWorkflowState)();
             const workflow = (0, state_1.getCurrentWorkflow)(state);
+            reporter.start(validatedInput, {
+                customDisplayTitle: getCustomNodeTitle(input, workflow.nodes),
+            });
             const node = (0, validation_1.validateNodeExists)(nodeId, workflow.nodes);
             if (!node) {
                 const error = (0, validation_1.createNodeNotFoundError)(nodeId);
@@ -131,5 +144,9 @@ function createUpdateNodeParametersTool(nodeTypes, llm, logger, instanceUrl) {
         description: 'Update the parameters of an existing node in the workflow based on natural language changes. This tool intelligently modifies only the specified parameters while preserving others. Examples: "Set the URL to https://api.example.com", "Add authentication header", "Change method to POST", "Set the condition to check if status equals success".',
         schema: updateNodeParametersSchema,
     });
+    return {
+        tool: dynamicTool,
+        displayTitle: DISPLAY_TITLE,
+    };
 }
 //# sourceMappingURL=update-node-parameters.tool.js.map

package/dist/tools/update-node-parameters.tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"update-node-parameters.tool.js","sourceRoot":"","sources":["../../src/tools/update-node-parameters.tool.ts"],"names":[],"mappings":";;AAoHA,wEAuHC;AA1OD,iDAA6C;AAE7C,6BAAwB;AAExB,mEAA0E;AAC1E,sCAAsF;AACtF,iDAA4E;AAC5E,iDAAgF;AAChF,2CAA6F;AAC7F,qDAK8B;AAC9B,2EAKwC;AAMxC,MAAM,0BAA0B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC3C,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;IAC3D,OAAO,EAAE,OAAC;SACR,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;SACjB,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACR,0IAA0I,CAC1I;CACF,CAAC,CAAC;AAKH,SAAS,mBAAmB,CAAC,IAAW,EAAE,OAAiB;IAC1D,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5D,OAAO,6CAA6C,IAAI,CAAC,IAAI,MAAM,IAAI,CAAC,IAAI,OAAO,WAAW,EAAE,CAAC;AAClG,CAAC;AAKD,KAAK,UAAU,uBAAuB,CACrC,IAAW,EACX,QAA8B,EAC9B,MAAc,EACd,OAAiB,EACjB,KAA0C,EAC1C,GAAkB,EAClB,MAAe,EACf,WAAoB;IAEpB,MAAM,QAAQ,GAAG,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC;IAG3C,MAAM,iBAAiB,GAAG,IAAA,8CAAqB,EAAC,IAAI,CAAC,CAAC;IAGtD,MAAM,gBAAgB,GAAG,IAAA,+CAAsB,EAAC,OAAO,CAAC,CAAC;IAGzD,MAAM,kBAAkB,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAG9E,MAAM,eAAe,GAAG,IAAA,+CAA2B,EAClD,GAAG,EACH;QACC,QAAQ,EAAE,IAAI,CAAC,IAAI;QACnB,cAAc,EAAE,QAAQ;QACxB,gBAAgB,EAAE,OAAO;KACzB,EACD,MAAM,CACN,CAAC;IAEF,MAAM,aAAa,GAAG,CAAC,MAAM,eAAe,CAAC,MAAM,CAAC;QACnD,aAAa,EAAE,QAAQ;QACvB,gBAAgB,EAAE,KAAK,CAAC,eAAe,EAAE,eAAe,IAAI,WAAW;QACvE,cAAc,EAAE,KAAK,CAAC,eAAe,EAAE,aAAa,IAAI,uBAAuB;QAC/E,OAAO,EAAE,MAAM;QACf,SAAS,EAAE,IAAI,CAAC,IAAI;QACpB,SAAS,EAAE,IAAI,CAAC,IAAI;QACpB,kBAAkB,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,eAAe,EAAE,kBAAkB;QACnC,OAAO,EAAE,gBAAgB;QACzB,WAAW,EAAE,WAAW,IAAI,EAAE;KAC9B,CAAC,CAAoB,CAAC;IAGvB,IAAI,CAAC,aAAa,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;QACzD,MAAM,IAAI,6BAAoB,CAAC,sCAAsC,EAAE;YACtE,MAAM;YACN,QAAQ,EAAE,IAAI,CAAC,IAAI;SACnB,CAAC,CAAC;IACJ,CAAC;IAGD,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,OAAO,aAAa,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QAC/E,MAAM,IAAI,6BAAoB,CAAC,gDAAgD,EAAE;YAChF,MAAM;YACN,QAAQ,EAAE,IAAI,CAAC,IAAI;SACnB,CAAC,CAAC;IACJ,CAAC;IAGD,OAAO,IAAA,8CAAqB,EAAC,aAAa,CAAC,UAAU,CAAoB,CAAC;AAC3E,CAAC;AAKD,SAAgB,8BAA8B,CAC7C,SAAiC,EACjC,GAAkB,EAClB,MAAe,EACf,WAAoB;IAEpB,OAAO,IAAA,YAAI,EACV,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;QACvB,MAAM,QAAQ,GAAG,IAAA,iCAAsB,EAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;QAE1E,IAAI,CAAC;YAEJ,MAAM,cAAc,GAAG,0BAA0B,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC/D,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,cAAc,CAAC;YAG3C,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YAG/B,MAAM,KAAK,GAAG,IAAA,wBAAgB,GAAE,CAAC;YACjC,MAAM,QAAQ,GAAG,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC;YAG3C,MAAM,IAAI,GAAG,IAAA,+BAAkB,EAAC,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YACxD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACX,MAAM,KAAK,GAAG,IAAA,oCAAuB,EAAC,MAAM,CAAC,CAAC;gBAC9C,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACtB,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC3C,CAAC;YAGD,MAAM,QAAQ,GAAG,IAAA,yBAAY,EAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YACpD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACf,MAAM,KAAK,GAAG,IAAA,wCAA2B,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACrD,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACtB,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC3C,CAAC;YAGD,IAAA,yBAAc,EAAC,QAAQ,EAAE,iCAAiC,IAAI,CAAC,IAAI,GAAG,EAAE;gBACvE,MAAM;gBACN,OAAO;aACP,CAAC,CAAC;YAEH,IAAI,CAAC;gBACJ,MAAM,iBAAiB,GAAG,MAAM,uBAAuB,CACtD,IAAI,EACJ,QAAQ,EACR,MAAM,EACN,OAAO,EACP,KAAK,EACL,GAAG,EACH,MAAM,EACN,WAAW,CACX,CAAC;gBAGF,MAAM,WAAW,GAAG,IAAA,iDAAwB,EAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;gBAGtE,MAAM,OAAO,GAAG,mBAAmB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAGnD,MAAM,MAAM,GAA+B;oBAC1C,MAAM;oBACN,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,iBAAiB;oBACjB,cAAc,EAAE,OAAO;oBACvB,OAAO;iBACP,CAAC;gBACF,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;gBAG1B,MAAM,YAAY,GAAG,IAAA,4BAAoB,EAAC,KAAK,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;gBACtE,OAAO,IAAA,gCAAqB,EAAC,MAAM,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;YAC7D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,IAAI,KAAK,YAAY,6BAAoB,EAAE,CAAC;oBAC3C,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACtB,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,KAAK,CAAC,CAAC;gBAC3C,CAAC;gBACD,MAAM,SAAS,GAAG,IAAI,2BAAkB,CACvC,qCAAqC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,EAC/F;oBACC,QAAQ,EAAE,wBAAwB;oBAClC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;iBACjD,CACD,CAAC;gBACF,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC1B,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAC/C,CAAC;QACF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEhB,IAAI,KAAK,YAAY,OAAC,CAAC,QAAQ,EAAE,CAAC;gBACjC,MAAM,eAAe,GAAG,IAAI,wBAAe,CAAC,0BAA0B,EAAE;oBACvE,KAAK,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE;iBAC/B,CAAC,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;gBAChC,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,eAAe,CAAC,CAAC;YACrD,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,2BAAkB,CACvC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,EACjE;gBACC,QAAQ,EAAE,wBAAwB;gBAClC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;aACjD,CACD,CAAC;YACF,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAC1B,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC/C,CAAC;IACF,CAAC,EACD;QACC,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EACV,uVAAuV;QACxV,MAAM,EAAE,0BAA0B;KAClC,CACD,CAAC;AACH,CAAC"}
+{"version":3,"file":"update-node-parameters.tool.js","sourceRoot":"","sources":["../../src/tools/update-node-parameters.tool.ts"],"names":[],"mappings":";;AAmIA,wEA8HC;AAhQD,iDAA6C;AAE7C,6BAAwB;AAExB,yEAAiE;AAEjE,mEAA0E;AAC1E,sCAAsF;AAEtF,iDAA4E;AAC5E,iDAAgF;AAChF,2CAA6F;AAC7F,qDAK8B;AAC9B,2EAKwC;AAExC,MAAM,aAAa,GAAG,0BAA0B,CAAC;AAKjD,MAAM,0BAA0B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC3C,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;IAC3D,OAAO,EAAE,OAAC;SACR,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;SACjB,GAAG,CAAC,CAAC,CAAC;SACN,QAAQ,CACR,0IAA0I,CAC1I;CACF,CAAC,CAAC;AAKH,SAAS,mBAAmB,CAAC,IAAW,EAAE,OAAiB;IAC1D,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5D,OAAO,6CAA6C,IAAI,CAAC,IAAI,MAAM,IAAI,CAAC,IAAI,OAAO,WAAW,EAAE,CAAC;AAClG,CAAC;AAKD,KAAK,UAAU,uBAAuB,CACrC,IAAW,EACX,QAA8B,EAC9B,MAAc,EACd,OAAiB,EACjB,KAA0C,EAC1C,GAAkB,EAClB,MAAe,EACf,WAAoB;IAEpB,MAAM,QAAQ,GAAG,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC;IAG3C,MAAM,iBAAiB,GAAG,IAAA,8CAAqB,EAAC,IAAI,CAAC,CAAC;IAGtD,MAAM,gBAAgB,GAAG,IAAA,+CAAsB,EAAC,OAAO,CAAC,CAAC;IAGzD,MAAM,kBAAkB,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAG9E,MAAM,eAAe,GAAG,IAAA,+CAA2B,EAClD,GAAG,EACH;QACC,QAAQ,EAAE,IAAI,CAAC,IAAI;QACnB,cAAc,EAAE,QAAQ;QACxB,gBAAgB,EAAE,OAAO;KACzB,EACD,MAAM,CACN,CAAC;IAEF,MAAM,aAAa,GAAG,CAAC,MAAM,eAAe,CAAC,MAAM,CAAC;QACnD,aAAa,EAAE,IAAA,wCAAgB,EAAC,QAAQ,CAAC;QACzC,gBAAgB,EAAE,KAAK,CAAC,eAAe,EAAE,eAAe,IAAI,WAAW;QACvE,cAAc,EAAE,KAAK,CAAC,eAAe,EAAE,aAAa,IAAI,uBAAuB;QAC/E,OAAO,EAAE,MAAM;QACf,SAAS,EAAE,IAAI,CAAC,IAAI;QACpB,SAAS,EAAE,IAAI,CAAC,IAAI;QACpB,kBAAkB,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,eAAe,EAAE,kBAAkB;QACnC,OAAO,EAAE,gBAAgB;QACzB,WAAW,EAAE,WAAW,IAAI,EAAE;KAC9B,CAAC,CAAoB,CAAC;IAGvB,IAAI,CAAC,aAAa,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;QACzD,MAAM,IAAI,6BAAoB,CAAC,sCAAsC,EAAE;YACtE,MAAM;YACN,QAAQ,EAAE,IAAI,CAAC,IAAI;SACnB,CAAC,CAAC;IACJ,CAAC;IAGD,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,OAAO,aAAa,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QAC/E,MAAM,IAAI,6BAAoB,CAAC,gDAAgD,EAAE;YAChF,MAAM;YACN,QAAQ,EAAE,IAAI,CAAC,IAAI;SACnB,CAAC,CAAC;IACJ,CAAC;IAGD,OAAO,IAAA,8CAAqB,EAAC,aAAa,CAAC,UAAU,CAAoB,CAAC;AAC3E,CAAC;AAED,SAAS,kBAAkB,CAAC,KAA8B,EAAE,KAAe;IAC1E,IAAI,QAAQ,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,QAAQ,CAAC,KAAK,QAAQ,EAAE,CAAC;QAC9D,MAAM,UAAU,GAAG,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,KAAK,CAAC,MAAM,CAAC,CAAC;QACnE,IAAI,UAAU,EAAE,CAAC;YAChB,OAAO,aAAa,UAAU,CAAC,IAAI,mBAAmB,CAAC;QACxD,CAAC;IACF,CAAC;IAED,OAAO,aAAa,CAAC;AACtB,CAAC;AAKD,SAAgB,8BAA8B,CAC7C,SAAiC,EACjC,GAAkB,EAClB,MAAe,EACf,WAAoB;IAEpB,MAAM,WAAW,GAAG,IAAA,YAAI,EACvB,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;QACvB,MAAM,QAAQ,GAAG,IAAA,iCAAsB,EAAC,MAAM,EAAE,wBAAwB,EAAE,aAAa,CAAC,CAAC;QAEzF,IAAI,CAAC;YAEJ,MAAM,cAAc,GAAG,0BAA0B,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC/D,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,cAAc,CAAC;YAG3C,MAAM,KAAK,GAAG,IAAA,wBAAgB,GAAE,CAAC;YACjC,MAAM,QAAQ,GAAG,IAAA,0BAAkB,EAAC,KAAK,CAAC,CAAC;YAG3C,QAAQ,CAAC,KAAK,CAAC,cAAc,EAAE;gBAC9B,kBAAkB,EAAE,kBAAkB,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC;aAC7D,CAAC,CAAC;YAGH,MAAM,IAAI,GAAG,IAAA,+BAAkB,EAAC,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YACxD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACX,MAAM,KAAK,GAAG,IAAA,oCAAuB,EAAC,MAAM,CAAC,CAAC;gBAC9C,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACtB,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC3C,CAAC;YAGD,MAAM,QAAQ,GAAG,IAAA,yBAAY,EAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YACpD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACf,MAAM,KAAK,GAAG,IAAA,wCAA2B,EAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACrD,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACtB,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC3C,CAAC;YAGD,IAAA,yBAAc,EAAC,QAAQ,EAAE,iCAAiC,IAAI,CAAC,IAAI,GAAG,EAAE;gBACvE,MAAM;gBACN,OAAO;aACP,CAAC,CAAC;YAEH,IAAI,CAAC;gBACJ,MAAM,iBAAiB,GAAG,MAAM,uBAAuB,CACtD,IAAI,EACJ,QAAQ,EACR,MAAM,EACN,OAAO,EACP,KAAK,EACL,GAAG,EACH,MAAM,EACN,WAAW,CACX,CAAC;gBAGF,MAAM,WAAW,GAAG,IAAA,iDAAwB,EAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;gBAGtE,MAAM,OAAO,GAAG,mBAAmB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAGnD,MAAM,MAAM,GAA+B;oBAC1C,MAAM;oBACN,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,iBAAiB;oBACjB,cAAc,EAAE,OAAO;oBACvB,OAAO;iBACP,CAAC;gBACF,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;gBAG1B,MAAM,YAAY,GAAG,IAAA,4BAAoB,EAAC,KAAK,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;gBACtE,OAAO,IAAA,gCAAqB,EAAC,MAAM,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;YAC7D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,IAAI,KAAK,YAAY,6BAAoB,EAAE,CAAC;oBAC3C,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACtB,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,KAAK,CAAC,CAAC;gBAC3C,CAAC;gBACD,MAAM,SAAS,GAAG,IAAI,2BAAkB,CACvC,qCAAqC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,EAC/F;oBACC,QAAQ,EAAE,wBAAwB;oBAClC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;iBACjD,CACD,CAAC;gBACF,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC1B,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAC/C,CAAC;QACF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEhB,IAAI,KAAK,YAAY,OAAC,CAAC,QAAQ,EAAE,CAAC;gBACjC,MAAM,eAAe,GAAG,IAAI,wBAAe,CAAC,0BAA0B,EAAE;oBACvE,KAAK,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE;iBAC/B,CAAC,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;gBAChC,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,eAAe,CAAC,CAAC;YACrD,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,2BAAkB,CACvC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,EACjE;gBACC,QAAQ,EAAE,wBAAwB;gBAClC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;aACjD,CACD,CAAC;YACF,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAC1B,OAAO,IAAA,8BAAmB,EAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC/C,CAAC;IACF,CAAC,EACD;QACC,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EACV,uVAAuV;QACxV,MAAM,EAAE,0BAA0B;KAClC,CACD,CAAC;IAEF,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,YAAY,EAAE,aAAa;KAC3B,CAAC;AACH,CAAC"}

package/dist/types/index.d.ts

@@ -1,8 +1,8 @@
-export * from './workflow';
-export * from './messages';
-export * from './tools';
-export * from './connections';
-export * from './streaming';
-export * from './nodes';
-export * from './config';
-export * from './utils';
+export type * from './workflow';
+export type * from './messages';
+export type * from './tools';
+export type * from './connections';
+export type * from './streaming';
+export type * from './nodes';
+export type * from './config';
+export type * from './utils';

package/dist/types/index.js

@@ -1,25 +1,3 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./workflow"), exports);
-__exportStar(require("./messages"), exports);
-__exportStar(require("./tools"), exports);
-__exportStar(require("./connections"), exports);
-__exportStar(require("./streaming"), exports);
-__exportStar(require("./nodes"), exports);
-__exportStar(require("./config"), exports);
-__exportStar(require("./utils"), exports);
 //# sourceMappingURL=index.js.map

package/dist/types/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAEA,6CAA2B;AAC3B,6CAA2B;AAC3B,0CAAwB;AACxB,gDAA8B;AAC9B,8CAA4B;AAC5B,0CAAwB;AACxB,2CAAyB;AACzB,0CAAwB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":""}

package/dist/types/langchain.d.ts

@@ -0,0 +1,2 @@
+import type { AIMessage, BaseMessage } from '@langchain/core/messages';
+export declare function isAIMessage(msg: BaseMessage): msg is AIMessage;

package/dist/types/langchain.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isAIMessage = isAIMessage;
+function isAIMessage(msg) {
+    return msg.getType() === 'ai';
+}
+//# sourceMappingURL=langchain.js.map

package/dist/types/langchain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"langchain.js","sourceRoot":"","sources":["../../src/types/langchain.ts"],"names":[],"mappings":";;AAEA,kCAEC;AAFD,SAAgB,WAAW,CAAC,GAAgB;IAC3C,OAAO,GAAG,CAAC,OAAO,EAAE,KAAK,IAAI,CAAC;AAC/B,CAAC"}

package/dist/types/messages.d.ts

@@ -27,12 +27,6 @@ export interface AgentChatMessage {
     title: string;
     text: string;
 }
-export interface PromptValidationMessage {
-    role: 'assistant';
-    type: 'prompt-validation';
-    isWorkflowPrompt: boolean;
-    id: string;
-}
-export type MessageResponse = ((AssistantChatMessage | AssistantSummaryMessage | AgentChatMessage | PromptValidationMessage) & {
+export type MessageResponse = ((AssistantChatMessage | AssistantSummaryMessage | AgentChatMessage) & {
     quickReplies?: QuickReplyOption[];
 }) | EndSessionMessage;

package/dist/types/streaming.d.ts

@@ -19,7 +19,12 @@ export interface ExecutionRequestChunk {
     type: 'execution-requested';
     reason: string;
 }
-export type StreamChunk = AgentMessageChunk | ToolProgressChunk | WorkflowUpdateChunk | ExecutionRequestChunk;
+export interface PlanChunk {
+    role: 'assistant';
+    type: 'plan';
+    plan: unknown;
+}
+export type StreamChunk = AgentMessageChunk | ToolProgressChunk | WorkflowUpdateChunk | ExecutionRequestChunk | PlanChunk;
 export interface StreamOutput {
     messages: StreamChunk[];
 }

package/dist/types/tools.d.ts

@@ -13,6 +13,8 @@ export interface ToolProgressMessage<TToolName extends string = string> {
     toolCallId?: string;
     status: 'running' | 'completed' | 'error';
     updates: ProgressUpdate[];
+    displayTitle?: string;
+    customDisplayTitle?: string;
 }
 export interface ToolError {
     message: string;
@@ -20,7 +22,9 @@ export interface ToolError {
     details?: ZodIssue[] | Record<string, unknown>;
 }
 export interface ProgressReporter {
-    start: <T>(input: T) => void;
+    start: <T>(input: T, options?: {
+        customDisplayTitle: string;
+    }) => void;
     progress: (message: string, data?: Record<string, unknown>) => void;
     complete: <T>(output: T) => void;
     error: (error: ToolError) => void;
@@ -74,3 +78,6 @@ export interface NodeSearchOutput {
     totalResults: number;
     message: string;
 }
+export interface GetNodeParameterOutput {
+    message: string;
+}

package/dist/utils/stream-processor.d.ts

@@ -1,6 +1,12 @@
 import { AIMessage, HumanMessage, ToolMessage } from '@langchain/core/messages';
+import type { DynamicStructuredTool } from '@langchain/core/tools';
 import type { StreamOutput } from '../types/streaming';
+export interface BuilderTool {
+    tool: DynamicStructuredTool;
+    displayTitle: string;
+    getCustomDisplayTitle?: (values: Record<string, unknown>) => string;
+}
 export declare const DEFAULT_WORKFLOW_UPDATE_TOOLS: string[];
 export declare function processStreamChunk(streamMode: string, chunk: unknown): StreamOutput | null;
 export declare function createStreamProcessor(stream: AsyncGenerator<[string, unknown], void, unknown>): AsyncGenerator<StreamOutput>;
-export declare function formatMessages(messages: Array<AIMessage | HumanMessage | ToolMessage>): Array<Record<string, unknown>>;
+export declare function formatMessages(messages: Array<AIMessage | HumanMessage | ToolMessage>, builderTools?: BuilderTool[]): Array<Record<string, unknown>>;

package/dist/utils/stream-processor.js

@@ -44,14 +44,46 @@ function processStreamChunk(streamMode, chunk) {
                 else {
                     content = lastMessage.content;
                 }
-                const messageChunk = {
-                    role: 'assistant',
-                    type: 'message',
-                    text: content,
-                };
-                return { messages: [messageChunk] };
+                if (content) {
+                    const messageChunk = {
+                        role: 'assistant',
+                        type: 'message',
+                        text: content,
+                    };
+                    return { messages: [messageChunk] };
+                }
+                return null;
             }
         }
+        if (agentChunk?.create_plan?.workflowPlan) {
+            const workflowPlan = agentChunk.create_plan.workflowPlan;
+            const planChunk = {
+                role: 'assistant',
+                type: 'plan',
+                plan: workflowPlan.plan,
+                message: workflowPlan.intro,
+            };
+            return { messages: [planChunk] };
+        }
+        else if ((agentChunk?.create_plan?.messages ?? []).length > 0) {
+            const lastMessage = agentChunk.create_plan.messages[agentChunk.create_plan.messages.length - 1];
+            const messageChunk = {
+                role: 'assistant',
+                type: 'message',
+                text: lastMessage.content,
+            };
+            return { messages: [messageChunk] };
+        }
+        if (agentChunk?.adjust_plan?.workflowPlan) {
+            const workflowPlan = agentChunk.adjust_plan.workflowPlan;
+            const planChunk = {
+                role: 'assistant',
+                type: 'plan',
+                plan: workflowPlan.plan,
+                message: workflowPlan.intro,
+            };
+            return { messages: [planChunk] };
+        }
         if (agentChunk?.process_operations) {
             const update = agentChunk.process_operations;
             if (update.workflowJSON && update.workflowOperations !== undefined) {
@@ -81,71 +113,97 @@ async function* createStreamProcessor(stream) {
         }
     }
 }
-function formatMessages(messages) {
+function formatHumanMessage(msg) {
+    return {
+        role: 'user',
+        type: 'message',
+        text: msg.content,
+    };
+}
+function processArrayContent(content) {
+    const textMessages = content.filter((c) => typeof c === 'object' && c !== null && 'type' in c && c.type === 'text' && 'text' in c);
+    return textMessages.map((textMessage) => ({
+        role: 'assistant',
+        type: 'message',
+        text: textMessage.text,
+    }));
+}
+function processAIMessageContent(msg) {
+    if (!msg.content) {
+        return [];
+    }
+    if (Array.isArray(msg.content)) {
+        return processArrayContent(msg.content);
+    }
+    return [
+        {
+            role: 'assistant',
+            type: 'message',
+            text: msg.content,
+        },
+    ];
+}
+function createToolCallMessage(toolCall, builderTool) {
+    if (toolCall.name === 'generate_workflow_plan') {
+        const workflowPlan = toolCall.args;
+        return {
+            role: 'assistant',
+            type: 'plan',
+            plan: workflowPlan.plan,
+            message: workflowPlan.intro,
+        };
+    }
+    return {
+        id: toolCall.id,
+        toolCallId: toolCall.id,
+        role: 'assistant',
+        type: 'tool',
+        toolName: toolCall.name,
+        displayTitle: builderTool?.displayTitle,
+        customDisplayTitle: toolCall.args && builderTool?.getCustomDisplayTitle?.(toolCall.args),
+        status: 'completed',
+        updates: [
+            {
+                type: 'input',
+                data: toolCall.args || {},
+            },
+        ],
+    };
+}
+function processToolCalls(toolCalls, builderTools) {
+    return toolCalls.map((toolCall) => {
+        const builderTool = builderTools?.find((bt) => bt.tool.name === toolCall.name);
+        return createToolCallMessage(toolCall, builderTool);
+    });
+}
+function processToolMessage(msg, formattedMessages) {
+    const toolCallId = msg.tool_call_id;
+    for (let i = formattedMessages.length - 1; i >= 0; i--) {
+        const m = formattedMessages[i];
+        if (m.type === 'tool' && m.id === toolCallId) {
+            m.updates ??= [];
+            m.updates.push({
+                type: 'output',
+                data: typeof msg.content === 'string' ? { result: msg.content } : msg.content,
+            });
+            break;
+        }
+    }
+}
+function formatMessages(messages, builderTools) {
     const formattedMessages = [];
     for (const msg of messages) {
         if (msg instanceof messages_1.HumanMessage) {
-            formattedMessages.push({
-                role: 'user',
-                type: 'message',
-                text: msg.content,
-            });
+            formattedMessages.push(formatHumanMessage(msg));
         }
         else if (msg instanceof messages_1.AIMessage) {
-            if (msg.content) {
-                if (Array.isArray(msg.content)) {
-                    const textMessages = msg.content.filter((c) => c.type === 'text');
-                    textMessages.forEach((textMessage) => {
-                        if (textMessage.type !== 'text') {
-                            return;
-                        }
-                        formattedMessages.push({
-                            role: 'assistant',
-                            type: 'message',
-                            text: textMessage.text,
-                        });
-                    });
-                }
-                else {
-                    formattedMessages.push({
-                        role: 'assistant',
-                        type: 'message',
-                        text: msg.content,
-                    });
-                }
-            }
-            if (msg.tool_calls && msg.tool_calls.length > 0) {
-                for (const toolCall of msg.tool_calls) {
-                    formattedMessages.push({
-                        id: toolCall.id,
-                        toolCallId: toolCall.id,
-                        role: 'assistant',
-                        type: 'tool',
-                        toolName: toolCall.name,
-                        status: 'completed',
-                        updates: [
-                            {
-                                type: 'input',
-                                data: toolCall.args || {},
-                            },
-                        ],
-                    });
-                }
+            formattedMessages.push(...processAIMessageContent(msg));
+            if (msg.tool_calls?.length) {
+                formattedMessages.push(...processToolCalls(msg.tool_calls, builderTools));
             }
         }
         else if (msg instanceof messages_1.ToolMessage) {
-            const toolCallId = msg.tool_call_id;
-            for (let i = formattedMessages.length - 1; i >= 0; i--) {
-                const m = formattedMessages[i];
-                if (m.type === 'tool' && m.id === toolCallId) {
-                    m.updates ??= [];
-                    m.updates.push({
-                        type: 'output',
-                        data: typeof msg.content === 'string' ? { result: msg.content } : msg.content,
-                    });
-                    break;
-                }
-            }
+            processToolMessage(msg, formattedMessages);
         }
     }
     return formattedMessages;

package/dist/utils/stream-processor.js.map

@@ -1 +1 @@
-{"version":3,"file":"stream-processor.js","sourceRoot":"","sources":["../../src/utils/stream-processor.ts"],"names":[],"mappings":";;;AAuBA,gDA2FC;AAKD,sDAUC;AAED,wCA4EC;AA/MD,uDAAgF;AAYnE,QAAA,6BAA6B,GAAG;IAC5C,WAAW;IACX,eAAe;IACf,wBAAwB;IACxB,aAAa;CACb,CAAC;AAMF,SAAgB,kBAAkB,CAAC,UAAkB,EAAE,KAAc;IACpE,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAE9B,MAAM,UAAU,GAAG,KAYlB,CAAC;QAEF,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9D,MAAM,YAAY,GAAsB;gBACvC,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,mBAAmB;aACzB,CAAC;YAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,UAAU,EAAE,gBAAgB,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/D,MAAM,WAAW,GAChB,UAAU,CAAC,gBAAiB,CAAC,QAAS,CAAC,UAAU,CAAC,gBAAiB,CAAC,QAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC3F,MAAM,YAAY,GAAsB;gBACvC,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,WAAW,CAAC,OAAiB;aACnC,CAAC;YAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,WAAW,GAAG,UAAU,CAAC,KAAM,CAAC,QAAS,CAAC,UAAU,CAAC,KAAM,CAAC,QAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACxF,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;gBACzB,IAAI,OAAe,CAAC;gBAGpB,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;oBACxC,OAAO,GAAG,WAAW,CAAC,OAAO;yBAC3B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;yBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;yBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,CAAC;qBAAM,CAAC;oBACP,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC;gBAC/B,CAAC;gBAED,MAAM,YAAY,GAAsB;oBACvC,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,OAAO;iBACb,CAAC;gBAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;YACrC,CAAC;QACF,CAAC;QAGD,IAAI,UAAU,EAAE,kBAAkB,EAAE,CAAC;YAEpC,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC;YAC7C,IAAI,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBAEpE,MAAM,mBAAmB,GAAwB;oBAChD,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,kBAAkB;oBACxB,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;iBACzD,CAAC;gBAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAC5C,CAAC;QACF,CAAC;IACF,CAAC;SAAM,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;QAEpC,MAAM,SAAS,GAAG,KAA0B,CAAC;QAE7C,IAAI,SAAS,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YAChC,MAAM,MAAM,GAAiB,EAAE,QAAQ,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC;YAEvD,OAAO,MAAM,CAAC;QACf,CAAC;IACF,CAAC;IAED,OAAO,IAAI,CAAC;AACb,CAAC;AAKM,KAAK,SAAS,CAAC,CAAC,qBAAqB,CAC3C,MAAwD;IAExD,IAAI,KAAK,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;QAChD,MAAM,MAAM,GAAG,kBAAkB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAErD,IAAI,MAAM,EAAE,CAAC;YACZ,MAAM,MAAM,CAAC;QACd,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAgB,cAAc,CAC7B,QAAuD;IAEvD,MAAM,iBAAiB,GAAmC,EAAE,CAAC;IAE7D,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,GAAG,YAAY,uBAAY,EAAE,CAAC;YACjC,iBAAiB,CAAC,IAAI,CAAC;gBACtB,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,GAAG,CAAC,OAAO;aACjB,CAAC,CAAC;QACJ,CAAC;aAAM,IAAI,GAAG,YAAY,oBAAS,EAAE,CAAC;YAErC,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBACjB,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAEhC,MAAM,YAAY,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;oBAElE,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;wBACpC,IAAI,WAAW,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BACjC,OAAO;wBACR,CAAC;wBACD,iBAAiB,CAAC,IAAI,CAAC;4BACtB,IAAI,EAAE,WAAW;4BACjB,IAAI,EAAE,SAAS;4BACf,IAAI,EAAE,WAAW,CAAC,IAAI;yBACtB,CAAC,CAAC;oBACJ,CAAC,CAAC,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACP,iBAAiB,CAAC,IAAI,CAAC;wBACtB,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,SAAS;wBACf,IAAI,EAAE,GAAG,CAAC,OAAO;qBACjB,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAED,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAEjD,KAAK,MAAM,QAAQ,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;oBACvC,iBAAiB,CAAC,IAAI,CAAC;wBACtB,EAAE,EAAE,QAAQ,CAAC,EAAE;wBACf,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,MAAM;wBACZ,QAAQ,EAAE,QAAQ,CAAC,IAAI;wBACvB,MAAM,EAAE,WAAW;wBACnB,OAAO,EAAE;4BACR;gCACC,IAAI,EAAE,OAAO;gCACb,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,EAAE;6BACzB;yBACD;qBACD,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;QACF,CAAC;aAAM,IAAI,GAAG,YAAY,sBAAW,EAAE,CAAC;YAEvC,MAAM,UAAU,GAAG,GAAG,CAAC,YAAY,CAAC;YACpC,KAAK,IAAI,CAAC,GAAG,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACxD,MAAM,CAAC,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;gBAC/B,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,EAAE,KAAK,UAAU,EAAE,CAAC;oBAE9C,CAAC,CAAC,OAAO,KAAK,EAAE,CAAC;oBAChB,CAAC,CAAC,OAA0C,CAAC,IAAI,CAAC;wBAClD,IAAI,EAAE,QAAQ;wBACd,IAAI,EAAE,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO;qBAC7E,CAAC,CAAC;oBACH,MAAM;gBACP,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,iBAAiB,CAAC;AAC1B,CAAC"}
+{"version":3,"file":"stream-processor.js","sourceRoot":"","sources":["../../src/utils/stream-processor.ts"],"names":[],"mappings":";;;AAgCA,gDAoJC;AAKD,sDAUC;AA0HD,wCAuBC;AApVD,uDAAgF;AAqBnE,QAAA,6BAA6B,GAAG;IAC5C,WAAW;IACX,eAAe;IACf,wBAAwB;IACxB,aAAa;CACb,CAAC;AAMF,SAAgB,kBAAkB,CAAC,UAAkB,EAAE,KAAc;IACpE,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAE9B,MAAM,UAAU,GAAG,KA8BlB,CAAC;QAEF,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9D,MAAM,YAAY,GAAsB;gBACvC,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,mBAAmB;aACzB,CAAC;YAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,UAAU,EAAE,gBAAgB,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/D,MAAM,WAAW,GAChB,UAAU,CAAC,gBAAiB,CAAC,QAAS,CAAC,UAAU,CAAC,gBAAiB,CAAC,QAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAE3F,MAAM,YAAY,GAAsB;gBACvC,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,WAAW,CAAC,OAAiB;aACnC,CAAC;YAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,WAAW,GAAG,UAAU,CAAC,KAAM,CAAC,QAAS,CAAC,UAAU,CAAC,KAAM,CAAC,QAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACxF,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;gBACzB,IAAI,OAAe,CAAC;gBAGpB,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;oBACxC,OAAO,GAAG,WAAW,CAAC,OAAO;yBAC3B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;yBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;yBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,CAAC;qBAAM,CAAC;oBACP,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC;gBAC/B,CAAC;gBAED,IAAI,OAAO,EAAE,CAAC;oBACb,MAAM,YAAY,GAAsB;wBACvC,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,SAAS;wBACf,IAAI,EAAE,OAAO;qBACb,CAAC;oBAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;gBACrC,CAAC;gBAED,OAAO,IAAI,CAAC;YACb,CAAC;QACF,CAAC;QAGD,IAAI,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;YAC3C,MAAM,YAAY,GAAG,UAAU,CAAC,WAAW,CAAC,YAA4B,CAAC;YACzE,MAAM,SAAS,GAAG;gBACjB,IAAI,EAAE,WAAoB;gBAC1B,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,YAAY,CAAC,IAAI;gBACvB,OAAO,EAAE,YAAY,CAAC,KAAK;aAC3B,CAAC;YACF,OAAO,EAAE,QAAQ,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC;QAClC,CAAC;aAAM,IAAI,CAAC,UAAU,EAAE,WAAW,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAEjE,MAAM,WAAW,GAChB,UAAU,CAAC,WAAY,CAAC,QAAS,CAAC,UAAU,CAAC,WAAY,CAAC,QAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACjF,MAAM,YAAY,GAAsB;gBACvC,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,WAAW,CAAC,OAAiB;aACnC,CAAC;YAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;YAC3C,MAAM,YAAY,GAAG,UAAU,CAAC,WAAW,CAAC,YAA4B,CAAC;YACzE,MAAM,SAAS,GAAG;gBACjB,IAAI,EAAE,WAAoB;gBAC1B,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,YAAY,CAAC,IAAI;gBACvB,OAAO,EAAE,YAAY,CAAC,KAAK;aAC3B,CAAC;YACF,OAAO,EAAE,QAAQ,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC;QAClC,CAAC;QAGD,IAAI,UAAU,EAAE,kBAAkB,EAAE,CAAC;YAEpC,MAAM,MAAM,GAAG,UAAU,CAAC,kBAAkB,CAAC;YAC7C,IAAI,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBAEpE,MAAM,mBAAmB,GAAwB;oBAChD,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,kBAAkB;oBACxB,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;iBACzD,CAAC;gBAEF,OAAO,EAAE,QAAQ,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAC5C,CAAC;QACF,CAAC;IACF,CAAC;SAAM,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;QAEpC,MAAM,SAAS,GAAG,KAA0B,CAAC;QAE7C,IAAI,SAAS,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YAChC,MAAM,MAAM,GAAiB,EAAE,QAAQ,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC;YAEvD,OAAO,MAAM,CAAC;QACf,CAAC;IACF,CAAC;IAED,OAAO,IAAI,CAAC;AACb,CAAC;AAKM,KAAK,SAAS,CAAC,CAAC,qBAAqB,CAC3C,MAAwD;IAExD,IAAI,KAAK,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;QAChD,MAAM,MAAM,GAAG,kBAAkB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAErD,IAAI,MAAM,EAAE,CAAC;YACZ,MAAM,MAAM,CAAC;QACd,CAAC;IACF,CAAC;AACF,CAAC;AAKD,SAAS,kBAAkB,CAAC,GAAiB;IAC5C,OAAO;QACN,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,GAAG,CAAC,OAAO;KACjB,CAAC;AACH,CAAC;AAKD,SAAS,mBAAmB,CAAC,OAAkB;IAC9C,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,CAClC,CAAC,CAAC,EAAuC,EAAE,CAC1C,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI,IAAI,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,MAAM,IAAI,CAAC,CACvF,CAAC;IAEF,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QACzC,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,WAAW,CAAC,IAAI;KACtB,CAAC,CAAC,CAAC;AACL,CAAC;AAKD,SAAS,uBAAuB,CAAC,GAAc;IAC9C,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAClB,OAAO,EAAE,CAAC;IACX,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAChC,OAAO,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,OAAO;QACN;YACC,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,GAAG,CAAC,OAAO;SACjB;KACD,CAAC;AACH,CAAC;AAKD,SAAS,qBAAqB,CAC7B,QAAkB,EAClB,WAAyB;IAEzB,IAAI,QAAQ,CAAC,IAAI,KAAK,wBAAwB,EAAE,CAAC;QAChD,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAoB,CAAC;QACnD,OAAO;YACN,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,YAAY,CAAC,IAAI;YACvB,OAAO,EAAE,YAAY,CAAC,KAAK;SAC3B,CAAC;IACH,CAAC;IAED,OAAO;QACN,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,UAAU,EAAE,QAAQ,CAAC,EAAE;QACvB,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,MAAM;QACZ,QAAQ,EAAE,QAAQ,CAAC,IAAI;QACvB,YAAY,EAAE,WAAW,EAAE,YAAY;QACvC,kBAAkB,EAAE,QAAQ,CAAC,IAAI,IAAI,WAAW,EAAE,qBAAqB,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;QACxF,MAAM,EAAE,WAAW;QACnB,OAAO,EAAE;YACR;gBACC,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,EAAE;aACzB;SACD;KACD,CAAC;AACH,CAAC;AAKD,SAAS,gBAAgB,CACxB,SAAqB,EACrB,YAA4B;IAE5B,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;QACjC,MAAM,WAAW,GAAG,YAAY,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC/E,OAAO,qBAAqB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACJ,CAAC;AAKD,SAAS,kBAAkB,CAC1B,GAAgB,EAChB,iBAAiD;IAEjD,MAAM,UAAU,GAAG,GAAG,CAAC,YAAY,CAAC;IAGpC,KAAK,IAAI,CAAC,GAAG,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACxD,MAAM,CAAC,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;QAC/B,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,EAAE,KAAK,UAAU,EAAE,CAAC;YAE9C,CAAC,CAAC,OAAO,KAAK,EAAE,CAAC;YAChB,CAAC,CAAC,OAA0C,CAAC,IAAI,CAAC;gBAClD,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO;aAC7E,CAAC,CAAC;YACH,MAAM;QACP,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAgB,cAAc,CAC7B,QAAuD,EACvD,YAA4B;IAE5B,MAAM,iBAAiB,GAAmC,EAAE,CAAC;IAE7D,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,GAAG,YAAY,uBAAY,EAAE,CAAC;YACjC,iBAAiB,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,CAAC;QACjD,CAAC;aAAM,IAAI,GAAG,YAAY,oBAAS,EAAE,CAAC;YAErC,iBAAiB,CAAC,IAAI,CAAC,GAAG,uBAAuB,CAAC,GAAG,CAAC,CAAC,CAAC;YAGxD,IAAI,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;gBAC5B,iBAAiB,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,GAAG,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC;YAC3E,CAAC;QACF,CAAC;aAAM,IAAI,GAAG,YAAY,sBAAW,EAAE,CAAC;YACvC,kBAAkB,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;QAC5C,CAAC;IACF,CAAC;IAED,OAAO,iBAAiB,CAAC;AAC1B,CAAC"}

package/dist/utils/token-usage.d.ts

@@ -1,3 +1,4 @@
+import type { BaseMessage } from '@langchain/core/messages';
 import { AIMessage } from '@langchain/core/messages';
 export type AIMessageWithUsageMetadata = AIMessage & {
     response_metadata: {
@@ -14,3 +15,5 @@ export interface TokenUsage {
     output_tokens: number;
 }
 export declare function extractLastTokenUsage(messages: unknown[]): TokenUsage | undefined;
+export declare function estimateTokenCountFromString(text: string): number;
+export declare function estimateTokenCountFromMessages(messages: BaseMessage[]): number;

package/dist/utils/token-usage.js

@@ -1,7 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.extractLastTokenUsage = extractLastTokenUsage;
+exports.estimateTokenCountFromString = estimateTokenCountFromString;
+exports.estimateTokenCountFromMessages = estimateTokenCountFromMessages;
 const messages_1 = require("@langchain/core/messages");
+const constants_1 = require("../constants");
 function extractLastTokenUsage(messages) {
     const lastAiAssistantMessage = messages.findLast((m) => m instanceof messages_1.AIMessage &&
         m.response_metadata?.usage !== undefined &&
@@ -12,4 +15,28 @@ function extractLastTokenUsage(messages) {
     }
     return lastAiAssistantMessage.response_metadata.usage;
 }
+function concatenateMessageContent(messages) {
+    return messages.reduce((acc, message) => {
+        if (typeof message.content === 'string') {
+            return acc + message.content;
+        }
+        else if (Array.isArray(message.content)) {
+            return (acc +
+                message.content.reduce((innerAcc, item) => {
+                    if (typeof item === 'object' && item !== null && 'text' in item) {
+                        return innerAcc + item.text;
+                    }
+                    return innerAcc;
+                }, ''));
+        }
+        return acc;
+    }, '');
+}
+function estimateTokenCountFromString(text) {
+    return Math.ceil(text.length / constants_1.AVG_CHARS_PER_TOKEN_ANTHROPIC);
+}
+function estimateTokenCountFromMessages(messages) {
+    const entireInput = concatenateMessageContent(messages);
+    return estimateTokenCountFromString(entireInput);
+}
 //# sourceMappingURL=token-usage.js.map

package/dist/utils/token-usage.js.map

@@ -1 +1 @@
-{"version":3,"file":"token-usage.js","sourceRoot":"","sources":["../../src/utils/token-usage.ts"],"names":[],"mappings":";;AAqBA,sDAcC;AAnCD,uDAAqD;AAqBrD,SAAgB,qBAAqB,CAAC,QAAmB;IACxD,MAAM,sBAAsB,GAAG,QAAQ,CAAC,QAAQ,CAC/C,CAAC,CAAC,EAAmC,EAAE,CACtC,CAAC,YAAY,oBAAS;QACtB,CAAC,CAAC,iBAAiB,EAAE,KAAK,KAAK,SAAS;QACxC,cAAc,IAAI,CAAC,CAAC,iBAAiB,CAAC,KAAK;QAC3C,eAAe,IAAI,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAC7C,CAAC;IAEF,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC7B,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,OAAO,sBAAsB,CAAC,iBAAiB,CAAC,KAAK,CAAC;AACvD,CAAC"}
+{"version":3,"file":"token-usage.js","sourceRoot":"","sources":["../../src/utils/token-usage.ts"],"names":[],"mappings":";;AAwBA,sDAcC;AAqBD,oEAEC;AAED,wEAIC;AAlED,uDAAqD;AAErD,2CAA4D;AAqB5D,SAAgB,qBAAqB,CAAC,QAAmB;IACxD,MAAM,sBAAsB,GAAG,QAAQ,CAAC,QAAQ,CAC/C,CAAC,CAAC,EAAmC,EAAE,CACtC,CAAC,YAAY,oBAAS;QACtB,CAAC,CAAC,iBAAiB,EAAE,KAAK,KAAK,SAAS;QACxC,cAAc,IAAI,CAAC,CAAC,iBAAiB,CAAC,KAAK;QAC3C,eAAe,IAAI,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAC7C,CAAC;IAEF,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC7B,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,OAAO,sBAAsB,CAAC,iBAAiB,CAAC,KAAK,CAAC;AACvD,CAAC;AAED,SAAS,yBAAyB,CAAC,QAAuB;IACzD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAW,EAAE,OAAO,EAAE,EAAE;QAC/C,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACzC,OAAO,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC;QAC9B,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3C,OAAO,CACN,GAAG;gBACH,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAgB,EAAE,IAAI,EAAE,EAAE;oBACjD,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;wBACjE,OAAO,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;oBAC7B,CAAC;oBACD,OAAO,QAAQ,CAAC;gBACjB,CAAC,EAAE,EAAE,CAAC,CACN,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAC;IACZ,CAAC,EAAE,EAAE,CAAC,CAAC;AACR,CAAC;AAED,SAAgB,4BAA4B,CAAC,IAAY;IACxD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,yCAA6B,CAAC,CAAC;AAC/D,CAAC;AAED,SAAgB,8BAA8B,CAAC,QAAuB;IACrE,MAAM,WAAW,GAAG,yBAAyB,CAAC,QAAQ,CAAC,CAAC;IAExD,OAAO,4BAA4B,CAAC,WAAW,CAAC,CAAC;AAClD,CAAC"}

package/dist/utils/trim-workflow-context.d.ts

@@ -0,0 +1,2 @@
+import type { SimpleWorkflow } from '../types';
+export declare function trimWorkflowJSON(workflow: SimpleWorkflow): SimpleWorkflow;

package/dist/utils/trim-workflow-context.js

@@ -0,0 +1,53 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.trimWorkflowJSON = trimWorkflowJSON;
+const constants_1 = require("../constants");
+const token_usage_1 = require("../utils/token-usage");
+const MAX_PARAMETER_VALUE_LENGTH_THRESHOLDS = [10000, 5000, 2000, 1000];
+function trimParameterValue(value, threshold) {
+    if (value === undefined || value === null) {
+        return value;
+    }
+    const valueStr = JSON.stringify(value);
+    if (valueStr.length > threshold) {
+        if (typeof value === 'string') {
+            return '[Large value omitted]';
+        }
+        else if (Array.isArray(value)) {
+            return '[Large array omitted]';
+        }
+        else if (typeof value === 'object' && value !== null) {
+            return '[Large object omitted]';
+        }
+    }
+    return value;
+}
+function trimWorkflowJsonWithThreshold(workflow, threshold) {
+    const simplifiedWorkflow = { ...workflow };
+    if (simplifiedWorkflow.nodes) {
+        simplifiedWorkflow.nodes = simplifiedWorkflow.nodes.map((node) => {
+            const simplifiedNode = { ...node };
+            if (simplifiedNode.parameters) {
+                const simplifiedParameters = {};
+                for (const [key, value] of Object.entries(simplifiedNode.parameters)) {
+                    simplifiedParameters[key] = trimParameterValue(value, threshold);
+                }
+                simplifiedNode.parameters = simplifiedParameters;
+            }
+            return simplifiedNode;
+        });
+    }
+    return simplifiedWorkflow;
+}
+function trimWorkflowJSON(workflow) {
+    for (const threshold of MAX_PARAMETER_VALUE_LENGTH_THRESHOLDS) {
+        const simplified = trimWorkflowJsonWithThreshold(workflow, threshold);
+        const workflowStr = JSON.stringify(simplified);
+        const estimatedTokens = (0, token_usage_1.estimateTokenCountFromString)(workflowStr);
+        if (estimatedTokens <= constants_1.MAX_WORKFLOW_LENGTH_TOKENS) {
+            return simplified;
+        }
+    }
+    return trimWorkflowJsonWithThreshold(workflow, MAX_PARAMETER_VALUE_LENGTH_THRESHOLDS[MAX_PARAMETER_VALUE_LENGTH_THRESHOLDS.length - 1]);
+}
+//# sourceMappingURL=trim-workflow-context.js.map

package/dist/utils/trim-workflow-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"trim-workflow-context.js","sourceRoot":"","sources":["../../src/utils/trim-workflow-context.ts"],"names":[],"mappings":";;AAoFA,4CAmBC;AArGD,2CAAyD;AAEzD,qDAAmE;AAMnE,MAAM,qCAAqC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAUxE,SAAS,kBAAkB,CAC1B,KAA6B,EAC7B,SAAiB;IAGjB,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC3C,OAAO,KAAK,CAAC;IACd,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACvC,IAAI,QAAQ,CAAC,MAAM,GAAG,SAAS,EAAE,CAAC;QAEjC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC/B,OAAO,uBAAuB,CAAC;QAChC,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,uBAAuB,CAAC;QAChC,CAAC;aAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YACxD,OAAO,wBAAwB,CAAC;QACjC,CAAC;IACF,CAAC;IAED,OAAO,KAAK,CAAC;AACd,CAAC;AAUD,SAAS,6BAA6B,CACrC,QAAwB,EACxB,SAAiB;IAEjB,MAAM,kBAAkB,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;IAC3C,IAAI,kBAAkB,CAAC,KAAK,EAAE,CAAC;QAC9B,kBAAkB,CAAC,KAAK,GAAG,kBAAkB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAChE,MAAM,cAAc,GAAG,EAAE,GAAG,IAAI,EAAE,CAAC;YAGnC,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC;gBAC/B,MAAM,oBAAoB,GAAoB,EAAE,CAAC;gBACjD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;oBACtE,oBAAoB,CAAC,GAAG,CAAC,GAAG,kBAAkB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;gBAClE,CAAC;gBACD,cAAc,CAAC,UAAU,GAAG,oBAAoB,CAAC;YAClD,CAAC;YAED,OAAO,cAAc,CAAC;QACvB,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,kBAAkB,CAAC;AAC3B,CAAC;AASD,SAAgB,gBAAgB,CAAC,QAAwB;IAExD,KAAK,MAAM,SAAS,IAAI,qCAAqC,EAAE,CAAC;QAC/D,MAAM,UAAU,GAAG,6BAA6B,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QACtE,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC/C,MAAM,eAAe,GAAG,IAAA,0CAA4B,EAAC,WAAW,CAAC,CAAC;QAGlE,IAAI,eAAe,IAAI,sCAA0B,EAAE,CAAC;YACnD,OAAO,UAAU,CAAC;QACnB,CAAC;IACF,CAAC;IAID,OAAO,6BAA6B,CACnC,QAAQ,EACR,qCAAqC,CAAC,qCAAqC,CAAC,MAAM,GAAG,CAAC,CAAC,CACvF,CAAC;AACH,CAAC"}