@illuma-ai/agents 1.0.90 → 1.0.94

package/src/tools/handlers.js

@@ -0,0 +1,306 @@
+/* eslint-disable no-console */
+// src/tools/handlers.ts
+import { nanoid } from 'nanoid';
+import { ToolMessage } from '@langchain/core/messages';
+import { ToolCallTypes, GraphEvents, StepTypes, Providers, Constants, } from '@/common';
+import { coerceAnthropicSearchResults, isAnthropicWebSearchResult, } from '@/tools/search/anthropic';
+import { formatResultsForLLM } from '@/tools/search/format';
+import { getMessageId } from '@/messages';
+export async function handleToolCallChunks({ graph, stepKey, toolCallChunks, metadata, }) {
+    let prevStepId;
+    let prevRunStep;
+    try {
+        prevStepId = graph.getStepIdByKey(stepKey);
+        prevRunStep = graph.getRunStep(prevStepId);
+    }
+    catch {
+        /** Edge Case: If no previous step exists, create a new message creation step */
+        const message_id = getMessageId(stepKey, graph, true) ?? '';
+        prevStepId = await graph.dispatchRunStep(stepKey, {
+            type: StepTypes.MESSAGE_CREATION,
+            message_creation: {
+                message_id,
+            },
+        }, metadata);
+        prevRunStep = graph.getRunStep(prevStepId);
+    }
+    const _stepId = graph.getStepIdByKey(stepKey);
+    /** Edge Case: Tool Call Run Step or `tool_call_ids` never dispatched */
+    const tool_calls = prevStepId && prevRunStep && prevRunStep.type === StepTypes.MESSAGE_CREATION
+        ? []
+        : undefined;
+    /**
+     * Feed streaming tool call buffer — accumulate raw arg strings for truncation recovery.
+     *
+     * Provider chunk patterns:
+     * - OpenAI/Anthropic: every chunk has {id, args} → direct append
+     * - Bedrock: START chunk has {id, name, index}, DELTA chunks have {args, index} (no id)
+     *   → use index-to-id mapping to resolve the target buffer entry
+     */
+    for (const toolCallChunk of toolCallChunks) {
+        const chunkIndex = toolCallChunk.index;
+        // START chunk: has id (and usually name). Store index→id mapping for future DELTA chunks.
+        if (toolCallChunk.id) {
+            if (typeof chunkIndex === 'number') {
+                graph.streamingToolCallBuffer.setIndexMapping(chunkIndex, toolCallChunk.id);
+            }
+            if (toolCallChunk.name) {
+                graph.streamingToolCallBuffer.setToolName(toolCallChunk.id, toolCallChunk.name);
+            }
+            // Append args if present on the same chunk (OpenAI/Anthropic pattern)
+            if (toolCallChunk.args) {
+                graph.streamingToolCallBuffer.append(toolCallChunk.id, toolCallChunk.args);
+            }
+        }
+        else if (toolCallChunk.args && typeof chunkIndex === 'number') {
+            // DELTA chunk: no id, but has args + index. Resolve id via index mapping (Bedrock pattern).
+            const resolvedId = graph.streamingToolCallBuffer.getIdByIndex(chunkIndex);
+            if (resolvedId) {
+                graph.streamingToolCallBuffer.append(resolvedId, toolCallChunk.args);
+            }
+        }
+    }
+    /** Edge Case: `id` and `name` fields cannot be empty strings */
+    for (const toolCallChunk of toolCallChunks) {
+        if (toolCallChunk.name === '') {
+            toolCallChunk.name = undefined;
+        }
+        if (toolCallChunk.id === '') {
+            toolCallChunk.id = undefined;
+        }
+        else if (tool_calls != null &&
+            toolCallChunk.id != null &&
+            toolCallChunk.name != null) {
+            tool_calls.push({
+                args: {},
+                id: toolCallChunk.id,
+                name: toolCallChunk.name,
+                type: ToolCallTypes.TOOL_CALL,
+            });
+        }
+    }
+    let stepId = _stepId;
+    const alreadyDispatched = prevRunStep?.type === StepTypes.MESSAGE_CREATION &&
+        graph.messageStepHasToolCalls.has(prevStepId);
+    if (prevRunStep?.type === StepTypes.TOOL_CALLS) {
+        /**
+         * If previous step is already a tool_calls step, use that step ID
+         * This ensures tool call deltas are dispatched to the correct step
+         */
+        stepId = prevStepId;
+    }
+    else if (!alreadyDispatched &&
+        prevRunStep?.type === StepTypes.MESSAGE_CREATION) {
+        /**
+         * Create tool_calls step as soon as we receive the first tool call chunk
+         * This ensures deltas are always associated with the correct step
+         *
+         * NOTE: We do NOT dispatch an empty text block here because:
+         * - Empty text blocks cause providers (Anthropic, Bedrock) to reject messages
+         * - The tool_calls themselves are sufficient for the step
+         * - Empty content with tool_call_ids gets stored in conversation history
+         *   and causes "messages must have non-empty content" errors on replay
+         */
+        graph.messageStepHasToolCalls.set(prevStepId, true);
+        stepId = await graph.dispatchRunStep(stepKey, {
+            type: StepTypes.TOOL_CALLS,
+            tool_calls: tool_calls ?? [],
+        }, metadata);
+    }
+    await graph.dispatchRunStepDelta(stepId, {
+        type: StepTypes.TOOL_CALLS,
+        tool_calls: toolCallChunks,
+    });
+}
+export const handleToolCalls = async (toolCalls, metadata, graph) => {
+    if (!graph || !metadata) {
+        console.warn('Graph or metadata not found in `handleToolCalls`');
+        return;
+    }
+    if (!toolCalls) {
+        return;
+    }
+    if (toolCalls.length === 0) {
+        return;
+    }
+    const stepKey = graph.getStepKey(metadata);
+    /**
+     * Track whether we've already reused an empty TOOL_CALLS step created by
+     * handleToolCallChunks during streaming. Only reuse it once (for the first
+     * tool call); subsequent parallel tool calls must create their own steps.
+     */
+    let reusedChunkStepId;
+    for (const tool_call of toolCalls) {
+        const toolCallId = tool_call.id ?? `toolu_${nanoid()}`;
+        tool_call.id = toolCallId;
+        // If this tool call ID was already tracked via handleToolCallChunks,
+        // the step exists but may lack the name (Bedrock sends name only at model end).
+        // Dispatch a delta with the complete data so the client can fill in the name.
+        if (toolCallId && graph.toolCallStepIds.has(toolCallId)) {
+            const existingStepId = graph.toolCallStepIds.get(toolCallId);
+            if (existingStepId != null && existingStepId !== '') {
+                const argsStr = typeof tool_call.args === 'string'
+                    ? tool_call.args
+                    : JSON.stringify(tool_call.args);
+                await graph.dispatchRunStepDelta(existingStepId, {
+                    type: StepTypes.TOOL_CALLS,
+                    tool_calls: [{ name: tool_call.name, args: argsStr, id: toolCallId }],
+                });
+            }
+            continue;
+        }
+        let prevStepId = '';
+        let prevRunStep;
+        try {
+            prevStepId = graph.getStepIdByKey(stepKey);
+            prevRunStep = graph.getRunStep(prevStepId);
+        }
+        catch {
+            // no previous step
+        }
+        // If the previous step is TOOL_CALLS (created by handleToolCallChunks),
+        // either reuse it (if empty) or dispatch a new TOOL_CALLS step directly —
+        // skip the intermediate MESSAGE_CREATION to avoid orphaned gaps.
+        if (prevRunStep?.type === StepTypes.TOOL_CALLS) {
+            const details = prevRunStep.stepDetails;
+            const isEmpty = !details.tool_calls || details.tool_calls.length === 0;
+            if (isEmpty && prevStepId !== reusedChunkStepId) {
+                graph.toolCallStepIds.set(toolCallId, prevStepId);
+                reusedChunkStepId = prevStepId;
+                continue;
+            }
+            await graph.dispatchRunStep(stepKey, { type: StepTypes.TOOL_CALLS, tool_calls: [tool_call] }, metadata);
+            continue;
+        }
+        /**
+         * NOTE: We do NOT dispatch empty text blocks with tool_call_ids because:
+         * - Empty text blocks cause providers (Anthropic, Bedrock) to reject messages
+         * - They get stored in conversation history and cause errors on replay:
+         *   "messages must have non-empty content" (Anthropic)
+         *   "The content field in the Message object is empty" (Bedrock)
+         * - The tool_calls themselves are sufficient
+         */
+        /* If the previous step exists and is a message creation */
+        if (prevStepId && prevRunStep) {
+            graph.messageStepHasToolCalls.set(prevStepId, true);
+            /* If the previous step doesn't exist */
+        }
+        else if (!prevRunStep) {
+            const messageId = getMessageId(stepKey, graph, true) ?? '';
+            const stepId = await graph.dispatchRunStep(stepKey, {
+                type: StepTypes.MESSAGE_CREATION,
+                message_creation: {
+                    message_id: messageId,
+                },
+            }, metadata);
+            graph.messageStepHasToolCalls.set(stepId, true);
+        }
+        await graph.dispatchRunStep(stepKey, {
+            type: StepTypes.TOOL_CALLS,
+            tool_calls: [tool_call],
+        }, metadata);
+    }
+};
+export const toolResultTypes = new Set([
+    // 'tool_use',
+    // 'server_tool_use',
+    // 'input_json_delta',
+    'tool_result',
+    'web_search_result',
+    'web_search_tool_result',
+]);
+/**
+ * Handles the result of a server tool call; in other words, a provider's built-in tool.
+ * As of 2025-07-06, only Anthropic handles server tool calls with this pattern.
+ */
+export async function handleServerToolResult({ graph, content, metadata, agentContext, }) {
+    let skipHandling = false;
+    if (agentContext?.provider !== Providers.ANTHROPIC) {
+        return skipHandling;
+    }
+    if (typeof content === 'string' ||
+        content == null ||
+        content.length === 0 ||
+        (content.length === 1 &&
+            content[0].tool_use_id == null)) {
+        return skipHandling;
+    }
+    for (const contentPart of content) {
+        const toolUseId = contentPart.tool_use_id;
+        if (toolUseId == null || toolUseId === '') {
+            continue;
+        }
+        const stepId = graph.toolCallStepIds.get(toolUseId);
+        if (stepId == null || stepId === '') {
+            console.warn(`Tool use ID ${toolUseId} not found in graph, cannot dispatch tool result.`);
+            continue;
+        }
+        const runStep = graph.getRunStep(stepId);
+        if (!runStep) {
+            console.warn(`Run step for ${stepId} does not exist, cannot dispatch tool result.`);
+            continue;
+        }
+        else if (runStep.type !== StepTypes.TOOL_CALLS) {
+            console.warn(`Run step for ${stepId} is not a tool call step, cannot dispatch tool result.`);
+            continue;
+        }
+        const toolCall = runStep.stepDetails.type === StepTypes.TOOL_CALLS
+            ? runStep.stepDetails.tool_calls?.find((toolCall) => toolCall.id === toolUseId)
+            : undefined;
+        if (!toolCall) {
+            continue;
+        }
+        if (contentPart.type === 'web_search_result' ||
+            contentPart.type === 'web_search_tool_result') {
+            await handleAnthropicSearchResults({
+                contentPart: contentPart,
+                toolCall,
+                metadata,
+                graph,
+            });
+        }
+        if (!skipHandling) {
+            skipHandling = true;
+        }
+    }
+    return skipHandling;
+}
+async function handleAnthropicSearchResults({ contentPart, toolCall, metadata, graph, }) {
+    if (!Array.isArray(contentPart.content)) {
+        console.warn(`Expected content to be an array, got ${typeof contentPart.content}`);
+        return;
+    }
+    if (!isAnthropicWebSearchResult(contentPart.content[0])) {
+        console.warn(`Expected content to be an Anthropic web search result, got ${JSON.stringify(contentPart.content)}`);
+        return;
+    }
+    const turn = graph.invokedToolIds?.size ?? 0;
+    const searchResultData = coerceAnthropicSearchResults({
+        turn,
+        results: contentPart.content,
+    });
+    const name = toolCall.name;
+    const input = toolCall.args ?? {};
+    const artifact = {
+        [Constants.WEB_SEARCH]: searchResultData,
+    };
+    const { output: formattedOutput } = formatResultsForLLM(turn, searchResultData);
+    const output = new ToolMessage({
+        name,
+        artifact,
+        content: formattedOutput,
+        tool_call_id: toolCall.id,
+    });
+    const toolEndData = {
+        input,
+        output,
+    };
+    await graph.handlerRegistry
+        ?.getHandler(GraphEvents.TOOL_END)
+        ?.handle(GraphEvents.TOOL_END, toolEndData, metadata, graph);
+    if (graph.invokedToolIds == null) {
+        graph.invokedToolIds = new Set();
+    }
+    graph.invokedToolIds.add(toolCall.id);
+}
+//# sourceMappingURL=handlers.js.map

package/src/tools/handlers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handlers.js","sourceRoot":"","sources":["handlers.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B,wBAAwB;AACxB,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAMvD,OAAO,EACL,aAAa,EACb,WAAW,EACX,SAAS,EACT,SAAS,EACT,SAAS,GACV,MAAM,UAAU,CAAC;AAClB,OAAO,EACL,4BAA4B,EAC5B,0BAA0B,GAC3B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,EACzC,KAAK,EACL,OAAO,EACP,cAAc,EACd,QAAQ,GAMT;IACC,IAAI,UAAkB,CAAC;IACvB,IAAI,WAAkC,CAAC;IACvC,IAAI,CAAC;QACH,UAAU,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAC3C,WAAW,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IAC7C,CAAC;IAAC,MAAM,CAAC;QACP,gFAAgF;QAChF,MAAM,UAAU,GAAG,YAAY,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5D,UAAU,GAAG,MAAM,KAAK,CAAC,eAAe,CACtC,OAAO,EACP;YACE,IAAI,EAAE,SAAS,CAAC,gBAAgB;YAChC,gBAAgB,EAAE;gBAChB,UAAU;aACX;SACF,EACD,QAAQ,CACT,CAAC;QACF,WAAW,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IAC7C,CAAC;IAED,MAAM,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IAE9C,wEAAwE;IACxE,MAAM,UAAU,GACd,UAAU,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC,gBAAgB;QAC1E,CAAC,CAAC,EAAE;QACJ,CAAC,CAAC,SAAS,CAAC;IAEhB;;;;;;;OAOG;IACH,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE,CAAC;QAC3C,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC;QAEvC,0FAA0F;QAC1F,IAAI,aAAa,CAAC,EAAE,EAAE,CAAC;YACrB,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;gBACnC,KAAK,CAAC,uBAAuB,CAAC,eAAe,CAAC,UAAU,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC;YAC9E,CAAC;YACD,IAAI,aAAa,CAAC,IAAI,EAAE,CAAC;gBACvB,KAAK,CAAC,uBAAuB,CAAC,WAAW,CAAC,aAAa,CAAC,EAAE,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;YAClF,CAAC;YACD,sEAAsE;YACtE,IAAI,aAAa,CAAC,IAAI,EAAE,CAAC;gBACvB,KAAK,CAAC,uBAAuB,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;aAAM,IAAI,aAAa,CAAC,IAAI,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;YAChE,4FAA4F;YAC5F,MAAM,UAAU,GAAG,KAAK,CAAC,uBAAuB,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YAC1E,IAAI,UAAU,EAAE,CAAC;gBACf,KAAK,CAAC,uBAAuB,CAAC,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC;YACvE,CAAC;QACH,CAAC;IACH,CAAC;IAED,gEAAgE;IAChE,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE,CAAC;QAC3C,IAAI,aAAa,CAAC,IAAI,KAAK,EAAE,EAAE,CAAC;YAC9B,aAAa,CAAC,IAAI,GAAG,SAAS,CAAC;QACjC,CAAC;QACD,IAAI,aAAa,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;YAC5B,aAAa,CAAC,EAAE,GAAG,SAAS,CAAC;QAC/B,CAAC;aAAM,IACL,UAAU,IAAI,IAAI;YAClB,aAAa,CAAC,EAAE,IAAI,IAAI;YACxB,aAAa,CAAC,IAAI,IAAI,IAAI,EAC1B,CAAC;YACD,UAAU,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,EAAE;gBACR,EAAE,EAAE,aAAa,CAAC,EAAE;gBACpB,IAAI,EAAE,aAAa,CAAC,IAAI;gBACxB,IAAI,EAAE,aAAa,CAAC,SAAS;aAC9B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,MAAM,GAAW,OAAO,CAAC;IAC7B,MAAM,iBAAiB,GACrB,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,gBAAgB;QAChD,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAEhD,IAAI,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,UAAU,EAAE,CAAC;QAC/C;;;WAGG;QACH,MAAM,GAAG,UAAU,CAAC;IACtB,CAAC;SAAM,IACL,CAAC,iBAAiB;QAClB,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,gBAAgB,EAChD,CAAC;QACD;;;;;;;;;WASG;QACH,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;QACpD,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAClC,OAAO,EACP;YACE,IAAI,EAAE,SAAS,CAAC,UAAU;YAC1B,UAAU,EAAE,UAAU,IAAI,EAAE;SAC7B,EACD,QAAQ,CACT,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,CAAC,oBAAoB,CAAC,MAAM,EAAE;QACvC,IAAI,EAAE,SAAS,CAAC,UAAU;QAC1B,UAAU,EAAE,cAAc;KAC3B,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,MAAM,eAAe,GAAG,KAAK,EAClC,SAAsB,EACtB,QAAkC,EAClC,KAA+C,EAChC,EAAE;IACjB,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,EAAE,CAAC;QACxB,OAAO,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;QACjE,OAAO;IACT,CAAC;IAED,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO;IACT,CAAC;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO;IACT,CAAC;IAED,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IAE3C;;;;OAIG;IACH,IAAI,iBAAqC,CAAC;IAE1C,KAAK,MAAM,SAAS,IAAI,SAAS,EAAE,CAAC;QAClC,MAAM,UAAU,GAAG,SAAS,CAAC,EAAE,IAAI,SAAS,MAAM,EAAE,EAAE,CAAC;QACvD,SAAS,CAAC,EAAE,GAAG,UAAU,CAAC;QAE1B,qEAAqE;QACrE,gFAAgF;QAChF,8EAA8E;QAC9E,IAAI,UAAU,IAAI,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YACxD,MAAM,cAAc,GAAG,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAC7D,IAAI,cAAc,IAAI,IAAI,IAAI,cAAc,KAAK,EAAE,EAAE,CAAC;gBACpD,MAAM,OAAO,GACX,OAAO,SAAS,CAAC,IAAI,KAAK,QAAQ;oBAChC,CAAC,CAAC,SAAS,CAAC,IAAI;oBAChB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBACrC,MAAM,KAAK,CAAC,oBAAoB,CAAC,cAAc,EAAE;oBAC/C,IAAI,EAAE,SAAS,CAAC,UAAU;oBAC1B,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC;iBACtE,CAAC,CAAC;YACL,CAAC;YACD,SAAS;QACX,CAAC;QAED,IAAI,UAAU,GAAG,EAAE,CAAC;QACpB,IAAI,WAAkC,CAAC;QACvC,IAAI,CAAC;YACH,UAAU,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAC3C,WAAW,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QAC7C,CAAC;QAAC,MAAM,CAAC;YACP,mBAAmB;QACrB,CAAC;QAED,wEAAwE;QACxE,0EAA0E;QAC1E,iEAAiE;QACjE,IAAI,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,UAAU,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,WAAW,CAAC,WAAiC,CAAC;YAC9D,MAAM,OAAO,GAAG,CAAC,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,CAAC;YACvE,IAAI,OAAO,IAAI,UAAU,KAAK,iBAAiB,EAAE,CAAC;gBAChD,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;gBAClD,iBAAiB,GAAG,UAAU,CAAC;gBAC/B,SAAS;YACX,CAAC;YACD,MAAM,KAAK,CAAC,eAAe,CACzB,OAAO,EACP,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC,SAAS,CAAC,EAAE,EACvD,QAAQ,CACT,CAAC;YACF,SAAS;QACX,CAAC;QAED;;;;;;;WAOG;QAEH,2DAA2D;QAC3D,IAAI,UAAU,IAAI,WAAW,EAAE,CAAC;YAC9B,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YACpD,wCAAwC;QAC1C,CAAC;aAAM,IAAI,CAAC,WAAW,EAAE,CAAC;YACxB,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CACxC,OAAO,EACP;gBACE,IAAI,EAAE,SAAS,CAAC,gBAAgB;gBAChC,gBAAgB,EAAE;oBAChB,UAAU,EAAE,SAAS;iBACtB;aACF,EACD,QAAQ,CACT,CAAC;YACF,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,KAAK,CAAC,eAAe,CACzB,OAAO,EACP;YACE,IAAI,EAAE,SAAS,CAAC,UAAU;YAC1B,UAAU,EAAE,CAAC,SAAS,CAAC;SACxB,EACD,QAAQ,CACT,CAAC;IACJ,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC;IACrC,cAAc;IACd,qBAAqB;IACrB,sBAAsB;IACtB,aAAa;IACb,mBAAmB;IACnB,wBAAwB;CACzB,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,EAC3C,KAAK,EACL,OAAO,EACP,QAAQ,EACR,YAAY,GAMb;IACC,IAAI,YAAY,GAAG,KAAK,CAAC;IACzB,IAAI,YAAY,EAAE,QAAQ,KAAK,SAAS,CAAC,SAAS,EAAE,CAAC;QACnD,OAAO,YAAY,CAAC;IACtB,CAAC;IACD,IACE,OAAO,OAAO,KAAK,QAAQ;QAC3B,OAAO,IAAI,IAAI;QACf,OAAO,CAAC,MAAM,KAAK,CAAC;QACpB,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC;YAClB,OAAO,CAAC,CAAC,CAAyB,CAAC,WAAW,IAAI,IAAI,CAAC,EAC1D,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,KAAK,MAAM,WAAW,IAAI,OAAO,EAAE,CAAC;QAClC,MAAM,SAAS,GAAI,WAAmC,CAAC,WAAW,CAAC;QACnE,IAAI,SAAS,IAAI,IAAI,IAAI,SAAS,KAAK,EAAE,EAAE,CAAC;YAC1C,SAAS;QACX,CAAC;QACD,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,MAAM,IAAI,IAAI,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YACpC,OAAO,CAAC,IAAI,CACV,eAAe,SAAS,mDAAmD,CAC5E,CAAC;YACF,SAAS;QACX,CAAC;QACD,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACzC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,IAAI,CACV,gBAAgB,MAAM,+CAA+C,CACtE,CAAC;YACF,SAAS;QACX,CAAC;aAAM,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,CAAC,UAAU,EAAE,CAAC;YACjD,OAAO,CAAC,IAAI,CACV,gBAAgB,MAAM,wDAAwD,CAC/E,CAAC;YACF,SAAS;QACX,CAAC;QAED,MAAM,QAAQ,GACZ,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC,UAAU;YAC/C,CAAC,CAAE,OAAO,CAAC,WAAW,CAAC,UAAU,EAAE,IAAI,CACrC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,SAAS,CAC3B;YACd,CAAC,CAAC,SAAS,CAAC;QAEhB,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,SAAS;QACX,CAAC;QAED,IACE,WAAW,CAAC,IAAI,KAAK,mBAAmB;YACxC,WAAW,CAAC,IAAI,KAAK,wBAAwB,EAC7C,CAAC;YACD,MAAM,4BAA4B,CAAC;gBACjC,WAAW,EAAE,WAAkC;gBAC/C,QAAQ;gBACR,QAAQ;gBACR,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,YAAY,GAAG,IAAI,CAAC;QACtB,CAAC;IACH,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,KAAK,UAAU,4BAA4B,CAAC,EAC1C,WAAW,EACX,QAAQ,EACR,QAAQ,EACR,KAAK,GAMN;IACC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;QACxC,OAAO,CAAC,IAAI,CACV,wCAAwC,OAAO,WAAW,CAAC,OAAO,EAAE,CACrE,CAAC;QACF,OAAO;IACT,CAAC;IAED,IAAI,CAAC,0BAA0B,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACxD,OAAO,CAAC,IAAI,CACV,8DAA8D,IAAI,CAAC,SAAS,CAC1E,WAAW,CAAC,OAAO,CACpB,EAAE,CACJ,CAAC;QACF,OAAO;IACT,CAAC;IAED,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,EAAE,IAAI,IAAI,CAAC,CAAC;IAC7C,MAAM,gBAAgB,GAAG,4BAA4B,CAAC;QACpD,IAAI;QACJ,OAAO,EAAE,WAAW,CAAC,OAA+C;KACrE,CAAC,CAAC;IAEH,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;IAC3B,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;IAClC,MAAM,QAAQ,GAAG;QACf,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,gBAAgB;KACzC,CAAC;IACF,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,mBAAmB,CACrD,IAAI,EACJ,gBAAgB,CACjB,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC;QAC7B,IAAI;QACJ,QAAQ;QACR,OAAO,EAAE,eAAe;QACxB,YAAY,EAAE,QAAQ,CAAC,EAAG;KAC3B,CAAC,CAAC;IACH,MAAM,WAAW,GAAkB;QACjC,KAAK;QACL,MAAM;KACP,CAAC;IACF,MAAM,KAAK,CAAC,eAAe;QACzB,EAAE,UAAU,CAAC,WAAW,CAAC,QAAQ,CAAC;QAClC,EAAE,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IAE/D,IAAI,KAAK,CAAC,cAAc,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAG,CAAC,CAAC;AACzC,CAAC"}

package/src/tools/handlers.ts

@@ -4,7 +4,7 @@ import { nanoid } from 'nanoid';
 import { ToolMessage } from '@langchain/core/messages';
 import type { AnthropicWebSearchResultBlockParam } from '@/llm/anthropic/types';
 import type { ToolCall, ToolCallChunk } from '@langchain/core/messages/tool';
-import type { MultiAgentGraph, StandardGraph } from '@/graphs';
+import type { Graph, MultiAgentGraph, StandardGraph } from '@/graphs';
 import type { AgentContext } from '@/agents/AgentContext';
 import type * as t from '@/types';
 import {
@@ -61,6 +61,74 @@ export async function handleToolCallChunks({
       ? []
       : undefined;
 
+  /**
+   * Feed streaming tool call buffer — accumulate raw arg strings for truncation recovery.
+   *
+   * Provider chunk patterns:
+   * - OpenAI/Anthropic: every chunk has {id, args} → direct append
+   * - Bedrock: START chunk has {id, name, index}, DELTA chunks have {args, index} (no id)
+   *   → use index-to-id mapping to resolve the target buffer entry
+   *
+   * Duplicate detection: LangChain's `streamEvents` can emit the same `on_chat_model_stream`
+   * event multiple times when the model is wrapped in nested runnables. The buffer's `append()`
+   * tracks the last fragment per tool call and returns false for exact duplicates.
+   * When ALL chunks in a batch are duplicates, we skip the entire delta dispatch.
+   */
+  let allDuplicates = true;
+  for (const toolCallChunk of toolCallChunks) {
+    const chunkIndex = toolCallChunk.index;
+
+    // START chunk: has id (and usually name). Store index→id mapping for future DELTA chunks.
+    if (toolCallChunk.id) {
+      if (typeof chunkIndex === 'number') {
+        graph.streamingToolCallBuffer.setIndexMapping(
+          chunkIndex,
+          toolCallChunk.id
+        );
+      }
+      if (toolCallChunk.name) {
+        graph.streamingToolCallBuffer.setToolName(
+          toolCallChunk.id,
+          toolCallChunk.name
+        );
+      }
+      // Append args if present on the same chunk (OpenAI/Anthropic pattern)
+      if (toolCallChunk.args) {
+        const appended = graph.streamingToolCallBuffer.append(
+          toolCallChunk.id,
+          toolCallChunk.args
+        );
+        if (appended) allDuplicates = false;
+      } else {
+        // START chunk without args is never a duplicate (it sets up id/name)
+        allDuplicates = false;
+      }
+    } else if (toolCallChunk.args && typeof chunkIndex === 'number') {
+      // DELTA chunk: no id, but has args + index. Resolve id via index mapping (Bedrock pattern).
+      const resolvedId = graph.streamingToolCallBuffer.getIdByIndex(chunkIndex);
+      if (resolvedId) {
+        const appended = graph.streamingToolCallBuffer.append(
+          resolvedId,
+          toolCallChunk.args
+        );
+        if (appended) allDuplicates = false;
+      }
+    } else {
+      // Chunks without args or index are not duplicates
+      allDuplicates = false;
+    }
+  }
+
+  // If every chunk in this batch was an exact duplicate, skip the delta dispatch entirely.
+  // This prevents the frontend from receiving doubled fragments that corrupt accumulated args.
+  if (allDuplicates && toolCallChunks.length > 0) {
+    // TEMP DEBUG: dedup fix active — remove after E2E verification
+    console.log(
+      '[handleToolCallChunks:DEDUP] All chunks duplicate — skipping delta dispatch'
+    );
+    return;
+  }
+
   /** Edge Case: `id` and `name` fields cannot be empty strings */
   for (const toolCallChunk of toolCallChunks) {
     if (toolCallChunk.name === '') {
@@ -117,6 +185,7 @@ export async function handleToolCallChunks({
       metadata
     );
   }
+
   await graph.dispatchRunStepDelta(stepId, {
     type: StepTypes.TOOL_CALLS,
     tool_calls: toolCallChunks,
@@ -126,10 +195,10 @@ export async function handleToolCallChunks({
 export const handleToolCalls = async (
   toolCalls?: ToolCall[],
   metadata?: Record<string, unknown>,
-  graph?: StandardGraph | MultiAgentGraph
+  graph?: Graph | StandardGraph | MultiAgentGraph
 ): Promise<void> => {
   if (!graph || !metadata) {
-    console.warn(`Graph or metadata not found in ${event} event`);
+    console.warn('Graph or metadata not found in `handleToolCalls`');
     return;
   }
 
@@ -143,10 +212,32 @@ export const handleToolCalls = async (
 
   const stepKey = graph.getStepKey(metadata);
 
+  /**
+   * Track whether we've already reused an empty TOOL_CALLS step created by
+   * handleToolCallChunks during streaming. Only reuse it once (for the first
+   * tool call); subsequent parallel tool calls must create their own steps.
+   */
+  let reusedChunkStepId: string | undefined;
+
   for (const tool_call of toolCalls) {
     const toolCallId = tool_call.id ?? `toolu_${nanoid()}`;
     tool_call.id = toolCallId;
-    if (!toolCallId || graph.toolCallStepIds.has(toolCallId)) {
+
+    // If this tool call ID was already tracked via handleToolCallChunks,
+    // the step exists but may lack the name (Bedrock sends name only at model end).
+    // Dispatch a delta with the complete data so the client can fill in the name.
+    if (toolCallId && graph.toolCallStepIds.has(toolCallId)) {
+      const existingStepId = graph.toolCallStepIds.get(toolCallId);
+      if (existingStepId != null && existingStepId !== '') {
+        const argsStr =
+          typeof tool_call.args === 'string'
+            ? tool_call.args
+            : JSON.stringify(tool_call.args);
+        await graph.dispatchRunStepDelta(existingStepId, {
+          type: StepTypes.TOOL_CALLS,
+          tool_calls: [{ name: tool_call.name, args: argsStr, id: toolCallId }],
+        });
+      }
       continue;
     }
 
@@ -159,6 +250,25 @@ export const handleToolCalls = async (
       // no previous step
     }
 
+    // If the previous step is TOOL_CALLS (created by handleToolCallChunks),
+    // either reuse it (if empty) or dispatch a new TOOL_CALLS step directly —
+    // skip the intermediate MESSAGE_CREATION to avoid orphaned gaps.
+    if (prevRunStep?.type === StepTypes.TOOL_CALLS) {
+      const details = prevRunStep.stepDetails as t.ToolCallsDetails;
+      const isEmpty = !details.tool_calls || details.tool_calls.length === 0;
+      if (isEmpty && prevStepId !== reusedChunkStepId) {
+        graph.toolCallStepIds.set(toolCallId, prevStepId);
+        reusedChunkStepId = prevStepId;
+        continue;
+      }
+      await graph.dispatchRunStep(
+        stepKey,
+        { type: StepTypes.TOOL_CALLS, tool_calls: [tool_call] },
+        metadata
+      );
+      continue;
+    }
+
     /**
      * NOTE: We do NOT dispatch empty text blocks with tool_call_ids because:
      * - Empty text blocks cause providers (Anthropic, Bedrock) to reject messages
@@ -169,17 +279,10 @@ export const handleToolCalls = async (
      */
 
     /* If the previous step exists and is a message creation */
-    if (
-      prevStepId &&
-      prevRunStep &&
-      prevRunStep.type === StepTypes.MESSAGE_CREATION
-    ) {
+    if (prevStepId && prevRunStep) {
       graph.messageStepHasToolCalls.set(prevStepId, true);
-      /* If the previous step doesn't exist or is not a message creation */
-    } else if (
-      !prevRunStep ||
-      prevRunStep.type !== StepTypes.MESSAGE_CREATION
-    ) {
+      /* If the previous step doesn't exist */
+    } else if (!prevRunStep) {
       const messageId = getMessageId(stepKey, graph, true) ?? '';
       const stepId = await graph.dispatchRunStep(
         stepKey,
@@ -271,8 +374,8 @@ export async function handleServerToolResult({
     const toolCall =
       runStep.stepDetails.type === StepTypes.TOOL_CALLS
         ? (runStep.stepDetails.tool_calls?.find(
-          (toolCall) => toolCall.id === toolUseId
-        ) as ToolCall)
+            (toolCall) => toolCall.id === toolUseId
+          ) as ToolCall)
         : undefined;
 
     if (!toolCall) {

package/src/tools/schema.js

@@ -0,0 +1,25 @@
+import { tool } from '@langchain/core/tools';
+/**
+ * Creates a schema-only tool for LLM binding in event-driven mode.
+ * These tools have valid schemas for the LLM to understand but should
+ * never be invoked directly - ToolNode handles execution via events.
+ */
+export function createSchemaOnlyTool(definition) {
+    const { name, description, parameters, responseFormat } = definition;
+    return tool(async () => {
+        throw new Error(`Tool "${name}" should not be invoked directly in event-driven mode. ` +
+            'ToolNode should dispatch ON_TOOL_EXECUTE events instead.');
+    }, {
+        name,
+        description: description ?? '',
+        schema: parameters ?? { type: 'object', properties: {} },
+        responseFormat: responseFormat ?? 'content_and_artifact',
+    });
+}
+/**
+ * Creates schema-only tools for all definitions in an array.
+ */
+export function createSchemaOnlyTools(definitions) {
+    return definitions.map((def) => createSchemaOnlyTool(def));
+}
+//# sourceMappingURL=schema.js.map

package/src/tools/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAgC,MAAM,uBAAuB,CAAC;AAG3E;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAClC,UAAkB;IAElB,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,cAAc,EAAE,GAAG,UAAU,CAAC;IAErE,OAAO,IAAI,CACT,KAAK,IAAI,EAAE;QACT,MAAM,IAAI,KAAK,CACb,SAAS,IAAI,yDAAyD;YACpE,0DAA0D,CAC7D,CAAC;IACJ,CAAC,EACD;QACE,IAAI;QACJ,WAAW,EAAE,WAAW,IAAI,EAAE;QAC9B,MAAM,EAAE,UAAU,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;QACxD,cAAc,EAAE,cAAc,IAAI,sBAAsB;KACzD,CACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CACnC,WAAqB;IAErB,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC;AAC7D,CAAC"}

package/src/tools/search/anthropic.js

@@ -0,0 +1,34 @@
+import { getAttribution } from './utils';
+/**
+ * Coerces Anthropic web search results to the SearchResultData format
+ * @param results - Array of Anthropic web search results
+ * @param turn - The turn number to associate with these results
+ * @returns SearchResultData with minimal ProcessedOrganic items
+ */
+export function coerceAnthropicSearchResults({ results, turn = 0, }) {
+    const organic = results
+        .filter((result) => result.type === 'web_search_result')
+        .map((result, index) => ({
+        link: result.url,
+        position: index + 1,
+        title: result.title,
+        date: result.page_age ?? undefined,
+        attribution: getAttribution(result.url),
+    }));
+    return {
+        turn,
+        organic,
+    };
+}
+/**
+ * Helper function to check if an object is an Anthropic web search result
+ */
+export function isAnthropicWebSearchResult(obj) {
+    return (typeof obj === 'object' &&
+        obj !== null &&
+        'type' in obj &&
+        obj.type === 'web_search_result' &&
+        'url' in obj &&
+        typeof obj.url === 'string');
+}
+//# sourceMappingURL=anthropic.js.map

package/src/tools/search/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["anthropic.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAEzC;;;;;GAKG;AACH,MAAM,UAAU,4BAA4B,CAAC,EAC3C,OAAO,EACP,IAAI,GAAG,CAAC,GAIT;IACC,MAAM,OAAO,GAAuB,OAAO;SACxC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,mBAAmB,CAAC;SACvD,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QACvB,IAAI,EAAE,MAAM,CAAC,GAAG;QAChB,QAAQ,EAAE,KAAK,GAAG,CAAC;QACnB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,IAAI,EAAE,MAAM,CAAC,QAAQ,IAAI,SAAS;QAClC,WAAW,EAAE,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC;KACxC,CAAC,CAAC,CAAC;IAEN,OAAO;QACL,IAAI;QACJ,OAAO;KACR,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,0BAA0B,CACxC,GAAY;IAEZ,OAAO,CACL,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,MAAM,IAAI,GAAG;QACb,GAAG,CAAC,IAAI,KAAK,mBAAmB;QAChC,KAAK,IAAI,GAAG;QACZ,OAAQ,GAA+B,CAAC,GAAG,KAAK,QAAQ,CACzD,CAAC;AACJ,CAAC"}