@librechat/agents 3.3.4 → 3.3.5

package/dist/cjs/common/constants.cjs

@@ -12,8 +12,29 @@ const ANTHROPIC_TOOL_TOKEN_MULTIPLIER = 2.6;
 * Empirically calibrated at ~1.4× the raw JSON token count.
 */
 const DEFAULT_TOOL_TOKEN_MULTIPLIER = 1.4;
+/**
+* Default ceiling on cooperative stream seals per run. Each seal costs one
+* extra superstep, so this also sizes the recursion-limit headroom a
+* preemption-enabled run reserves.
+*/
+const DEFAULT_MAX_SEALS = 8;
+/**
+* Per-hook timeout for `PreemptBoundary`, deliberately far above
+* `DEFAULT_HOOK_TIMEOUT_MS`. A host drain has already popped its queue and
+* persisted content by the time this fires, so a timeout does not cancel the
+* work — it only loses the messages the run was about to inject.
+*/
+const PREEMPT_BOUNDARY_HOOK_TIMEOUT_MS = 12e4;
+/**
+* Default LangGraph recursion limit for a run. Callers may override it via
+* the stream config; a preemption-enabled run adds its seal budget on top.
+*/
+const DEFAULT_RECURSION_LIMIT = 50;
 //#endregion
 exports.ANTHROPIC_TOOL_TOKEN_MULTIPLIER = ANTHROPIC_TOOL_TOKEN_MULTIPLIER;
+exports.DEFAULT_MAX_SEALS = DEFAULT_MAX_SEALS;
+exports.DEFAULT_RECURSION_LIMIT = DEFAULT_RECURSION_LIMIT;
 exports.DEFAULT_TOOL_TOKEN_MULTIPLIER = DEFAULT_TOOL_TOKEN_MULTIPLIER;
+exports.PREEMPT_BOUNDARY_HOOK_TIMEOUT_MS = PREEMPT_BOUNDARY_HOOK_TIMEOUT_MS;
 
 //# sourceMappingURL=constants.cjs.map

package/dist/cjs/common/constants.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"constants.cjs","names":[],"sources":["../../../src/common/constants.ts"],"sourcesContent":["/**\n * Anthropic direct API tool schema overhead multiplier.\n * Empirically calibrated against real MCP tool sets (29 tools).\n * Accounts for Anthropic's internal XML-like tool encoding plus\n * a ~300-token hidden tool-system preamble.\n */\nexport const ANTHROPIC_TOOL_TOKEN_MULTIPLIER = 2.6;\n\n/**\n * Default tool schema overhead multiplier for all non-Anthropic providers.\n * Covers OpenAI function-calling format, Bedrock, and other providers.\n * Empirically calibrated at ~1.4× the raw JSON token count.\n */\nexport const DEFAULT_TOOL_TOKEN_MULTIPLIER = 1.4;\n"],"mappings":";;;;;;;AAMA,MAAa,kCAAkC;;;;;;AAO/C,MAAa,gCAAgC"}
+{"version":3,"file":"constants.cjs","names":[],"sources":["../../../src/common/constants.ts"],"sourcesContent":["/**\n * Anthropic direct API tool schema overhead multiplier.\n * Empirically calibrated against real MCP tool sets (29 tools).\n * Accounts for Anthropic's internal XML-like tool encoding plus\n * a ~300-token hidden tool-system preamble.\n */\nexport const ANTHROPIC_TOOL_TOKEN_MULTIPLIER = 2.6;\n\n/**\n * Default tool schema overhead multiplier for all non-Anthropic providers.\n * Covers OpenAI function-calling format, Bedrock, and other providers.\n * Empirically calibrated at ~1.4× the raw JSON token count.\n */\nexport const DEFAULT_TOOL_TOKEN_MULTIPLIER = 1.4;\n\n/**\n * Default ceiling on cooperative stream seals per run. Each seal costs one\n * extra superstep, so this also sizes the recursion-limit headroom a\n * preemption-enabled run reserves.\n */\nexport const DEFAULT_MAX_SEALS = 8;\n\n/**\n * Per-hook timeout for `PreemptBoundary`, deliberately far above\n * `DEFAULT_HOOK_TIMEOUT_MS`. A host drain has already popped its queue and\n * persisted content by the time this fires, so a timeout does not cancel the\n * work — it only loses the messages the run was about to inject.\n */\nexport const PREEMPT_BOUNDARY_HOOK_TIMEOUT_MS = 120_000;\n\n/**\n * Default LangGraph recursion limit for a run. Callers may override it via\n * the stream config; a preemption-enabled run adds its seal budget on top.\n */\nexport const DEFAULT_RECURSION_LIMIT = 50;\n"],"mappings":";;;;;;;AAMA,MAAa,kCAAkC;;;;;;AAO/C,MAAa,gCAAgC;;;;;;AAO7C,MAAa,oBAAoB;;;;;;;AAQjC,MAAa,mCAAmC;;;;;AAMhD,MAAa,0BAA0B"}

package/dist/cjs/events.cjs

@@ -1,5 +1,6 @@
 require("./common/enum.cjs");
 require("./common/index.cjs");
+const require_stream = require("./stream.cjs");
 //#region src/events.ts
 var HandlerRegistry = class {
 	handlers = /* @__PURE__ */ new Map();
@@ -20,10 +21,18 @@ function composeEventHandlers(...handlerSets) {
 				composed[eventType] = handler;
 				continue;
 			}
-			composed[eventType] = { handle: async (...args) => {
+			const wrapper = { handle: async (...args) => {
 				await previous.handle(...args);
 				await handler.handle(...args);
 			} };
+			/**
+			* Carry the stream-dispatch brand across the wrapper. Without this a
+			* composed handler fails every capability check made about it — notably
+			* the one deciding whether a run may seal — even though it still drives
+			* the SDK's content-part dispatch.
+			*/
+			if (require_stream.dispatchesChatModelStream(previous) || require_stream.dispatchesChatModelStream(handler)) Object.defineProperty(wrapper, require_stream.SDK_STREAM_DISPATCH, { value: true });
+			composed[eventType] = wrapper;
 		}
 	}
 	return composed;

package/dist/cjs/events.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"events.cjs","names":[],"sources":["../../src/events.ts"],"sourcesContent":["/* eslint-disable no-console */\n// src/events.ts\nimport type {\n  BaseMessageFields,\n  UsageMetadata,\n} from '@langchain/core/messages';\nimport type { Logger } from 'winston';\nimport type { MultiAgentGraph, StandardGraph } from '@/graphs';\nimport type * as t from '@/types';\nimport { Constants } from '@/common';\n\nexport class HandlerRegistry {\n  private handlers: Map<string, t.EventHandler> = new Map();\n\n  register(eventType: string, handler: t.EventHandler): void {\n    this.handlers.set(eventType, handler);\n  }\n\n  getHandler(eventType: string): t.EventHandler | undefined {\n    return this.handlers.get(eventType);\n  }\n}\n\nexport function composeEventHandlers(\n  ...handlerSets: Array<Record<string, t.EventHandler> | undefined>\n): Record<string, t.EventHandler> {\n  const composed: Partial<Record<string, t.EventHandler>> = {};\n\n  for (const handlerSet of handlerSets) {\n    if (!handlerSet) {\n      continue;\n    }\n    for (const [eventType, handler] of Object.entries(handlerSet)) {\n      const previous = composed[eventType];\n      if (previous === undefined) {\n        composed[eventType] = handler;\n        continue;\n      }\n      composed[eventType] = {\n        handle: async (\n          ...args: Parameters<t.EventHandler['handle']>\n        ): Promise<void> => {\n          await previous.handle(...args);\n          await handler.handle(...args);\n        },\n      };\n    }\n  }\n\n  return composed as Record<string, t.EventHandler>;\n}\n\nexport class ModelEndHandler implements t.EventHandler {\n  collectedUsage?: UsageMetadata[];\n  constructor(collectedUsage?: UsageMetadata[]) {\n    if (collectedUsage && !Array.isArray(collectedUsage)) {\n      throw new Error('collectedUsage must be an array');\n    }\n    this.collectedUsage = collectedUsage;\n  }\n\n  async handle(\n    event: string,\n    data: t.ModelEndData,\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    const usage = data?.output?.usage_metadata;\n    if (usage != null && this.collectedUsage != null) {\n      this.collectedUsage.push(usage);\n    }\n  }\n}\n\nexport class ToolEndHandler implements t.EventHandler {\n  private callback?: t.ToolEndCallback;\n  private logger?: Logger;\n  constructor(callback?: t.ToolEndCallback, logger?: Logger) {\n    this.callback = callback;\n    this.logger = logger;\n  }\n\n  /**\n   * Handles on_tool_end events from the for-await stream consumer.\n   *\n   * This handler is now purely a consumer callback — tool completion\n   * (ON_RUN_STEP_COMPLETED dispatch + session context storage) is handled\n   * in graph context by ToolNode directly, eliminating the race between\n   * the stream consumer and graph execution.\n   */\n  async handle(\n    event: string,\n    data: t.StreamEventData | undefined,\n    metadata?: Record<string, unknown>,\n    graph?: StandardGraph | MultiAgentGraph\n  ): Promise<void> {\n    try {\n      if (!graph || !metadata) {\n        if (this.logger) {\n          this.logger.warn(`Graph or metadata not found in ${event} event`);\n        } else {\n          console.warn(`Graph or metadata not found in ${event} event`);\n        }\n        return;\n      }\n\n      const toolEndData = data as t.ToolEndData | undefined;\n      if (!toolEndData?.output) {\n        if (this.logger) {\n          this.logger.warn('No output found in tool_end event');\n        } else {\n          console.warn('No output found in tool_end event');\n        }\n        return;\n      }\n\n      if (\n        metadata[Constants.PROGRAMMATIC_TOOL_CALLING] === true ||\n        metadata[Constants.BASH_PROGRAMMATIC_TOOL_CALLING] === true\n      ) {\n        return;\n      }\n\n      if (this.callback) {\n        await this.callback(toolEndData, metadata);\n      }\n    } catch (error) {\n      if (this.logger) {\n        this.logger.error('Error handling tool_end event:', error);\n      } else {\n        console.error('Error handling tool_end event:', error);\n      }\n    }\n  }\n}\n\nexport class TestLLMStreamHandler implements t.EventHandler {\n  handle(event: string, data: t.StreamEventData | undefined): void {\n    const chunk = data?.chunk;\n    const isMessageChunk = !!(chunk && 'message' in chunk);\n    const msg = isMessageChunk ? chunk.message : undefined;\n    if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {\n      console.log(msg.tool_call_chunks);\n    } else if (msg && typeof msg.content === 'string') {\n      process.stdout.write(msg.content);\n    }\n  }\n}\n\nexport class TestChatStreamHandler implements t.EventHandler {\n  handle(event: string, data: t.StreamEventData | undefined): void {\n    const chunk = data?.chunk;\n    const isContentChunk = !!(chunk && 'content' in chunk);\n    if (!isContentChunk) {\n      return;\n    }\n\n    const content = chunk.content;\n\n    if (chunk.tool_call_chunks && chunk.tool_call_chunks.length > 0) {\n      console.dir(chunk.tool_call_chunks, { depth: null });\n    }\n\n    if (typeof content === 'string') {\n      process.stdout.write(content);\n    } else {\n      console.dir(content, { depth: null });\n    }\n  }\n}\n\nexport class LLMStreamHandler implements t.EventHandler {\n  handle(\n    event: string,\n    data: t.StreamEventData | undefined,\n    metadata?: Record<string, unknown>\n  ): void {\n    const chunk = data?.chunk;\n    const isMessageChunk = !!(chunk && 'message' in chunk);\n    const msg = isMessageChunk ? chunk.message : undefined;\n    if (metadata) {\n      console.log(metadata);\n    }\n    if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {\n      console.log(msg.tool_call_chunks);\n    } else if (msg && typeof msg.content === 'string') {\n      process.stdout.write(msg.content);\n    }\n  }\n}\n\nexport const createMetadataAggregator = (\n  _collected?: Record<\n    string,\n    NonNullable<BaseMessageFields['response_metadata']>\n  >[]\n): t.MetadataAggregatorResult => {\n  const collected = _collected || [];\n\n  const handleLLMEnd: t.HandleLLMEnd = (output) => {\n    const { generations } = output;\n    const lastMessageOutput = (\n      generations[generations.length - 1] as\n        | (t.StreamGeneration | undefined)[]\n        | undefined\n    )?.[0];\n    if (!lastMessageOutput) {\n      return;\n    }\n    const { message } = lastMessageOutput;\n    if (message?.response_metadata) {\n      collected.push(message.response_metadata);\n    }\n  };\n\n  return { handleLLMEnd, collected };\n};\n"],"mappings":";;;AAWA,IAAa,kBAAb,MAA6B;CAC3B,2BAAgD,IAAI,IAAI;CAExD,SAAS,WAAmB,SAA+B;EACzD,KAAK,SAAS,IAAI,WAAW,OAAO;CACtC;CAEA,WAAW,WAA+C;EACxD,OAAO,KAAK,SAAS,IAAI,SAAS;CACpC;AACF;AAEA,SAAgB,qBACd,GAAG,aAC6B;CAChC,MAAM,WAAoD,CAAC;CAE3D,KAAK,MAAM,cAAc,aAAa;EACpC,IAAI,CAAC,YACH;EAEF,KAAK,MAAM,CAAC,WAAW,YAAY,OAAO,QAAQ,UAAU,GAAG;GAC7D,MAAM,WAAW,SAAS;GAC1B,IAAI,aAAa,KAAA,GAAW;IAC1B,SAAS,aAAa;IACtB;GACF;GACA,SAAS,aAAa,EACpB,QAAQ,OACN,GAAG,SACe;IAClB,MAAM,SAAS,OAAO,GAAG,IAAI;IAC7B,MAAM,QAAQ,OAAO,GAAG,IAAI;GAC9B,EACF;EACF;CACF;CAEA,OAAO;AACT;AAEA,IAAa,kBAAb,MAAuD;CACrD;CACA,YAAY,gBAAkC;EAC5C,IAAI,kBAAkB,CAAC,MAAM,QAAQ,cAAc,GACjD,MAAM,IAAI,MAAM,iCAAiC;EAEnD,KAAK,iBAAiB;CACxB;CAEA,MAAM,OACJ,OACA,MACA,UACA,OACe;EACf,IAAI,CAAC,SAAS,CAAC,UAAU;GACvB,QAAQ,KAAK,kCAAkC,MAAM,OAAO;GAC5D;EACF;EAEA,MAAM,QAAQ,MAAM,QAAQ;EAC5B,IAAI,SAAS,QAAQ,KAAK,kBAAkB,MAC1C,KAAK,eAAe,KAAK,KAAK;CAElC;AACF;AAEA,IAAa,iBAAb,MAAsD;CACpD;CACA;CACA,YAAY,UAA8B,QAAiB;EACzD,KAAK,WAAW;EAChB,KAAK,SAAS;CAChB;;;;;;;;;CAUA,MAAM,OACJ,OACA,MACA,UACA,OACe;EACf,IAAI;GACF,IAAI,CAAC,SAAS,CAAC,UAAU;IACvB,IAAI,KAAK,QACP,KAAK,OAAO,KAAK,kCAAkC,MAAM,OAAO;SAEhE,QAAQ,KAAK,kCAAkC,MAAM,OAAO;IAE9D;GACF;GAEA,MAAM,cAAc;GACpB,IAAI,CAAC,aAAa,QAAQ;IACxB,IAAI,KAAK,QACP,KAAK,OAAO,KAAK,mCAAmC;SAEpD,QAAQ,KAAK,mCAAmC;IAElD;GACF;GAEA,IACE,SAAA,2BAAkD,QAClD,SAAA,2BAAuD,MAEvD;GAGF,IAAI,KAAK,UACP,MAAM,KAAK,SAAS,aAAa,QAAQ;EAE7C,SAAS,OAAO;GACd,IAAI,KAAK,QACP,KAAK,OAAO,MAAM,kCAAkC,KAAK;QAEzD,QAAQ,MAAM,kCAAkC,KAAK;EAEzD;CACF;AACF;AAEA,IAAa,uBAAb,MAA4D;CAC1D,OAAO,OAAe,MAA2C;EAC/D,MAAM,QAAQ,MAAM;EAEpB,MAAM,MAAM,CADY,EAAE,SAAS,aAAa,SACnB,MAAM,UAAU,KAAA;EAC7C,IAAI,OAAO,IAAI,oBAAoB,IAAI,iBAAiB,SAAS,GAC/D,QAAQ,IAAI,IAAI,gBAAgB;OAC3B,IAAI,OAAO,OAAO,IAAI,YAAY,UACvC,QAAQ,OAAO,MAAM,IAAI,OAAO;CAEpC;AACF;AAEA,IAAa,wBAAb,MAA6D;CAC3D,OAAO,OAAe,MAA2C;EAC/D,MAAM,QAAQ,MAAM;EAEpB,IAAI,CAAC,CADmB,EAAE,SAAS,aAAa,QAE9C;EAGF,MAAM,UAAU,MAAM;EAEtB,IAAI,MAAM,oBAAoB,MAAM,iBAAiB,SAAS,GAC5D,QAAQ,IAAI,MAAM,kBAAkB,EAAE,OAAO,KAAK,CAAC;EAGrD,IAAI,OAAO,YAAY,UACrB,QAAQ,OAAO,MAAM,OAAO;OAE5B,QAAQ,IAAI,SAAS,EAAE,OAAO,KAAK,CAAC;CAExC;AACF;AAEA,IAAa,mBAAb,MAAwD;CACtD,OACE,OACA,MACA,UACM;EACN,MAAM,QAAQ,MAAM;EAEpB,MAAM,MAAM,CADY,EAAE,SAAS,aAAa,SACnB,MAAM,UAAU,KAAA;EAC7C,IAAI,UACF,QAAQ,IAAI,QAAQ;EAEtB,IAAI,OAAO,IAAI,oBAAoB,IAAI,iBAAiB,SAAS,GAC/D,QAAQ,IAAI,IAAI,gBAAgB;OAC3B,IAAI,OAAO,OAAO,IAAI,YAAY,UACvC,QAAQ,OAAO,MAAM,IAAI,OAAO;CAEpC;AACF;AAEA,MAAa,4BACX,eAI+B;CAC/B,MAAM,YAAY,cAAc,CAAC;CAEjC,MAAM,gBAAgC,WAAW;EAC/C,MAAM,EAAE,gBAAgB;EACxB,MAAM,oBACJ,YAAY,YAAY,SAAS,EAAE,GAGjC;EACJ,IAAI,CAAC,mBACH;EAEF,MAAM,EAAE,YAAY;EACpB,IAAI,SAAS,mBACX,UAAU,KAAK,QAAQ,iBAAiB;CAE5C;CAEA,OAAO;EAAE;EAAc;CAAU;AACnC"}
+{"version":3,"file":"events.cjs","names":["dispatchesChatModelStream","SDK_STREAM_DISPATCH"],"sources":["../../src/events.ts"],"sourcesContent":["/* eslint-disable no-console */\n// src/events.ts\nimport type {\n  BaseMessageFields,\n  UsageMetadata,\n} from '@langchain/core/messages';\nimport type { Logger } from 'winston';\nimport type { MultiAgentGraph, StandardGraph } from '@/graphs';\nimport type * as t from '@/types';\nimport { dispatchesChatModelStream, SDK_STREAM_DISPATCH } from '@/stream';\nimport { Constants } from '@/common';\n\nexport class HandlerRegistry {\n  private handlers: Map<string, t.EventHandler> = new Map();\n\n  register(eventType: string, handler: t.EventHandler): void {\n    this.handlers.set(eventType, handler);\n  }\n\n  getHandler(eventType: string): t.EventHandler | undefined {\n    return this.handlers.get(eventType);\n  }\n}\n\nexport function composeEventHandlers(\n  ...handlerSets: Array<Record<string, t.EventHandler> | undefined>\n): Record<string, t.EventHandler> {\n  const composed: Partial<Record<string, t.EventHandler>> = {};\n\n  for (const handlerSet of handlerSets) {\n    if (!handlerSet) {\n      continue;\n    }\n    for (const [eventType, handler] of Object.entries(handlerSet)) {\n      const previous = composed[eventType];\n      if (previous === undefined) {\n        composed[eventType] = handler;\n        continue;\n      }\n      const wrapper: t.EventHandler = {\n        handle: async (\n          ...args: Parameters<t.EventHandler['handle']>\n        ): Promise<void> => {\n          await previous.handle(...args);\n          await handler.handle(...args);\n        },\n      };\n      /**\n       * Carry the stream-dispatch brand across the wrapper. Without this a\n       * composed handler fails every capability check made about it — notably\n       * the one deciding whether a run may seal — even though it still drives\n       * the SDK's content-part dispatch.\n       */\n      if (\n        dispatchesChatModelStream(previous) ||\n        dispatchesChatModelStream(handler)\n      ) {\n        Object.defineProperty(wrapper, SDK_STREAM_DISPATCH, { value: true });\n      }\n      composed[eventType] = wrapper;\n    }\n  }\n\n  return composed as Record<string, t.EventHandler>;\n}\n\nexport class ModelEndHandler implements t.EventHandler {\n  collectedUsage?: UsageMetadata[];\n  constructor(collectedUsage?: UsageMetadata[]) {\n    if (collectedUsage && !Array.isArray(collectedUsage)) {\n      throw new Error('collectedUsage must be an array');\n    }\n    this.collectedUsage = collectedUsage;\n  }\n\n  async handle(\n    event: string,\n    data: t.ModelEndData,\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    const usage = data?.output?.usage_metadata;\n    if (usage != null && this.collectedUsage != null) {\n      this.collectedUsage.push(usage);\n    }\n  }\n}\n\nexport class ToolEndHandler implements t.EventHandler {\n  private callback?: t.ToolEndCallback;\n  private logger?: Logger;\n  constructor(callback?: t.ToolEndCallback, logger?: Logger) {\n    this.callback = callback;\n    this.logger = logger;\n  }\n\n  /**\n   * Handles on_tool_end events from the for-await stream consumer.\n   *\n   * This handler is now purely a consumer callback — tool completion\n   * (ON_RUN_STEP_COMPLETED dispatch + session context storage) is handled\n   * in graph context by ToolNode directly, eliminating the race between\n   * the stream consumer and graph execution.\n   */\n  async handle(\n    event: string,\n    data: t.StreamEventData | undefined,\n    metadata?: Record<string, unknown>,\n    graph?: StandardGraph | MultiAgentGraph\n  ): Promise<void> {\n    try {\n      if (!graph || !metadata) {\n        if (this.logger) {\n          this.logger.warn(`Graph or metadata not found in ${event} event`);\n        } else {\n          console.warn(`Graph or metadata not found in ${event} event`);\n        }\n        return;\n      }\n\n      const toolEndData = data as t.ToolEndData | undefined;\n      if (!toolEndData?.output) {\n        if (this.logger) {\n          this.logger.warn('No output found in tool_end event');\n        } else {\n          console.warn('No output found in tool_end event');\n        }\n        return;\n      }\n\n      if (\n        metadata[Constants.PROGRAMMATIC_TOOL_CALLING] === true ||\n        metadata[Constants.BASH_PROGRAMMATIC_TOOL_CALLING] === true\n      ) {\n        return;\n      }\n\n      if (this.callback) {\n        await this.callback(toolEndData, metadata);\n      }\n    } catch (error) {\n      if (this.logger) {\n        this.logger.error('Error handling tool_end event:', error);\n      } else {\n        console.error('Error handling tool_end event:', error);\n      }\n    }\n  }\n}\n\nexport class TestLLMStreamHandler implements t.EventHandler {\n  handle(event: string, data: t.StreamEventData | undefined): void {\n    const chunk = data?.chunk;\n    const isMessageChunk = !!(chunk && 'message' in chunk);\n    const msg = isMessageChunk ? chunk.message : undefined;\n    if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {\n      console.log(msg.tool_call_chunks);\n    } else if (msg && typeof msg.content === 'string') {\n      process.stdout.write(msg.content);\n    }\n  }\n}\n\nexport class TestChatStreamHandler implements t.EventHandler {\n  handle(event: string, data: t.StreamEventData | undefined): void {\n    const chunk = data?.chunk;\n    const isContentChunk = !!(chunk && 'content' in chunk);\n    if (!isContentChunk) {\n      return;\n    }\n\n    const content = chunk.content;\n\n    if (chunk.tool_call_chunks && chunk.tool_call_chunks.length > 0) {\n      console.dir(chunk.tool_call_chunks, { depth: null });\n    }\n\n    if (typeof content === 'string') {\n      process.stdout.write(content);\n    } else {\n      console.dir(content, { depth: null });\n    }\n  }\n}\n\nexport class LLMStreamHandler implements t.EventHandler {\n  handle(\n    event: string,\n    data: t.StreamEventData | undefined,\n    metadata?: Record<string, unknown>\n  ): void {\n    const chunk = data?.chunk;\n    const isMessageChunk = !!(chunk && 'message' in chunk);\n    const msg = isMessageChunk ? chunk.message : undefined;\n    if (metadata) {\n      console.log(metadata);\n    }\n    if (msg && msg.tool_call_chunks && msg.tool_call_chunks.length > 0) {\n      console.log(msg.tool_call_chunks);\n    } else if (msg && typeof msg.content === 'string') {\n      process.stdout.write(msg.content);\n    }\n  }\n}\n\nexport const createMetadataAggregator = (\n  _collected?: Record<\n    string,\n    NonNullable<BaseMessageFields['response_metadata']>\n  >[]\n): t.MetadataAggregatorResult => {\n  const collected = _collected || [];\n\n  const handleLLMEnd: t.HandleLLMEnd = (output) => {\n    const { generations } = output;\n    const lastMessageOutput = (\n      generations[generations.length - 1] as\n        | (t.StreamGeneration | undefined)[]\n        | undefined\n    )?.[0];\n    if (!lastMessageOutput) {\n      return;\n    }\n    const { message } = lastMessageOutput;\n    if (message?.response_metadata) {\n      collected.push(message.response_metadata);\n    }\n  };\n\n  return { handleLLMEnd, collected };\n};\n"],"mappings":";;;;AAYA,IAAa,kBAAb,MAA6B;CAC3B,2BAAgD,IAAI,IAAI;CAExD,SAAS,WAAmB,SAA+B;EACzD,KAAK,SAAS,IAAI,WAAW,OAAO;CACtC;CAEA,WAAW,WAA+C;EACxD,OAAO,KAAK,SAAS,IAAI,SAAS;CACpC;AACF;AAEA,SAAgB,qBACd,GAAG,aAC6B;CAChC,MAAM,WAAoD,CAAC;CAE3D,KAAK,MAAM,cAAc,aAAa;EACpC,IAAI,CAAC,YACH;EAEF,KAAK,MAAM,CAAC,WAAW,YAAY,OAAO,QAAQ,UAAU,GAAG;GAC7D,MAAM,WAAW,SAAS;GAC1B,IAAI,aAAa,KAAA,GAAW;IAC1B,SAAS,aAAa;IACtB;GACF;GACA,MAAM,UAA0B,EAC9B,QAAQ,OACN,GAAG,SACe;IAClB,MAAM,SAAS,OAAO,GAAG,IAAI;IAC7B,MAAM,QAAQ,OAAO,GAAG,IAAI;GAC9B,EACF;;;;;;;GAOA,IACEA,eAAAA,0BAA0B,QAAQ,KAClCA,eAAAA,0BAA0B,OAAO,GAEjC,OAAO,eAAe,SAASC,eAAAA,qBAAqB,EAAE,OAAO,KAAK,CAAC;GAErE,SAAS,aAAa;EACxB;CACF;CAEA,OAAO;AACT;AAEA,IAAa,kBAAb,MAAuD;CACrD;CACA,YAAY,gBAAkC;EAC5C,IAAI,kBAAkB,CAAC,MAAM,QAAQ,cAAc,GACjD,MAAM,IAAI,MAAM,iCAAiC;EAEnD,KAAK,iBAAiB;CACxB;CAEA,MAAM,OACJ,OACA,MACA,UACA,OACe;EACf,IAAI,CAAC,SAAS,CAAC,UAAU;GACvB,QAAQ,KAAK,kCAAkC,MAAM,OAAO;GAC5D;EACF;EAEA,MAAM,QAAQ,MAAM,QAAQ;EAC5B,IAAI,SAAS,QAAQ,KAAK,kBAAkB,MAC1C,KAAK,eAAe,KAAK,KAAK;CAElC;AACF;AAEA,IAAa,iBAAb,MAAsD;CACpD;CACA;CACA,YAAY,UAA8B,QAAiB;EACzD,KAAK,WAAW;EAChB,KAAK,SAAS;CAChB;;;;;;;;;CAUA,MAAM,OACJ,OACA,MACA,UACA,OACe;EACf,IAAI;GACF,IAAI,CAAC,SAAS,CAAC,UAAU;IACvB,IAAI,KAAK,QACP,KAAK,OAAO,KAAK,kCAAkC,MAAM,OAAO;SAEhE,QAAQ,KAAK,kCAAkC,MAAM,OAAO;IAE9D;GACF;GAEA,MAAM,cAAc;GACpB,IAAI,CAAC,aAAa,QAAQ;IACxB,IAAI,KAAK,QACP,KAAK,OAAO,KAAK,mCAAmC;SAEpD,QAAQ,KAAK,mCAAmC;IAElD;GACF;GAEA,IACE,SAAA,2BAAkD,QAClD,SAAA,2BAAuD,MAEvD;GAGF,IAAI,KAAK,UACP,MAAM,KAAK,SAAS,aAAa,QAAQ;EAE7C,SAAS,OAAO;GACd,IAAI,KAAK,QACP,KAAK,OAAO,MAAM,kCAAkC,KAAK;QAEzD,QAAQ,MAAM,kCAAkC,KAAK;EAEzD;CACF;AACF;AAEA,IAAa,uBAAb,MAA4D;CAC1D,OAAO,OAAe,MAA2C;EAC/D,MAAM,QAAQ,MAAM;EAEpB,MAAM,MAAM,CADY,EAAE,SAAS,aAAa,SACnB,MAAM,UAAU,KAAA;EAC7C,IAAI,OAAO,IAAI,oBAAoB,IAAI,iBAAiB,SAAS,GAC/D,QAAQ,IAAI,IAAI,gBAAgB;OAC3B,IAAI,OAAO,OAAO,IAAI,YAAY,UACvC,QAAQ,OAAO,MAAM,IAAI,OAAO;CAEpC;AACF;AAEA,IAAa,wBAAb,MAA6D;CAC3D,OAAO,OAAe,MAA2C;EAC/D,MAAM,QAAQ,MAAM;EAEpB,IAAI,CAAC,CADmB,EAAE,SAAS,aAAa,QAE9C;EAGF,MAAM,UAAU,MAAM;EAEtB,IAAI,MAAM,oBAAoB,MAAM,iBAAiB,SAAS,GAC5D,QAAQ,IAAI,MAAM,kBAAkB,EAAE,OAAO,KAAK,CAAC;EAGrD,IAAI,OAAO,YAAY,UACrB,QAAQ,OAAO,MAAM,OAAO;OAE5B,QAAQ,IAAI,SAAS,EAAE,OAAO,KAAK,CAAC;CAExC;AACF;AAEA,IAAa,mBAAb,MAAwD;CACtD,OACE,OACA,MACA,UACM;EACN,MAAM,QAAQ,MAAM;EAEpB,MAAM,MAAM,CADY,EAAE,SAAS,aAAa,SACnB,MAAM,UAAU,KAAA;EAC7C,IAAI,UACF,QAAQ,IAAI,QAAQ;EAEtB,IAAI,OAAO,IAAI,oBAAoB,IAAI,iBAAiB,SAAS,GAC/D,QAAQ,IAAI,IAAI,gBAAgB;OAC3B,IAAI,OAAO,OAAO,IAAI,YAAY,UACvC,QAAQ,OAAO,MAAM,IAAI,OAAO;CAEpC;AACF;AAEA,MAAa,4BACX,eAI+B;CAC/B,MAAM,YAAY,cAAc,CAAC;CAEjC,MAAM,gBAAgC,WAAW;EAC/C,MAAM,EAAE,gBAAgB;EACxB,MAAM,oBACJ,YAAY,YAAY,SAAS,EAAE,GAGjC;EACJ,IAAI,CAAC,mBACH;EAEF,MAAM,EAAE,YAAY;EACpB,IAAI,SAAS,mBACX,UAAU,KAAK,QAAQ,iBAAiB;CAE5C;CAEA,OAAO;EAAE;EAAc;CAAU;AACnC"}