@librechat/agents 3.1.37 → 3.1.38

package/dist/esm/tools/handlers.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"handlers.mjs","sources":["../../../src/tools/handlers.ts"],"sourcesContent":["/* eslint-disable no-console */\n// src/tools/handlers.ts\nimport { nanoid } from 'nanoid';\nimport { ToolMessage } from '@langchain/core/messages';\nimport type { AnthropicWebSearchResultBlockParam } from '@/llm/anthropic/types';\nimport type { ToolCall, ToolCallChunk } from '@langchain/core/messages/tool';\nimport type { MultiAgentGraph, StandardGraph } from '@/graphs';\nimport type { AgentContext } from '@/agents/AgentContext';\nimport type * as t from '@/types';\nimport {\n  ToolCallTypes,\n  GraphEvents,\n  StepTypes,\n  Providers,\n  Constants,\n} from '@/common';\nimport {\n  coerceAnthropicSearchResults,\n  isAnthropicWebSearchResult,\n} from '@/tools/search/anthropic';\nimport { formatResultsForLLM } from '@/tools/search/format';\nimport { getMessageId } from '@/messages';\n\nexport async function handleToolCallChunks({\n  graph,\n  stepKey,\n  toolCallChunks,\n  metadata,\n}: {\n  graph: StandardGraph | MultiAgentGraph;\n  stepKey: string;\n  toolCallChunks: ToolCallChunk[];\n  metadata?: Record<string, unknown>;\n}): Promise<void> {\n  let prevStepId: string;\n  let prevRunStep: t.RunStep | undefined;\n  try {\n    prevStepId = graph.getStepIdByKey(stepKey);\n    prevRunStep = graph.getRunStep(prevStepId);\n  } catch {\n    /** Edge Case: If no previous step exists, create a new message creation step */\n    const message_id = getMessageId(stepKey, graph, true) ?? '';\n    prevStepId = await graph.dispatchRunStep(\n      stepKey,\n      {\n        type: StepTypes.MESSAGE_CREATION,\n        message_creation: {\n          message_id,\n        },\n      },\n      metadata\n    );\n    prevRunStep = graph.getRunStep(prevStepId);\n  }\n\n  const _stepId = graph.getStepIdByKey(stepKey);\n\n  /** Edge Case: Tool Call Run Step or `tool_call_ids` never dispatched */\n  const tool_calls: ToolCall[] | undefined =\n    prevStepId && prevRunStep && prevRunStep.type === StepTypes.MESSAGE_CREATION\n      ? []\n      : undefined;\n\n  /** Edge Case: `id` and `name` fields cannot be empty strings */\n  for (const toolCallChunk of toolCallChunks) {\n    if (toolCallChunk.name === '') {\n      toolCallChunk.name = undefined;\n    }\n    if (toolCallChunk.id === '') {\n      toolCallChunk.id = undefined;\n    } else if (\n      tool_calls != null &&\n      toolCallChunk.id != null &&\n      toolCallChunk.name != null\n    ) {\n      tool_calls.push({\n        args: {},\n        id: toolCallChunk.id,\n        name: toolCallChunk.name,\n        type: ToolCallTypes.TOOL_CALL,\n      });\n    }\n  }\n\n  let stepId: string = _stepId;\n  const alreadyDispatched =\n    prevRunStep?.type === StepTypes.MESSAGE_CREATION &&\n    graph.messageStepHasToolCalls.has(prevStepId);\n\n  if (prevRunStep?.type === StepTypes.TOOL_CALLS) {\n    /**\n     * If previous step is already a tool_calls step, use that step ID\n     * This ensures tool call deltas are dispatched to the correct step\n     */\n    stepId = prevStepId;\n  } else if (\n    !alreadyDispatched &&\n    prevRunStep?.type === StepTypes.MESSAGE_CREATION\n  ) {\n    /**\n     * Create tool_calls step as soon as we receive the first tool call chunk\n     * This ensures deltas are always associated with the correct step\n     *\n     * NOTE: We do NOT dispatch an empty text block here because:\n     * - Empty text blocks cause providers (Anthropic, Bedrock) to reject messages\n     * - The tool_calls themselves are sufficient for the step\n     * - Empty content with tool_call_ids gets stored in conversation history\n     *   and causes \"messages must have non-empty content\" errors on replay\n     */\n    graph.messageStepHasToolCalls.set(prevStepId, true);\n    stepId = await graph.dispatchRunStep(\n      stepKey,\n      {\n        type: StepTypes.TOOL_CALLS,\n        tool_calls: tool_calls ?? [],\n      },\n      metadata\n    );\n  }\n  await graph.dispatchRunStepDelta(stepId, {\n    type: StepTypes.TOOL_CALLS,\n    tool_calls: toolCallChunks,\n  });\n}\n\nexport const handleToolCalls = async (\n  toolCalls?: ToolCall[],\n  metadata?: Record<string, unknown>,\n  graph?: StandardGraph | MultiAgentGraph\n): Promise<void> => {\n  if (!graph || !metadata) {\n    console.warn(`Graph or metadata not found in ${event} event`);\n    return;\n  }\n\n  if (!toolCalls) {\n    return;\n  }\n\n  if (toolCalls.length === 0) {\n    return;\n  }\n\n  const stepKey = graph.getStepKey(metadata);\n\n  for (const tool_call of toolCalls) {\n    const toolCallId = tool_call.id ?? `toolu_${nanoid()}`;\n    tool_call.id = toolCallId;\n    if (!toolCallId || graph.toolCallStepIds.has(toolCallId)) {\n      continue;\n    }\n\n    let prevStepId = '';\n    let prevRunStep: t.RunStep | undefined;\n    try {\n      prevStepId = graph.getStepIdByKey(stepKey);\n      prevRunStep = graph.getRunStep(prevStepId);\n    } catch {\n      // no previous step\n    }\n\n    /**\n     * NOTE: We do NOT dispatch empty text blocks with tool_call_ids because:\n     * - Empty text blocks cause providers (Anthropic, Bedrock) to reject messages\n     * - They get stored in conversation history and cause errors on replay:\n     *   \"messages must have non-empty content\" (Anthropic)\n     *   \"The content field in the Message object is empty\" (Bedrock)\n     * - The tool_calls themselves are sufficient\n     */\n\n    /* If the previous step exists and is a message creation */\n    if (\n      prevStepId &&\n      prevRunStep &&\n      prevRunStep.type === StepTypes.MESSAGE_CREATION\n    ) {\n      graph.messageStepHasToolCalls.set(prevStepId, true);\n      /* If the previous step doesn't exist or is not a message creation */\n    } else if (\n      !prevRunStep ||\n      prevRunStep.type !== StepTypes.MESSAGE_CREATION\n    ) {\n      const messageId = getMessageId(stepKey, graph, true) ?? '';\n      const stepId = await graph.dispatchRunStep(\n        stepKey,\n        {\n          type: StepTypes.MESSAGE_CREATION,\n          message_creation: {\n            message_id: messageId,\n          },\n        },\n        metadata\n      );\n      graph.messageStepHasToolCalls.set(stepId, true);\n    }\n\n    await graph.dispatchRunStep(\n      stepKey,\n      {\n        type: StepTypes.TOOL_CALLS,\n        tool_calls: [tool_call],\n      },\n      metadata\n    );\n  }\n};\n\nexport const toolResultTypes = new Set([\n  // 'tool_use',\n  // 'server_tool_use',\n  // 'input_json_delta',\n  'tool_result',\n  'web_search_result',\n  'web_search_tool_result',\n]);\n\n/**\n * Handles the result of a server tool call; in other words, a provider's built-in tool.\n * As of 2025-07-06, only Anthropic handles server tool calls with this pattern.\n */\nexport async function handleServerToolResult({\n  graph,\n  content,\n  metadata,\n  agentContext,\n}: {\n  graph: StandardGraph | MultiAgentGraph;\n  content?: string | t.MessageContentComplex[];\n  metadata?: Record<string, unknown>;\n  agentContext?: AgentContext;\n}): Promise<boolean> {\n  let skipHandling = false;\n  if (agentContext?.provider !== Providers.ANTHROPIC) {\n    return skipHandling;\n  }\n  if (\n    typeof content === 'string' ||\n    content == null ||\n    content.length === 0 ||\n    (content.length === 1 &&\n      (content[0] as t.ToolResultContent).tool_use_id == null)\n  ) {\n    return skipHandling;\n  }\n\n  for (const contentPart of content) {\n    const toolUseId = (contentPart as t.ToolResultContent).tool_use_id;\n    if (toolUseId == null || toolUseId === '') {\n      continue;\n    }\n    const stepId = graph.toolCallStepIds.get(toolUseId);\n    if (stepId == null || stepId === '') {\n      console.warn(\n        `Tool use ID ${toolUseId} not found in graph, cannot dispatch tool result.`\n      );\n      continue;\n    }\n    const runStep = graph.getRunStep(stepId);\n    if (!runStep) {\n      console.warn(\n        `Run step for ${stepId} does not exist, cannot dispatch tool result.`\n      );\n      continue;\n    } else if (runStep.type !== StepTypes.TOOL_CALLS) {\n      console.warn(\n        `Run step for ${stepId} is not a tool call step, cannot dispatch tool result.`\n      );\n      continue;\n    }\n\n    const toolCall =\n      runStep.stepDetails.type === StepTypes.TOOL_CALLS\n        ? (runStep.stepDetails.tool_calls?.find(\n          (toolCall) => toolCall.id === toolUseId\n        ) as ToolCall)\n        : undefined;\n\n    if (!toolCall) {\n      continue;\n    }\n\n    if (\n      contentPart.type === 'web_search_result' ||\n      contentPart.type === 'web_search_tool_result'\n    ) {\n      await handleAnthropicSearchResults({\n        contentPart: contentPart as t.ToolResultContent,\n        toolCall,\n        metadata,\n        graph,\n      });\n    }\n\n    if (!skipHandling) {\n      skipHandling = true;\n    }\n  }\n\n  return skipHandling;\n}\n\nasync function handleAnthropicSearchResults({\n  contentPart,\n  toolCall,\n  metadata,\n  graph,\n}: {\n  contentPart: t.ToolResultContent;\n  toolCall: Partial<ToolCall>;\n  metadata?: Record<string, unknown>;\n  graph: StandardGraph | MultiAgentGraph;\n}): Promise<void> {\n  if (!Array.isArray(contentPart.content)) {\n    console.warn(\n      `Expected content to be an array, got ${typeof contentPart.content}`\n    );\n    return;\n  }\n\n  if (!isAnthropicWebSearchResult(contentPart.content[0])) {\n    console.warn(\n      `Expected content to be an Anthropic web search result, got ${JSON.stringify(\n        contentPart.content\n      )}`\n    );\n    return;\n  }\n\n  const turn = graph.invokedToolIds?.size ?? 0;\n  const searchResultData = coerceAnthropicSearchResults({\n    turn,\n    results: contentPart.content as AnthropicWebSearchResultBlockParam[],\n  });\n\n  const name = toolCall.name;\n  const input = toolCall.args ?? {};\n  const artifact = {\n    [Constants.WEB_SEARCH]: searchResultData,\n  };\n  const { output: formattedOutput } = formatResultsForLLM(\n    turn,\n    searchResultData\n  );\n  const output = new ToolMessage({\n    name,\n    artifact,\n    content: formattedOutput,\n    tool_call_id: toolCall.id!,\n  });\n  const toolEndData: t.ToolEndData = {\n    input,\n    output,\n  };\n  await graph.handlerRegistry\n    ?.getHandler(GraphEvents.TOOL_END)\n    ?.handle(GraphEvents.TOOL_END, toolEndData, metadata, graph);\n\n  if (graph.invokedToolIds == null) {\n    graph.invokedToolIds = new Set<string>();\n  }\n\n  graph.invokedToolIds.add(toolCall.id!);\n}\n"],"names":[],"mappings":";;;;;;;;AAAA;AACA;AAsBO,eAAe,oBAAoB,CAAC,EACzC,KAAK,EACL,OAAO,EACP,cAAc,EACd,QAAQ,GAMT,EAAA;AACC,IAAA,IAAI,UAAkB;AACtB,IAAA,IAAI,WAAkC;AACtC,IAAA,IAAI;AACF,QAAA,UAAU,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC;AAC1C,QAAA,WAAW,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;;AAC1C,IAAA,MAAM;;AAEN,QAAA,MAAM,UAAU,GAAG,YAAY,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE;AAC3D,QAAA,UAAU,GAAG,MAAM,KAAK,CAAC,eAAe,CACtC,OAAO,EACP;YACE,IAAI,EAAE,SAAS,CAAC,gBAAgB;AAChC,YAAA,gBAAgB,EAAE;gBAChB,UAAU;AACX,aAAA;SACF,EACD,QAAQ,CACT;AACD,QAAA,WAAW,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;;IAG5C,MAAM,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC;;AAG7C,IAAA,MAAM,UAAU,GACd,UAAU,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC;AAC1D,UAAE;UACA,SAAS;;AAGf,IAAA,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;AAC1C,QAAA,IAAI,aAAa,CAAC,IAAI,KAAK,EAAE,EAAE;AAC7B,YAAA,aAAa,CAAC,IAAI,GAAG,SAAS;;AAEhC,QAAA,IAAI,aAAa,CAAC,EAAE,KAAK,EAAE,EAAE;AAC3B,YAAA,aAAa,CAAC,EAAE,GAAG,SAAS;;aACvB,IACL,UAAU,IAAI,IAAI;YAClB,aAAa,CAAC,EAAE,IAAI,IAAI;AACxB,YAAA,aAAa,CAAC,IAAI,IAAI,IAAI,EAC1B;YACA,UAAU,CAAC,IAAI,CAAC;AACd,gBAAA,IAAI,EAAE,EAAE;gBACR,EAAE,EAAE,aAAa,CAAC,EAAE;gBACpB,IAAI,EAAE,aAAa,CAAC,IAAI;gBACxB,IAAI,EAAE,aAAa,CAAC,SAAS;AAC9B,aAAA,CAAC;;;IAIN,IAAI,MAAM,GAAW,OAAO;IAC5B,MAAM,iBAAiB,GACrB,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,gBAAgB;AAChD,QAAA,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,CAAC;IAE/C,IAAI,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,UAAU,EAAE;AAC9C;;;AAGG;QACH,MAAM,GAAG,UAAU;;AACd,SAAA,IACL,CAAC,iBAAiB;AAClB,QAAA,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,gBAAgB,EAChD;AACA;;;;;;;;;AASG;QACH,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC;AACnD,QAAA,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;;AAEH,IAAA,MAAM,KAAK,CAAC,oBAAoB,CAAC,MAAM,EAAE;QACvC,IAAI,EAAE,SAAS,CAAC,UAAU;AAC1B,QAAA,UAAU,EAAE,cAAc;AAC3B,KAAA,CAAC;AACJ;AAEO,MAAM,eAAe,GAAG,OAC7B,SAAsB,EACtB,QAAkC,EAClC,KAAuC,KACtB;AACjB,IAAA,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,EAAE;AACvB,QAAA,OAAO,CAAC,IAAI,CAAC,kCAAkC,KAAK,CAAA,MAAA,CAAQ,CAAC;QAC7D;;IAGF,IAAI,CAAC,SAAS,EAAE;QACd;;AAGF,IAAA,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;QAC1B;;IAGF,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC;AAE1C,IAAA,KAAK,MAAM,SAAS,IAAI,SAAS,EAAE;QACjC,MAAM,UAAU,GAAG,SAAS,CAAC,EAAE,IAAI,CAAS,MAAA,EAAA,MAAM,EAAE,CAAA,CAAE;AACtD,QAAA,SAAS,CAAC,EAAE,GAAG,UAAU;AACzB,QAAA,IAAI,CAAC,UAAU,IAAI,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;YACxD;;QAGF,IAAI,UAAU,GAAG,EAAE;AACnB,QAAA,IAAI,WAAkC;AACtC,QAAA,IAAI;AACF,YAAA,UAAU,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC;AAC1C,YAAA,WAAW,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;;AAC1C,QAAA,MAAM;;;AAIR;;;;;;;AAOG;;AAGH,QAAA,IACE,UAAU;YACV,WAAW;AACX,YAAA,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC,gBAAgB,EAC/C;YACA,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC;;;AAE9C,aAAA,IACL,CAAC,WAAW;AACZ,YAAA,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC,gBAAgB,EAC/C;AACA,YAAA,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE;YAC1D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CACxC,OAAO,EACP;gBACE,IAAI,EAAE,SAAS,CAAC,gBAAgB;AAChC,gBAAA,gBAAgB,EAAE;AAChB,oBAAA,UAAU,EAAE,SAAS;AACtB,iBAAA;aACF,EACD,QAAQ,CACT;YACD,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC;;AAGjD,QAAA,MAAM,KAAK,CAAC,eAAe,CACzB,OAAO,EACP;YACE,IAAI,EAAE,SAAS,CAAC,UAAU;YAC1B,UAAU,EAAE,CAAC,SAAS,CAAC;SACxB,EACD,QAAQ,CACT;;AAEL;AAEa,MAAA,eAAe,GAAG,IAAI,GAAG,CAAC;;;;IAIrC,aAAa;IACb,mBAAmB;IACnB,wBAAwB;AACzB,CAAA;AAED;;;AAGG;AACI,eAAe,sBAAsB,CAAC,EAC3C,KAAK,EACL,OAAO,EACP,QAAQ,EACR,YAAY,GAMb,EAAA;IACC,IAAI,YAAY,GAAG,KAAK;IACxB,IAAI,YAAY,EAAE,QAAQ,KAAK,SAAS,CAAC,SAAS,EAAE;AAClD,QAAA,OAAO,YAAY;;IAErB,IACE,OAAO,OAAO,KAAK,QAAQ;AAC3B,QAAA,OAAO,IAAI,IAAI;QACf,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,SAAC,OAAO,CAAC,MAAM,KAAK,CAAC;YAClB,OAAO,CAAC,CAAC,CAAyB,CAAC,WAAW,IAAI,IAAI,CAAC,EAC1D;AACA,QAAA,OAAO,YAAY;;AAGrB,IAAA,KAAK,MAAM,WAAW,IAAI,OAAO,EAAE;AACjC,QAAA,MAAM,SAAS,GAAI,WAAmC,CAAC,WAAW;QAClE,IAAI,SAAS,IAAI,IAAI,IAAI,SAAS,KAAK,EAAE,EAAE;YACzC;;QAEF,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC;QACnD,IAAI,MAAM,IAAI,IAAI,IAAI,MAAM,KAAK,EAAE,EAAE;AACnC,YAAA,OAAO,CAAC,IAAI,CACV,eAAe,SAAS,CAAA,iDAAA,CAAmD,CAC5E;YACD;;QAEF,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC;QACxC,IAAI,CAAC,OAAO,EAAE;AACZ,YAAA,OAAO,CAAC,IAAI,CACV,gBAAgB,MAAM,CAAA,6CAAA,CAA+C,CACtE;YACD;;aACK,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,CAAC,UAAU,EAAE;AAChD,YAAA,OAAO,CAAC,IAAI,CACV,gBAAgB,MAAM,CAAA,sDAAA,CAAwD,CAC/E;YACD;;QAGF,MAAM,QAAQ,GACZ,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC;AACrC,cAAG,OAAO,CAAC,WAAW,CAAC,UAAU,EAAE,IAAI,CACrC,CAAC,QAAQ,KAAK,QAAQ,CAAC,EAAE,KAAK,SAAS;cAEvC,SAAS;QAEf,IAAI,CAAC,QAAQ,EAAE;YACb;;AAGF,QAAA,IACE,WAAW,CAAC,IAAI,KAAK,mBAAmB;AACxC,YAAA,WAAW,CAAC,IAAI,KAAK,wBAAwB,EAC7C;AACA,YAAA,MAAM,4BAA4B,CAAC;AACjC,gBAAA,WAAW,EAAE,WAAkC;gBAC/C,QAAQ;gBACR,QAAQ;gBACR,KAAK;AACN,aAAA,CAAC;;QAGJ,IAAI,CAAC,YAAY,EAAE;YACjB,YAAY,GAAG,IAAI;;;AAIvB,IAAA,OAAO,YAAY;AACrB;AAEA,eAAe,4BAA4B,CAAC,EAC1C,WAAW,EACX,QAAQ,EACR,QAAQ,EACR,KAAK,GAMN,EAAA;IACC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE;QACvC,OAAO,CAAC,IAAI,CACV,CAAwC,qCAAA,EAAA,OAAO,WAAW,CAAC,OAAO,CAAE,CAAA,CACrE;QACD;;IAGF,IAAI,CAAC,0BAA0B,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE;AACvD,QAAA,OAAO,CAAC,IAAI,CACV,CAAA,2DAAA,EAA8D,IAAI,CAAC,SAAS,CAC1E,WAAW,CAAC,OAAO,CACpB,CAAA,CAAE,CACJ;QACD;;IAGF,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,EAAE,IAAI,IAAI,CAAC;IAC5C,MAAM,gBAAgB,GAAG,4BAA4B,CAAC;QACpD,IAAI;QACJ,OAAO,EAAE,WAAW,CAAC,OAA+C;AACrE,KAAA,CAAC;AAEF,IAAA,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI;AAC1B,IAAA,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,IAAI,EAAE;AACjC,IAAA,MAAM,QAAQ,GAAG;AACf,QAAA,CAAC,SAAS,CAAC,UAAU,GAAG,gBAAgB;KACzC;AACD,IAAA,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,mBAAmB,CACrD,IAAI,EACJ,gBAAgB,CACjB;AACD,IAAA,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC;QAC7B,IAAI;QACJ,QAAQ;AACR,QAAA,OAAO,EAAE,eAAe;QACxB,YAAY,EAAE,QAAQ,CAAC,EAAG;AAC3B,KAAA,CAAC;AACF,IAAA,MAAM,WAAW,GAAkB;QACjC,KAAK;QACL,MAAM;KACP;IACD,MAAM,KAAK,CAAC;AACV,UAAE,UAAU,CAAC,WAAW,CAAC,QAAQ;AACjC,UAAE,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,CAAC;AAE9D,IAAA,IAAI,KAAK,CAAC,cAAc,IAAI,IAAI,EAAE;AAChC,QAAA,KAAK,CAAC,cAAc,GAAG,IAAI,GAAG,EAAU;;IAG1C,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAG,CAAC;AACxC;;;;"}
+{"version":3,"file":"handlers.mjs","sources":["../../../src/tools/handlers.ts"],"sourcesContent":["/* eslint-disable no-console */\n// src/tools/handlers.ts\nimport { nanoid } from 'nanoid';\nimport { ToolMessage } from '@langchain/core/messages';\nimport type { AnthropicWebSearchResultBlockParam } from '@/llm/anthropic/types';\nimport type { ToolCall, ToolCallChunk } from '@langchain/core/messages/tool';\nimport type { MultiAgentGraph, StandardGraph } from '@/graphs';\nimport type { AgentContext } from '@/agents/AgentContext';\nimport type * as t from '@/types';\nimport {\n  ToolCallTypes,\n  GraphEvents,\n  StepTypes,\n  Providers,\n  Constants,\n} from '@/common';\nimport {\n  coerceAnthropicSearchResults,\n  isAnthropicWebSearchResult,\n} from '@/tools/search/anthropic';\nimport { formatResultsForLLM } from '@/tools/search/format';\nimport { getMessageId } from '@/messages';\n\nexport async function handleToolCallChunks({\n  graph,\n  stepKey,\n  toolCallChunks,\n  metadata,\n}: {\n  graph: StandardGraph | MultiAgentGraph;\n  stepKey: string;\n  toolCallChunks: ToolCallChunk[];\n  metadata?: Record<string, unknown>;\n}): Promise<void> {\n  let prevStepId: string;\n  let prevRunStep: t.RunStep | undefined;\n  try {\n    prevStepId = graph.getStepIdByKey(stepKey);\n    prevRunStep = graph.getRunStep(prevStepId);\n  } catch {\n    /** Edge Case: If no previous step exists, create a new message creation step */\n    const message_id = getMessageId(stepKey, graph, true) ?? '';\n    prevStepId = await graph.dispatchRunStep(\n      stepKey,\n      {\n        type: StepTypes.MESSAGE_CREATION,\n        message_creation: {\n          message_id,\n        },\n      },\n      metadata\n    );\n    prevRunStep = graph.getRunStep(prevStepId);\n  }\n\n  const _stepId = graph.getStepIdByKey(stepKey);\n\n  /** Edge Case: Tool Call Run Step or `tool_call_ids` never dispatched */\n  const tool_calls: ToolCall[] | undefined =\n    prevStepId && prevRunStep && prevRunStep.type === StepTypes.MESSAGE_CREATION\n      ? []\n      : undefined;\n\n  /** Edge Case: `id` and `name` fields cannot be empty strings */\n  for (const toolCallChunk of toolCallChunks) {\n    if (toolCallChunk.name === '') {\n      toolCallChunk.name = undefined;\n    }\n    if (toolCallChunk.id === '') {\n      toolCallChunk.id = undefined;\n    } else if (\n      tool_calls != null &&\n      toolCallChunk.id != null &&\n      toolCallChunk.name != null\n    ) {\n      tool_calls.push({\n        args: {},\n        id: toolCallChunk.id,\n        name: toolCallChunk.name,\n        type: ToolCallTypes.TOOL_CALL,\n      });\n    }\n  }\n\n  let stepId: string = _stepId;\n  const alreadyDispatched =\n    prevRunStep?.type === StepTypes.MESSAGE_CREATION &&\n    graph.messageStepHasToolCalls.has(prevStepId);\n\n  if (prevRunStep?.type === StepTypes.TOOL_CALLS) {\n    /**\n     * If previous step is already a tool_calls step, use that step ID\n     * This ensures tool call deltas are dispatched to the correct step\n     */\n    stepId = prevStepId;\n  } else if (\n    !alreadyDispatched &&\n    prevRunStep?.type === StepTypes.MESSAGE_CREATION\n  ) {\n    /**\n     * Create tool_calls step as soon as we receive the first tool call chunk\n     * This ensures deltas are always associated with the correct step\n     *\n     * NOTE: We do NOT dispatch an empty text block here because:\n     * - Empty text blocks cause providers (Anthropic, Bedrock) to reject messages\n     * - The tool_calls themselves are sufficient for the step\n     * - Empty content with tool_call_ids gets stored in conversation history\n     *   and causes \"messages must have non-empty content\" errors on replay\n     */\n    graph.messageStepHasToolCalls.set(prevStepId, true);\n    stepId = await graph.dispatchRunStep(\n      stepKey,\n      {\n        type: StepTypes.TOOL_CALLS,\n        tool_calls: tool_calls ?? [],\n      },\n      metadata\n    );\n  }\n\n  await graph.dispatchRunStepDelta(stepId, {\n    type: StepTypes.TOOL_CALLS,\n    tool_calls: toolCallChunks,\n  });\n}\n\nexport const handleToolCalls = async (\n  toolCalls?: ToolCall[],\n  metadata?: Record<string, unknown>,\n  graph?: StandardGraph | MultiAgentGraph\n): Promise<void> => {\n  if (!graph || !metadata) {\n    console.warn('Graph or metadata not found in `handleToolCalls`');\n    return;\n  }\n\n  if (!toolCalls) {\n    return;\n  }\n\n  if (toolCalls.length === 0) {\n    return;\n  }\n\n  const stepKey = graph.getStepKey(metadata);\n\n  /**\n   * Track whether we've already reused an empty TOOL_CALLS step created by\n   * handleToolCallChunks during streaming. Only reuse it once (for the first\n   * tool call); subsequent parallel tool calls must create their own steps.\n   */\n  let reusedChunkStepId: string | undefined;\n\n  for (const tool_call of toolCalls) {\n    const toolCallId = tool_call.id ?? `toolu_${nanoid()}`;\n    tool_call.id = toolCallId;\n    if (!toolCallId || graph.toolCallStepIds.has(toolCallId)) {\n      continue;\n    }\n\n    let prevStepId = '';\n    let prevRunStep: t.RunStep | undefined;\n    try {\n      prevStepId = graph.getStepIdByKey(stepKey);\n      prevRunStep = graph.getRunStep(prevStepId);\n    } catch {\n      // no previous step\n    }\n\n    /**\n     * If the previous step is TOOL_CALLS (from handleToolCallChunks or a prior\n     * iteration), either reuse it (if empty) or dispatch a new TOOL_CALLS step\n     * directly — skip the intermediate MESSAGE_CREATION to avoid orphaned gaps.\n     */\n    if (prevRunStep?.type === StepTypes.TOOL_CALLS) {\n      const details = prevRunStep.stepDetails as t.ToolCallsDetails;\n      const isEmpty = !details.tool_calls || details.tool_calls.length === 0;\n      if (isEmpty && prevStepId !== reusedChunkStepId) {\n        graph.toolCallStepIds.set(toolCallId, prevStepId);\n        reusedChunkStepId = prevStepId;\n        continue;\n      }\n      await graph.dispatchRunStep(\n        stepKey,\n        { type: StepTypes.TOOL_CALLS, tool_calls: [tool_call] },\n        metadata\n      );\n      continue;\n    }\n\n    /**\n     * NOTE: We do NOT dispatch empty text blocks with tool_call_ids because:\n     * - Empty text blocks cause providers (Anthropic, Bedrock) to reject messages\n     * - They get stored in conversation history and cause errors on replay:\n     *   \"messages must have non-empty content\" (Anthropic)\n     *   \"The content field in the Message object is empty\" (Bedrock)\n     * - The tool_calls themselves are sufficient\n     */\n    if (prevStepId && prevRunStep) {\n      graph.messageStepHasToolCalls.set(prevStepId, true);\n    } else if (!prevRunStep) {\n      const messageId = getMessageId(stepKey, graph, true) ?? '';\n      const stepId = await graph.dispatchRunStep(\n        stepKey,\n        {\n          type: StepTypes.MESSAGE_CREATION,\n          message_creation: {\n            message_id: messageId,\n          },\n        },\n        metadata\n      );\n      graph.messageStepHasToolCalls.set(stepId, true);\n    }\n\n    await graph.dispatchRunStep(\n      stepKey,\n      {\n        type: StepTypes.TOOL_CALLS,\n        tool_calls: [tool_call],\n      },\n      metadata\n    );\n  }\n};\n\nexport const toolResultTypes = new Set([\n  // 'tool_use',\n  // 'server_tool_use',\n  // 'input_json_delta',\n  'tool_result',\n  'web_search_result',\n  'web_search_tool_result',\n]);\n\n/**\n * Handles the result of a server tool call; in other words, a provider's built-in tool.\n * As of 2025-07-06, only Anthropic handles server tool calls with this pattern.\n */\nexport async function handleServerToolResult({\n  graph,\n  content,\n  metadata,\n  agentContext,\n}: {\n  graph: StandardGraph | MultiAgentGraph;\n  content?: string | t.MessageContentComplex[];\n  metadata?: Record<string, unknown>;\n  agentContext?: AgentContext;\n}): Promise<boolean> {\n  let skipHandling = false;\n  if (agentContext?.provider !== Providers.ANTHROPIC) {\n    return skipHandling;\n  }\n  if (\n    typeof content === 'string' ||\n    content == null ||\n    content.length === 0 ||\n    (content.length === 1 &&\n      (content[0] as t.ToolResultContent).tool_use_id == null)\n  ) {\n    return skipHandling;\n  }\n\n  for (const contentPart of content) {\n    const toolUseId = (contentPart as t.ToolResultContent).tool_use_id;\n    if (toolUseId == null || toolUseId === '') {\n      continue;\n    }\n    const stepId = graph.toolCallStepIds.get(toolUseId);\n    if (stepId == null || stepId === '') {\n      console.warn(\n        `Tool use ID ${toolUseId} not found in graph, cannot dispatch tool result.`\n      );\n      continue;\n    }\n    const runStep = graph.getRunStep(stepId);\n    if (!runStep) {\n      console.warn(\n        `Run step for ${stepId} does not exist, cannot dispatch tool result.`\n      );\n      continue;\n    } else if (runStep.type !== StepTypes.TOOL_CALLS) {\n      console.warn(\n        `Run step for ${stepId} is not a tool call step, cannot dispatch tool result.`\n      );\n      continue;\n    }\n\n    const toolCall =\n      runStep.stepDetails.type === StepTypes.TOOL_CALLS\n        ? (runStep.stepDetails.tool_calls?.find(\n          (toolCall) => toolCall.id === toolUseId\n        ) as ToolCall)\n        : undefined;\n\n    if (!toolCall) {\n      continue;\n    }\n\n    if (\n      contentPart.type === 'web_search_result' ||\n      contentPart.type === 'web_search_tool_result'\n    ) {\n      await handleAnthropicSearchResults({\n        contentPart: contentPart as t.ToolResultContent,\n        toolCall,\n        metadata,\n        graph,\n      });\n    }\n\n    if (!skipHandling) {\n      skipHandling = true;\n    }\n  }\n\n  return skipHandling;\n}\n\nasync function handleAnthropicSearchResults({\n  contentPart,\n  toolCall,\n  metadata,\n  graph,\n}: {\n  contentPart: t.ToolResultContent;\n  toolCall: Partial<ToolCall>;\n  metadata?: Record<string, unknown>;\n  graph: StandardGraph | MultiAgentGraph;\n}): Promise<void> {\n  if (!Array.isArray(contentPart.content)) {\n    console.warn(\n      `Expected content to be an array, got ${typeof contentPart.content}`\n    );\n    return;\n  }\n\n  if (!isAnthropicWebSearchResult(contentPart.content[0])) {\n    console.warn(\n      `Expected content to be an Anthropic web search result, got ${JSON.stringify(\n        contentPart.content\n      )}`\n    );\n    return;\n  }\n\n  const turn = graph.invokedToolIds?.size ?? 0;\n  const searchResultData = coerceAnthropicSearchResults({\n    turn,\n    results: contentPart.content as AnthropicWebSearchResultBlockParam[],\n  });\n\n  const name = toolCall.name;\n  const input = toolCall.args ?? {};\n  const artifact = {\n    [Constants.WEB_SEARCH]: searchResultData,\n  };\n  const { output: formattedOutput } = formatResultsForLLM(\n    turn,\n    searchResultData\n  );\n  const output = new ToolMessage({\n    name,\n    artifact,\n    content: formattedOutput,\n    tool_call_id: toolCall.id!,\n  });\n  const toolEndData: t.ToolEndData = {\n    input,\n    output,\n  };\n  await graph.handlerRegistry\n    ?.getHandler(GraphEvents.TOOL_END)\n    ?.handle(GraphEvents.TOOL_END, toolEndData, metadata, graph);\n\n  if (graph.invokedToolIds == null) {\n    graph.invokedToolIds = new Set<string>();\n  }\n\n  graph.invokedToolIds.add(toolCall.id!);\n}\n"],"names":[],"mappings":";;;;;;;;AAAA;AACA;AAsBO,eAAe,oBAAoB,CAAC,EACzC,KAAK,EACL,OAAO,EACP,cAAc,EACd,QAAQ,GAMT,EAAA;AACC,IAAA,IAAI,UAAkB;AACtB,IAAA,IAAI,WAAkC;AACtC,IAAA,IAAI;AACF,QAAA,UAAU,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC;AAC1C,QAAA,WAAW,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;;AAC1C,IAAA,MAAM;;AAEN,QAAA,MAAM,UAAU,GAAG,YAAY,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE;AAC3D,QAAA,UAAU,GAAG,MAAM,KAAK,CAAC,eAAe,CACtC,OAAO,EACP;YACE,IAAI,EAAE,SAAS,CAAC,gBAAgB;AAChC,YAAA,gBAAgB,EAAE;gBAChB,UAAU;AACX,aAAA;SACF,EACD,QAAQ,CACT;AACD,QAAA,WAAW,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;;IAG5C,MAAM,OAAO,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC;;AAG7C,IAAA,MAAM,UAAU,GACd,UAAU,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC;AAC1D,UAAE;UACA,SAAS;;AAGf,IAAA,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;AAC1C,QAAA,IAAI,aAAa,CAAC,IAAI,KAAK,EAAE,EAAE;AAC7B,YAAA,aAAa,CAAC,IAAI,GAAG,SAAS;;AAEhC,QAAA,IAAI,aAAa,CAAC,EAAE,KAAK,EAAE,EAAE;AAC3B,YAAA,aAAa,CAAC,EAAE,GAAG,SAAS;;aACvB,IACL,UAAU,IAAI,IAAI;YAClB,aAAa,CAAC,EAAE,IAAI,IAAI;AACxB,YAAA,aAAa,CAAC,IAAI,IAAI,IAAI,EAC1B;YACA,UAAU,CAAC,IAAI,CAAC;AACd,gBAAA,IAAI,EAAE,EAAE;gBACR,EAAE,EAAE,aAAa,CAAC,EAAE;gBACpB,IAAI,EAAE,aAAa,CAAC,IAAI;gBACxB,IAAI,EAAE,aAAa,CAAC,SAAS;AAC9B,aAAA,CAAC;;;IAIN,IAAI,MAAM,GAAW,OAAO;IAC5B,MAAM,iBAAiB,GACrB,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,gBAAgB;AAChD,QAAA,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,CAAC;IAE/C,IAAI,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,UAAU,EAAE;AAC9C;;;AAGG;QACH,MAAM,GAAG,UAAU;;AACd,SAAA,IACL,CAAC,iBAAiB;AAClB,QAAA,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,gBAAgB,EAChD;AACA;;;;;;;;;AASG;QACH,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC;AACnD,QAAA,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;;AAGH,IAAA,MAAM,KAAK,CAAC,oBAAoB,CAAC,MAAM,EAAE;QACvC,IAAI,EAAE,SAAS,CAAC,UAAU;AAC1B,QAAA,UAAU,EAAE,cAAc;AAC3B,KAAA,CAAC;AACJ;AAEO,MAAM,eAAe,GAAG,OAC7B,SAAsB,EACtB,QAAkC,EAClC,KAAuC,KACtB;AACjB,IAAA,IAAI,CAAC,KAAK,IAAI,CAAC,QAAQ,EAAE;AACvB,QAAA,OAAO,CAAC,IAAI,CAAC,kDAAkD,CAAC;QAChE;;IAGF,IAAI,CAAC,SAAS,EAAE;QACd;;AAGF,IAAA,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;QAC1B;;IAGF,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC;AAE1C;;;;AAIG;AACH,IAAA,IAAI,iBAAqC;AAEzC,IAAA,KAAK,MAAM,SAAS,IAAI,SAAS,EAAE;QACjC,MAAM,UAAU,GAAG,SAAS,CAAC,EAAE,IAAI,CAAS,MAAA,EAAA,MAAM,EAAE,CAAA,CAAE;AACtD,QAAA,SAAS,CAAC,EAAE,GAAG,UAAU;AACzB,QAAA,IAAI,CAAC,UAAU,IAAI,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;YACxD;;QAGF,IAAI,UAAU,GAAG,EAAE;AACnB,QAAA,IAAI,WAAkC;AACtC,QAAA,IAAI;AACF,YAAA,UAAU,GAAG,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC;AAC1C,YAAA,WAAW,GAAG,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;;AAC1C,QAAA,MAAM;;;AAIR;;;;AAIG;QACH,IAAI,WAAW,EAAE,IAAI,KAAK,SAAS,CAAC,UAAU,EAAE;AAC9C,YAAA,MAAM,OAAO,GAAG,WAAW,CAAC,WAAiC;AAC7D,YAAA,MAAM,OAAO,GAAG,CAAC,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC;AACtE,YAAA,IAAI,OAAO,IAAI,UAAU,KAAK,iBAAiB,EAAE;gBAC/C,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAAC;gBACjD,iBAAiB,GAAG,UAAU;gBAC9B;;YAEF,MAAM,KAAK,CAAC,eAAe,CACzB,OAAO,EACP,EAAE,IAAI,EAAE,SAAS,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC,SAAS,CAAC,EAAE,EACvD,QAAQ,CACT;YACD;;AAGF;;;;;;;AAOG;AACH,QAAA,IAAI,UAAU,IAAI,WAAW,EAAE;YAC7B,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC;;aAC9C,IAAI,CAAC,WAAW,EAAE;AACvB,YAAA,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE;YAC1D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CACxC,OAAO,EACP;gBACE,IAAI,EAAE,SAAS,CAAC,gBAAgB;AAChC,gBAAA,gBAAgB,EAAE;AAChB,oBAAA,UAAU,EAAE,SAAS;AACtB,iBAAA;aACF,EACD,QAAQ,CACT;YACD,KAAK,CAAC,uBAAuB,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC;;AAGjD,QAAA,MAAM,KAAK,CAAC,eAAe,CACzB,OAAO,EACP;YACE,IAAI,EAAE,SAAS,CAAC,UAAU;YAC1B,UAAU,EAAE,CAAC,SAAS,CAAC;SACxB,EACD,QAAQ,CACT;;AAEL;AAEa,MAAA,eAAe,GAAG,IAAI,GAAG,CAAC;;;;IAIrC,aAAa;IACb,mBAAmB;IACnB,wBAAwB;AACzB,CAAA;AAED;;;AAGG;AACI,eAAe,sBAAsB,CAAC,EAC3C,KAAK,EACL,OAAO,EACP,QAAQ,EACR,YAAY,GAMb,EAAA;IACC,IAAI,YAAY,GAAG,KAAK;IACxB,IAAI,YAAY,EAAE,QAAQ,KAAK,SAAS,CAAC,SAAS,EAAE;AAClD,QAAA,OAAO,YAAY;;IAErB,IACE,OAAO,OAAO,KAAK,QAAQ;AAC3B,QAAA,OAAO,IAAI,IAAI;QACf,OAAO,CAAC,MAAM,KAAK,CAAC;AACpB,SAAC,OAAO,CAAC,MAAM,KAAK,CAAC;YAClB,OAAO,CAAC,CAAC,CAAyB,CAAC,WAAW,IAAI,IAAI,CAAC,EAC1D;AACA,QAAA,OAAO,YAAY;;AAGrB,IAAA,KAAK,MAAM,WAAW,IAAI,OAAO,EAAE;AACjC,QAAA,MAAM,SAAS,GAAI,WAAmC,CAAC,WAAW;QAClE,IAAI,SAAS,IAAI,IAAI,IAAI,SAAS,KAAK,EAAE,EAAE;YACzC;;QAEF,MAAM,MAAM,GAAG,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC;QACnD,IAAI,MAAM,IAAI,IAAI,IAAI,MAAM,KAAK,EAAE,EAAE;AACnC,YAAA,OAAO,CAAC,IAAI,CACV,eAAe,SAAS,CAAA,iDAAA,CAAmD,CAC5E;YACD;;QAEF,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC;QACxC,IAAI,CAAC,OAAO,EAAE;AACZ,YAAA,OAAO,CAAC,IAAI,CACV,gBAAgB,MAAM,CAAA,6CAAA,CAA+C,CACtE;YACD;;aACK,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,CAAC,UAAU,EAAE;AAChD,YAAA,OAAO,CAAC,IAAI,CACV,gBAAgB,MAAM,CAAA,sDAAA,CAAwD,CAC/E;YACD;;QAGF,MAAM,QAAQ,GACZ,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC;AACrC,cAAG,OAAO,CAAC,WAAW,CAAC,UAAU,EAAE,IAAI,CACrC,CAAC,QAAQ,KAAK,QAAQ,CAAC,EAAE,KAAK,SAAS;cAEvC,SAAS;QAEf,IAAI,CAAC,QAAQ,EAAE;YACb;;AAGF,QAAA,IACE,WAAW,CAAC,IAAI,KAAK,mBAAmB;AACxC,YAAA,WAAW,CAAC,IAAI,KAAK,wBAAwB,EAC7C;AACA,YAAA,MAAM,4BAA4B,CAAC;AACjC,gBAAA,WAAW,EAAE,WAAkC;gBAC/C,QAAQ;gBACR,QAAQ;gBACR,KAAK;AACN,aAAA,CAAC;;QAGJ,IAAI,CAAC,YAAY,EAAE;YACjB,YAAY,GAAG,IAAI;;;AAIvB,IAAA,OAAO,YAAY;AACrB;AAEA,eAAe,4BAA4B,CAAC,EAC1C,WAAW,EACX,QAAQ,EACR,QAAQ,EACR,KAAK,GAMN,EAAA;IACC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE;QACvC,OAAO,CAAC,IAAI,CACV,CAAwC,qCAAA,EAAA,OAAO,WAAW,CAAC,OAAO,CAAE,CAAA,CACrE;QACD;;IAGF,IAAI,CAAC,0BAA0B,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE;AACvD,QAAA,OAAO,CAAC,IAAI,CACV,CAAA,2DAAA,EAA8D,IAAI,CAAC,SAAS,CAC1E,WAAW,CAAC,OAAO,CACpB,CAAA,CAAE,CACJ;QACD;;IAGF,MAAM,IAAI,GAAG,KAAK,CAAC,cAAc,EAAE,IAAI,IAAI,CAAC;IAC5C,MAAM,gBAAgB,GAAG,4BAA4B,CAAC;QACpD,IAAI;QACJ,OAAO,EAAE,WAAW,CAAC,OAA+C;AACrE,KAAA,CAAC;AAEF,IAAA,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI;AAC1B,IAAA,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,IAAI,EAAE;AACjC,IAAA,MAAM,QAAQ,GAAG;AACf,QAAA,CAAC,SAAS,CAAC,UAAU,GAAG,gBAAgB;KACzC;AACD,IAAA,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,mBAAmB,CACrD,IAAI,EACJ,gBAAgB,CACjB;AACD,IAAA,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC;QAC7B,IAAI;QACJ,QAAQ;AACR,QAAA,OAAO,EAAE,eAAe;QACxB,YAAY,EAAE,QAAQ,CAAC,EAAG;AAC3B,KAAA,CAAC;AACF,IAAA,MAAM,WAAW,GAAkB;QACjC,KAAK;QACL,MAAM;KACP;IACD,MAAM,KAAK,CAAC;AACV,UAAE,UAAU,CAAC,WAAW,CAAC,QAAQ;AACjC,UAAE,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,CAAC;AAE9D,IAAA,IAAI,KAAK,CAAC,cAAc,IAAI,IAAI,EAAE;AAChC,QAAA,KAAK,CAAC,cAAc,GAAG,IAAI,GAAG,EAAU;;IAG1C,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAG,CAAC;AACxC;;;;"}

package/dist/types/agents/AgentContext.d.ts

@@ -64,6 +64,8 @@ export declare class AgentContext {
     lastToken?: string;
     /** Token type switch state */
     tokenTypeSwitch?: 'reasoning' | 'content';
+    /** Tracks how many reasoning→text transitions have occurred (ensures unique post-reasoning step keys) */
+    reasoningTransitionCount: number;
     /** Current token type being processed */
     currentTokenType: ContentTypes.TEXT | ContentTypes.THINK | 'think_and_text';
     /** Whether tools should end the workflow */

package/dist/types/types/tools.d.ts

@@ -48,6 +48,7 @@ export type ToolEndEvent = {
     tool_call: ToolCall;
     /** The content index of the tool call */
     index: number;
+    type?: 'tool_call';
 };
 export type CodeEnvFile = {
     id: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@librechat/agents",
-  "version": "3.1.37",
+  "version": "3.1.38",
   "main": "./dist/cjs/main.cjs",
   "module": "./dist/esm/main.mjs",
   "types": "./dist/types/index.d.ts",

package/src/agents/AgentContext.ts

@@ -142,6 +142,8 @@ export class AgentContext {
   lastToken?: string;
   /** Token type switch state */
   tokenTypeSwitch?: 'reasoning' | 'content';
+  /** Tracks how many reasoning→text transitions have occurred (ensures unique post-reasoning step keys) */
+  reasoningTransitionCount = 0;
   /** Current token type being processed */
   currentTokenType: ContentTypes.TEXT | ContentTypes.THINK | 'think_and_text' =
     ContentTypes.TEXT;
@@ -462,6 +464,7 @@ export class AgentContext {
     this.pruneMessages = undefined;
     this.lastStreamCall = undefined;
     this.tokenTypeSwitch = undefined;
+    this.reasoningTransitionCount = 0;
     this.currentTokenType = ContentTypes.TEXT;
     this.discoveredToolNames.clear();
     this.handoffContext = undefined;

package/src/graphs/Graph.ts

@@ -314,7 +314,7 @@ export class StandardGraph extends Graph<t.BaseGraphState, t.GraphNode> {
     ) {
       keyList.push('reasoning');
     } else if (agentContext.tokenTypeSwitch === 'content') {
-      keyList.push('post-reasoning');
+      keyList.push(`post-reasoning-${agentContext.reasoningTransitionCount}`);
     }
 
     if (this.invokedToolIds != null && this.invokedToolIds.size > 0) {

package/src/scripts/bedrock-content-aggregation-test.ts

@@ -0,0 +1,265 @@
+import { config } from 'dotenv';
+config();
+import { HumanMessage, BaseMessage } from '@langchain/core/messages';
+import type { UsageMetadata } from '@langchain/core/messages';
+import * as t from '@/types';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { createCodeExecutionTool } from '@/tools/CodeExecutor';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { GraphEvents, ContentTypes, Providers } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { Run } from '@/run';
+
+const conversationHistory: BaseMessage[] = [];
+let _contentParts: t.MessageContentComplex[] = [];
+const collectedUsage: UsageMetadata[] = [];
+
+async function testBedrockContentAggregation(): Promise<void> {
+  const instructions =
+    'You are a helpful AI assistant with coding capabilities. When answering questions, be thorough in your reasoning.';
+  const { contentParts, aggregateContent } = createContentAggregator();
+  _contentParts = contentParts as t.MessageContentComplex[];
+
+  const customHandlers = {
+    [GraphEvents.TOOL_END]: new ToolEndHandler(),
+    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
+    [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
+    [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_COMPLETED,
+        data: t.StreamEventData
+      ): void => {
+        const result = (data as unknown as { result: t.ToolEndEvent }).result;
+        console.log(
+          `[ON_RUN_STEP_COMPLETED] stepId=${result.id} index=${result.index} type=${result.type} tool=${result.tool_call?.name ?? 'n/a'}`
+        );
+        aggregateContent({
+          event,
+          data: data as unknown as { result: t.ToolEndEvent },
+        });
+      },
+    },
+    [GraphEvents.ON_RUN_STEP]: {
+      handle: (event: GraphEvents.ON_RUN_STEP, data: t.RunStep) => {
+        const toolCalls =
+          data.stepDetails.type === 'tool_calls' && data.stepDetails.tool_calls
+            ? (
+                data.stepDetails.tool_calls as Array<{
+                  name?: string;
+                  id?: string;
+                }>
+              )
+                .map((tc) => `${tc.name ?? '?'}(${tc.id ?? '?'})`)
+                .join(', ')
+            : 'none';
+        console.log(
+          `[ON_RUN_STEP] stepId=${data.id} index=${data.index} type=${data.type} stepIndex=${data.stepIndex} toolCalls=[${toolCalls}]`
+        );
+        aggregateContent({ event, data });
+      },
+    },
+    [GraphEvents.ON_RUN_STEP_DELTA]: {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_DELTA,
+        data: t.RunStepDeltaEvent
+      ) => {
+        const tcNames =
+          data.delta.tool_calls
+            ?.map(
+              (tc) =>
+                `${tc.name ?? '?'}(args=${(tc.args ?? '').substring(0, 30)}...)`
+            )
+            .join(', ') ?? 'none';
+        console.log(
+          `[ON_RUN_STEP_DELTA] stepId=${data.id} type=${data.delta.type} toolCalls=[${tcNames}]`
+        );
+        aggregateContent({ event, data });
+      },
+    },
+    [GraphEvents.ON_MESSAGE_DELTA]: {
+      handle: (
+        event: GraphEvents.ON_MESSAGE_DELTA,
+        data: t.MessageDeltaEvent
+      ) => {
+        const preview = Array.isArray(data.delta.content)
+          ? data.delta.content
+              .map(
+                (c) =>
+                  `${c.type}:"${String((c as Record<string, unknown>).text ?? (c as Record<string, unknown>).think ?? '').substring(0, 40)}"`
+              )
+              .join(', ')
+          : String(data.delta.content).substring(0, 40);
+        console.log(
+          `[ON_MESSAGE_DELTA] stepId=${data.id} content=[${preview}]`
+        );
+        aggregateContent({ event, data });
+      },
+    },
+    [GraphEvents.ON_REASONING_DELTA]: {
+      handle: (
+        event: GraphEvents.ON_REASONING_DELTA,
+        data: t.ReasoningDeltaEvent
+      ) => {
+        const preview = Array.isArray(data.delta.content)
+          ? data.delta.content
+              .map(
+                (c) =>
+                  `${c.type}:"${String((c as Record<string, unknown>).think ?? '').substring(0, 40)}"`
+              )
+              .join(', ')
+          : '?';
+        console.log(
+          `[ON_REASONING_DELTA] stepId=${data.id} content=[${preview}]`
+        );
+        aggregateContent({ event, data });
+      },
+    },
+  };
+
+  const baseLlmConfig = getLLMConfig(Providers.BEDROCK);
+
+  const llmConfig = {
+    ...baseLlmConfig,
+    model: 'global.anthropic.claude-opus-4-6-v1',
+    maxTokens: 16000,
+    additionalModelRequestFields: {
+      thinking: { type: 'enabled', budget_tokens: 10000 },
+    },
+  };
+
+  const run = await Run.create<t.IState>({
+    runId: 'bedrock-content-aggregation-test',
+    graphConfig: {
+      instructions,
+      type: 'standard',
+      tools: [createCodeExecutionTool()],
+      llmConfig,
+    },
+    returnContent: true,
+    customHandlers: customHandlers as t.RunConfig['customHandlers'],
+  });
+
+  const streamConfig = {
+    configurable: {
+      thread_id: 'bedrock-content-aggregation-thread',
+    },
+    streamMode: 'values',
+    version: 'v2' as const,
+  };
+
+  const userMessage = `im testing edge cases with our code interpreter. i know we can persist files, but what happens when we put them in directories?`;
+  conversationHistory.push(new HumanMessage(userMessage));
+
+  console.log('Running Bedrock content aggregation test...\n');
+  console.log(`Prompt: "${userMessage}"\n`);
+
+  const inputs = { messages: [...conversationHistory] };
+  await run.processStream(inputs, streamConfig);
+
+  console.log('\n\n========== CONTENT PARTS ANALYSIS ==========\n');
+
+  let hasEmptyToolCall = false;
+  let hasReasoningOrderIssue = false;
+
+  for (let i = 0; i < _contentParts.length; i++) {
+    const part = _contentParts[i];
+    if (!part) {
+      console.log(`  [${i}] undefined`);
+      continue;
+    }
+
+    const partType = part.type;
+    if (partType === ContentTypes.TOOL_CALL) {
+      const tc = (part as t.ToolCallContent).tool_call;
+      if (!tc || !tc.name) {
+        hasEmptyToolCall = true;
+        console.log(`  [${i}] TOOL_CALL *** EMPTY (no tool_call data) ***`);
+      } else {
+        const outputPreview = tc.output
+          ? `output=${(tc.output as string).substring(0, 80)}...`
+          : 'no output';
+        console.log(`  [${i}] TOOL_CALL name=${tc.name} ${outputPreview}`);
+      }
+    } else if (partType === ContentTypes.THINK) {
+      const think = (part as t.ReasoningContentText).think ?? '';
+      console.log(
+        `  [${i}] THINK (${think.length} chars): "${think.substring(0, 80)}..."`
+      );
+    } else if (partType === ContentTypes.TEXT) {
+      const text = (part as t.MessageDeltaUpdate).text ?? '';
+      console.log(
+        `  [${i}] TEXT (${text.length} chars): "${text.substring(0, 80)}..."`
+      );
+    } else {
+      console.log(`  [${i}] ${partType}`);
+    }
+  }
+
+  /**
+   * Check reasoning ordering within a single invocation cycle.
+   * A tool_call resets the cycle — text before think across different
+   * invocations (e.g., text from invocation 2, think from invocation 3) is valid.
+   */
+  let lastTextInCycle: number | null = null;
+  for (let i = 0; i < _contentParts.length; i++) {
+    const part = _contentParts[i];
+    if (!part) continue;
+
+    if (part.type === ContentTypes.TOOL_CALL) {
+      lastTextInCycle = null;
+      continue;
+    }
+
+    if (part.type === ContentTypes.TEXT) {
+      lastTextInCycle = i;
+    } else if (part.type === ContentTypes.THINK && lastTextInCycle !== null) {
+      const prevText = _contentParts[lastTextInCycle] as t.MessageDeltaUpdate;
+      const thinkContent = (part as t.ReasoningContentText).think ?? '';
+      if (
+        prevText?.text &&
+        prevText.text.trim().length > 5 &&
+        thinkContent.length > 0
+      ) {
+        hasReasoningOrderIssue = true;
+        console.log(
+          `\n  *** ORDERING ISSUE (same invocation): TEXT at [${lastTextInCycle}] appears before THINK at [${i}]`
+        );
+        console.log(
+          `      Text ends with: "...${prevText.text.substring(prevText.text.length - 60)}"`
+        );
+        console.log(
+          `      Think starts with: "${thinkContent.substring(0, 60)}..."`
+        );
+      }
+    }
+  }
+
+  console.log('\n========== SUMMARY ==========\n');
+  console.log(`Total content parts: ${_contentParts.filter(Boolean).length}`);
+  console.log(
+    `Empty tool_call parts: ${hasEmptyToolCall ? 'YES (BUG)' : 'No'}`
+  );
+  console.log(
+    `Reasoning order issues: ${hasReasoningOrderIssue ? 'YES (BUG)' : 'No'}`
+  );
+  console.log('\nFull contentParts dump:');
+  console.dir(_contentParts, { depth: null });
+}
+
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+  console.log('Content parts:');
+  console.dir(_contentParts, { depth: null });
+  process.exit(1);
+});
+
+process.on('uncaughtException', (err) => {
+  console.error('Uncaught Exception:', err);
+});
+
+testBedrockContentAggregation().catch((err) => {
+  console.error(err);
+  console.log('Content parts:');
+  console.dir(_contentParts, { depth: null });
+  process.exit(1);
+});

package/src/scripts/bedrock-parallel-tools-test.ts

@@ -0,0 +1,203 @@
+import { config } from 'dotenv';
+config();
+import { HumanMessage, BaseMessage } from '@langchain/core/messages';
+import type { UsageMetadata } from '@langchain/core/messages';
+import type { StandardGraph } from '@/graphs';
+import * as t from '@/types';
+import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
+import { GraphEvents, ContentTypes, Providers } from '@/common';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { Calculator } from '@/tools/Calculator';
+import { Run } from '@/run';
+
+const conversationHistory: BaseMessage[] = [];
+let _contentParts: t.MessageContentComplex[] = [];
+const collectedUsage: UsageMetadata[] = [];
+
+async function testParallelToolCalls(): Promise<void> {
+  const { contentParts, aggregateContent } = createContentAggregator();
+  _contentParts = contentParts as t.MessageContentComplex[];
+
+  const customHandlers = {
+    [GraphEvents.TOOL_END]: new ToolEndHandler(),
+    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
+    [GraphEvents.CHAT_MODEL_STREAM]: {
+      handle: async (
+        event: string,
+        data: t.StreamEventData,
+        metadata?: Record<string, unknown>,
+        graph?: unknown
+      ): Promise<void> => {
+        const chunk = data.chunk as Record<string, unknown> | undefined;
+        const tcc = chunk?.tool_call_chunks as
+          | Array<{ id?: string; name?: string; index?: number }>
+          | undefined;
+        if (tcc && tcc.length > 0) {
+          console.log(
+            `[CHAT_MODEL_STREAM] tool_call_chunks: ${JSON.stringify(tcc.map((c) => ({ id: c.id, name: c.name, index: c.index })))}`
+          );
+        }
+        const handler = new ChatModelStreamHandler();
+        return handler.handle(event, data, metadata, graph as StandardGraph);
+      },
+    },
+    [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_COMPLETED,
+        data: t.StreamEventData
+      ): void => {
+        const result = (data as unknown as { result: t.ToolEndEvent }).result;
+        console.log(
+          `[ON_RUN_STEP_COMPLETED] stepId=${result.id} index=${result.index} type=${result.type} tool=${result.tool_call?.name ?? 'n/a'}`
+        );
+        aggregateContent({
+          event,
+          data: data as unknown as { result: t.ToolEndEvent },
+        });
+      },
+    },
+    [GraphEvents.ON_RUN_STEP]: {
+      handle: (event: GraphEvents.ON_RUN_STEP, data: t.RunStep) => {
+        const toolCalls =
+          data.stepDetails.type === 'tool_calls' && data.stepDetails.tool_calls
+            ? (
+                data.stepDetails.tool_calls as Array<{
+                  name?: string;
+                  id?: string;
+                }>
+              )
+                .map((tc) => `${tc.name ?? '?'}(${tc.id ?? '?'})`)
+                .join(', ')
+            : 'none';
+        console.log(
+          `[ON_RUN_STEP] stepId=${data.id} index=${data.index} type=${data.type} stepIndex=${data.stepIndex} toolCalls=[${toolCalls}]`
+        );
+        aggregateContent({ event, data });
+      },
+    },
+    [GraphEvents.ON_RUN_STEP_DELTA]: {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_DELTA,
+        data: t.RunStepDeltaEvent
+      ) => {
+        aggregateContent({ event, data });
+      },
+    },
+    [GraphEvents.ON_MESSAGE_DELTA]: {
+      handle: (
+        event: GraphEvents.ON_MESSAGE_DELTA,
+        data: t.MessageDeltaEvent
+      ) => {
+        aggregateContent({ event, data });
+      },
+    },
+    [GraphEvents.ON_REASONING_DELTA]: {
+      handle: (
+        event: GraphEvents.ON_REASONING_DELTA,
+        data: t.ReasoningDeltaEvent
+      ) => {
+        aggregateContent({ event, data });
+      },
+    },
+  };
+
+  const baseLlmConfig = getLLMConfig(Providers.BEDROCK);
+
+  const llmConfig = {
+    ...baseLlmConfig,
+    model: 'global.anthropic.claude-opus-4-6-v1',
+    maxTokens: 16000,
+    additionalModelRequestFields: {
+      thinking: { type: 'enabled', budget_tokens: 10000 },
+    },
+  };
+
+  const run = await Run.create<t.IState>({
+    runId: 'bedrock-parallel-tools-test',
+    graphConfig: {
+      instructions:
+        'You are a math assistant. When asked to calculate multiple things, use the calculator tool for ALL of them in parallel. Do NOT chain calculations sequentially.',
+      type: 'standard',
+      tools: [new Calculator()],
+      llmConfig,
+    },
+    returnContent: true,
+    customHandlers: customHandlers as t.RunConfig['customHandlers'],
+  });
+
+  const streamConfig = {
+    configurable: { thread_id: 'bedrock-parallel-tools-thread' },
+    streamMode: 'values',
+    version: 'v2' as const,
+  };
+
+  const userMessage =
+    'Calculate these 3 things at the same time using the calculator: 1) 123 * 456, 2) sqrt(144) + 7, 3) 2^10 - 24';
+  conversationHistory.push(new HumanMessage(userMessage));
+
+  console.log('Running Bedrock parallel tool calls test...\n');
+  console.log(`Prompt: "${userMessage}"\n`);
+
+  const inputs = { messages: [...conversationHistory] };
+  await run.processStream(inputs, streamConfig);
+
+  console.log('\n\n========== ANALYSIS ==========\n');
+
+  let toolCallCount = 0;
+  const toolCallNames: string[] = [];
+  let hasUndefined = false;
+
+  for (let i = 0; i < _contentParts.length; i++) {
+    const part = _contentParts[i];
+    if (!part) {
+      hasUndefined = true;
+      console.log(`  [${i}] *** UNDEFINED ***`);
+      continue;
+    }
+    if (part.type === ContentTypes.TOOL_CALL) {
+      toolCallCount++;
+      const tc = (part as t.ToolCallContent).tool_call;
+      const hasData = tc && tc.name;
+      if (!hasData) {
+        console.log(`  [${i}] TOOL_CALL *** EMPTY ***`);
+      } else {
+        toolCallNames.push(tc.name ?? '');
+        console.log(
+          `  [${i}] TOOL_CALL name=${tc.name} id=${tc.id} output=${String(tc.output ?? '').substring(0, 40)}`
+        );
+      }
+    } else if (part.type === ContentTypes.THINK) {
+      const think = (part as t.ReasoningContentText).think ?? '';
+      console.log(`  [${i}] THINK (${think.length} chars)`);
+    } else if (part.type === ContentTypes.TEXT) {
+      const text = (part as t.MessageDeltaUpdate).text ?? '';
+      console.log(
+        `  [${i}] TEXT (${text.length} chars): "${text.substring(0, 80)}..."`
+      );
+    }
+  }
+
+  console.log('\n========== SUMMARY ==========\n');
+  console.log(`Total content parts: ${_contentParts.filter(Boolean).length}`);
+  console.log(`Tool calls found: ${toolCallCount}`);
+  console.log(`Tool call names: [${toolCallNames.join(', ')}]`);
+  console.log(`Undefined gaps: ${hasUndefined ? 'YES (BUG)' : 'No'}`);
+  console.log(
+    `Expected 3 tool calls: ${toolCallCount >= 3 ? 'PASS' : 'FAIL (only ' + toolCallCount + ')'}`
+  );
+  console.log('\nFull contentParts dump:');
+  console.dir(_contentParts, { depth: null });
+}
+
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+  console.dir(_contentParts, { depth: null });
+  process.exit(1);
+});
+
+testParallelToolCalls().catch((err) => {
+  console.error(err);
+  console.dir(_contentParts, { depth: null });
+  process.exit(1);
+});

package/src/scripts/tools.ts

@@ -25,16 +25,7 @@ async function testStandardStreaming(): Promise<void> {
         return true;
       }
     ),
-    [GraphEvents.CHAT_MODEL_END]: {
-      handle: (
-        _event: string,
-        _data: t.StreamEventData,
-        metadata?: Record<string, unknown>
-      ): void => {
-        console.log('\n====== CHAT_MODEL_END METADATA ======');
-        console.dir(metadata, { depth: null });
-      },
-    },
+    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
     [GraphEvents.CHAT_MODEL_START]: {
       handle: (
         _event: string,
@@ -158,9 +149,9 @@ async function testStandardStreaming(): Promise<void> {
     conversationHistory.push(...finalMessages);
     console.dir(conversationHistory, { depth: null });
   }
-  console.dir(finalContentParts, { depth: null });
+  // console.dir(finalContentParts, { depth: null });
   console.log('\n\n====================\n\n');
-  // console.dir(contentParts, { depth: null });
+  console.dir(contentParts, { depth: null });
 }
 
 process.on('unhandledRejection', (reason, promise) => {

package/src/stream.ts

@@ -411,6 +411,7 @@ hasToolCallChunks: ${hasToolCallChunks}
     ) {
       agentContext.currentTokenType = ContentTypes.TEXT;
       agentContext.tokenTypeSwitch = 'content';
+      agentContext.reasoningTransitionCount++;
     } else if (
       chunk.content != null &&
       typeof chunk.content === 'string' &&
@@ -465,7 +466,7 @@ export function createContentAggregator(): t.ContentAggregatorResult {
       return;
     }
 
-    if (!contentParts[index]) {
+    if (!contentParts[index] && partType !== ContentTypes.TOOL_CALL) {
       contentParts[index] = { type: partType };
     }