@librechat/agents 3.2.21 → 3.2.31

package/dist/cjs/session/handlers.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"handlers.cjs","sources":["../../../src/session/handlers.ts"],"sourcesContent":["import type { UsageMetadata } from '@langchain/core/messages';\nimport { GraphEvents } from '@/common';\nimport { ModelEndHandler, ToolEndHandler } from '@/events';\nimport { createContentAggregator } from '@/stream';\nimport type * as t from '@/types';\nimport type {\n  AgentSessionHandlersResult,\n  AgentSessionStreamEvent,\n  AgentSessionUsage,\n} from './types';\nimport { createTimestamp } from './ids';\nimport { toJsonValue } from './messageSerialization';\n\ntype CompletedRunStepResult =\n  | t.ToolEndEvent\n  | (t.SummaryCompleted & { id: string; index: number });\n\nfunction isToolCompletion(\n  result: CompletedRunStepResult\n): result is t.ToolEndEvent {\n  return 'tool_call' in result;\n}\n\nfunction createEventFactory(params: {\n  runId: string;\n  threadId: string;\n}): (\n  type: AgentSessionStreamEvent['type'],\n  data?: unknown\n) => AgentSessionStreamEvent {\n  let sequence = 0;\n  return (type, data) => ({\n    type,\n    sequence: sequence++,\n    runId: params.runId,\n    threadId: params.threadId,\n    timestamp: createTimestamp(),\n    ...(typeof data !== 'undefined' ? { data: toJsonValue(data) } : {}),\n  });\n}\n\nfunction getTokenCount(value: number | null | undefined): number {\n  return typeof value === 'number' && Number.isFinite(value) ? value : 0;\n}\n\nfunction updateUsage(usage: AgentSessionUsage, data: t.ModelEndData): void {\n  const metadata = data?.output?.usage_metadata as\n    | Partial<UsageMetadata>\n    | undefined;\n  if (!metadata) {\n    return;\n  }\n  const inputTokens = getTokenCount(metadata.input_tokens);\n  const outputTokens = getTokenCount(metadata.output_tokens);\n  const totalTokens =\n    metadata.total_tokens == null\n      ? inputTokens + outputTokens\n      : getTokenCount(metadata.total_tokens);\n  usage.inputTokens += inputTokens;\n  usage.outputTokens += outputTokens;\n  usage.totalTokens += totalTokens;\n}\n\nasync function callUserHandler(params: {\n  userHandlers?: Record<string, t.EventHandler>;\n  event: string;\n  data: Parameters<t.EventHandler['handle']>[1];\n  metadata?: Record<string, unknown>;\n  graph?: Parameters<t.EventHandler['handle']>[3];\n}): Promise<void> {\n  const handler = params.userHandlers?.[params.event];\n  if (!handler) {\n    return;\n  }\n  await handler.handle(\n    params.event,\n    params.data,\n    params.metadata,\n    params.graph\n  );\n}\n\nexport function createRunHandlers(params: {\n  runId: string;\n  threadId: string;\n  userHandlers?: Record<string, t.EventHandler>;\n  onEvent?: (event: AgentSessionStreamEvent) => void;\n}): AgentSessionHandlersResult {\n  const { contentParts, aggregateContent } = createContentAggregator();\n  const steps: t.RunStep[] = [];\n  const usage: AgentSessionUsage = {\n    inputTokens: 0,\n    outputTokens: 0,\n    totalTokens: 0,\n  };\n  const events: AgentSessionStreamEvent[] = [];\n  const createEvent = createEventFactory({\n    runId: params.runId,\n    threadId: params.threadId,\n  });\n  const emitEvent = (event: AgentSessionStreamEvent): void => {\n    events.push(event);\n    params.onEvent?.(event);\n  };\n  const toolEndHandler = new ToolEndHandler();\n  const modelEndHandler = new ModelEndHandler();\n\n  emitEvent(createEvent('run.started'));\n\n  const handlers: Record<string, t.EventHandler> = {\n    [GraphEvents.CHAT_MODEL_STREAM]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.CHAT_MODEL_END]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        await modelEndHandler.handle(\n          event,\n          data as t.ModelEndData,\n          metadata,\n          graph\n        );\n        updateUsage(usage, data as t.ModelEndData);\n        emitEvent(createEvent('usage.updated', usage));\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.TOOL_END]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        await toolEndHandler.handle(\n          event,\n          data as t.StreamEventData,\n          metadata,\n          graph\n        );\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_RUN_STEP]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const runStep = data as t.RunStep;\n        steps.push(runStep);\n        aggregateContent({ event: GraphEvents.ON_RUN_STEP, data: runStep });\n        if (runStep.stepDetails.type === 'tool_calls') {\n          emitEvent(createEvent('tool.started', runStep));\n        }\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_RUN_STEP_DELTA]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const delta = data as t.RunStepDeltaEvent;\n        aggregateContent({ event: GraphEvents.ON_RUN_STEP_DELTA, data: delta });\n        emitEvent(createEvent('tool.delta', delta));\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_RUN_STEP_COMPLETED]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const completed = data as unknown as { result: CompletedRunStepResult };\n        aggregateContent({\n          event: GraphEvents.ON_RUN_STEP_COMPLETED,\n          data: completed as { result: t.ToolEndEvent },\n        });\n        if (isToolCompletion(completed.result)) {\n          emitEvent(createEvent('tool.completed', completed));\n        }\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_MESSAGE_DELTA]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const delta = data as t.MessageDeltaEvent;\n        aggregateContent({ event: GraphEvents.ON_MESSAGE_DELTA, data: delta });\n        emitEvent(createEvent('message.delta', delta));\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_REASONING_DELTA]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const delta = data as t.ReasoningDeltaEvent;\n        aggregateContent({\n          event: GraphEvents.ON_REASONING_DELTA,\n          data: delta,\n        });\n        emitEvent(createEvent('reasoning.delta', delta));\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_SUMMARIZE_DELTA]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        aggregateContent({\n          event: GraphEvents.ON_SUMMARIZE_DELTA,\n          data: data as t.SummarizeDeltaData,\n        });\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_SUMMARIZE_COMPLETE]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        aggregateContent({\n          event: GraphEvents.ON_SUMMARIZE_COMPLETE,\n          data: data as t.SummarizeCompleteEvent,\n        });\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n  };\n\n  return { contentParts, steps, usage, events, handlers };\n}\n"],"names":["createTimestamp","toJsonValue","createContentAggregator","events","ToolEndHandler","ModelEndHandler","GraphEvents"],"mappings":";;;;;;;;AAiBA,SAAS,gBAAgB,CACvB,MAA8B,EAAA;IAE9B,OAAO,WAAW,IAAI,MAAM;AAC9B;AAEA,SAAS,kBAAkB,CAAC,MAG3B,EAAA;IAIC,IAAI,QAAQ,GAAG,CAAC;AAChB,IAAA,OAAO,CAAC,IAAI,EAAE,IAAI,MAAM;QACtB,IAAI;QACJ,QAAQ,EAAE,QAAQ,EAAE;QACpB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,SAAS,EAAEA,mBAAe,EAAE;QAC5B,IAAI,OAAO,IAAI,KAAK,WAAW,GAAG,EAAE,IAAI,EAAEC,gCAAW,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;AACpE,KAAA,CAAC;AACJ;AAEA,SAAS,aAAa,CAAC,KAAgC,EAAA;AACrD,IAAA,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,KAAK,GAAG,CAAC;AACxE;AAEA,SAAS,WAAW,CAAC,KAAwB,EAAE,IAAoB,EAAA;AACjE,IAAA,MAAM,QAAQ,GAAG,IAAI,EAAE,MAAM,EAAE,cAElB;IACb,IAAI,CAAC,QAAQ,EAAE;QACb;IACF;IACA,MAAM,WAAW,GAAG,aAAa,CAAC,QAAQ,CAAC,YAAY,CAAC;IACxD,MAAM,YAAY,GAAG,aAAa,CAAC,QAAQ,CAAC,aAAa,CAAC;AAC1D,IAAA,MAAM,WAAW,GACf,QAAQ,CAAC,YAAY,IAAI;UACrB,WAAW,GAAG;AAChB,UAAE,aAAa,CAAC,QAAQ,CAAC,YAAY,CAAC;AAC1C,IAAA,KAAK,CAAC,WAAW,IAAI,WAAW;AAChC,IAAA,KAAK,CAAC,YAAY,IAAI,YAAY;AAClC,IAAA,KAAK,CAAC,WAAW,IAAI,WAAW;AAClC;AAEA,eAAe,eAAe,CAAC,MAM9B,EAAA;IACC,MAAM,OAAO,GAAG,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC;IACnD,IAAI,CAAC,OAAO,EAAE;QACZ;IACF;IACA,MAAM,OAAO,CAAC,MAAM,CAClB,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,IAAI,EACX,MAAM,CAAC,QAAQ,EACf,MAAM,CAAC,KAAK,CACb;AACH;AAEM,SAAU,iBAAiB,CAAC,MAKjC,EAAA;IACC,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAGC,8BAAuB,EAAE;IACpE,MAAM,KAAK,GAAgB,EAAE;AAC7B,IAAA,MAAM,KAAK,GAAsB;AAC/B,QAAA,WAAW,EAAE,CAAC;AACd,QAAA,YAAY,EAAE,CAAC;AACf,QAAA,WAAW,EAAE,CAAC;KACf;IACD,MAAMC,QAAM,GAA8B,EAAE;IAC5C,MAAM,WAAW,GAAG,kBAAkB,CAAC;QACrC,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;AAC1B,KAAA,CAAC;AACF,IAAA,MAAM,SAAS,GAAG,CAAC,KAA8B,KAAU;AACzD,QAAAA,QAAM,CAAC,IAAI,CAAC,KAAK,CAAC;AAClB,QAAA,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;AACzB,IAAA,CAAC;AACD,IAAA,MAAM,cAAc,GAAG,IAAIC,qBAAc,EAAE;AAC3C,IAAA,MAAM,eAAe,GAAG,IAAIC,sBAAe,EAAE;AAE7C,IAAA,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;AAErC,IAAA,MAAM,QAAQ,GAAmC;AAC/C,QAAA,CAACC,iBAAW,CAAC,iBAAiB,GAAG;YAC/B,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,cAAc,GAAG;YAC5B,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,MAAM,eAAe,CAAC,MAAM,CAC1B,KAAK,EACL,IAAsB,EACtB,QAAQ,EACR,KAAK,CACN;AACD,gBAAA,WAAW,CAAC,KAAK,EAAE,IAAsB,CAAC;gBAC1C,SAAS,CAAC,WAAW,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;AAC9C,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,QAAQ,GAAG;YACtB,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,MAAM,cAAc,CAAC,MAAM,CACzB,KAAK,EACL,IAAyB,EACzB,QAAQ,EACR,KAAK,CACN;AACD,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,WAAW,GAAG;YACzB,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,OAAO,GAAG,IAAiB;AACjC,gBAAA,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;AACnB,gBAAA,gBAAgB,CAAC,EAAE,KAAK,EAAEA,iBAAW,CAAC,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBACnE,IAAI,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,YAAY,EAAE;oBAC7C,SAAS,CAAC,WAAW,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;gBACjD;AACA,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,iBAAiB,GAAG;YAC/B,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,KAAK,GAAG,IAA2B;AACzC,gBAAA,gBAAgB,CAAC,EAAE,KAAK,EAAEA,iBAAW,CAAC,iBAAiB,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBACvE,SAAS,CAAC,WAAW,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;AAC3C,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,qBAAqB,GAAG;YACnC,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,SAAS,GAAG,IAAqD;AACvE,gBAAA,gBAAgB,CAAC;oBACf,KAAK,EAAEA,iBAAW,CAAC,qBAAqB;AACxC,oBAAA,IAAI,EAAE,SAAuC;AAC9C,iBAAA,CAAC;AACF,gBAAA,IAAI,gBAAgB,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;oBACtC,SAAS,CAAC,WAAW,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;gBACrD;AACA,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,gBAAgB,GAAG;YAC9B,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,KAAK,GAAG,IAA2B;AACzC,gBAAA,gBAAgB,CAAC,EAAE,KAAK,EAAEA,iBAAW,CAAC,gBAAgB,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBACtE,SAAS,CAAC,WAAW,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;AAC9C,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,kBAAkB,GAAG;YAChC,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,KAAK,GAAG,IAA6B;AAC3C,gBAAA,gBAAgB,CAAC;oBACf,KAAK,EAAEA,iBAAW,CAAC,kBAAkB;AACrC,oBAAA,IAAI,EAAE,KAAK;AACZ,iBAAA,CAAC;gBACF,SAAS,CAAC,WAAW,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAC;AAChD,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,kBAAkB,GAAG;YAChC,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,gBAAgB,CAAC;oBACf,KAAK,EAAEA,iBAAW,CAAC,kBAAkB;AACrC,oBAAA,IAAI,EAAE,IAA4B;AACnC,iBAAA,CAAC;AACF,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,qBAAqB,GAAG;YACnC,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,gBAAgB,CAAC;oBACf,KAAK,EAAEA,iBAAW,CAAC,qBAAqB;AACxC,oBAAA,IAAI,EAAE,IAAgC;AACvC,iBAAA,CAAC;AACF,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;KACF;IAED,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,UAAEH,QAAM,EAAE,QAAQ,EAAE;AACzD;;;;"}
+{"version":3,"file":"handlers.cjs","sources":["../../../src/session/handlers.ts"],"sourcesContent":["import type { UsageMetadata } from '@langchain/core/messages';\nimport type {\n  AgentSessionHandlersResult,\n  AgentSessionStreamEvent,\n  AgentSessionUsage,\n} from './types';\nimport type * as t from '@/types';\nimport { ModelEndHandler, ToolEndHandler } from '@/events';\nimport { toJsonValue } from './messageSerialization';\nimport { createContentAggregator } from '@/stream';\nimport { createTimestamp } from './ids';\nimport { GraphEvents } from '@/common';\n\ntype CompletedRunStepResult =\n  | t.ToolEndEvent\n  | (t.SummaryCompleted & { id: string; index: number });\n\nfunction isToolCompletion(\n  result: CompletedRunStepResult\n): result is t.ToolEndEvent {\n  return 'tool_call' in result;\n}\n\nfunction createEventFactory(params: {\n  runId: string;\n  threadId: string;\n}): (\n  type: AgentSessionStreamEvent['type'],\n  data?: unknown\n) => AgentSessionStreamEvent {\n  let sequence = 0;\n  return (type, data) => ({\n    type,\n    sequence: sequence++,\n    runId: params.runId,\n    threadId: params.threadId,\n    timestamp: createTimestamp(),\n    ...(typeof data !== 'undefined' ? { data: toJsonValue(data) } : {}),\n  });\n}\n\nfunction getTokenCount(value: number | null | undefined): number {\n  return typeof value === 'number' && Number.isFinite(value) ? value : 0;\n}\n\nfunction updateUsage(usage: AgentSessionUsage, data: t.ModelEndData): void {\n  const metadata = data?.output?.usage_metadata as\n    | Partial<UsageMetadata>\n    | undefined;\n  if (!metadata) {\n    return;\n  }\n  const inputTokens = getTokenCount(metadata.input_tokens);\n  const outputTokens = getTokenCount(metadata.output_tokens);\n  const totalTokens =\n    metadata.total_tokens == null\n      ? inputTokens + outputTokens\n      : getTokenCount(metadata.total_tokens);\n  usage.inputTokens += inputTokens;\n  usage.outputTokens += outputTokens;\n  usage.totalTokens += totalTokens;\n}\n\nasync function callUserHandler(params: {\n  userHandlers?: Record<string, t.EventHandler>;\n  event: string;\n  data: Parameters<t.EventHandler['handle']>[1];\n  metadata?: Record<string, unknown>;\n  graph?: Parameters<t.EventHandler['handle']>[3];\n}): Promise<void> {\n  const handler = params.userHandlers?.[params.event];\n  if (!handler) {\n    return;\n  }\n  await handler.handle(\n    params.event,\n    params.data,\n    params.metadata,\n    params.graph\n  );\n}\n\nexport function createRunHandlers(params: {\n  runId: string;\n  threadId: string;\n  userHandlers?: Record<string, t.EventHandler>;\n  onEvent?: (event: AgentSessionStreamEvent) => void;\n}): AgentSessionHandlersResult {\n  const { contentParts, aggregateContent } = createContentAggregator();\n  const steps: t.RunStep[] = [];\n  const usage: AgentSessionUsage = {\n    inputTokens: 0,\n    outputTokens: 0,\n    totalTokens: 0,\n  };\n  const events: AgentSessionStreamEvent[] = [];\n  const createEvent = createEventFactory({\n    runId: params.runId,\n    threadId: params.threadId,\n  });\n  const emitEvent = (event: AgentSessionStreamEvent): void => {\n    events.push(event);\n    params.onEvent?.(event);\n  };\n  const toolEndHandler = new ToolEndHandler();\n  const modelEndHandler = new ModelEndHandler();\n\n  emitEvent(createEvent('run.started'));\n\n  const handlers: Record<string, t.EventHandler> = {\n    [GraphEvents.CHAT_MODEL_STREAM]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.CHAT_MODEL_END]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        await modelEndHandler.handle(\n          event,\n          data as t.ModelEndData,\n          metadata,\n          graph\n        );\n        updateUsage(usage, data as t.ModelEndData);\n        emitEvent(createEvent('usage.updated', usage));\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.TOOL_END]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        await toolEndHandler.handle(\n          event,\n          data as t.StreamEventData,\n          metadata,\n          graph\n        );\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_RUN_STEP]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const runStep = data as t.RunStep;\n        steps.push(runStep);\n        aggregateContent({ event: GraphEvents.ON_RUN_STEP, data: runStep });\n        if (runStep.stepDetails.type === 'tool_calls') {\n          emitEvent(createEvent('tool.started', runStep));\n        }\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_RUN_STEP_DELTA]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const delta = data as t.RunStepDeltaEvent;\n        aggregateContent({ event: GraphEvents.ON_RUN_STEP_DELTA, data: delta });\n        emitEvent(createEvent('tool.delta', delta));\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_RUN_STEP_COMPLETED]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const completed = data as unknown as { result: CompletedRunStepResult };\n        aggregateContent({\n          event: GraphEvents.ON_RUN_STEP_COMPLETED,\n          data: completed as { result: t.ToolEndEvent },\n        });\n        if (isToolCompletion(completed.result)) {\n          emitEvent(createEvent('tool.completed', completed));\n        }\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_MESSAGE_DELTA]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const delta = data as t.MessageDeltaEvent;\n        aggregateContent({ event: GraphEvents.ON_MESSAGE_DELTA, data: delta });\n        emitEvent(createEvent('message.delta', delta));\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_REASONING_DELTA]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        const delta = data as t.ReasoningDeltaEvent;\n        aggregateContent({\n          event: GraphEvents.ON_REASONING_DELTA,\n          data: delta,\n        });\n        emitEvent(createEvent('reasoning.delta', delta));\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_SUMMARIZE_DELTA]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        aggregateContent({\n          event: GraphEvents.ON_SUMMARIZE_DELTA,\n          data: data as t.SummarizeDeltaData,\n        });\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n    [GraphEvents.ON_SUMMARIZE_COMPLETE]: {\n      handle: async (event, data, metadata, graph): Promise<void> => {\n        aggregateContent({\n          event: GraphEvents.ON_SUMMARIZE_COMPLETE,\n          data: data as t.SummarizeCompleteEvent,\n        });\n        await callUserHandler({\n          userHandlers: params.userHandlers,\n          event,\n          data,\n          metadata,\n          graph,\n        });\n      },\n    },\n  };\n\n  return { contentParts, steps, usage, events, handlers };\n}\n"],"names":["createTimestamp","toJsonValue","createContentAggregator","events","ToolEndHandler","ModelEndHandler","GraphEvents"],"mappings":";;;;;;;;AAiBA,SAAS,gBAAgB,CACvB,MAA8B,EAAA;IAE9B,OAAO,WAAW,IAAI,MAAM;AAC9B;AAEA,SAAS,kBAAkB,CAAC,MAG3B,EAAA;IAIC,IAAI,QAAQ,GAAG,CAAC;AAChB,IAAA,OAAO,CAAC,IAAI,EAAE,IAAI,MAAM;QACtB,IAAI;QACJ,QAAQ,EAAE,QAAQ,EAAE;QACpB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,SAAS,EAAEA,mBAAe,EAAE;QAC5B,IAAI,OAAO,IAAI,KAAK,WAAW,GAAG,EAAE,IAAI,EAAEC,gCAAW,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;AACpE,KAAA,CAAC;AACJ;AAEA,SAAS,aAAa,CAAC,KAAgC,EAAA;AACrD,IAAA,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,KAAK,GAAG,CAAC;AACxE;AAEA,SAAS,WAAW,CAAC,KAAwB,EAAE,IAAoB,EAAA;AACjE,IAAA,MAAM,QAAQ,GAAG,IAAI,EAAE,MAAM,EAAE,cAElB;IACb,IAAI,CAAC,QAAQ,EAAE;QACb;IACF;IACA,MAAM,WAAW,GAAG,aAAa,CAAC,QAAQ,CAAC,YAAY,CAAC;IACxD,MAAM,YAAY,GAAG,aAAa,CAAC,QAAQ,CAAC,aAAa,CAAC;AAC1D,IAAA,MAAM,WAAW,GACf,QAAQ,CAAC,YAAY,IAAI;UACrB,WAAW,GAAG;AAChB,UAAE,aAAa,CAAC,QAAQ,CAAC,YAAY,CAAC;AAC1C,IAAA,KAAK,CAAC,WAAW,IAAI,WAAW;AAChC,IAAA,KAAK,CAAC,YAAY,IAAI,YAAY;AAClC,IAAA,KAAK,CAAC,WAAW,IAAI,WAAW;AAClC;AAEA,eAAe,eAAe,CAAC,MAM9B,EAAA;IACC,MAAM,OAAO,GAAG,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC;IACnD,IAAI,CAAC,OAAO,EAAE;QACZ;IACF;IACA,MAAM,OAAO,CAAC,MAAM,CAClB,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,IAAI,EACX,MAAM,CAAC,QAAQ,EACf,MAAM,CAAC,KAAK,CACb;AACH;AAEM,SAAU,iBAAiB,CAAC,MAKjC,EAAA;IACC,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAGC,8BAAuB,EAAE;IACpE,MAAM,KAAK,GAAgB,EAAE;AAC7B,IAAA,MAAM,KAAK,GAAsB;AAC/B,QAAA,WAAW,EAAE,CAAC;AACd,QAAA,YAAY,EAAE,CAAC;AACf,QAAA,WAAW,EAAE,CAAC;KACf;IACD,MAAMC,QAAM,GAA8B,EAAE;IAC5C,MAAM,WAAW,GAAG,kBAAkB,CAAC;QACrC,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;AAC1B,KAAA,CAAC;AACF,IAAA,MAAM,SAAS,GAAG,CAAC,KAA8B,KAAU;AACzD,QAAAA,QAAM,CAAC,IAAI,CAAC,KAAK,CAAC;AAClB,QAAA,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;AACzB,IAAA,CAAC;AACD,IAAA,MAAM,cAAc,GAAG,IAAIC,qBAAc,EAAE;AAC3C,IAAA,MAAM,eAAe,GAAG,IAAIC,sBAAe,EAAE;AAE7C,IAAA,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;AAErC,IAAA,MAAM,QAAQ,GAAmC;AAC/C,QAAA,CAACC,iBAAW,CAAC,iBAAiB,GAAG;YAC/B,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,cAAc,GAAG;YAC5B,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,MAAM,eAAe,CAAC,MAAM,CAC1B,KAAK,EACL,IAAsB,EACtB,QAAQ,EACR,KAAK,CACN;AACD,gBAAA,WAAW,CAAC,KAAK,EAAE,IAAsB,CAAC;gBAC1C,SAAS,CAAC,WAAW,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;AAC9C,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,QAAQ,GAAG;YACtB,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,MAAM,cAAc,CAAC,MAAM,CACzB,KAAK,EACL,IAAyB,EACzB,QAAQ,EACR,KAAK,CACN;AACD,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,WAAW,GAAG;YACzB,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,OAAO,GAAG,IAAiB;AACjC,gBAAA,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;AACnB,gBAAA,gBAAgB,CAAC,EAAE,KAAK,EAAEA,iBAAW,CAAC,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBACnE,IAAI,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,YAAY,EAAE;oBAC7C,SAAS,CAAC,WAAW,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;gBACjD;AACA,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,iBAAiB,GAAG;YAC/B,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,KAAK,GAAG,IAA2B;AACzC,gBAAA,gBAAgB,CAAC,EAAE,KAAK,EAAEA,iBAAW,CAAC,iBAAiB,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBACvE,SAAS,CAAC,WAAW,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;AAC3C,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,qBAAqB,GAAG;YACnC,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,SAAS,GAAG,IAAqD;AACvE,gBAAA,gBAAgB,CAAC;oBACf,KAAK,EAAEA,iBAAW,CAAC,qBAAqB;AACxC,oBAAA,IAAI,EAAE,SAAuC;AAC9C,iBAAA,CAAC;AACF,gBAAA,IAAI,gBAAgB,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;oBACtC,SAAS,CAAC,WAAW,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;gBACrD;AACA,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,gBAAgB,GAAG;YAC9B,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,KAAK,GAAG,IAA2B;AACzC,gBAAA,gBAAgB,CAAC,EAAE,KAAK,EAAEA,iBAAW,CAAC,gBAAgB,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBACtE,SAAS,CAAC,WAAW,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;AAC9C,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,kBAAkB,GAAG;YAChC,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;gBAC5D,MAAM,KAAK,GAAG,IAA6B;AAC3C,gBAAA,gBAAgB,CAAC;oBACf,KAAK,EAAEA,iBAAW,CAAC,kBAAkB;AACrC,oBAAA,IAAI,EAAE,KAAK;AACZ,iBAAA,CAAC;gBACF,SAAS,CAAC,WAAW,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAC;AAChD,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,kBAAkB,GAAG;YAChC,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,gBAAgB,CAAC;oBACf,KAAK,EAAEA,iBAAW,CAAC,kBAAkB;AACrC,oBAAA,IAAI,EAAE,IAA4B;AACnC,iBAAA,CAAC;AACF,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;AACD,QAAA,CAACA,iBAAW,CAAC,qBAAqB,GAAG;YACnC,MAAM,EAAE,OAAO,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,KAAmB;AAC5D,gBAAA,gBAAgB,CAAC;oBACf,KAAK,EAAEA,iBAAW,CAAC,qBAAqB;AACxC,oBAAA,IAAI,EAAE,IAAgC;AACvC,iBAAA,CAAC;AACF,gBAAA,MAAM,eAAe,CAAC;oBACpB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK;oBACL,IAAI;oBACJ,QAAQ;oBACR,KAAK;AACN,iBAAA,CAAC;YACJ,CAAC;AACF,SAAA;KACF;IAED,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,UAAEH,QAAM,EAAE,QAAQ,EAAE;AACzD;;;;"}

package/dist/cjs/stream.cjs

@@ -1,16 +1,18 @@
 'use strict';
 
 var _enum = require('./common/enum.cjs');
+var streamedToolCallSeals = require('./tools/streamedToolCallSeals.cjs');
+var eagerEventExecution = require('./tools/eagerEventExecution.cjs');
 var handlers = require('./tools/handlers.cjs');
+var truncation = require('./utils/truncation.cjs');
+var toolOutputReferences = require('./tools/toolOutputReferences.cjs');
+var events = require('./utils/events.cjs');
+var llm = require('./utils/llm.cjs');
 require('./messages/core.cjs');
 var ids = require('./messages/ids.cjs');
 require('@langchain/core/messages');
-var truncation = require('./utils/truncation.cjs');
-var events = require('./utils/events.cjs');
+require('node:crypto');
 require('uuid');
-var eagerEventExecution = require('./tools/eagerEventExecution.cjs');
-var streamedToolCallSeals = require('./tools/streamedToolCallSeals.cjs');
-var toolOutputReferences = require('./tools/toolOutputReferences.cjs');
 
 const LOCAL_CODING_BUNDLE_NAME_SET = new Set(_enum.LOCAL_CODING_BUNDLE_NAMES);
 /**
@@ -206,6 +208,120 @@ function hasDirectToolCallChunkStateInStep(args) {
     }
     return false;
 }
+function isGoogleServerSideToolContentPart(contentPart) {
+    return contentPart.type === 'toolCall' || contentPart.type === 'toolResponse';
+}
+function isTextContentPart(contentPart) {
+    return contentPart.type?.startsWith(_enum.ContentTypes.TEXT) ?? false;
+}
+function isReasoningContentPart(contentPart) {
+    return ((contentPart.type?.startsWith(_enum.ContentTypes.THINKING) ?? false) ||
+        (contentPart.type?.startsWith(_enum.ContentTypes.REASONING) ?? false) ||
+        (contentPart.type?.startsWith(_enum.ContentTypes.REASONING_CONTENT) ?? false) ||
+        contentPart.type === 'redacted_thinking');
+}
+function getReasoningTextFromContentPart(contentPart) {
+    return (contentPart.thinking ??
+        contentPart.reasoning ??
+        contentPart.reasoningText
+            ?.text ??
+        '');
+}
+function getReasoningTextFromChunk(chunk, agentContext) {
+    const reasoning = chunk.additional_kwargs?.[agentContext.reasoningKey];
+    if (typeof reasoning === 'string') {
+        return reasoning;
+    }
+    return reasoning?.summary?.[0]?.text ?? '';
+}
+const googleServerSideToolStepIdsByGraph = new WeakMap();
+function markGoogleServerSideToolMessageStep(graph, stepId) {
+    const stepIds = googleServerSideToolStepIdsByGraph.get(graph) ?? new Set();
+    stepIds.add(stepId);
+    googleServerSideToolStepIdsByGraph.set(graph, stepIds);
+}
+function isGoogleServerSideToolMessageStep(graph, stepId) {
+    return googleServerSideToolStepIdsByGraph.get(graph)?.has(stepId) === true;
+}
+function shouldStartFreshMessageStepAfterGoogleServerSideTool({ graph, stepId, runStep, content, }) {
+    if (runStep?.type !== _enum.StepTypes.MESSAGE_CREATION ||
+        !isGoogleServerSideToolMessageStep(graph, stepId)) {
+        return false;
+    }
+    if (typeof content === 'string') {
+        return true;
+    }
+    return (content.every((c) => isTextContentPart(c)) ||
+        content.every((c) => isReasoningContentPart(c)));
+}
+async function dispatchMessageCreationStep({ graph, stepKey, metadata, }) {
+    const messageId = ids.getMessageId(stepKey, graph, true) ?? '';
+    return graph.dispatchRunStep(stepKey, {
+        type: _enum.StepTypes.MESSAGE_CREATION,
+        message_creation: {
+            message_id: messageId,
+        },
+    }, metadata);
+}
+async function dispatchMessageContentParts({ graph, stepKey, content, metadata, }) {
+    for (const contentPart of content) {
+        const currentStepId = await dispatchMessageCreationStep({
+            graph,
+            stepKey,
+            metadata,
+        });
+        if (isGoogleServerSideToolContentPart(contentPart)) {
+            markGoogleServerSideToolMessageStep(graph, currentStepId);
+        }
+        await graph.dispatchMessageDelta(currentStepId, {
+            content: [contentPart],
+        }, metadata);
+    }
+}
+async function dispatchReasoningContentParts({ graph, stepKey, content, metadata, }) {
+    if (content.length === 0) {
+        return;
+    }
+    const currentStepId = await dispatchMessageCreationStep({
+        graph,
+        stepKey,
+        metadata,
+    });
+    await graph.dispatchReasoningDelta(currentStepId, {
+        content,
+    }, metadata);
+}
+async function dispatchGoogleServerSideToolStreamContent({ graph, stepKey, chunk, agentContext, content, metadata, }) {
+    const reasoningContent = [];
+    const reasoningText = getReasoningTextFromChunk(chunk, agentContext);
+    if (reasoningText !== '') {
+        reasoningContent.push({
+            type: _enum.ContentTypes.THINK,
+            think: reasoningText,
+        });
+    }
+    reasoningContent.push(...content
+        .filter((contentPart) => isReasoningContentPart(contentPart))
+        .map((contentPart) => ({
+        type: _enum.ContentTypes.THINK,
+        think: getReasoningTextFromContentPart(contentPart),
+    }))
+        .filter((contentPart) => contentPart.think !== ''));
+    await dispatchReasoningContentParts({
+        graph,
+        stepKey,
+        content: reasoningContent,
+        metadata,
+    });
+    const messageContent = content.filter((contentPart) => isTextContentPart(contentPart) ||
+        isGoogleServerSideToolContentPart(contentPart));
+    await dispatchMessageContentParts({
+        graph,
+        stepKey,
+        content: messageContent,
+        metadata,
+    });
+}
 function createEagerToolExecutionPlan(args) {
     const { graph, metadata, agentContext, toolCalls, skipExisting = false, } = args;
     if (!isEagerToolExecutionEnabledForBatch({
@@ -595,21 +711,16 @@ function startReadyStreamedEagerToolExecutions(args) {
     });
 }
 function getChunkContent({ chunk, provider, reasoningKey, }) {
+    if (llm.isGoogleLike(provider) &&
+        Array.isArray(chunk?.content) &&
+        chunk.content.some((c) => isGoogleServerSideToolContentPart(c))) {
+        return chunk.content;
+    }
     if ((provider === _enum.Providers.OPENAI || provider === _enum.Providers.AZURE) &&
         chunk?.additional_kwargs?.reasoning?.summary?.[0]?.text != null &&
         (chunk?.additional_kwargs?.reasoning?.summary?.[0]?.text?.length ?? 0) > 0) {
         return chunk?.additional_kwargs?.reasoning?.summary?.[0]?.text;
     }
-    /**
-     * For OpenRouter, reasoning is stored in additional_kwargs.reasoning (not reasoning_content).
-     * NOTE: We intentionally do NOT extract text from reasoning_details here.
-     * The reasoning_details array contains the FULL accumulated reasoning text (set only on final chunk),
-     * but individual reasoning tokens are already streamed via additional_kwargs.reasoning.
-     * Extracting from reasoning_details would cause duplication.
-     * The reasoning_details is only used for:
-     * 1. Detecting reasoning mode in handleReasoning()
-     * 2. Final message storage (for thought signatures)
-     */
     if (provider === _enum.Providers.OPENROUTER) {
         // Content presence signals end of reasoning phase - prefer content over reasoning
         // This handles transitional chunks that may have both reasoning and content
@@ -620,10 +731,84 @@ function getChunkContent({ chunk, provider, reasoningKey, }) {
         if (reasoning != null && reasoning !== '') {
             return reasoning;
         }
+        const reasoningContent = chunk?.additional_kwargs?.reasoning_content;
+        if (reasoningContent != null && reasoningContent !== '') {
+            return reasoningContent;
+        }
         return chunk?.content;
     }
-    return ((chunk?.additional_kwargs?.[reasoningKey] ?? '') ||
-        chunk?.content);
+    const keyedReasoning = chunk?.additional_kwargs?.[reasoningKey];
+    if (typeof chunk?.content === 'string' &&
+        chunk.content !== '' &&
+        keyedReasoning != null &&
+        keyedReasoning !== '') {
+        return chunk.content;
+    }
+    return (keyedReasoning ?? '') || chunk?.content;
+}
+function isDisableStreamingEnabled(clientOptions) {
+    return (clientOptions != null &&
+        'disableStreaming' in clientOptions &&
+        clientOptions.disableStreaming === true);
+}
+function hasReasoningContent(value) {
+    if (typeof value === 'string') {
+        return value !== '';
+    }
+    if (Array.isArray(value)) {
+        return value.length > 0;
+    }
+    if (value == null) {
+        return false;
+    }
+    return (value.summary?.some((summary) => summary.text != null && summary.text.length > 0) === true);
+}
+function shouldDeferMixedFinalReasoningChunk({ chunk, agentContext, }) {
+    if ((chunk.tool_calls?.length ?? 0) > 0 ||
+        (chunk.tool_call_chunks?.length ?? 0) > 0 ||
+        typeof chunk.content !== 'string' ||
+        chunk.content === '') {
+        return false;
+    }
+    const additionalKwargs = chunk.additional_kwargs;
+    if (agentContext.provider === _enum.Providers.OPENROUTER &&
+        hasReasoningContent(additionalKwargs?.reasoning_details)) {
+        return true;
+    }
+    if (!isDisableStreamingEnabled(agentContext.clientOptions)) {
+        return false;
+    }
+    return (hasReasoningContent(additionalKwargs?.[agentContext.reasoningKey]) ||
+        hasReasoningContent(additionalKwargs?.reasoning_content) ||
+        hasReasoningContent(additionalKwargs?.reasoning) ||
+        hasReasoningContent(additionalKwargs?.reasoning_details));
+}
+function hasCurrentTextDeltaStep({ graph, metadata, }) {
+    if (metadata == null) {
+        return false;
+    }
+    const baseStepKey = graph.getStepBaseKey(metadata);
+    for (const [stepKey, stepIds] of graph.stepKeyIds) {
+        if (stepKey !== baseStepKey && !stepKey.startsWith(`${baseStepKey}_`)) {
+            continue;
+        }
+        if (stepIds.some((stepId) => graph.messageStepHasTextDeltas.has(stepId))) {
+            return true;
+        }
+    }
+    return false;
+}
+function shouldSkipLateOpenRouterReasoningChunk({ chunk, agentContext, graph, metadata, }) {
+    if (agentContext.provider !== _enum.Providers.OPENROUTER ||
+        (chunk.tool_calls?.length ?? 0) > 0 ||
+        (chunk.tool_call_chunks?.length ?? 0) > 0 ||
+        (chunk.content != null && chunk.content !== '')) {
+        return false;
+    }
+    return ((hasReasoningContent(chunk.additional_kwargs?.reasoning) ||
+        hasReasoningContent(chunk.additional_kwargs?.reasoning_content) ||
+        hasReasoningContent(chunk.additional_kwargs?.reasoning_details)) &&
+        hasCurrentTextDeltaStep({ graph, metadata }));
 }
 class ChatModelStreamHandler {
     async handle(event, data, metadata, graph) {
@@ -653,10 +838,34 @@ class ChatModelStreamHandler {
         if (skipHandling) {
             return;
         }
+        if (shouldDeferMixedFinalReasoningChunk({ chunk, agentContext })) {
+            return;
+        }
+        if (shouldSkipLateOpenRouterReasoningChunk({
+            chunk,
+            agentContext,
+            graph,
+            metadata,
+        })) {
+            return;
+        }
         this.handleReasoning(chunk, agentContext);
         const stepKey = graph.getStepKey(metadata);
         let hasToolCalls = false;
         const hasToolCallChunks = (chunk.tool_call_chunks && chunk.tool_call_chunks.length > 0) ?? false;
+        const hasGoogleServerSideToolContent = llm.isGoogleLike(agentContext.provider) &&
+            Array.isArray(content) &&
+            content.some((c) => isGoogleServerSideToolContentPart(c));
+        if (hasGoogleServerSideToolContent && Array.isArray(content)) {
+            await dispatchGoogleServerSideToolStreamContent({
+                graph,
+                stepKey,
+                chunk,
+                agentContext,
+                content,
+                metadata,
+            });
+        }
         if (chunk.tool_calls &&
             chunk.tool_calls.length > 0 &&
             chunk.tool_calls.every((tc) => tc.id != null &&
@@ -729,6 +938,9 @@ class ChatModelStreamHandler {
         if (isEmptyContent) {
             return;
         }
+        if (hasGoogleServerSideToolContent) {
+            return;
+        }
         const message_id = ids.getMessageId(stepKey, graph) ?? '';
         if (message_id) {
             await graph.dispatchRunStep(stepKey, {
@@ -738,8 +950,17 @@ class ChatModelStreamHandler {
                 },
             }, metadata);
         }
-        const stepId = graph.getStepIdByKey(stepKey);
-        const runStep = graph.getRunStep(stepId);
+        let stepId = graph.getStepIdByKey(stepKey);
+        let runStep = graph.getRunStep(stepId);
+        if (shouldStartFreshMessageStepAfterGoogleServerSideTool({
+            graph,
+            stepId,
+            runStep,
+            content,
+        })) {
+            stepId = await dispatchMessageCreationStep({ graph, stepKey, metadata });
+            runStep = graph.getRunStep(stepId);
+        }
         if (!runStep) {
             console.warn(`\n
 ==============================================================
@@ -822,15 +1043,12 @@ hasToolCallChunks: ${hasToolCallChunks}
                 }, metadata);
             }
         }
-        else if (content.every((c) => c.type?.startsWith(_enum.ContentTypes.TEXT) ?? false)) {
+        else if (content.every((c) => isTextContentPart(c))) {
             await graph.dispatchMessageDelta(stepId, {
                 content,
             }, metadata);
         }
-        else if (content.every((c) => (c.type?.startsWith(_enum.ContentTypes.THINKING) ?? false) ||
-            (c.type?.startsWith(_enum.ContentTypes.REASONING) ?? false) ||
-            (c.type?.startsWith(_enum.ContentTypes.REASONING_CONTENT) ?? false) ||
-            c.type === 'redacted_thinking')) {
+        else if (content.every((c) => isReasoningContentPart(c))) {
             await graph.dispatchReasoningDelta(stepId, {
                 content: content.map((c) => ({
                     type: _enum.ContentTypes.THINK,
@@ -868,7 +1086,9 @@ hasToolCallChunks: ${hasToolCallChunks}
                 Array.isArray(chunk.additional_kwargs.reasoning_details) &&
                 chunk.additional_kwargs.reasoning_details.length > 0) ||
                 (typeof chunk.additional_kwargs?.reasoning === 'string' &&
-                    chunk.additional_kwargs.reasoning !== ''))) {
+                    chunk.additional_kwargs.reasoning !== '') ||
+                (typeof chunk.additional_kwargs?.reasoning_content === 'string' &&
+                    chunk.additional_kwargs.reasoning_content !== ''))) {
             reasoning_content = 'valid';
         }
         if (reasoning_content != null &&
@@ -975,6 +1195,9 @@ function createContentAggregator() {
             };
             contentParts[index] = update;
         }
+        else if (partType === 'toolCall' || partType === 'toolResponse') {
+            contentParts[index] = contentPart;
+        }
         else if (partType === _enum.ContentTypes.SUMMARY) {
             const currentSummary = contentParts[index];
             const incoming = contentPart;