@atom8n/ai-workflow-builder 1.2.0

package/dist/utils/state-reducers.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.appendArrayReducer = appendArrayReducer;
+exports.cachedTemplatesReducer = cachedTemplatesReducer;
+function appendArrayReducer(current, update) {
+    return update && update.length > 0 ? [...current, ...update] : current;
+}
+function cachedTemplatesReducer(current, update) {
+    if (!update || update.length === 0) {
+        return current;
+    }
+    const existingById = new Map(current.map((wf) => [wf.templateId, wf]));
+    for (const workflow of update) {
+        if (!existingById.has(workflow.templateId)) {
+            existingById.set(workflow.templateId, workflow);
+        }
+    }
+    return Array.from(existingById.values());
+}
+//# sourceMappingURL=state-reducers.js.map

package/dist/utils/state-reducers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"state-reducers.js","sourceRoot":"","sources":["../../src/utils/state-reducers.ts"],"names":[],"mappings":";;AAMA,gDAEC;AAMD,wDAmBC;AA3BD,SAAgB,kBAAkB,CAAI,OAAY,EAAE,MAA8B;IACjF,OAAO,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;AACxE,CAAC;AAMD,SAAgB,sBAAsB,CACrC,OAA2B,EAC3B,MAA6C;IAE7C,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACpC,OAAO,OAAO,CAAC;IAChB,CAAC;IAGD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;IAGvE,KAAK,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;QAC/B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC5C,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACjD,CAAC;IACF,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;AAC1C,CAAC"}

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

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

package/dist/utils/stream-processor.js

@@ -0,0 +1,277 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_WORKFLOW_UPDATE_TOOLS = void 0;
+exports.cleanContextTags = cleanContextTags;
+exports.processStreamChunk = processStreamChunk;
+exports.createStreamProcessor = createStreamProcessor;
+exports.formatMessages = formatMessages;
+const messages_1 = require("@langchain/core/messages");
+exports.DEFAULT_WORKFLOW_UPDATE_TOOLS = [
+    'add_nodes',
+    'connect_nodes',
+    'update_node_parameters',
+    'remove_node',
+];
+const EMITTING_NODES = ['agent', 'responder'];
+const SKIPPED_NODES = [
+    'supervisor',
+    'tools',
+    'cleanup_dangling_tool_calls',
+    'create_workflow_name',
+    'auto_compact_messages',
+    'configurator_subgraph',
+    'discovery_subgraph',
+    'builder_subgraph',
+];
+const SKIPPED_SUBGRAPH_PREFIXES = [
+    'discovery_subgraph',
+    'builder_subgraph',
+    'configurator_subgraph',
+];
+function isFromSkippedSubgraph(namespace) {
+    return namespace.some((ns) => SKIPPED_SUBGRAPH_PREFIXES.some((prefix) => ns.startsWith(prefix)));
+}
+function shouldSkipNode(nodeName) {
+    return SKIPPED_NODES.includes(nodeName);
+}
+function shouldEmitFromNode(nodeName) {
+    return EMITTING_NODES.includes(nodeName);
+}
+function hasMessageInUpdate(update) {
+    const typed = update;
+    return Array.isArray(typed?.messages) && typed.messages.length > 0;
+}
+function shouldFilterSubgraphUpdate(namespace, data) {
+    if (!isFromSkippedSubgraph(namespace))
+        return false;
+    return Object.entries(data).some(([nodeName, update]) => {
+        if (shouldSkipNode(nodeName))
+            return false;
+        return hasMessageInUpdate(update);
+    });
+}
+function isSubgraphEvent(event) {
+    return Array.isArray(event) && event.length === 3 && Array.isArray(event[0]);
+}
+function isParentEvent(event) {
+    return Array.isArray(event) && event.length === 2 && typeof event[0] === 'string';
+}
+function extractMessageContent(messages) {
+    if (messages.length === 0)
+        return null;
+    const lastMessage = messages[messages.length - 1];
+    if (!lastMessage.content)
+        return null;
+    if (Array.isArray(lastMessage.content)) {
+        const textContent = lastMessage.content
+            .filter((c) => c.type === 'text')
+            .map((c) => c.text)
+            .join('\n');
+        return textContent || null;
+    }
+    return lastMessage.content;
+}
+function cleanContextTags(text) {
+    return text.replace(/\n*<current_workflow_json>[\s\S]*?<\/current_execution_nodes_schemas>/, '');
+}
+function processOperationsUpdate(update) {
+    const typed = update;
+    if (!typed?.workflowJSON || typed.workflowOperations === undefined)
+        return null;
+    const workflowUpdateChunk = {
+        role: 'assistant',
+        type: 'workflow-updated',
+        codeSnippet: JSON.stringify(typed.workflowJSON, null, 2),
+    };
+    return { messages: [workflowUpdateChunk] };
+}
+function processAgentNodeUpdate(nodeName, update) {
+    if (!shouldEmitFromNode(nodeName))
+        return null;
+    const typed = update;
+    if (!typed?.messages?.length)
+        return null;
+    const content = extractMessageContent(typed.messages);
+    if (!content?.trim() || content.includes('<current_workflow_json>'))
+        return null;
+    const messageChunk = {
+        role: 'assistant',
+        type: 'message',
+        text: content,
+    };
+    return { messages: [messageChunk] };
+}
+function processToolChunk(chunk) {
+    const typed = chunk;
+    if (typed?.type !== 'tool')
+        return null;
+    return { messages: [typed] };
+}
+function processUpdatesChunk(nodeUpdate) {
+    if (!nodeUpdate || typeof nodeUpdate !== 'object')
+        return null;
+    if (nodeUpdate.delete_messages || nodeUpdate.compact_messages) {
+        return null;
+    }
+    if (nodeUpdate.process_operations) {
+        return processOperationsUpdate(nodeUpdate.process_operations);
+    }
+    for (const [nodeName, update] of Object.entries(nodeUpdate)) {
+        if (shouldSkipNode(nodeName))
+            continue;
+        const result = processAgentNodeUpdate(nodeName, update);
+        if (result)
+            return result;
+    }
+    return null;
+}
+function processStreamChunk(streamMode, chunk) {
+    if (streamMode === 'updates') {
+        return processUpdatesChunk(chunk);
+    }
+    if (streamMode === 'custom') {
+        return processToolChunk(chunk);
+    }
+    return null;
+}
+function processSubgraphEvent(event) {
+    const [namespace, streamMode, data] = event;
+    if (streamMode === 'updates' &&
+        shouldFilterSubgraphUpdate(namespace, data)) {
+        return null;
+    }
+    return processStreamChunk(streamMode, data);
+}
+function processParentEvent(event) {
+    const [streamMode, chunk] = event;
+    if (!streamMode || typeof streamMode !== 'string')
+        return null;
+    return processStreamChunk(streamMode, chunk);
+}
+function processEvent(event) {
+    if (isSubgraphEvent(event)) {
+        return processSubgraphEvent(event);
+    }
+    if (isParentEvent(event)) {
+        return processParentEvent(event);
+    }
+    return null;
+}
+async function* createStreamProcessor(stream) {
+    for await (const event of stream) {
+        const result = processEvent(event);
+        if (result) {
+            yield result;
+        }
+    }
+}
+function extractHumanMessageText(content) {
+    if (typeof content === 'string') {
+        return content;
+    }
+    if (Array.isArray(content)) {
+        return content
+            .filter((c) => typeof c === 'object' && c !== null && 'type' in c && c.type === 'text' && 'text' in c)
+            .map((c) => c.text)
+            .join('\n');
+    }
+    return '';
+}
+function formatHumanMessage(msg) {
+    const rawText = extractHumanMessageText(msg.content);
+    const cleanedText = cleanContextTags(rawText);
+    const result = {
+        role: 'user',
+        type: 'message',
+        text: cleanedText,
+    };
+    const versionId = msg.additional_kwargs?.versionId;
+    if (typeof versionId === 'string') {
+        result.revertVersionId = versionId;
+    }
+    const messageId = msg.additional_kwargs?.messageId;
+    if (typeof messageId === 'string') {
+        result.id = messageId;
+    }
+    return result;
+}
+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) {
+    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.toolName === 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(formatHumanMessage(msg));
+        }
+        else if (msg instanceof messages_1.AIMessage) {
+            formattedMessages.push(...processAIMessageContent(msg));
+            if (msg.tool_calls?.length) {
+                formattedMessages.push(...processToolCalls(msg.tool_calls, builderTools));
+            }
+        }
+        else if (msg instanceof messages_1.ToolMessage) {
+            processToolMessage(msg, formattedMessages);
+        }
+    }
+    return formattedMessages;
+}
+//# sourceMappingURL=stream-processor.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"stream-processor.js","sourceRoot":"","sources":["../../src/utils/stream-processor.ts"],"names":[],"mappings":";;;AAyJA,4CAEC;AA2ED,gDAUC;AA6CD,sDASC;AA6ID,wCAuBC;AAtcD,uDAAgF;AAsCnE,QAAA,6BAA6B,GAAG;IAC5C,WAAW;IACX,eAAe;IACf,wBAAwB;IACxB,aAAa;CACb,CAAC;AAOF,MAAM,cAAc,GAAG,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;AAG9C,MAAM,aAAa,GAAG;IACrB,YAAY;IACZ,OAAO;IACP,6BAA6B;IAC7B,sBAAsB;IACtB,uBAAuB;IACvB,uBAAuB;IACvB,oBAAoB;IACpB,kBAAkB;CAClB,CAAC;AAMF,MAAM,yBAAyB,GAAG;IACjC,oBAAoB;IACpB,kBAAkB;IAClB,uBAAuB;CACvB,CAAC;AAOF,SAAS,qBAAqB,CAAC,SAAmB;IACjD,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;AAClG,CAAC;AAGD,SAAS,cAAc,CAAC,QAAgB;IACvC,OAAO,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AACzC,CAAC;AAGD,SAAS,kBAAkB,CAAC,QAAgB;IAC3C,OAAO,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AAC1C,CAAC;AAGD,SAAS,kBAAkB,CAAC,MAAe;IAC1C,MAAM,KAAK,GAAG,MAAkC,CAAC;IACjD,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;AACpE,CAAC;AAGD,SAAS,0BAA0B,CAAC,SAAmB,EAAE,IAA6B;IACrF,IAAI,CAAC,qBAAqB,CAAC,SAAS,CAAC;QAAE,OAAO,KAAK,CAAC;IAEpD,OAAO,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE;QACvD,IAAI,cAAc,CAAC,QAAQ,CAAC;YAAE,OAAO,KAAK,CAAC;QAC3C,OAAO,kBAAkB,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;AACJ,CAAC;AAGD,SAAS,eAAe,CAAC,KAAc;IACtC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;AAC9E,CAAC;AAGD,SAAS,aAAa,CAAC,KAAc;IACpC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC;AACnF,CAAC;AAOD,SAAS,qBAAqB,CAAC,QAA0B;IACxD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAEvC,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClD,IAAI,CAAC,WAAW,CAAC,OAAO;QAAE,OAAO,IAAI,CAAC;IAGtC,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;QACxC,MAAM,WAAW,GAAG,WAAW,CAAC,OAAO;aACrC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAClB,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,OAAO,WAAW,IAAI,IAAI,CAAC;IAC5B,CAAC;IAED,OAAO,WAAW,CAAC,OAAO,CAAC;AAC5B,CAAC;AASD,SAAgB,gBAAgB,CAAC,IAAY;IAC5C,OAAO,IAAI,CAAC,OAAO,CAAC,uEAAuE,EAAE,EAAE,CAAC,CAAC;AAClG,CAAC;AAOD,SAAS,uBAAuB,CAAC,MAAe;IAC/C,MAAM,KAAK,GAAG,MAA8E,CAAC;IAC7F,IAAI,CAAC,KAAK,EAAE,YAAY,IAAI,KAAK,CAAC,kBAAkB,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IAEhF,MAAM,mBAAmB,GAAwB;QAChD,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;KACxD,CAAC;IACF,OAAO,EAAE,QAAQ,EAAE,CAAC,mBAAmB,CAAC,EAAE,CAAC;AAC5C,CAAC;AAGD,SAAS,sBAAsB,CAAC,QAAgB,EAAE,MAAe;IAChE,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC;QAAE,OAAO,IAAI,CAAC;IAE/C,MAAM,KAAK,GAAG,MAAqD,CAAC;IACpE,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM;QAAE,OAAO,IAAI,CAAC;IAE1C,MAAM,OAAO,GAAG,qBAAqB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAEtD,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,QAAQ,CAAC,yBAAyB,CAAC;QAAE,OAAO,IAAI,CAAC;IAEjF,MAAM,YAAY,GAAsB;QACvC,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,OAAO;KACb,CAAC;IACF,OAAO,EAAE,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;AACrC,CAAC;AAGD,SAAS,gBAAgB,CAAC,KAAc;IACvC,MAAM,KAAK,GAAG,KAA0B,CAAC;IACzC,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM;QAAE,OAAO,IAAI,CAAC;IAExC,OAAO,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;AAC9B,CAAC;AAOD,SAAS,mBAAmB,CAAC,UAAmC;IAC/D,IAAI,CAAC,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAE/D,IAAI,UAAU,CAAC,eAAe,IAAI,UAAU,CAAC,gBAAgB,EAAE,CAAC;QAC/D,OAAO,IAAI,CAAC;IACb,CAAC;IAGD,IAAI,UAAU,CAAC,kBAAkB,EAAE,CAAC;QACnC,OAAO,uBAAuB,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;IAC/D,CAAC;IAGD,KAAK,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC7D,IAAI,cAAc,CAAC,QAAQ,CAAC;YAAE,SAAS;QAEvC,MAAM,MAAM,GAAG,sBAAsB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACxD,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;IAC3B,CAAC;IAED,OAAO,IAAI,CAAC;AACb,CAAC;AAGD,SAAgB,kBAAkB,CAAC,UAAkB,EAAE,KAAc;IACpE,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC9B,OAAO,mBAAmB,CAAC,KAAgC,CAAC,CAAC;IAC9D,CAAC;IAED,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,IAAI,CAAC;AACb,CAAC;AAGD,SAAS,oBAAoB,CAAC,KAAoB;IACjD,MAAM,CAAC,SAAS,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,KAAK,CAAC;IAG5C,IACC,UAAU,KAAK,SAAS;QACxB,0BAA0B,CAAC,SAAS,EAAE,IAA+B,CAAC,EACrE,CAAC;QACF,OAAO,IAAI,CAAC;IACb,CAAC;IAED,OAAO,kBAAkB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;AAC7C,CAAC;AAGD,SAAS,kBAAkB,CAAC,KAAkB;IAC7C,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,GAAG,KAAK,CAAC;IAClC,IAAI,CAAC,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAE/D,OAAO,kBAAkB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;AAC9C,CAAC;AAGD,SAAS,YAAY,CAAC,KAAkB;IACvC,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,oBAAoB,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,OAAO,IAAI,CAAC;AACb,CAAC;AASM,KAAK,SAAS,CAAC,CAAC,qBAAqB,CAC3C,MAAkC;IAElC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QACnC,IAAI,MAAM,EAAE,CAAC;YACZ,MAAM,MAAM,CAAC;QACd,CAAC;IACF,CAAC;AACF,CAAC;AAOD,SAAS,uBAAuB,CAAC,OAAgC;IAChE,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,OAAO,CAAC;IAChB,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,OAAO,OAAO;aACZ,MAAM,CACN,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;aACA,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAClB,IAAI,CAAC,IAAI,CAAC,CAAC;IACd,CAAC;IAED,OAAO,EAAE,CAAC;AACX,CAAC;AAGD,SAAS,kBAAkB,CAAC,GAAiB;IAC5C,MAAM,OAAO,GAAG,uBAAuB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACrD,MAAM,WAAW,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAE9C,MAAM,MAAM,GAA4B;QACvC,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,WAAW;KACjB,CAAC;IAGF,MAAM,SAAS,GAAG,GAAG,CAAC,iBAAiB,EAAE,SAAS,CAAC;IACnD,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;QACnC,MAAM,CAAC,eAAe,GAAG,SAAS,CAAC;IACpC,CAAC;IAGD,MAAM,SAAS,GAAG,GAAG,CAAC,iBAAiB,EAAE,SAAS,CAAC;IACnD,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;QACnC,MAAM,CAAC,EAAE,GAAG,SAAS,CAAC;IACvB,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAGD,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;AAGD,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;AAGD,SAAS,qBAAqB,CAC7B,QAAkB,EAClB,WAA6B;IAE7B,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;AAGD,SAAS,gBAAgB,CACxB,SAAqB,EACrB,YAAgC;IAEhC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;QACjC,MAAM,WAAW,GAAG,YAAY,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC9E,OAAO,qBAAqB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACJ,CAAC;AAGD,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;AAGD,SAAgB,cAAc,CAC7B,QAAuD,EACvD,YAAgC;IAEhC,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/subgraph-helpers.d.ts

@@ -0,0 +1,16 @@
+import type { BaseMessage } from '@langchain/core/messages';
+import type { StructuredTool } from '@langchain/core/tools';
+import type { WorkflowMetadata } from '../types/tools';
+import type { WorkflowOperation } from '../types/workflow';
+export declare function executeSubgraphTools(state: {
+    messages: BaseMessage[];
+}, toolMap: Map<string, StructuredTool>): Promise<{
+    messages?: BaseMessage[];
+    workflowOperations?: WorkflowOperation[] | null;
+    templateIds?: number[];
+    cachedTemplates?: WorkflowMetadata[];
+}>;
+export declare function extractUserRequest(messages: BaseMessage[], defaultValue?: string): string;
+export declare function createStandardShouldContinue(): (state: {
+    messages: BaseMessage[];
+}) => "tools" | "__end__";

package/dist/utils/subgraph-helpers.js

@@ -0,0 +1,117 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.executeSubgraphTools = executeSubgraphTools;
+exports.extractUserRequest = extractUserRequest;
+exports.createStandardShouldContinue = createStandardShouldContinue;
+const messages_1 = require("@langchain/core/messages");
+const langgraph_1 = require("@langchain/langgraph");
+const langchain_1 = require("../types/langchain");
+function isCommandUpdate(value) {
+    if (typeof value !== 'object' || value === null) {
+        return false;
+    }
+    const obj = value;
+    if ('messages' in obj && obj.messages !== undefined && !Array.isArray(obj.messages)) {
+        return false;
+    }
+    if ('workflowOperations' in obj &&
+        obj.workflowOperations !== undefined &&
+        !Array.isArray(obj.workflowOperations)) {
+        return false;
+    }
+    if ('templateIds' in obj && obj.templateIds !== undefined && !Array.isArray(obj.templateIds)) {
+        return false;
+    }
+    if ('cachedTemplates' in obj &&
+        obj.cachedTemplates !== undefined &&
+        !Array.isArray(obj.cachedTemplates)) {
+        return false;
+    }
+    return true;
+}
+async function executeSubgraphTools(state, toolMap) {
+    const lastMessage = state.messages[state.messages.length - 1];
+    if (!lastMessage || !(0, messages_1.isAIMessage)(lastMessage) || !lastMessage.tool_calls?.length) {
+        return {};
+    }
+    const toolResults = await Promise.all(lastMessage.tool_calls.map(async (toolCall) => {
+        const tool = toolMap.get(toolCall.name);
+        if (!tool) {
+            return new messages_1.ToolMessage({
+                content: `Tool ${toolCall.name} not found`,
+                tool_call_id: toolCall.id ?? '',
+            });
+        }
+        try {
+            const result = await tool.invoke(toolCall.args ?? {}, {
+                toolCall: {
+                    id: toolCall.id,
+                    name: toolCall.name,
+                    args: toolCall.args ?? {},
+                },
+            });
+            return result;
+        }
+        catch (error) {
+            return new messages_1.ToolMessage({
+                content: `Tool failed: ${error instanceof Error ? error.message : 'Unknown error'}`,
+                tool_call_id: toolCall.id ?? '',
+            });
+        }
+    }));
+    const messages = [];
+    const operations = [];
+    const templateIds = [];
+    const cachedTemplates = [];
+    for (const result of toolResults) {
+        if ((0, langgraph_1.isCommand)(result)) {
+            if (isCommandUpdate(result.update)) {
+                if (result.update.messages) {
+                    messages.push(...result.update.messages);
+                }
+                if (result.update.workflowOperations) {
+                    operations.push(...result.update.workflowOperations);
+                }
+                if (result.update.templateIds) {
+                    templateIds.push(...result.update.templateIds);
+                }
+                if (result.update.cachedTemplates) {
+                    cachedTemplates.push(...result.update.cachedTemplates);
+                }
+            }
+        }
+        else if ((0, langchain_1.isBaseMessage)(result)) {
+            messages.push(result);
+        }
+    }
+    const stateUpdate = {};
+    if (messages.length > 0) {
+        stateUpdate.messages = messages;
+    }
+    if (operations.length > 0) {
+        stateUpdate.workflowOperations = operations;
+    }
+    if (templateIds.length > 0) {
+        stateUpdate.templateIds = templateIds;
+    }
+    if (cachedTemplates.length > 0) {
+        stateUpdate.cachedTemplates = cachedTemplates;
+    }
+    return stateUpdate;
+}
+function extractUserRequest(messages, defaultValue = '') {
+    const humanMessages = messages.filter((m) => m instanceof messages_1.HumanMessage);
+    const lastUserMessage = humanMessages[humanMessages.length - 1];
+    return typeof lastUserMessage?.content === 'string' ? lastUserMessage.content : defaultValue;
+}
+function createStandardShouldContinue() {
+    return (state) => {
+        const lastMessage = state.messages[state.messages.length - 1];
+        const hasToolCalls = lastMessage &&
+            'tool_calls' in lastMessage &&
+            Array.isArray(lastMessage.tool_calls) &&
+            lastMessage.tool_calls.length > 0;
+        return hasToolCalls ? 'tools' : langgraph_1.END;
+    };
+}
+//# sourceMappingURL=subgraph-helpers.js.map

package/dist/utils/subgraph-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"subgraph-helpers.js","sourceRoot":"","sources":["../../src/utils/subgraph-helpers.ts"],"names":[],"mappings":";;AA6DA,oDAmGC;AAMD,gDAKC;AAMD,oEAWC;AA3LD,uDAAkF;AAElF,oDAAsD;AAEtD,kDAAmD;AAcnD,SAAS,eAAe,CAAC,KAAc;IACtC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACjD,OAAO,KAAK,CAAC;IACd,CAAC;IACD,MAAM,GAAG,GAAG,KAAgC,CAAC;IAE7C,IAAI,UAAU,IAAI,GAAG,IAAI,GAAG,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrF,OAAO,KAAK,CAAC;IACd,CAAC;IAED,IACC,oBAAoB,IAAI,GAAG;QAC3B,GAAG,CAAC,kBAAkB,KAAK,SAAS;QACpC,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,EACrC,CAAC;QACF,OAAO,KAAK,CAAC;IACd,CAAC;IAED,IAAI,aAAa,IAAI,GAAG,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9F,OAAO,KAAK,CAAC;IACd,CAAC;IAED,IACC,iBAAiB,IAAI,GAAG;QACxB,GAAG,CAAC,eAAe,KAAK,SAAS;QACjC,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,EAClC,CAAC;QACF,OAAO,KAAK,CAAC;IACd,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAYM,KAAK,UAAU,oBAAoB,CACzC,KAAkC,EAClC,OAAoC;IAOpC,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAE9D,IAAI,CAAC,WAAW,IAAI,CAAC,IAAA,sBAAW,EAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;QAClF,OAAO,EAAE,CAAC;IACX,CAAC;IAGD,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;QAC7C,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,OAAO,IAAI,sBAAW,CAAC;gBACtB,OAAO,EAAE,QAAQ,QAAQ,CAAC,IAAI,YAAY;gBAC1C,YAAY,EAAE,QAAQ,CAAC,EAAE,IAAI,EAAE;aAC/B,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACJ,MAAM,MAAM,GAAY,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,EAAE;gBAC9D,QAAQ,EAAE;oBACT,EAAE,EAAE,QAAQ,CAAC,EAAE;oBACf,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,EAAE;iBACzB;aACD,CAAC,CAAC;YAGH,OAAO,MAAM,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,IAAI,sBAAW,CAAC;gBACtB,OAAO,EAAE,gBAAgB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE;gBACnF,YAAY,EAAE,QAAQ,CAAC,EAAE,IAAI,EAAE;aAC/B,CAAC,CAAC;QACJ,CAAC;IACF,CAAC,CAAC,CACF,CAAC;IAGF,MAAM,QAAQ,GAAkB,EAAE,CAAC;IACnC,MAAM,UAAU,GAAwB,EAAE,CAAC;IAC3C,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,MAAM,eAAe,GAAuB,EAAE,CAAC;IAE/C,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,IAAA,qBAAS,EAAC,MAAM,CAAC,EAAE,CAAC;YAEvB,IAAI,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;gBACpC,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;oBAC5B,QAAQ,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBAC1C,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;oBACtC,UAAU,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;gBACtD,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;oBAC/B,WAAW,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;gBAChD,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;oBACnC,eAAe,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;gBACxD,CAAC;YACF,CAAC;QACF,CAAC;aAAM,IAAI,IAAA,yBAAa,EAAC,MAAM,CAAC,EAAE,CAAC;YAElC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC;IACF,CAAC;IAED,MAAM,WAAW,GAKb,EAAE,CAAC;IAEP,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,WAAW,CAAC,QAAQ,GAAG,QAAQ,CAAC;IACjC,CAAC;IAED,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,WAAW,CAAC,kBAAkB,GAAG,UAAU,CAAC;IAC7C,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,WAAW,CAAC,WAAW,GAAG,WAAW,CAAC;IACvC,CAAC;IAED,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,WAAW,CAAC,eAAe,GAAG,eAAe,CAAC;IAC/C,CAAC;IAED,OAAO,WAAW,CAAC;AACpB,CAAC;AAMD,SAAgB,kBAAkB,CAAC,QAAuB,EAAE,YAAY,GAAG,EAAE;IAE5E,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,uBAAY,CAAC,CAAC;IACxE,MAAM,eAAe,GAAG,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAChE,OAAO,OAAO,eAAe,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC;AAC9F,CAAC;AAMD,SAAgB,4BAA4B;IAC3C,OAAO,CAAC,KAAkC,EAAE,EAAE;QAC7C,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC9D,MAAM,YAAY,GACjB,WAAW;YACX,YAAY,IAAI,WAAW;YAC3B,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC;YACrC,WAAW,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC;QAEnC,OAAO,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,eAAG,CAAC;IACrC,CAAC,CAAC;AACH,CAAC"}

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

@@ -0,0 +1,21 @@
+import type { BaseMessage } from '@langchain/core/messages';
+import { AIMessage } from '@langchain/core/messages';
+export type AIMessageWithUsageMetadata = AIMessage & {
+    response_metadata: {
+        usage: {
+            input_tokens: number;
+            output_tokens: number;
+            cache_read_input_tokens?: number;
+            cache_creation_input_tokens?: number;
+        };
+    };
+};
+export interface TokenUsage {
+    input_tokens: number;
+    output_tokens: number;
+    cache_read_input_tokens?: number;
+    cache_creation_input_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

@@ -0,0 +1,45 @@
+"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 &&
+        'input_tokens' in m.response_metadata.usage &&
+        'output_tokens' in m.response_metadata.usage);
+    if (!lastAiAssistantMessage) {
+        return undefined;
+    }
+    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 &&
+                        typeof item.text === 'string') {
+                        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

@@ -0,0 +1 @@
+{"version":3,"file":"token-usage.js","sourceRoot":"","sources":["../../src/utils/token-usage.ts"],"names":[],"mappings":";;AA0BA,sDAcC;AA0BD,oEAEC;AAED,wEAIC;AAzED,uDAAqD;AAErD,2CAA4D;AAuB5D,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,IACC,OAAO,IAAI,KAAK,QAAQ;wBACxB,IAAI,KAAK,IAAI;wBACb,MAAM,IAAI,IAAI;wBACd,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAC5B,CAAC;wBACF,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/tool-executor.d.ts

@@ -0,0 +1,3 @@
+import type { ToolExecutorOptions } from '../types/config';
+import type { WorkflowState } from '../workflow-state';
+export declare function executeToolsInParallel(options: ToolExecutorOptions): Promise<Partial<typeof WorkflowState.State>>;

package/dist/utils/tool-executor.js

@@ -0,0 +1,108 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.executeToolsInParallel = executeToolsInParallel;
+const messages_1 = require("@langchain/core/messages");
+const tools_1 = require("@langchain/core/tools");
+const langgraph_1 = require("@langchain/langgraph");
+const errors_1 = require("../errors");
+function isArray(value) {
+    return Array.isArray(value);
+}
+function collectArrayFromUpdates(updates, key) {
+    const result = [];
+    for (const update of updates) {
+        const value = update[key];
+        if (isArray(value)) {
+            for (const item of value) {
+                result.push(item);
+            }
+        }
+    }
+    return result;
+}
+function createToolErrorMessage(toolName, toolCallId, error) {
+    const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred';
+    const isParsingError = error instanceof tools_1.ToolInputParsingException || errorMessage.includes('expected schema');
+    const errorContent = isParsingError
+        ? `Invalid input for tool ${toolName}: ${errorMessage}`
+        : `Tool ${toolName} failed: ${errorMessage}`;
+    return new messages_1.ToolMessage({
+        content: errorContent,
+        tool_call_id: toolCallId,
+        name: toolName,
+        additional_kwargs: { error: true },
+    });
+}
+async function executeToolsInParallel(options) {
+    const { state, toolMap } = options;
+    const lastMessage = state.messages.at(-1);
+    if (!lastMessage || !(0, messages_1.isAIMessage)(lastMessage)) {
+        const error = new errors_1.WorkflowStateError('Most recent message must be an AIMessage with tool calls');
+        throw error;
+    }
+    const aiMessage = lastMessage;
+    if (!aiMessage.tool_calls?.length) {
+        const error = new errors_1.WorkflowStateError('AIMessage must have tool calls');
+        throw error;
+    }
+    const toolResults = await Promise.all(aiMessage.tool_calls.map(async (toolCall) => {
+        try {
+            const tool = toolMap.get(toolCall.name);
+            if (!tool) {
+                throw new errors_1.ToolExecutionError(`Tool ${toolCall.name} not found`, {
+                    toolName: toolCall.name,
+                });
+            }
+            const result = await tool.invoke(toolCall.args ?? {}, {
+                toolCall: {
+                    id: toolCall.id,
+                    name: toolCall.name,
+                    args: toolCall.args ?? {},
+                },
+            });
+            return result;
+        }
+        catch (error) {
+            return createToolErrorMessage(toolCall.name, toolCall.id ?? '', error);
+        }
+    }));
+    const allMessages = [];
+    const stateUpdates = [];
+    toolResults.forEach((result) => {
+        if ((0, langgraph_1.isCommand)(result)) {
+            const update = result.update;
+            if (update) {
+                stateUpdates.push(update);
+            }
+        }
+        else {
+            allMessages.push(result);
+        }
+    });
+    allMessages.push(...collectArrayFromUpdates(stateUpdates, 'messages'));
+    const allOperations = collectArrayFromUpdates(stateUpdates, 'workflowOperations');
+    const allTechniqueCategories = collectArrayFromUpdates(stateUpdates, 'techniqueCategories');
+    const allValidationHistory = collectArrayFromUpdates(stateUpdates, 'validationHistory');
+    const allTemplateIds = collectArrayFromUpdates(stateUpdates, 'templateIds');
+    const allCachedTemplates = collectArrayFromUpdates(stateUpdates, 'cachedTemplates');
+    const finalUpdate = {
+        messages: allMessages,
+    };
+    if (allOperations.length > 0) {
+        finalUpdate.workflowOperations = allOperations;
+    }
+    if (allTechniqueCategories.length > 0) {
+        finalUpdate.techniqueCategories = allTechniqueCategories;
+    }
+    if (allValidationHistory.length > 0) {
+        finalUpdate.validationHistory = allValidationHistory;
+    }
+    if (allTemplateIds.length > 0) {
+        finalUpdate.templateIds = allTemplateIds;
+    }
+    if (allCachedTemplates.length > 0) {
+        finalUpdate.cachedTemplates = allCachedTemplates;
+    }
+    return finalUpdate;
+}
+//# sourceMappingURL=tool-executor.js.map