@livekit/agents 0.0.0-20260120144724

package/dist/voice/generation.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/voice/generation.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2025 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport type { AudioFrame } from '@livekit/rtc-node';\nimport { AudioResampler } from '@livekit/rtc-node';\nimport type { Span } from '@opentelemetry/api';\nimport { context as otelContext } from '@opentelemetry/api';\nimport type { ReadableStream, ReadableStreamDefaultReader } from 'stream/web';\nimport {\n  type ChatContext,\n  ChatMessage,\n  FunctionCall,\n  FunctionCallOutput,\n} from '../llm/chat_context.js';\nimport type { ChatChunk } from '../llm/llm.js';\nimport {\n  type ToolChoice,\n  type ToolContext,\n  isAgentHandoff,\n  isFunctionTool,\n  isToolError,\n} from '../llm/tool_context.js';\nimport { isZodSchema, parseZodSchema } from '../llm/zod-utils.js';\nimport { log } from '../log.js';\nimport { IdentityTransform } from '../stream/identity_transform.js';\nimport { traceTypes, tracer } from '../telemetry/index.js';\nimport { Future, Task, shortuuid, toError, waitForAbort } from '../utils.js';\nimport { type Agent, type ModelSettings, asyncLocalStorage, isStopResponse } from './agent.js';\nimport type { AgentSession } from './agent_session.js';\nimport type { AudioOutput, LLMNode, TTSNode, TextOutput } from './io.js';\nimport { RunContext } from './run_context.js';\nimport type { SpeechHandle } from './speech_handle.js';\n\n/** @internal */\nexport class _LLMGenerationData {\n  generatedText: string = '';\n  generatedToolCalls: FunctionCall[];\n  id: string;\n\n  constructor(\n    public readonly textStream: ReadableStream<string>,\n    public readonly toolCallStream: ReadableStream<FunctionCall>,\n  ) {\n    this.id = shortuuid('item_');\n    this.generatedToolCalls = [];\n  }\n}\n\n// TODO(brian): remove this class in favor of ToolOutput\nexport class _ToolOutput {\n  output: _JsOutput[];\n  firstToolFut: Future;\n\n  constructor() {\n    this.output = [];\n    this.firstToolFut = new Future();\n  }\n}\n\n// TODO(brian): remove this class in favor of ToolExecutionOutput\nexport class _SanitizedOutput {\n  toolCall: FunctionCall;\n  toolCallOutput?: FunctionCallOutput;\n  replyRequired: boolean;\n  agentTask?: Agent;\n\n  constructor(\n    toolCall: FunctionCall,\n    toolCallOutput: FunctionCallOutput | undefined,\n    replyRequired: boolean,\n    agentTask: Agent | undefined,\n  ) {\n    this.toolCall = toolCall;\n    this.toolCallOutput = toolCallOutput;\n    this.replyRequired = replyRequired;\n    this.agentTask = agentTask;\n  }\n\n  static create(params: {\n    toolCall: FunctionCall;\n    toolCallOutput?: FunctionCallOutput;\n    replyRequired?: boolean;\n    agentTask?: Agent;\n  }) {\n    const { toolCall, toolCallOutput, replyRequired = true, agentTask } = params;\n    return new _SanitizedOutput(toolCall, toolCallOutput, replyRequired, agentTask);\n  }\n}\n\nfunction isValidToolOutput(toolOutput: unknown): boolean {\n  const validTypes = ['string', 'number', 'boolean'];\n\n  if (validTypes.includes(typeof toolOutput)) {\n    return true;\n  }\n\n  if (toolOutput === undefined || toolOutput === null) {\n    return true;\n  }\n\n  if (Array.isArray(toolOutput)) {\n    return toolOutput.every(isValidToolOutput);\n  }\n\n  if (toolOutput instanceof Set) {\n    return Array.from(toolOutput).every(isValidToolOutput);\n  }\n\n  if (toolOutput instanceof Map) {\n    return Array.from(toolOutput.values()).every(isValidToolOutput);\n  }\n\n  if (toolOutput instanceof Object) {\n    return Object.entries(toolOutput).every(\n      ([key, value]) => validTypes.includes(typeof key) && isValidToolOutput(value),\n    );\n  }\n\n  return false;\n}\n\nexport class ToolExecutionOutput {\n  constructor(\n    public readonly toolCall: FunctionCall,\n    public readonly toolCallOutput: FunctionCallOutput | undefined,\n    public readonly agentTask: Agent | undefined,\n    public readonly rawOutput: unknown,\n    public readonly rawException: Error | undefined,\n    public readonly replyRequired: boolean,\n  ) {}\n\n  static create(params: {\n    toolCall: FunctionCall;\n    toolCallOutput?: FunctionCallOutput;\n    agentTask?: Agent;\n    rawOutput: unknown;\n    rawException?: Error;\n    replyRequired?: boolean;\n  }) {\n    const {\n      toolCall,\n      toolCallOutput,\n      agentTask,\n      rawOutput,\n      rawException,\n      replyRequired = true,\n    } = params;\n    return new ToolExecutionOutput(\n      toolCall,\n      toolCallOutput,\n      agentTask,\n      rawOutput,\n      rawException,\n      replyRequired,\n    );\n  }\n}\n\nexport interface ToolOutput {\n  output: ToolExecutionOutput[];\n  firstToolStartedFuture: Future<void>;\n}\n\n// TODO(brian): remove this class in favor of ToolExecutionOutput\nexport class _JsOutput {\n  toolCall: FunctionCall;\n  output: unknown;\n  exception?: Error;\n\n  #logger = log();\n\n  constructor(toolCall: FunctionCall, output: unknown, exception: Error | undefined) {\n    this.toolCall = toolCall;\n    this.output = output;\n    this.exception = exception;\n  }\n\n  static create(params: { toolCall: FunctionCall; output?: unknown; exception?: Error }) {\n    const { toolCall, output = undefined, exception = undefined } = params;\n    return new _JsOutput(toolCall, output, exception);\n  }\n\n  sanitize(): _SanitizedOutput {\n    if (isToolError(this.exception)) {\n      return _SanitizedOutput.create({\n        toolCall: FunctionCall.create({ ...this.toolCall }),\n        toolCallOutput: FunctionCallOutput.create({\n          name: this.toolCall.name,\n          callId: this.toolCall.callId,\n          output: this.exception.message,\n          isError: true,\n        }),\n      });\n    }\n\n    if (isStopResponse(this.exception)) {\n      return _SanitizedOutput.create({\n        toolCall: FunctionCall.create({ ...this.toolCall }),\n      });\n    }\n\n    if (this.exception !== undefined) {\n      return _SanitizedOutput.create({\n        toolCall: FunctionCall.create({ ...this.toolCall }),\n        toolCallOutput: FunctionCallOutput.create({\n          name: this.toolCall.name,\n          callId: this.toolCall.callId,\n          output: 'An internal error occurred while executing the tool.', // Don't send the actual error message, as it may contain sensitive information\n          isError: true,\n        }),\n      });\n    }\n\n    let agentTask: Agent | undefined = undefined;\n    let toolOutput: unknown = this.output;\n    if (isAgentHandoff(this.output)) {\n      agentTask = this.output.agent;\n      toolOutput = this.output.returns;\n    }\n\n    if (!isValidToolOutput(toolOutput)) {\n      this.#logger.error(\n        {\n          callId: this.toolCall.callId,\n          function: this.toolCall.name,\n        },\n        `AI function ${this.toolCall.name} returned an invalid output`,\n      );\n      return _SanitizedOutput.create({\n        toolCall: FunctionCall.create({ ...this.toolCall }),\n        toolCallOutput: undefined,\n      });\n    }\n\n    return _SanitizedOutput.create({\n      toolCall: FunctionCall.create({ ...this.toolCall }),\n      toolCallOutput: FunctionCallOutput.create({\n        name: this.toolCall.name,\n        callId: this.toolCall.callId,\n        output: toolOutput !== undefined ? JSON.stringify(toolOutput) : '', // take the string representation of the output\n        isError: false,\n      }),\n      replyRequired: toolOutput !== undefined, // require a reply if the tool returned an output\n      agentTask,\n    });\n  }\n}\n\nexport function createToolOutput(params: {\n  toolCall: FunctionCall;\n  output?: unknown;\n  exception?: Error;\n}): ToolExecutionOutput {\n  const { toolCall, output, exception } = params;\n  const logger = log();\n\n  // support returning Exception instead of raising them (for devex purposes inside evals)\n  let finalOutput = output;\n  let finalException = exception;\n  if (output instanceof Error) {\n    finalException = output;\n    finalOutput = undefined;\n  }\n\n  if (isToolError(finalException)) {\n    return ToolExecutionOutput.create({\n      toolCall: FunctionCall.create({ ...toolCall }),\n      toolCallOutput: FunctionCallOutput.create({\n        name: toolCall.name,\n        callId: toolCall.callId,\n        output: finalException.message,\n        isError: true,\n      }),\n      rawOutput: finalOutput,\n      rawException: finalException,\n    });\n  }\n\n  if (isStopResponse(finalException)) {\n    return ToolExecutionOutput.create({\n      toolCall: FunctionCall.create({ ...toolCall }),\n      rawOutput: finalOutput,\n      rawException: finalException,\n    });\n  }\n\n  if (finalException !== undefined) {\n    return ToolExecutionOutput.create({\n      toolCall: FunctionCall.create({ ...toolCall }),\n      toolCallOutput: FunctionCallOutput.create({\n        name: toolCall.name,\n        callId: toolCall.callId,\n        output: 'An internal error occurred', // Don't send the actual error message, as it may contain sensitive information\n        isError: true,\n      }),\n      rawOutput: finalOutput,\n      rawException: finalException,\n    });\n  }\n\n  let agentTask: Agent | undefined = undefined;\n  let toolOutput: unknown = finalOutput;\n  if (isAgentHandoff(finalOutput)) {\n    agentTask = finalOutput.agent;\n    toolOutput = finalOutput.returns;\n  }\n\n  if (!isValidToolOutput(toolOutput)) {\n    logger.error(\n      {\n        callId: toolCall.callId,\n        output: finalOutput,\n      },\n      `AI function ${toolCall.name} returned an invalid output`,\n    );\n    return ToolExecutionOutput.create({\n      toolCall: FunctionCall.create({ ...toolCall }),\n      rawOutput: finalOutput,\n      rawException: finalException,\n    });\n  }\n\n  return ToolExecutionOutput.create({\n    toolCall: FunctionCall.create({ ...toolCall }),\n    toolCallOutput: FunctionCallOutput.create({\n      name: toolCall.name,\n      callId: toolCall.callId,\n      output: toolOutput !== undefined ? JSON.stringify(toolOutput) : '', // take the string representation of the output\n      isError: false,\n    }),\n    replyRequired: toolOutput !== undefined, // require a reply if the tool returned an output\n    agentTask,\n    rawOutput: finalOutput,\n    rawException: finalException,\n  });\n}\n\nconst INSTRUCTIONS_MESSAGE_ID = 'lk.agent_task.instructions';\n\n/**\n * Update the instruction message in the chat context or insert a new one if missing.\n *\n * This function looks for an existing instruction message in the chat context using the identifier\n * 'INSTRUCTIONS_MESSAGE_ID'.\n *\n * @param options - The options for updating the instructions.\n * @param options.chatCtx - The chat context to update.\n * @param options.instructions - The instructions to add.\n * @param options.addIfMissing - Whether to add the instructions if they are missing.\n */\nexport function updateInstructions(options: {\n  chatCtx: ChatContext;\n  instructions: string;\n  addIfMissing: boolean;\n}) {\n  const { chatCtx, instructions, addIfMissing } = options;\n\n  const idx = chatCtx.indexById(INSTRUCTIONS_MESSAGE_ID);\n  if (idx !== undefined) {\n    if (chatCtx.items[idx]!.type === 'message') {\n      // create a new instance to avoid mutating the original\n      chatCtx.items[idx] = ChatMessage.create({\n        id: INSTRUCTIONS_MESSAGE_ID,\n        role: 'system',\n        content: [instructions],\n        createdAt: chatCtx.items[idx]!.createdAt,\n      });\n    } else {\n      throw new Error('expected the instructions inside the chatCtx to be of type \"message\"');\n    }\n  } else if (addIfMissing) {\n    // insert the instructions at the beginning of the chat context\n    chatCtx.items.unshift(\n      ChatMessage.create({\n        id: INSTRUCTIONS_MESSAGE_ID,\n        role: 'system',\n        content: [instructions],\n      }),\n    );\n  }\n}\n\nexport function performLLMInference(\n  node: LLMNode,\n  chatCtx: ChatContext,\n  toolCtx: ToolContext,\n  modelSettings: ModelSettings,\n  controller: AbortController,\n): [Task<void>, _LLMGenerationData] {\n  const textStream = new IdentityTransform<string>();\n  const toolCallStream = new IdentityTransform<FunctionCall>();\n\n  const textWriter = textStream.writable.getWriter();\n  const toolCallWriter = toolCallStream.writable.getWriter();\n  const data = new _LLMGenerationData(textStream.readable, toolCallStream.readable);\n\n  const _performLLMInferenceImpl = async (signal: AbortSignal, span: Span) => {\n    span.setAttribute(\n      traceTypes.ATTR_CHAT_CTX,\n      JSON.stringify(chatCtx.toJSON({ excludeTimestamp: false })),\n    );\n    span.setAttribute(traceTypes.ATTR_FUNCTION_TOOLS, JSON.stringify(Object.keys(toolCtx)));\n\n    let llmStreamReader: ReadableStreamDefaultReader<string | ChatChunk> | null = null;\n    let llmStream: ReadableStream<string | ChatChunk> | null = null;\n\n    try {\n      llmStream = await node(chatCtx, toolCtx, modelSettings);\n      if (llmStream === null) {\n        await textWriter.close();\n        return;\n      }\n\n      const abortPromise = waitForAbort(signal);\n\n      // TODO(brian): add support for dynamic tools\n\n      llmStreamReader = llmStream.getReader();\n      while (true) {\n        if (signal.aborted) break;\n\n        const result = await Promise.race([llmStreamReader.read(), abortPromise]);\n        if (result === undefined) break;\n\n        const { done, value: chunk } = result;\n        if (done) break;\n\n        if (typeof chunk === 'string') {\n          data.generatedText += chunk;\n          await textWriter.write(chunk);\n          // TODO(shubhra): better way to check??\n        } else {\n          if (chunk.delta === undefined) {\n            continue;\n          }\n\n          if (chunk.delta.toolCalls) {\n            for (const tool of chunk.delta.toolCalls) {\n              if (tool.type !== 'function_call') continue;\n\n              const toolCall = FunctionCall.create({\n                callId: `${data.id}/fnc_${data.generatedToolCalls.length}`,\n                name: tool.name,\n                args: tool.args,\n                // Preserve thought signature for Gemini 3+ thinking mode\n                thoughtSignature: tool.thoughtSignature,\n                extra: tool.extra || {},\n              });\n\n              data.generatedToolCalls.push(toolCall);\n              await toolCallWriter.write(toolCall);\n            }\n          }\n\n          if (chunk.delta.content) {\n            data.generatedText += chunk.delta.content;\n            await textWriter.write(chunk.delta.content);\n          }\n        }\n\n        // No need to check if chunk is of type other than ChatChunk or string like in\n        // Python since chunk is defined in the type ChatChunk | string in TypeScript\n      }\n\n      span.setAttribute(traceTypes.ATTR_RESPONSE_TEXT, data.generatedText);\n    } catch (error) {\n      if (error instanceof DOMException && error.name === 'AbortError') {\n        // Abort signal was triggered, handle gracefully\n        return;\n      }\n      throw error;\n    } finally {\n      llmStreamReader?.releaseLock();\n      await llmStream?.cancel();\n      await textWriter.close();\n      await toolCallWriter.close();\n    }\n  };\n\n  // Capture the current context (agent_turn) to ensure llm_node is properly parented\n  const currentContext = otelContext.active();\n\n  const inferenceTask = async (signal: AbortSignal) =>\n    tracer.startActiveSpan(async (span) => _performLLMInferenceImpl(signal, span), {\n      name: 'llm_node',\n      context: currentContext,\n    });\n\n  return [\n    Task.from((controller) => inferenceTask(controller.signal), controller, 'performLLMInference'),\n    data,\n  ];\n}\n\nexport function performTTSInference(\n  node: TTSNode,\n  text: ReadableStream<string>,\n  modelSettings: ModelSettings,\n  controller: AbortController,\n): [Task<void>, ReadableStream<AudioFrame>] {\n  const audioStream = new IdentityTransform<AudioFrame>();\n  const outputWriter = audioStream.writable.getWriter();\n  const audioOutputStream = audioStream.readable;\n\n  const _performTTSInferenceImpl = async (signal: AbortSignal) => {\n    let ttsStreamReader: ReadableStreamDefaultReader<AudioFrame> | null = null;\n    let ttsStream: ReadableStream<AudioFrame> | null = null;\n\n    try {\n      ttsStream = await node(text, modelSettings);\n      if (ttsStream === null) {\n        await outputWriter.close();\n        return;\n      }\n\n      ttsStreamReader = ttsStream.getReader();\n      while (true) {\n        if (signal.aborted) {\n          break;\n        }\n        const { done, value: chunk } = await ttsStreamReader.read();\n        if (done) {\n          break;\n        }\n        await outputWriter.write(chunk);\n      }\n    } catch (error) {\n      if (error instanceof DOMException && error.name === 'AbortError') {\n        // Abort signal was triggered, handle gracefully\n        return;\n      }\n      throw error;\n    } finally {\n      ttsStreamReader?.releaseLock();\n      await ttsStream?.cancel();\n      await outputWriter.close();\n    }\n  };\n\n  // Capture the current context (agent_turn) to ensure tts_node is properly parented\n  const currentContext = otelContext.active();\n\n  const inferenceTask = async (signal: AbortSignal) =>\n    tracer.startActiveSpan(async () => _performTTSInferenceImpl(signal), {\n      name: 'tts_node',\n      context: currentContext,\n    });\n\n  return [\n    Task.from((controller) => inferenceTask(controller.signal), controller, 'performTTSInference'),\n    audioOutputStream,\n  ];\n}\n\nexport interface _TextOut {\n  text: string;\n  firstTextFut: Future;\n}\n\nasync function forwardText(\n  source: ReadableStream<string>,\n  out: _TextOut,\n  signal: AbortSignal,\n  textOutput: TextOutput | null,\n): Promise<void> {\n  const reader = source.getReader();\n  try {\n    while (true) {\n      if (signal.aborted) {\n        break;\n      }\n      const { done, value: delta } = await reader.read();\n      if (done) break;\n      out.text += delta;\n      if (textOutput !== null) {\n        await textOutput.captureText(delta);\n      }\n      if (!out.firstTextFut.done) {\n        out.firstTextFut.resolve();\n      }\n    }\n  } finally {\n    if (textOutput !== null) {\n      textOutput.flush();\n    }\n    reader?.releaseLock();\n  }\n}\n\nexport function performTextForwarding(\n  source: ReadableStream<string>,\n  controller: AbortController,\n  textOutput: TextOutput | null,\n): [Task<void>, _TextOut] {\n  const out = {\n    text: '',\n    firstTextFut: new Future(),\n  };\n  return [\n    Task.from(\n      (controller) => forwardText(source, out, controller.signal, textOutput),\n      controller,\n      'performTextForwarding',\n    ),\n    out,\n  ];\n}\n\nexport interface _AudioOut {\n  audio: Array<AudioFrame>;\n  firstFrameFut: Future;\n}\n\nasync function forwardAudio(\n  ttsStream: ReadableStream<AudioFrame>,\n  audioOuput: AudioOutput,\n  out: _AudioOut,\n  signal?: AbortSignal,\n): Promise<void> {\n  const reader = ttsStream.getReader();\n  let resampler: AudioResampler | null = null;\n\n  try {\n    while (true) {\n      if (signal?.aborted) {\n        break;\n      }\n\n      const { done, value: frame } = await reader.read();\n      if (done) break;\n\n      out.audio.push(frame);\n\n      if (\n        !out.firstFrameFut.done &&\n        audioOuput.sampleRate &&\n        audioOuput.sampleRate !== frame.sampleRate &&\n        !resampler\n      ) {\n        resampler = new AudioResampler(frame.sampleRate, audioOuput.sampleRate, 1);\n      }\n\n      if (resampler) {\n        for (const f of resampler.push(frame)) {\n          await audioOuput.captureFrame(f);\n        }\n      } else {\n        await audioOuput.captureFrame(frame);\n      }\n\n      // set the first frame future if not already set\n      // (after completing the first frame)\n      if (!out.firstFrameFut.done) {\n        out.firstFrameFut.resolve();\n      }\n    }\n  } finally {\n    reader?.releaseLock();\n    if (resampler) {\n      for (const f of resampler.flush()) {\n        await audioOuput.captureFrame(f);\n      }\n    }\n    audioOuput.flush();\n  }\n}\n\nexport function performAudioForwarding(\n  ttsStream: ReadableStream<AudioFrame>,\n  audioOutput: AudioOutput,\n  controller: AbortController,\n): [Task<void>, _AudioOut] {\n  const out = {\n    audio: [],\n    firstFrameFut: new Future(),\n  };\n  return [\n    Task.from(\n      (controller) => forwardAudio(ttsStream, audioOutput, out, controller.signal),\n      controller,\n      'performAudioForwarding',\n    ),\n    out,\n  ];\n}\n\n// function_tool span is already implemented in tracableToolExecution below (line ~796)\nexport function performToolExecutions({\n  session,\n  speechHandle,\n  toolCtx,\n  toolChoice,\n  toolCallStream,\n  onToolExecutionStarted = () => {},\n  onToolExecutionCompleted = () => {},\n  controller,\n}: {\n  session: AgentSession;\n  speechHandle: SpeechHandle;\n  toolCtx: ToolContext;\n  toolChoice?: ToolChoice;\n  toolCallStream: ReadableStream<FunctionCall>;\n  onToolExecutionStarted?: (toolCall: FunctionCall) => void;\n  onToolExecutionCompleted?: (toolExecutionOutput: ToolExecutionOutput) => void;\n  controller: AbortController;\n}): [Task<void>, ToolOutput] {\n  const logger = log();\n  const toolOutput: ToolOutput = {\n    output: [],\n    firstToolStartedFuture: new Future(),\n  };\n\n  const toolCompleted = (out: ToolExecutionOutput) => {\n    onToolExecutionCompleted(out);\n    toolOutput.output.push(out);\n  };\n\n  const executeToolsTask = async (controller: AbortController) => {\n    const signal = controller.signal;\n    const reader = toolCallStream.getReader();\n\n    const tasks: Promise<any>[] = [];\n    while (!signal.aborted) {\n      const { done, value: toolCall } = await reader.read();\n      if (signal.aborted) break;\n      if (done) break;\n\n      if (toolChoice === 'none') {\n        logger.error(\n          {\n            function: toolCall.name,\n            speech_id: speechHandle.id,\n          },\n          \"received a tool call with toolChoice set to 'none', ignoring\",\n        );\n        continue;\n      }\n\n      // TODO(brian): assert other toolChoice values\n\n      const tool = toolCtx[toolCall.name];\n      if (!tool) {\n        logger.warn(\n          {\n            function: toolCall.name,\n            speech_id: speechHandle.id,\n          },\n          `unknown AI function ${toolCall.name}`,\n        );\n        continue;\n      }\n\n      if (!isFunctionTool(tool)) {\n        logger.error(\n          {\n            function: toolCall.name,\n            speech_id: speechHandle.id,\n          },\n          `unknown tool type: ${typeof tool}`,\n        );\n        continue;\n      }\n\n      let parsedArgs: object | undefined;\n\n      // Ensure valid arguments\n      try {\n        const jsonArgs = JSON.parse(toolCall.args);\n\n        if (isZodSchema(tool.parameters)) {\n          const result = await parseZodSchema<object>(tool.parameters, jsonArgs);\n          if (result.success) {\n            parsedArgs = result.data;\n          } else {\n            throw result.error;\n          }\n        } else {\n          parsedArgs = jsonArgs;\n        }\n      } catch (rawError) {\n        const error = toError(rawError);\n        logger.error(\n          {\n            function: toolCall.name,\n            arguments: toolCall.args,\n            speech_id: speechHandle.id,\n            error: error.message,\n          },\n          `tried to call AI function ${toolCall.name} with invalid arguments`,\n        );\n        toolCompleted(\n          createToolOutput({\n            toolCall,\n            exception: error,\n          }),\n        );\n        continue;\n      }\n\n      if (!toolOutput.firstToolStartedFuture.done) {\n        toolOutput.firstToolStartedFuture.resolve();\n      }\n\n      onToolExecutionStarted(toolCall);\n\n      logger.info(\n        {\n          function: toolCall.name,\n          arguments: parsedArgs,\n          speech_id: speechHandle.id,\n        },\n        'Executing LLM tool call',\n      );\n\n      const toolExecution = asyncLocalStorage.run({ functionCall: toolCall }, async () => {\n        return await tool.execute(parsedArgs, {\n          ctx: new RunContext(session, speechHandle, toolCall),\n          toolCallId: toolCall.callId,\n          abortSignal: signal,\n        });\n      });\n\n      const _tracableToolExecutionImpl = async (toolExecTask: Promise<unknown>, span: Span) => {\n        span.setAttribute(traceTypes.ATTR_FUNCTION_TOOL_NAME, toolCall.name);\n        span.setAttribute(traceTypes.ATTR_FUNCTION_TOOL_ARGS, toolCall.args);\n\n        // await for task to complete, if task is aborted, set exception\n        let toolOutput: ToolExecutionOutput | undefined;\n        try {\n          const { result, isAborted } = await waitUntilAborted(toolExecTask, signal);\n          toolOutput = createToolOutput({\n            toolCall,\n            exception: isAborted ? new Error('tool call was aborted') : undefined,\n            output: isAborted ? undefined : result,\n          });\n\n          if (toolOutput.toolCallOutput) {\n            span.setAttribute(\n              traceTypes.ATTR_FUNCTION_TOOL_OUTPUT,\n              toolOutput.toolCallOutput.output,\n            );\n            span.setAttribute(\n              traceTypes.ATTR_FUNCTION_TOOL_IS_ERROR,\n              toolOutput.toolCallOutput.isError,\n            );\n          }\n        } catch (rawError) {\n          logger.error(\n            {\n              function: toolCall.name,\n              speech_id: speechHandle.id,\n              error: toError(rawError).message,\n            },\n            'exception occurred while executing tool',\n          );\n          toolOutput = createToolOutput({\n            toolCall,\n            exception: toError(rawError),\n          });\n\n          if (toolOutput.toolCallOutput) {\n            span.setAttribute(\n              traceTypes.ATTR_FUNCTION_TOOL_OUTPUT,\n              toolOutput.toolCallOutput.output,\n            );\n            span.setAttribute(traceTypes.ATTR_FUNCTION_TOOL_IS_ERROR, true);\n          }\n        } finally {\n          if (!toolOutput) throw new Error('toolOutput is undefined');\n          toolCompleted(toolOutput);\n        }\n      };\n\n      const tracableToolExecution = (toolExecTask: Promise<unknown>) =>\n        tracer.startActiveSpan(async (span) => _tracableToolExecutionImpl(toolExecTask, span), {\n          name: 'function_tool',\n        });\n\n      // wait, not cancelling all tool calling tasks\n      tasks.push(tracableToolExecution(toolExecution));\n    }\n\n    await Promise.allSettled(tasks);\n    if (toolOutput.output.length > 0) {\n      logger.debug(\n        {\n          speech_id: speechHandle.id,\n        },\n        'tools execution completed',\n      );\n    }\n  };\n\n  return [Task.from(executeToolsTask, controller, 'performToolExecutions'), toolOutput];\n}\n\ntype Aborted<T> =\n  | {\n      result: T;\n      isAborted: false;\n    }\n  | {\n      result: undefined;\n      isAborted: true;\n    };\n\nasync function waitUntilAborted<T>(promise: Promise<T>, signal: AbortSignal): Promise<Aborted<T>> {\n  const abortFut = new Future<Aborted<T>>();\n\n  const resolveAbort = () => {\n    if (!abortFut.done) {\n      abortFut.resolve({ result: undefined, isAborted: true });\n    }\n  };\n\n  signal.addEventListener('abort', resolveAbort);\n\n  promise\n    .then((r) => {\n      if (!abortFut.done) {\n        abortFut.resolve({ result: r, isAborted: false });\n      }\n    })\n    .catch((e) => {\n      if (!abortFut.done) {\n        abortFut.reject(e);\n      }\n    })\n    .finally(() => {\n      signal.removeEventListener('abort', resolveAbort);\n    });\n\n  return await abortFut.await;\n}\n\nexport function removeInstructions(chatCtx: ChatContext) {\n  // loop in case there are items with the same id (shouldn't happen!)\n  while (true) {\n    const idx = chatCtx.indexById(INSTRUCTIONS_MESSAGE_ID);\n    if (idx !== undefined) {\n      chatCtx.items.splice(idx, 1);\n    } else {\n      break;\n    }\n  }\n}\n"],"mappings":"AAIA,SAAS,sBAAsB;AAE/B,SAAS,WAAW,mBAAmB;AAEvC;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAEP;AAAA,EAGE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,aAAa,sBAAsB;AAC5C,SAAS,WAAW;AACpB,SAAS,yBAAyB;AAClC,SAAS,YAAY,cAAc;AACnC,SAAS,QAAQ,MAAM,WAAW,SAAS,oBAAoB;AAC/D,SAAyC,mBAAmB,sBAAsB;AAGlF,SAAS,kBAAkB;AAIpB,MAAM,mBAAmB;AAAA,EAK9B,YACkB,YACA,gBAChB;AAFgB;AACA;AAEhB,SAAK,KAAK,UAAU,OAAO;AAC3B,SAAK,qBAAqB,CAAC;AAAA,EAC7B;AAAA,EAVA,gBAAwB;AAAA,EACxB;AAAA,EACA;AASF;AAGO,MAAM,YAAY;AAAA,EACvB;AAAA,EACA;AAAA,EAEA,cAAc;AACZ,SAAK,SAAS,CAAC;AACf,SAAK,eAAe,IAAI,OAAO;AAAA,EACjC;AACF;AAGO,MAAM,iBAAiB;AAAA,EAC5B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAEA,YACE,UACA,gBACA,eACA,WACA;AACA,SAAK,WAAW;AAChB,SAAK,iBAAiB;AACtB,SAAK,gBAAgB;AACrB,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,OAAO,OAAO,QAKX;AACD,UAAM,EAAE,UAAU,gBAAgB,gBAAgB,MAAM,UAAU,IAAI;AACtE,WAAO,IAAI,iBAAiB,UAAU,gBAAgB,eAAe,SAAS;AAAA,EAChF;AACF;AAEA,SAAS,kBAAkB,YAA8B;AACvD,QAAM,aAAa,CAAC,UAAU,UAAU,SAAS;AAEjD,MAAI,WAAW,SAAS,OAAO,UAAU,GAAG;AAC1C,WAAO;AAAA,EACT;AAEA,MAAI,eAAe,UAAa,eAAe,MAAM;AACnD,WAAO;AAAA,EACT;AAEA,MAAI,MAAM,QAAQ,UAAU,GAAG;AAC7B,WAAO,WAAW,MAAM,iBAAiB;AAAA,EAC3C;AAEA,MAAI,sBAAsB,KAAK;AAC7B,WAAO,MAAM,KAAK,UAAU,EAAE,MAAM,iBAAiB;AAAA,EACvD;AAEA,MAAI,sBAAsB,KAAK;AAC7B,WAAO,MAAM,KAAK,WAAW,OAAO,CAAC,EAAE,MAAM,iBAAiB;AAAA,EAChE;AAEA,MAAI,sBAAsB,QAAQ;AAChC,WAAO,OAAO,QAAQ,UAAU,EAAE;AAAA,MAChC,CAAC,CAAC,KAAK,KAAK,MAAM,WAAW,SAAS,OAAO,GAAG,KAAK,kBAAkB,KAAK;AAAA,IAC9E;AAAA,EACF;AAEA,SAAO;AACT;AAEO,MAAM,oBAAoB;AAAA,EAC/B,YACkB,UACA,gBACA,WACA,WACA,cACA,eAChB;AANgB;AACA;AACA;AACA;AACA;AACA;AAAA,EACf;AAAA,EAEH,OAAO,OAAO,QAOX;AACD,UAAM;AAAA,MACJ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,gBAAgB;AAAA,IAClB,IAAI;AACJ,WAAO,IAAI;AAAA,MACT;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AAQO,MAAM,UAAU;AAAA,EACrB;AAAA,EACA;AAAA,EACA;AAAA,EAEA,UAAU,IAAI;AAAA,EAEd,YAAY,UAAwB,QAAiB,WAA8B;AACjF,SAAK,WAAW;AAChB,SAAK,SAAS;AACd,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,OAAO,OAAO,QAAyE;AACrF,UAAM,EAAE,UAAU,SAAS,QAAW,YAAY,OAAU,IAAI;AAChE,WAAO,IAAI,UAAU,UAAU,QAAQ,SAAS;AAAA,EAClD;AAAA,EAEA,WAA6B;AAC3B,QAAI,YAAY,KAAK,SAAS,GAAG;AAC/B,aAAO,iBAAiB,OAAO;AAAA,QAC7B,UAAU,aAAa,OAAO,EAAE,GAAG,KAAK,SAAS,CAAC;AAAA,QAClD,gBAAgB,mBAAmB,OAAO;AAAA,UACxC,MAAM,KAAK,SAAS;AAAA,UACpB,QAAQ,KAAK,SAAS;AAAA,UACtB,QAAQ,KAAK,UAAU;AAAA,UACvB,SAAS;AAAA,QACX,CAAC;AAAA,MACH,CAAC;AAAA,IACH;AAEA,QAAI,eAAe,KAAK,SAAS,GAAG;AAClC,aAAO,iBAAiB,OAAO;AAAA,QAC7B,UAAU,aAAa,OAAO,EAAE,GAAG,KAAK,SAAS,CAAC;AAAA,MACpD,CAAC;AAAA,IACH;AAEA,QAAI,KAAK,cAAc,QAAW;AAChC,aAAO,iBAAiB,OAAO;AAAA,QAC7B,UAAU,aAAa,OAAO,EAAE,GAAG,KAAK,SAAS,CAAC;AAAA,QAClD,gBAAgB,mBAAmB,OAAO;AAAA,UACxC,MAAM,KAAK,SAAS;AAAA,UACpB,QAAQ,KAAK,SAAS;AAAA,UACtB,QAAQ;AAAA;AAAA,UACR,SAAS;AAAA,QACX,CAAC;AAAA,MACH,CAAC;AAAA,IACH;AAEA,QAAI,YAA+B;AACnC,QAAI,aAAsB,KAAK;AAC/B,QAAI,eAAe,KAAK,MAAM,GAAG;AAC/B,kBAAY,KAAK,OAAO;AACxB,mBAAa,KAAK,OAAO;AAAA,IAC3B;AAEA,QAAI,CAAC,kBAAkB,UAAU,GAAG;AAClC,WAAK,QAAQ;AAAA,QACX;AAAA,UACE,QAAQ,KAAK,SAAS;AAAA,UACtB,UAAU,KAAK,SAAS;AAAA,QAC1B;AAAA,QACA,eAAe,KAAK,SAAS,IAAI;AAAA,MACnC;AACA,aAAO,iBAAiB,OAAO;AAAA,QAC7B,UAAU,aAAa,OAAO,EAAE,GAAG,KAAK,SAAS,CAAC;AAAA,QAClD,gBAAgB;AAAA,MAClB,CAAC;AAAA,IACH;AAEA,WAAO,iBAAiB,OAAO;AAAA,MAC7B,UAAU,aAAa,OAAO,EAAE,GAAG,KAAK,SAAS,CAAC;AAAA,MAClD,gBAAgB,mBAAmB,OAAO;AAAA,QACxC,MAAM,KAAK,SAAS;AAAA,QACpB,QAAQ,KAAK,SAAS;AAAA,QACtB,QAAQ,eAAe,SAAY,KAAK,UAAU,UAAU,IAAI;AAAA;AAAA,QAChE,SAAS;AAAA,MACX,CAAC;AAAA,MACD,eAAe,eAAe;AAAA;AAAA,MAC9B;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEO,SAAS,iBAAiB,QAIT;AACtB,QAAM,EAAE,UAAU,QAAQ,UAAU,IAAI;AACxC,QAAM,SAAS,IAAI;AAGnB,MAAI,cAAc;AAClB,MAAI,iBAAiB;AACrB,MAAI,kBAAkB,OAAO;AAC3B,qBAAiB;AACjB,kBAAc;AAAA,EAChB;AAEA,MAAI,YAAY,cAAc,GAAG;AAC/B,WAAO,oBAAoB,OAAO;AAAA,MAChC,UAAU,aAAa,OAAO,EAAE,GAAG,SAAS,CAAC;AAAA,MAC7C,gBAAgB,mBAAmB,OAAO;AAAA,QACxC,MAAM,SAAS;AAAA,QACf,QAAQ,SAAS;AAAA,QACjB,QAAQ,eAAe;AAAA,QACvB,SAAS;AAAA,MACX,CAAC;AAAA,MACD,WAAW;AAAA,MACX,cAAc;AAAA,IAChB,CAAC;AAAA,EACH;AAEA,MAAI,eAAe,cAAc,GAAG;AAClC,WAAO,oBAAoB,OAAO;AAAA,MAChC,UAAU,aAAa,OAAO,EAAE,GAAG,SAAS,CAAC;AAAA,MAC7C,WAAW;AAAA,MACX,cAAc;AAAA,IAChB,CAAC;AAAA,EACH;AAEA,MAAI,mBAAmB,QAAW;AAChC,WAAO,oBAAoB,OAAO;AAAA,MAChC,UAAU,aAAa,OAAO,EAAE,GAAG,SAAS,CAAC;AAAA,MAC7C,gBAAgB,mBAAmB,OAAO;AAAA,QACxC,MAAM,SAAS;AAAA,QACf,QAAQ,SAAS;AAAA,QACjB,QAAQ;AAAA;AAAA,QACR,SAAS;AAAA,MACX,CAAC;AAAA,MACD,WAAW;AAAA,MACX,cAAc;AAAA,IAChB,CAAC;AAAA,EACH;AAEA,MAAI,YAA+B;AACnC,MAAI,aAAsB;AAC1B,MAAI,eAAe,WAAW,GAAG;AAC/B,gBAAY,YAAY;AACxB,iBAAa,YAAY;AAAA,EAC3B;AAEA,MAAI,CAAC,kBAAkB,UAAU,GAAG;AAClC,WAAO;AAAA,MACL;AAAA,QACE,QAAQ,SAAS;AAAA,QACjB,QAAQ;AAAA,MACV;AAAA,MACA,eAAe,SAAS,IAAI;AAAA,IAC9B;AACA,WAAO,oBAAoB,OAAO;AAAA,MAChC,UAAU,aAAa,OAAO,EAAE,GAAG,SAAS,CAAC;AAAA,MAC7C,WAAW;AAAA,MACX,cAAc;AAAA,IAChB,CAAC;AAAA,EACH;AAEA,SAAO,oBAAoB,OAAO;AAAA,IAChC,UAAU,aAAa,OAAO,EAAE,GAAG,SAAS,CAAC;AAAA,IAC7C,gBAAgB,mBAAmB,OAAO;AAAA,MACxC,MAAM,SAAS;AAAA,MACf,QAAQ,SAAS;AAAA,MACjB,QAAQ,eAAe,SAAY,KAAK,UAAU,UAAU,IAAI;AAAA;AAAA,MAChE,SAAS;AAAA,IACX,CAAC;AAAA,IACD,eAAe,eAAe;AAAA;AAAA,IAC9B;AAAA,IACA,WAAW;AAAA,IACX,cAAc;AAAA,EAChB,CAAC;AACH;AAEA,MAAM,0BAA0B;AAazB,SAAS,mBAAmB,SAIhC;AACD,QAAM,EAAE,SAAS,cAAc,aAAa,IAAI;AAEhD,QAAM,MAAM,QAAQ,UAAU,uBAAuB;AACrD,MAAI,QAAQ,QAAW;AACrB,QAAI,QAAQ,MAAM,GAAG,EAAG,SAAS,WAAW;AAE1C,cAAQ,MAAM,GAAG,IAAI,YAAY,OAAO;AAAA,QACtC,IAAI;AAAA,QACJ,MAAM;AAAA,QACN,SAAS,CAAC,YAAY;AAAA,QACtB,WAAW,QAAQ,MAAM,GAAG,EAAG;AAAA,MACjC,CAAC;AAAA,IACH,OAAO;AACL,YAAM,IAAI,MAAM,sEAAsE;AAAA,IACxF;AAAA,EACF,WAAW,cAAc;AAEvB,YAAQ,MAAM;AAAA,MACZ,YAAY,OAAO;AAAA,QACjB,IAAI;AAAA,QACJ,MAAM;AAAA,QACN,SAAS,CAAC,YAAY;AAAA,MACxB,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEO,SAAS,oBACd,MACA,SACA,SACA,eACA,YACkC;AAClC,QAAM,aAAa,IAAI,kBAA0B;AACjD,QAAM,iBAAiB,IAAI,kBAAgC;AAE3D,QAAM,aAAa,WAAW,SAAS,UAAU;AACjD,QAAM,iBAAiB,eAAe,SAAS,UAAU;AACzD,QAAM,OAAO,IAAI,mBAAmB,WAAW,UAAU,eAAe,QAAQ;AAEhF,QAAM,2BAA2B,OAAO,QAAqB,SAAe;AAC1E,SAAK;AAAA,MACH,WAAW;AAAA,MACX,KAAK,UAAU,QAAQ,OAAO,EAAE,kBAAkB,MAAM,CAAC,CAAC;AAAA,IAC5D;AACA,SAAK,aAAa,WAAW,qBAAqB,KAAK,UAAU,OAAO,KAAK,OAAO,CAAC,CAAC;AAEtF,QAAI,kBAA0E;AAC9E,QAAI,YAAuD;AAE3D,QAAI;AACF,kBAAY,MAAM,KAAK,SAAS,SAAS,aAAa;AACtD,UAAI,cAAc,MAAM;AACtB,cAAM,WAAW,MAAM;AACvB;AAAA,MACF;AAEA,YAAM,eAAe,aAAa,MAAM;AAIxC,wBAAkB,UAAU,UAAU;AACtC,aAAO,MAAM;AACX,YAAI,OAAO,QAAS;AAEpB,cAAM,SAAS,MAAM,QAAQ,KAAK,CAAC,gBAAgB,KAAK,GAAG,YAAY,CAAC;AACxE,YAAI,WAAW,OAAW;AAE1B,cAAM,EAAE,MAAM,OAAO,MAAM,IAAI;AAC/B,YAAI,KAAM;AAEV,YAAI,OAAO,UAAU,UAAU;AAC7B,eAAK,iBAAiB;AACtB,gBAAM,WAAW,MAAM,KAAK;AAAA,QAE9B,OAAO;AACL,cAAI,MAAM,UAAU,QAAW;AAC7B;AAAA,UACF;AAEA,cAAI,MAAM,MAAM,WAAW;AACzB,uBAAW,QAAQ,MAAM,MAAM,WAAW;AACxC,kBAAI,KAAK,SAAS,gBAAiB;AAEnC,oBAAM,WAAW,aAAa,OAAO;AAAA,gBACnC,QAAQ,GAAG,KAAK,EAAE,QAAQ,KAAK,mBAAmB,MAAM;AAAA,gBACxD,MAAM,KAAK;AAAA,gBACX,MAAM,KAAK;AAAA;AAAA,gBAEX,kBAAkB,KAAK;AAAA,gBACvB,OAAO,KAAK,SAAS,CAAC;AAAA,cACxB,CAAC;AAED,mBAAK,mBAAmB,KAAK,QAAQ;AACrC,oBAAM,eAAe,MAAM,QAAQ;AAAA,YACrC;AAAA,UACF;AAEA,cAAI,MAAM,MAAM,SAAS;AACvB,iBAAK,iBAAiB,MAAM,MAAM;AAClC,kBAAM,WAAW,MAAM,MAAM,MAAM,OAAO;AAAA,UAC5C;AAAA,QACF;AAAA,MAIF;AAEA,WAAK,aAAa,WAAW,oBAAoB,KAAK,aAAa;AAAA,IACrE,SAAS,OAAO;AACd,UAAI,iBAAiB,gBAAgB,MAAM,SAAS,cAAc;AAEhE;AAAA,MACF;AACA,YAAM;AAAA,IACR,UAAE;AACA,yDAAiB;AACjB,aAAM,uCAAW;AACjB,YAAM,WAAW,MAAM;AACvB,YAAM,eAAe,MAAM;AAAA,IAC7B;AAAA,EACF;AAGA,QAAM,iBAAiB,YAAY,OAAO;AAE1C,QAAM,gBAAgB,OAAO,WAC3B,OAAO,gBAAgB,OAAO,SAAS,yBAAyB,QAAQ,IAAI,GAAG;AAAA,IAC7E,MAAM;AAAA,IACN,SAAS;AAAA,EACX,CAAC;AAEH,SAAO;AAAA,IACL,KAAK,KAAK,CAACA,gBAAe,cAAcA,YAAW,MAAM,GAAG,YAAY,qBAAqB;AAAA,IAC7F;AAAA,EACF;AACF;AAEO,SAAS,oBACd,MACA,MACA,eACA,YAC0C;AAC1C,QAAM,cAAc,IAAI,kBAA8B;AACtD,QAAM,eAAe,YAAY,SAAS,UAAU;AACpD,QAAM,oBAAoB,YAAY;AAEtC,QAAM,2BAA2B,OAAO,WAAwB;AAC9D,QAAI,kBAAkE;AACtE,QAAI,YAA+C;AAEnD,QAAI;AACF,kBAAY,MAAM,KAAK,MAAM,aAAa;AAC1C,UAAI,cAAc,MAAM;AACtB,cAAM,aAAa,MAAM;AACzB;AAAA,MACF;AAEA,wBAAkB,UAAU,UAAU;AACtC,aAAO,MAAM;AACX,YAAI,OAAO,SAAS;AAClB;AAAA,QACF;AACA,cAAM,EAAE,MAAM,OAAO,MAAM,IAAI,MAAM,gBAAgB,KAAK;AAC1D,YAAI,MAAM;AACR;AAAA,QACF;AACA,cAAM,aAAa,MAAM,KAAK;AAAA,MAChC;AAAA,IACF,SAAS,OAAO;AACd,UAAI,iBAAiB,gBAAgB,MAAM,SAAS,cAAc;AAEhE;AAAA,MACF;AACA,YAAM;AAAA,IACR,UAAE;AACA,yDAAiB;AACjB,aAAM,uCAAW;AACjB,YAAM,aAAa,MAAM;AAAA,IAC3B;AAAA,EACF;AAGA,QAAM,iBAAiB,YAAY,OAAO;AAE1C,QAAM,gBAAgB,OAAO,WAC3B,OAAO,gBAAgB,YAAY,yBAAyB,MAAM,GAAG;AAAA,IACnE,MAAM;AAAA,IACN,SAAS;AAAA,EACX,CAAC;AAEH,SAAO;AAAA,IACL,KAAK,KAAK,CAACA,gBAAe,cAAcA,YAAW,MAAM,GAAG,YAAY,qBAAqB;AAAA,IAC7F;AAAA,EACF;AACF;AAOA,eAAe,YACb,QACA,KACA,QACA,YACe;AACf,QAAM,SAAS,OAAO,UAAU;AAChC,MAAI;AACF,WAAO,MAAM;AACX,UAAI,OAAO,SAAS;AAClB;AAAA,MACF;AACA,YAAM,EAAE,MAAM,OAAO,MAAM,IAAI,MAAM,OAAO,KAAK;AACjD,UAAI,KAAM;AACV,UAAI,QAAQ;AACZ,UAAI,eAAe,MAAM;AACvB,cAAM,WAAW,YAAY,KAAK;AAAA,MACpC;AACA,UAAI,CAAC,IAAI,aAAa,MAAM;AAC1B,YAAI,aAAa,QAAQ;AAAA,MAC3B;AAAA,IACF;AAAA,EACF,UAAE;AACA,QAAI,eAAe,MAAM;AACvB,iBAAW,MAAM;AAAA,IACnB;AACA,qCAAQ;AAAA,EACV;AACF;AAEO,SAAS,sBACd,QACA,YACA,YACwB;AACxB,QAAM,MAAM;AAAA,IACV,MAAM;AAAA,IACN,cAAc,IAAI,OAAO;AAAA,EAC3B;AACA,SAAO;AAAA,IACL,KAAK;AAAA,MACH,CAACA,gBAAe,YAAY,QAAQ,KAAKA,YAAW,QAAQ,UAAU;AAAA,MACtE;AAAA,MACA;AAAA,IACF;AAAA,IACA;AAAA,EACF;AACF;AAOA,eAAe,aACb,WACA,YACA,KACA,QACe;AACf,QAAM,SAAS,UAAU,UAAU;AACnC,MAAI,YAAmC;AAEvC,MAAI;AACF,WAAO,MAAM;AACX,UAAI,iCAAQ,SAAS;AACnB;AAAA,MACF;AAEA,YAAM,EAAE,MAAM,OAAO,MAAM,IAAI,MAAM,OAAO,KAAK;AACjD,UAAI,KAAM;AAEV,UAAI,MAAM,KAAK,KAAK;AAEpB,UACE,CAAC,IAAI,cAAc,QACnB,WAAW,cACX,WAAW,eAAe,MAAM,cAChC,CAAC,WACD;AACA,oBAAY,IAAI,eAAe,MAAM,YAAY,WAAW,YAAY,CAAC;AAAA,MAC3E;AAEA,UAAI,WAAW;AACb,mBAAW,KAAK,UAAU,KAAK,KAAK,GAAG;AACrC,gBAAM,WAAW,aAAa,CAAC;AAAA,QACjC;AAAA,MACF,OAAO;AACL,cAAM,WAAW,aAAa,KAAK;AAAA,MACrC;AAIA,UAAI,CAAC,IAAI,cAAc,MAAM;AAC3B,YAAI,cAAc,QAAQ;AAAA,MAC5B;AAAA,IACF;AAAA,EACF,UAAE;AACA,qCAAQ;AACR,QAAI,WAAW;AACb,iBAAW,KAAK,UAAU,MAAM,GAAG;AACjC,cAAM,WAAW,aAAa,CAAC;AAAA,MACjC;AAAA,IACF;AACA,eAAW,MAAM;AAAA,EACnB;AACF;AAEO,SAAS,uBACd,WACA,aACA,YACyB;AACzB,QAAM,MAAM;AAAA,IACV,OAAO,CAAC;AAAA,IACR,eAAe,IAAI,OAAO;AAAA,EAC5B;AACA,SAAO;AAAA,IACL,KAAK;AAAA,MACH,CAACA,gBAAe,aAAa,WAAW,aAAa,KAAKA,YAAW,MAAM;AAAA,MAC3E;AAAA,MACA;AAAA,IACF;AAAA,IACA;AAAA,EACF;AACF;AAGO,SAAS,sBAAsB;AAAA,EACpC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,yBAAyB,MAAM;AAAA,EAAC;AAAA,EAChC,2BAA2B,MAAM;AAAA,EAAC;AAAA,EAClC;AACF,GAS6B;AAC3B,QAAM,SAAS,IAAI;AACnB,QAAM,aAAyB;AAAA,IAC7B,QAAQ,CAAC;AAAA,IACT,wBAAwB,IAAI,OAAO;AAAA,EACrC;AAEA,QAAM,gBAAgB,CAAC,QAA6B;AAClD,6BAAyB,GAAG;AAC5B,eAAW,OAAO,KAAK,GAAG;AAAA,EAC5B;AAEA,QAAM,mBAAmB,OAAOA,gBAAgC;AAC9D,UAAM,SAASA,YAAW;AAC1B,UAAM,SAAS,eAAe,UAAU;AAExC,UAAM,QAAwB,CAAC;AAC/B,WAAO,CAAC,OAAO,SAAS;AACtB,YAAM,EAAE,MAAM,OAAO,SAAS,IAAI,MAAM,OAAO,KAAK;AACpD,UAAI,OAAO,QAAS;AACpB,UAAI,KAAM;AAEV,UAAI,eAAe,QAAQ;AACzB,eAAO;AAAA,UACL;AAAA,YACE,UAAU,SAAS;AAAA,YACnB,WAAW,aAAa;AAAA,UAC1B;AAAA,UACA;AAAA,QACF;AACA;AAAA,MACF;AAIA,YAAM,OAAO,QAAQ,SAAS,IAAI;AAClC,UAAI,CAAC,MAAM;AACT,eAAO;AAAA,UACL;AAAA,YACE,UAAU,SAAS;AAAA,YACnB,WAAW,aAAa;AAAA,UAC1B;AAAA,UACA,uBAAuB,SAAS,IAAI;AAAA,QACtC;AACA;AAAA,MACF;AAEA,UAAI,CAAC,eAAe,IAAI,GAAG;AACzB,eAAO;AAAA,UACL;AAAA,YACE,UAAU,SAAS;AAAA,YACnB,WAAW,aAAa;AAAA,UAC1B;AAAA,UACA,sBAAsB,OAAO,IAAI;AAAA,QACnC;AACA;AAAA,MACF;AAEA,UAAI;AAGJ,UAAI;AACF,cAAM,WAAW,KAAK,MAAM,SAAS,IAAI;AAEzC,YAAI,YAAY,KAAK,UAAU,GAAG;AAChC,gBAAM,SAAS,MAAM,eAAuB,KAAK,YAAY,QAAQ;AACrE,cAAI,OAAO,SAAS;AAClB,yBAAa,OAAO;AAAA,UACtB,OAAO;AACL,kBAAM,OAAO;AAAA,UACf;AAAA,QACF,OAAO;AACL,uBAAa;AAAA,QACf;AAAA,MACF,SAAS,UAAU;AACjB,cAAM,QAAQ,QAAQ,QAAQ;AAC9B,eAAO;AAAA,UACL;AAAA,YACE,UAAU,SAAS;AAAA,YACnB,WAAW,SAAS;AAAA,YACpB,WAAW,aAAa;AAAA,YACxB,OAAO,MAAM;AAAA,UACf;AAAA,UACA,6BAA6B,SAAS,IAAI;AAAA,QAC5C;AACA;AAAA,UACE,iBAAiB;AAAA,YACf;AAAA,YACA,WAAW;AAAA,UACb,CAAC;AAAA,QACH;AACA;AAAA,MACF;AAEA,UAAI,CAAC,WAAW,uBAAuB,MAAM;AAC3C,mBAAW,uBAAuB,QAAQ;AAAA,MAC5C;AAEA,6BAAuB,QAAQ;AAE/B,aAAO;AAAA,QACL;AAAA,UACE,UAAU,SAAS;AAAA,UACnB,WAAW;AAAA,UACX,WAAW,aAAa;AAAA,QAC1B;AAAA,QACA;AAAA,MACF;AAEA,YAAM,gBAAgB,kBAAkB,IAAI,EAAE,cAAc,SAAS,GAAG,YAAY;AAClF,eAAO,MAAM,KAAK,QAAQ,YAAY;AAAA,UACpC,KAAK,IAAI,WAAW,SAAS,cAAc,QAAQ;AAAA,UACnD,YAAY,SAAS;AAAA,UACrB,aAAa;AAAA,QACf,CAAC;AAAA,MACH,CAAC;AAED,YAAM,6BAA6B,OAAO,cAAgC,SAAe;AACvF,aAAK,aAAa,WAAW,yBAAyB,SAAS,IAAI;AACnE,aAAK,aAAa,WAAW,yBAAyB,SAAS,IAAI;AAGnE,YAAIC;AACJ,YAAI;AACF,gBAAM,EAAE,QAAQ,UAAU,IAAI,MAAM,iBAAiB,cAAc,MAAM;AACzE,UAAAA,cAAa,iBAAiB;AAAA,YAC5B;AAAA,YACA,WAAW,YAAY,IAAI,MAAM,uBAAuB,IAAI;AAAA,YAC5D,QAAQ,YAAY,SAAY;AAAA,UAClC,CAAC;AAED,cAAIA,YAAW,gBAAgB;AAC7B,iBAAK;AAAA,cACH,WAAW;AAAA,cACXA,YAAW,eAAe;AAAA,YAC5B;AACA,iBAAK;AAAA,cACH,WAAW;AAAA,cACXA,YAAW,eAAe;AAAA,YAC5B;AAAA,UACF;AAAA,QACF,SAAS,UAAU;AACjB,iBAAO;AAAA,YACL;AAAA,cACE,UAAU,SAAS;AAAA,cACnB,WAAW,aAAa;AAAA,cACxB,OAAO,QAAQ,QAAQ,EAAE;AAAA,YAC3B;AAAA,YACA;AAAA,UACF;AACA,UAAAA,cAAa,iBAAiB;AAAA,YAC5B;AAAA,YACA,WAAW,QAAQ,QAAQ;AAAA,UAC7B,CAAC;AAED,cAAIA,YAAW,gBAAgB;AAC7B,iBAAK;AAAA,cACH,WAAW;AAAA,cACXA,YAAW,eAAe;AAAA,YAC5B;AACA,iBAAK,aAAa,WAAW,6BAA6B,IAAI;AAAA,UAChE;AAAA,QACF,UAAE;AACA,cAAI,CAACA,YAAY,OAAM,IAAI,MAAM,yBAAyB;AAC1D,wBAAcA,WAAU;AAAA,QAC1B;AAAA,MACF;AAEA,YAAM,wBAAwB,CAAC,iBAC7B,OAAO,gBAAgB,OAAO,SAAS,2BAA2B,cAAc,IAAI,GAAG;AAAA,QACrF,MAAM;AAAA,MACR,CAAC;AAGH,YAAM,KAAK,sBAAsB,aAAa,CAAC;AAAA,IACjD;AAEA,UAAM,QAAQ,WAAW,KAAK;AAC9B,QAAI,WAAW,OAAO,SAAS,GAAG;AAChC,aAAO;AAAA,QACL;AAAA,UACE,WAAW,aAAa;AAAA,QAC1B;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,SAAO,CAAC,KAAK,KAAK,kBAAkB,YAAY,uBAAuB,GAAG,UAAU;AACtF;AAYA,eAAe,iBAAoB,SAAqB,QAA0C;AAChG,QAAM,WAAW,IAAI,OAAmB;AAExC,QAAM,eAAe,MAAM;AACzB,QAAI,CAAC,SAAS,MAAM;AAClB,eAAS,QAAQ,EAAE,QAAQ,QAAW,WAAW,KAAK,CAAC;AAAA,IACzD;AAAA,EACF;AAEA,SAAO,iBAAiB,SAAS,YAAY;AAE7C,UACG,KAAK,CAAC,MAAM;AACX,QAAI,CAAC,SAAS,MAAM;AAClB,eAAS,QAAQ,EAAE,QAAQ,GAAG,WAAW,MAAM,CAAC;AAAA,IAClD;AAAA,EACF,CAAC,EACA,MAAM,CAAC,MAAM;AACZ,QAAI,CAAC,SAAS,MAAM;AAClB,eAAS,OAAO,CAAC;AAAA,IACnB;AAAA,EACF,CAAC,EACA,QAAQ,MAAM;AACb,WAAO,oBAAoB,SAAS,YAAY;AAAA,EAClD,CAAC;AAEH,SAAO,MAAM,SAAS;AACxB;AAEO,SAAS,mBAAmB,SAAsB;AAEvD,SAAO,MAAM;AACX,UAAM,MAAM,QAAQ,UAAU,uBAAuB;AACrD,QAAI,QAAQ,QAAW;AACrB,cAAQ,MAAM,OAAO,KAAK,CAAC;AAAA,IAC7B,OAAO;AACL;AAAA,IACF;AAAA,EACF;AACF;","names":["controller","toolOutput"]}

package/dist/voice/generation_tools.test.cjs

@@ -0,0 +1,236 @@
+"use strict";
+var import_web = require("stream/web");
+var import_vitest = require("vitest");
+var import_zod = require("zod");
+var import_llm = require("../llm/index.cjs");
+var import_log = require("../log.cjs");
+var import_utils = require("../utils.cjs");
+var import_generation = require("./generation.cjs");
+function createStringStream(chunks, delayMs = 0) {
+  return new import_web.ReadableStream({
+    async start(controller) {
+      for (const c of chunks) {
+        if (delayMs > 0) {
+          await (0, import_utils.delay)(delayMs);
+        }
+        controller.enqueue(c);
+      }
+      controller.close();
+    }
+  });
+}
+function createFunctionCallStream(fc) {
+  return new import_web.ReadableStream({
+    start(controller) {
+      controller.enqueue(fc);
+      controller.close();
+    }
+  });
+}
+function createFunctionCallStreamFromArray(fcs) {
+  return new import_web.ReadableStream({
+    start(controller) {
+      for (const fc of fcs) {
+        controller.enqueue(fc);
+      }
+      controller.close();
+    }
+  });
+}
+(0, import_vitest.describe)("Generation + Tool Execution", () => {
+  (0, import_log.initializeLogger)({ pretty: false, level: "silent" });
+  (0, import_vitest.it)("should not abort tool when preamble forwarders are cleaned up", async () => {
+    var _a, _b;
+    const replyAbortController = new AbortController();
+    const forwarderController = new AbortController();
+    const chunks = Array.from({ length: 50 }, () => `Hi.`);
+    const fullPreambleText = chunks.join("");
+    const preamble = createStringStream(chunks, 20);
+    const [textForwardTask, textOut] = (0, import_generation.performTextForwarding)(
+      preamble,
+      forwarderController,
+      null
+    );
+    let toolAborted = false;
+    const getWeather = (0, import_llm.tool)({
+      description: "weather",
+      parameters: import_zod.z.object({ location: import_zod.z.string() }),
+      execute: async ({ location }, { abortSignal }) => {
+        if (abortSignal) {
+          abortSignal.addEventListener("abort", () => {
+            toolAborted = true;
+          });
+        }
+        await (0, import_utils.delay)(6e3);
+        return `Sunny in ${location}`;
+      }
+    });
+    const fc = import_llm.FunctionCall.create({
+      callId: "call_1",
+      name: "getWeather",
+      args: JSON.stringify({ location: "San Francisco" })
+    });
+    const toolCallStream = createFunctionCallStream(fc);
+    const [execTask, toolOutput] = (0, import_generation.performToolExecutions)({
+      session: {},
+      speechHandle: { id: "speech_test", _itemAdded: () => {
+      } },
+      toolCtx: { getWeather },
+      toolCallStream,
+      controller: replyAbortController,
+      onToolExecutionStarted: () => {
+      },
+      onToolExecutionCompleted: () => {
+      }
+    });
+    await toolOutput.firstToolStartedFuture.await;
+    await (0, import_utils.delay)(100);
+    await (0, import_utils.cancelAndWait)([textForwardTask], 5e3);
+    await execTask.result;
+    (0, import_vitest.expect)(toolOutput.output.length).toBe(1);
+    const out = toolOutput.output[0];
+    (0, import_vitest.expect)((_a = out.toolCallOutput) == null ? void 0 : _a.isError).toBe(false);
+    (0, import_vitest.expect)((_b = out.toolCallOutput) == null ? void 0 : _b.output).toContain("Sunny in San Francisco");
+    (0, import_vitest.expect)(textOut.text).not.toBe(fullPreambleText);
+    (0, import_vitest.expect)(toolAborted).toBe(false);
+  }, 3e4);
+  (0, import_vitest.it)("should return basic tool execution output", async () => {
+    var _a, _b;
+    const replyAbortController = new AbortController();
+    const echo = (0, import_llm.tool)({
+      description: "echo",
+      parameters: import_zod.z.object({ msg: import_zod.z.string() }),
+      execute: async ({ msg }) => `echo: ${msg}`
+    });
+    const fc = import_llm.FunctionCall.create({
+      callId: "call_2",
+      name: "echo",
+      args: JSON.stringify({ msg: "hello" })
+    });
+    const toolCallStream = createFunctionCallStream(fc);
+    const [execTask, toolOutput] = (0, import_generation.performToolExecutions)({
+      session: {},
+      speechHandle: { id: "speech_test2", _itemAdded: () => {
+      } },
+      toolCtx: { echo },
+      toolCallStream,
+      controller: replyAbortController
+    });
+    await execTask.result;
+    (0, import_vitest.expect)(toolOutput.output.length).toBe(1);
+    const out = toolOutput.output[0];
+    (0, import_vitest.expect)((_a = out == null ? void 0 : out.toolCallOutput) == null ? void 0 : _a.isError).toBe(false);
+    (0, import_vitest.expect)((_b = out == null ? void 0 : out.toolCallOutput) == null ? void 0 : _b.output).toContain("echo: hello");
+  });
+  (0, import_vitest.it)("should abort tool when reply is aborted mid-execution", async () => {
+    var _a;
+    const replyAbortController = new AbortController();
+    let aborted = false;
+    const longOp = (0, import_llm.tool)({
+      description: "longOp",
+      parameters: import_zod.z.object({ ms: import_zod.z.number() }),
+      execute: async ({ ms }, { abortSignal }) => {
+        if (abortSignal) {
+          abortSignal.addEventListener("abort", () => {
+            aborted = true;
+          });
+        }
+        await (0, import_utils.delay)(ms);
+        return "done";
+      }
+    });
+    const fc = import_llm.FunctionCall.create({
+      callId: "call_abort_1",
+      name: "longOp",
+      args: JSON.stringify({ ms: 5e3 })
+    });
+    const toolCallStream = createFunctionCallStream(fc);
+    const [execTask, toolOutput] = (0, import_generation.performToolExecutions)({
+      session: {},
+      speechHandle: { id: "speech_abort", _itemAdded: () => {
+      } },
+      toolCtx: { longOp },
+      toolCallStream,
+      controller: replyAbortController
+    });
+    await toolOutput.firstToolStartedFuture.await;
+    replyAbortController.abort();
+    await execTask.result;
+    (0, import_vitest.expect)(aborted).toBe(true);
+    (0, import_vitest.expect)(toolOutput.output.length).toBe(1);
+    const out = toolOutput.output[0];
+    (0, import_vitest.expect)((_a = out == null ? void 0 : out.toolCallOutput) == null ? void 0 : _a.isError).toBe(true);
+  }, 2e4);
+  (0, import_vitest.it)("should return error output on invalid tool args (zod validation failure)", async () => {
+    var _a;
+    const replyAbortController = new AbortController();
+    const echo = (0, import_llm.tool)({
+      description: "echo",
+      parameters: import_zod.z.object({ msg: import_zod.z.string() }),
+      execute: async ({ msg }) => `echo: ${msg}`
+    });
+    const fc = import_llm.FunctionCall.create({
+      callId: "call_invalid_args",
+      name: "echo",
+      args: JSON.stringify({ msg: 123 })
+    });
+    const toolCallStream = createFunctionCallStream(fc);
+    const [execTask, toolOutput] = (0, import_generation.performToolExecutions)({
+      session: {},
+      speechHandle: { id: "speech_invalid", _itemAdded: () => {
+      } },
+      toolCtx: { echo },
+      toolCallStream,
+      controller: replyAbortController
+    });
+    await execTask.result;
+    (0, import_vitest.expect)(toolOutput.output.length).toBe(1);
+    const out = toolOutput.output[0];
+    (0, import_vitest.expect)((_a = out == null ? void 0 : out.toolCallOutput) == null ? void 0 : _a.isError).toBe(true);
+  });
+  (0, import_vitest.it)("should handle multiple tool calls within a single stream", async () => {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    const replyAbortController = new AbortController();
+    const sum = (0, import_llm.tool)({
+      description: "sum",
+      parameters: import_zod.z.object({ a: import_zod.z.number(), b: import_zod.z.number() }),
+      execute: async ({ a, b }) => a + b
+    });
+    const upper = (0, import_llm.tool)({
+      description: "upper",
+      parameters: import_zod.z.object({ s: import_zod.z.string() }),
+      execute: async ({ s }) => s.toUpperCase()
+    });
+    const fc1 = import_llm.FunctionCall.create({
+      callId: "call_multi_1",
+      name: "sum",
+      args: JSON.stringify({ a: 2, b: 3 })
+    });
+    const fc2 = import_llm.FunctionCall.create({
+      callId: "call_multi_2",
+      name: "upper",
+      args: JSON.stringify({ s: "hey" })
+    });
+    const toolCallStream = createFunctionCallStreamFromArray([fc1, fc2]);
+    const [execTask, toolOutput] = (0, import_generation.performToolExecutions)({
+      session: {},
+      speechHandle: { id: "speech_multi", _itemAdded: () => {
+      } },
+      toolCtx: { sum, upper },
+      toolCallStream,
+      controller: replyAbortController
+    });
+    await execTask.result;
+    (0, import_vitest.expect)(toolOutput.output.length).toBe(2);
+    const sorted = [...toolOutput.output].sort(
+      (a, b) => a.toolCall.callId.localeCompare(b.toolCall.callId)
+    );
+    (0, import_vitest.expect)((_a = sorted[0]) == null ? void 0 : _a.toolCall.name).toBe("sum");
+    (0, import_vitest.expect)((_c = (_b = sorted[0]) == null ? void 0 : _b.toolCallOutput) == null ? void 0 : _c.isError).toBe(false);
+    (0, import_vitest.expect)((_e = (_d = sorted[0]) == null ? void 0 : _d.toolCallOutput) == null ? void 0 : _e.output).toBe("5");
+    (0, import_vitest.expect)((_f = sorted[1]) == null ? void 0 : _f.toolCall.name).toBe("upper");
+    (0, import_vitest.expect)((_h = (_g = sorted[1]) == null ? void 0 : _g.toolCallOutput) == null ? void 0 : _h.isError).toBe(false);
+    (0, import_vitest.expect)((_j = (_i = sorted[1]) == null ? void 0 : _i.toolCallOutput) == null ? void 0 : _j.output).toBe('"HEY"');
+  });
+});
+//# sourceMappingURL=generation_tools.test.cjs.map

package/dist/voice/generation_tools.test.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/voice/generation_tools.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2025 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { ReadableStream as NodeReadableStream } from 'stream/web';\nimport { describe, expect, it } from 'vitest';\nimport { z } from 'zod';\nimport { FunctionCall, tool } from '../llm/index.js';\nimport { initializeLogger } from '../log.js';\nimport type { Task } from '../utils.js';\nimport { cancelAndWait, delay } from '../utils.js';\nimport { type _TextOut, performTextForwarding, performToolExecutions } from './generation.js';\n\nfunction createStringStream(chunks: string[], delayMs: number = 0): NodeReadableStream<string> {\n  return new NodeReadableStream<string>({\n    async start(controller) {\n      for (const c of chunks) {\n        if (delayMs > 0) {\n          await delay(delayMs);\n        }\n        controller.enqueue(c);\n      }\n      controller.close();\n    },\n  });\n}\n\nfunction createFunctionCallStream(fc: FunctionCall): NodeReadableStream<FunctionCall> {\n  return new NodeReadableStream<FunctionCall>({\n    start(controller) {\n      controller.enqueue(fc);\n      controller.close();\n    },\n  });\n}\n\nfunction createFunctionCallStreamFromArray(fcs: FunctionCall[]): NodeReadableStream<FunctionCall> {\n  return new NodeReadableStream<FunctionCall>({\n    start(controller) {\n      for (const fc of fcs) {\n        controller.enqueue(fc);\n      }\n      controller.close();\n    },\n  });\n}\n\ndescribe('Generation + Tool Execution', () => {\n  initializeLogger({ pretty: false, level: 'silent' });\n\n  it('should not abort tool when preamble forwarders are cleaned up', async () => {\n    const replyAbortController = new AbortController();\n    const forwarderController = new AbortController();\n\n    const chunks = Array.from({ length: 50 }, () => `Hi.`);\n    const fullPreambleText = chunks.join('');\n    const preamble = createStringStream(chunks, 20);\n    const [textForwardTask, textOut]: [Task<void>, _TextOut] = performTextForwarding(\n      preamble,\n      forwarderController,\n      null,\n    );\n\n    // Tool that takes > 5 seconds\n    let toolAborted = false;\n    const getWeather = tool({\n      description: 'weather',\n      parameters: z.object({ location: z.string() }),\n      execute: async ({ location }, { abortSignal }) => {\n        if (abortSignal) {\n          abortSignal.addEventListener('abort', () => {\n            toolAborted = true;\n          });\n        }\n        // 6s delay\n        await delay(6000);\n        return `Sunny in ${location}`;\n      },\n    });\n\n    const fc = FunctionCall.create({\n      callId: 'call_1',\n      name: 'getWeather',\n      args: JSON.stringify({ location: 'San Francisco' }),\n    });\n    const toolCallStream = createFunctionCallStream(fc);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_test', _itemAdded: () => {} } as any,\n      toolCtx: { getWeather } as any,\n      toolCallStream,\n      controller: replyAbortController,\n      onToolExecutionStarted: () => {},\n      onToolExecutionCompleted: () => {},\n    });\n\n    // Ensure tool has started, then cancel forwarders mid-stream (without aborting parent AbortController)\n    await toolOutput.firstToolStartedFuture.await;\n    await delay(100);\n    await cancelAndWait([textForwardTask], 5000);\n\n    await execTask.result;\n\n    expect(toolOutput.output.length).toBe(1);\n    const out = toolOutput.output[0]!;\n    expect(out.toolCallOutput?.isError).toBe(false);\n    expect(out.toolCallOutput?.output).toContain('Sunny in San Francisco');\n    // Forwarder should have been cancelled before finishing all preamble chunks\n    expect(textOut.text).not.toBe(fullPreambleText);\n    // Tool's abort signal must not have fired\n    expect(toolAborted).toBe(false);\n  }, 30_000);\n\n  it('should return basic tool execution output', async () => {\n    const replyAbortController = new AbortController();\n\n    const echo = tool({\n      description: 'echo',\n      parameters: z.object({ msg: z.string() }),\n      execute: async ({ msg }) => `echo: ${msg}`,\n    });\n\n    const fc = FunctionCall.create({\n      callId: 'call_2',\n      name: 'echo',\n      args: JSON.stringify({ msg: 'hello' }),\n    });\n    const toolCallStream = createFunctionCallStream(fc);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_test2', _itemAdded: () => {} } as any,\n      toolCtx: { echo } as any,\n      toolCallStream,\n      controller: replyAbortController,\n    });\n\n    await execTask.result;\n    expect(toolOutput.output.length).toBe(1);\n    const out = toolOutput.output[0];\n    expect(out?.toolCallOutput?.isError).toBe(false);\n    expect(out?.toolCallOutput?.output).toContain('echo: hello');\n  });\n\n  it('should abort tool when reply is aborted mid-execution', async () => {\n    const replyAbortController = new AbortController();\n\n    let aborted = false;\n    const longOp = tool({\n      description: 'longOp',\n      parameters: z.object({ ms: z.number() }),\n      execute: async ({ ms }, { abortSignal }) => {\n        if (abortSignal) {\n          abortSignal.addEventListener('abort', () => {\n            aborted = true;\n          });\n        }\n        await delay(ms);\n        return 'done';\n      },\n    });\n\n    const fc = FunctionCall.create({\n      callId: 'call_abort_1',\n      name: 'longOp',\n      args: JSON.stringify({ ms: 5000 }),\n    });\n    const toolCallStream = createFunctionCallStream(fc);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_abort', _itemAdded: () => {} } as any,\n      toolCtx: { longOp } as any,\n      toolCallStream,\n      controller: replyAbortController,\n    });\n\n    await toolOutput.firstToolStartedFuture.await;\n    replyAbortController.abort();\n    await execTask.result;\n\n    expect(aborted).toBe(true);\n    expect(toolOutput.output.length).toBe(1);\n    const out = toolOutput.output[0];\n    expect(out?.toolCallOutput?.isError).toBe(true);\n  }, 20_000);\n\n  it('should return error output on invalid tool args (zod validation failure)', async () => {\n    const replyAbortController = new AbortController();\n\n    const echo = tool({\n      description: 'echo',\n      parameters: z.object({ msg: z.string() }),\n      execute: async ({ msg }) => `echo: ${msg}`,\n    });\n\n    // invalid: msg should be string\n    const fc = FunctionCall.create({\n      callId: 'call_invalid_args',\n      name: 'echo',\n      args: JSON.stringify({ msg: 123 }),\n    });\n    const toolCallStream = createFunctionCallStream(fc);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_invalid', _itemAdded: () => {} } as any,\n      toolCtx: { echo } as any,\n      toolCallStream,\n      controller: replyAbortController,\n    });\n\n    await execTask.result;\n    expect(toolOutput.output.length).toBe(1);\n    const out = toolOutput.output[0];\n    expect(out?.toolCallOutput?.isError).toBe(true);\n  });\n\n  it('should handle multiple tool calls within a single stream', async () => {\n    const replyAbortController = new AbortController();\n\n    const sum = tool({\n      description: 'sum',\n      parameters: z.object({ a: z.number(), b: z.number() }),\n      execute: async ({ a, b }) => a + b,\n    });\n    const upper = tool({\n      description: 'upper',\n      parameters: z.object({ s: z.string() }),\n      execute: async ({ s }) => s.toUpperCase(),\n    });\n\n    const fc1 = FunctionCall.create({\n      callId: 'call_multi_1',\n      name: 'sum',\n      args: JSON.stringify({ a: 2, b: 3 }),\n    });\n    const fc2 = FunctionCall.create({\n      callId: 'call_multi_2',\n      name: 'upper',\n      args: JSON.stringify({ s: 'hey' }),\n    });\n    const toolCallStream = createFunctionCallStreamFromArray([fc1, fc2]);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_multi', _itemAdded: () => {} } as any,\n      toolCtx: { sum, upper } as any,\n      toolCallStream,\n      controller: replyAbortController,\n    });\n\n    await execTask.result;\n    expect(toolOutput.output.length).toBe(2);\n\n    // sort by callId to assert deterministically\n    const sorted = [...toolOutput.output].sort((a, b) =>\n      a.toolCall.callId.localeCompare(b.toolCall.callId),\n    );\n\n    expect(sorted[0]?.toolCall.name).toBe('sum');\n    expect(sorted[0]?.toolCallOutput?.isError).toBe(false);\n    expect(sorted[0]?.toolCallOutput?.output).toBe('5');\n    expect(sorted[1]?.toolCall.name).toBe('upper');\n    expect(sorted[1]?.toolCallOutput?.isError).toBe(false);\n    expect(sorted[1]?.toolCallOutput?.output).toBe('\"HEY\"');\n  });\n});\n"],"mappings":";AAGA,iBAAqD;AACrD,oBAAqC;AACrC,iBAAkB;AAClB,iBAAmC;AACnC,iBAAiC;AAEjC,mBAAqC;AACrC,wBAA4E;AAE5E,SAAS,mBAAmB,QAAkB,UAAkB,GAA+B;AAC7F,SAAO,IAAI,WAAAA,eAA2B;AAAA,IACpC,MAAM,MAAM,YAAY;AACtB,iBAAW,KAAK,QAAQ;AACtB,YAAI,UAAU,GAAG;AACf,oBAAM,oBAAM,OAAO;AAAA,QACrB;AACA,mBAAW,QAAQ,CAAC;AAAA,MACtB;AACA,iBAAW,MAAM;AAAA,IACnB;AAAA,EACF,CAAC;AACH;AAEA,SAAS,yBAAyB,IAAoD;AACpF,SAAO,IAAI,WAAAA,eAAiC;AAAA,IAC1C,MAAM,YAAY;AAChB,iBAAW,QAAQ,EAAE;AACrB,iBAAW,MAAM;AAAA,IACnB;AAAA,EACF,CAAC;AACH;AAEA,SAAS,kCAAkC,KAAuD;AAChG,SAAO,IAAI,WAAAA,eAAiC;AAAA,IAC1C,MAAM,YAAY;AAChB,iBAAW,MAAM,KAAK;AACpB,mBAAW,QAAQ,EAAE;AAAA,MACvB;AACA,iBAAW,MAAM;AAAA,IACnB;AAAA,EACF,CAAC;AACH;AAAA,IAEA,wBAAS,+BAA+B,MAAM;AAC5C,mCAAiB,EAAE,QAAQ,OAAO,OAAO,SAAS,CAAC;AAEnD,wBAAG,iEAAiE,YAAY;AAjDlF;AAkDI,UAAM,uBAAuB,IAAI,gBAAgB;AACjD,UAAM,sBAAsB,IAAI,gBAAgB;AAEhD,UAAM,SAAS,MAAM,KAAK,EAAE,QAAQ,GAAG,GAAG,MAAM,KAAK;AACrD,UAAM,mBAAmB,OAAO,KAAK,EAAE;AACvC,UAAM,WAAW,mBAAmB,QAAQ,EAAE;AAC9C,UAAM,CAAC,iBAAiB,OAAO,QAA4B;AAAA,MACzD;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAGA,QAAI,cAAc;AAClB,UAAM,iBAAa,iBAAK;AAAA,MACtB,aAAa;AAAA,MACb,YAAY,aAAE,OAAO,EAAE,UAAU,aAAE,OAAO,EAAE,CAAC;AAAA,MAC7C,SAAS,OAAO,EAAE,SAAS,GAAG,EAAE,YAAY,MAAM;AAChD,YAAI,aAAa;AACf,sBAAY,iBAAiB,SAAS,MAAM;AAC1C,0BAAc;AAAA,UAChB,CAAC;AAAA,QACH;AAEA,kBAAM,oBAAM,GAAI;AAChB,eAAO,YAAY,QAAQ;AAAA,MAC7B;AAAA,IACF,CAAC;AAED,UAAM,KAAK,wBAAa,OAAO;AAAA,MAC7B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,UAAU,gBAAgB,CAAC;AAAA,IACpD,CAAC;AACD,UAAM,iBAAiB,yBAAyB,EAAE;AAElD,UAAM,CAAC,UAAU,UAAU,QAAI,yCAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,eAAe,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MACxD,SAAS,EAAE,WAAW;AAAA,MACtB;AAAA,MACA,YAAY;AAAA,MACZ,wBAAwB,MAAM;AAAA,MAAC;AAAA,MAC/B,0BAA0B,MAAM;AAAA,MAAC;AAAA,IACnC,CAAC;AAGD,UAAM,WAAW,uBAAuB;AACxC,cAAM,oBAAM,GAAG;AACf,cAAM,4BAAc,CAAC,eAAe,GAAG,GAAI;AAE3C,UAAM,SAAS;AAEf,8BAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AACvC,UAAM,MAAM,WAAW,OAAO,CAAC;AAC/B,+BAAO,SAAI,mBAAJ,mBAAoB,OAAO,EAAE,KAAK,KAAK;AAC9C,+BAAO,SAAI,mBAAJ,mBAAoB,MAAM,EAAE,UAAU,wBAAwB;AAErE,8BAAO,QAAQ,IAAI,EAAE,IAAI,KAAK,gBAAgB;AAE9C,8BAAO,WAAW,EAAE,KAAK,KAAK;AAAA,EAChC,GAAG,GAAM;AAET,wBAAG,6CAA6C,YAAY;AAjH9D;AAkHI,UAAM,uBAAuB,IAAI,gBAAgB;AAEjD,UAAM,WAAO,iBAAK;AAAA,MAChB,aAAa;AAAA,MACb,YAAY,aAAE,OAAO,EAAE,KAAK,aAAE,OAAO,EAAE,CAAC;AAAA,MACxC,SAAS,OAAO,EAAE,IAAI,MAAM,SAAS,GAAG;AAAA,IAC1C,CAAC;AAED,UAAM,KAAK,wBAAa,OAAO;AAAA,MAC7B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,KAAK,QAAQ,CAAC;AAAA,IACvC,CAAC;AACD,UAAM,iBAAiB,yBAAyB,EAAE;AAElD,UAAM,CAAC,UAAU,UAAU,QAAI,yCAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,gBAAgB,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MACzD,SAAS,EAAE,KAAK;AAAA,MAChB;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,UAAM,SAAS;AACf,8BAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AACvC,UAAM,MAAM,WAAW,OAAO,CAAC;AAC/B,+BAAO,gCAAK,mBAAL,mBAAqB,OAAO,EAAE,KAAK,KAAK;AAC/C,+BAAO,gCAAK,mBAAL,mBAAqB,MAAM,EAAE,UAAU,aAAa;AAAA,EAC7D,CAAC;AAED,wBAAG,yDAAyD,YAAY;AAhJ1E;AAiJI,UAAM,uBAAuB,IAAI,gBAAgB;AAEjD,QAAI,UAAU;AACd,UAAM,aAAS,iBAAK;AAAA,MAClB,aAAa;AAAA,MACb,YAAY,aAAE,OAAO,EAAE,IAAI,aAAE,OAAO,EAAE,CAAC;AAAA,MACvC,SAAS,OAAO,EAAE,GAAG,GAAG,EAAE,YAAY,MAAM;AAC1C,YAAI,aAAa;AACf,sBAAY,iBAAiB,SAAS,MAAM;AAC1C,sBAAU;AAAA,UACZ,CAAC;AAAA,QACH;AACA,kBAAM,oBAAM,EAAE;AACd,eAAO;AAAA,MACT;AAAA,IACF,CAAC;AAED,UAAM,KAAK,wBAAa,OAAO;AAAA,MAC7B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,IAAI,IAAK,CAAC;AAAA,IACnC,CAAC;AACD,UAAM,iBAAiB,yBAAyB,EAAE;AAElD,UAAM,CAAC,UAAU,UAAU,QAAI,yCAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,gBAAgB,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MACzD,SAAS,EAAE,OAAO;AAAA,MAClB;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,UAAM,WAAW,uBAAuB;AACxC,yBAAqB,MAAM;AAC3B,UAAM,SAAS;AAEf,8BAAO,OAAO,EAAE,KAAK,IAAI;AACzB,8BAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AACvC,UAAM,MAAM,WAAW,OAAO,CAAC;AAC/B,+BAAO,gCAAK,mBAAL,mBAAqB,OAAO,EAAE,KAAK,IAAI;AAAA,EAChD,GAAG,GAAM;AAET,wBAAG,4EAA4E,YAAY;AA3L7F;AA4LI,UAAM,uBAAuB,IAAI,gBAAgB;AAEjD,UAAM,WAAO,iBAAK;AAAA,MAChB,aAAa;AAAA,MACb,YAAY,aAAE,OAAO,EAAE,KAAK,aAAE,OAAO,EAAE,CAAC;AAAA,MACxC,SAAS,OAAO,EAAE,IAAI,MAAM,SAAS,GAAG;AAAA,IAC1C,CAAC;AAGD,UAAM,KAAK,wBAAa,OAAO;AAAA,MAC7B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,KAAK,IAAI,CAAC;AAAA,IACnC,CAAC;AACD,UAAM,iBAAiB,yBAAyB,EAAE;AAElD,UAAM,CAAC,UAAU,UAAU,QAAI,yCAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,kBAAkB,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MAC3D,SAAS,EAAE,KAAK;AAAA,MAChB;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,UAAM,SAAS;AACf,8BAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AACvC,UAAM,MAAM,WAAW,OAAO,CAAC;AAC/B,+BAAO,gCAAK,mBAAL,mBAAqB,OAAO,EAAE,KAAK,IAAI;AAAA,EAChD,CAAC;AAED,wBAAG,4DAA4D,YAAY;AA1N7E;AA2NI,UAAM,uBAAuB,IAAI,gBAAgB;AAEjD,UAAM,UAAM,iBAAK;AAAA,MACf,aAAa;AAAA,MACb,YAAY,aAAE,OAAO,EAAE,GAAG,aAAE,OAAO,GAAG,GAAG,aAAE,OAAO,EAAE,CAAC;AAAA,MACrD,SAAS,OAAO,EAAE,GAAG,EAAE,MAAM,IAAI;AAAA,IACnC,CAAC;AACD,UAAM,YAAQ,iBAAK;AAAA,MACjB,aAAa;AAAA,MACb,YAAY,aAAE,OAAO,EAAE,GAAG,aAAE,OAAO,EAAE,CAAC;AAAA,MACtC,SAAS,OAAO,EAAE,EAAE,MAAM,EAAE,YAAY;AAAA,IAC1C,CAAC;AAED,UAAM,MAAM,wBAAa,OAAO;AAAA,MAC9B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,GAAG,GAAG,GAAG,EAAE,CAAC;AAAA,IACrC,CAAC;AACD,UAAM,MAAM,wBAAa,OAAO;AAAA,MAC9B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,GAAG,MAAM,CAAC;AAAA,IACnC,CAAC;AACD,UAAM,iBAAiB,kCAAkC,CAAC,KAAK,GAAG,CAAC;AAEnE,UAAM,CAAC,UAAU,UAAU,QAAI,yCAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,gBAAgB,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MACzD,SAAS,EAAE,KAAK,MAAM;AAAA,MACtB;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,UAAM,SAAS;AACf,8BAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AAGvC,UAAM,SAAS,CAAC,GAAG,WAAW,MAAM,EAAE;AAAA,MAAK,CAAC,GAAG,MAC7C,EAAE,SAAS,OAAO,cAAc,EAAE,SAAS,MAAM;AAAA,IACnD;AAEA,+BAAO,YAAO,CAAC,MAAR,mBAAW,SAAS,IAAI,EAAE,KAAK,KAAK;AAC3C,+BAAO,kBAAO,CAAC,MAAR,mBAAW,mBAAX,mBAA2B,OAAO,EAAE,KAAK,KAAK;AACrD,+BAAO,kBAAO,CAAC,MAAR,mBAAW,mBAAX,mBAA2B,MAAM,EAAE,KAAK,GAAG;AAClD,+BAAO,YAAO,CAAC,MAAR,mBAAW,SAAS,IAAI,EAAE,KAAK,OAAO;AAC7C,+BAAO,kBAAO,CAAC,MAAR,mBAAW,mBAAX,mBAA2B,OAAO,EAAE,KAAK,KAAK;AACrD,+BAAO,kBAAO,CAAC,MAAR,mBAAW,mBAAX,mBAA2B,MAAM,EAAE,KAAK,OAAO;AAAA,EACxD,CAAC;AACH,CAAC;","names":["NodeReadableStream"]}

package/dist/voice/generation_tools.test.js

@@ -0,0 +1,235 @@
+import { ReadableStream as NodeReadableStream } from "stream/web";
+import { describe, expect, it } from "vitest";
+import { z } from "zod";
+import { FunctionCall, tool } from "../llm/index.js";
+import { initializeLogger } from "../log.js";
+import { cancelAndWait, delay } from "../utils.js";
+import { performTextForwarding, performToolExecutions } from "./generation.js";
+function createStringStream(chunks, delayMs = 0) {
+  return new NodeReadableStream({
+    async start(controller) {
+      for (const c of chunks) {
+        if (delayMs > 0) {
+          await delay(delayMs);
+        }
+        controller.enqueue(c);
+      }
+      controller.close();
+    }
+  });
+}
+function createFunctionCallStream(fc) {
+  return new NodeReadableStream({
+    start(controller) {
+      controller.enqueue(fc);
+      controller.close();
+    }
+  });
+}
+function createFunctionCallStreamFromArray(fcs) {
+  return new NodeReadableStream({
+    start(controller) {
+      for (const fc of fcs) {
+        controller.enqueue(fc);
+      }
+      controller.close();
+    }
+  });
+}
+describe("Generation + Tool Execution", () => {
+  initializeLogger({ pretty: false, level: "silent" });
+  it("should not abort tool when preamble forwarders are cleaned up", async () => {
+    var _a, _b;
+    const replyAbortController = new AbortController();
+    const forwarderController = new AbortController();
+    const chunks = Array.from({ length: 50 }, () => `Hi.`);
+    const fullPreambleText = chunks.join("");
+    const preamble = createStringStream(chunks, 20);
+    const [textForwardTask, textOut] = performTextForwarding(
+      preamble,
+      forwarderController,
+      null
+    );
+    let toolAborted = false;
+    const getWeather = tool({
+      description: "weather",
+      parameters: z.object({ location: z.string() }),
+      execute: async ({ location }, { abortSignal }) => {
+        if (abortSignal) {
+          abortSignal.addEventListener("abort", () => {
+            toolAborted = true;
+          });
+        }
+        await delay(6e3);
+        return `Sunny in ${location}`;
+      }
+    });
+    const fc = FunctionCall.create({
+      callId: "call_1",
+      name: "getWeather",
+      args: JSON.stringify({ location: "San Francisco" })
+    });
+    const toolCallStream = createFunctionCallStream(fc);
+    const [execTask, toolOutput] = performToolExecutions({
+      session: {},
+      speechHandle: { id: "speech_test", _itemAdded: () => {
+      } },
+      toolCtx: { getWeather },
+      toolCallStream,
+      controller: replyAbortController,
+      onToolExecutionStarted: () => {
+      },
+      onToolExecutionCompleted: () => {
+      }
+    });
+    await toolOutput.firstToolStartedFuture.await;
+    await delay(100);
+    await cancelAndWait([textForwardTask], 5e3);
+    await execTask.result;
+    expect(toolOutput.output.length).toBe(1);
+    const out = toolOutput.output[0];
+    expect((_a = out.toolCallOutput) == null ? void 0 : _a.isError).toBe(false);
+    expect((_b = out.toolCallOutput) == null ? void 0 : _b.output).toContain("Sunny in San Francisco");
+    expect(textOut.text).not.toBe(fullPreambleText);
+    expect(toolAborted).toBe(false);
+  }, 3e4);
+  it("should return basic tool execution output", async () => {
+    var _a, _b;
+    const replyAbortController = new AbortController();
+    const echo = tool({
+      description: "echo",
+      parameters: z.object({ msg: z.string() }),
+      execute: async ({ msg }) => `echo: ${msg}`
+    });
+    const fc = FunctionCall.create({
+      callId: "call_2",
+      name: "echo",
+      args: JSON.stringify({ msg: "hello" })
+    });
+    const toolCallStream = createFunctionCallStream(fc);
+    const [execTask, toolOutput] = performToolExecutions({
+      session: {},
+      speechHandle: { id: "speech_test2", _itemAdded: () => {
+      } },
+      toolCtx: { echo },
+      toolCallStream,
+      controller: replyAbortController
+    });
+    await execTask.result;
+    expect(toolOutput.output.length).toBe(1);
+    const out = toolOutput.output[0];
+    expect((_a = out == null ? void 0 : out.toolCallOutput) == null ? void 0 : _a.isError).toBe(false);
+    expect((_b = out == null ? void 0 : out.toolCallOutput) == null ? void 0 : _b.output).toContain("echo: hello");
+  });
+  it("should abort tool when reply is aborted mid-execution", async () => {
+    var _a;
+    const replyAbortController = new AbortController();
+    let aborted = false;
+    const longOp = tool({
+      description: "longOp",
+      parameters: z.object({ ms: z.number() }),
+      execute: async ({ ms }, { abortSignal }) => {
+        if (abortSignal) {
+          abortSignal.addEventListener("abort", () => {
+            aborted = true;
+          });
+        }
+        await delay(ms);
+        return "done";
+      }
+    });
+    const fc = FunctionCall.create({
+      callId: "call_abort_1",
+      name: "longOp",
+      args: JSON.stringify({ ms: 5e3 })
+    });
+    const toolCallStream = createFunctionCallStream(fc);
+    const [execTask, toolOutput] = performToolExecutions({
+      session: {},
+      speechHandle: { id: "speech_abort", _itemAdded: () => {
+      } },
+      toolCtx: { longOp },
+      toolCallStream,
+      controller: replyAbortController
+    });
+    await toolOutput.firstToolStartedFuture.await;
+    replyAbortController.abort();
+    await execTask.result;
+    expect(aborted).toBe(true);
+    expect(toolOutput.output.length).toBe(1);
+    const out = toolOutput.output[0];
+    expect((_a = out == null ? void 0 : out.toolCallOutput) == null ? void 0 : _a.isError).toBe(true);
+  }, 2e4);
+  it("should return error output on invalid tool args (zod validation failure)", async () => {
+    var _a;
+    const replyAbortController = new AbortController();
+    const echo = tool({
+      description: "echo",
+      parameters: z.object({ msg: z.string() }),
+      execute: async ({ msg }) => `echo: ${msg}`
+    });
+    const fc = FunctionCall.create({
+      callId: "call_invalid_args",
+      name: "echo",
+      args: JSON.stringify({ msg: 123 })
+    });
+    const toolCallStream = createFunctionCallStream(fc);
+    const [execTask, toolOutput] = performToolExecutions({
+      session: {},
+      speechHandle: { id: "speech_invalid", _itemAdded: () => {
+      } },
+      toolCtx: { echo },
+      toolCallStream,
+      controller: replyAbortController
+    });
+    await execTask.result;
+    expect(toolOutput.output.length).toBe(1);
+    const out = toolOutput.output[0];
+    expect((_a = out == null ? void 0 : out.toolCallOutput) == null ? void 0 : _a.isError).toBe(true);
+  });
+  it("should handle multiple tool calls within a single stream", async () => {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    const replyAbortController = new AbortController();
+    const sum = tool({
+      description: "sum",
+      parameters: z.object({ a: z.number(), b: z.number() }),
+      execute: async ({ a, b }) => a + b
+    });
+    const upper = tool({
+      description: "upper",
+      parameters: z.object({ s: z.string() }),
+      execute: async ({ s }) => s.toUpperCase()
+    });
+    const fc1 = FunctionCall.create({
+      callId: "call_multi_1",
+      name: "sum",
+      args: JSON.stringify({ a: 2, b: 3 })
+    });
+    const fc2 = FunctionCall.create({
+      callId: "call_multi_2",
+      name: "upper",
+      args: JSON.stringify({ s: "hey" })
+    });
+    const toolCallStream = createFunctionCallStreamFromArray([fc1, fc2]);
+    const [execTask, toolOutput] = performToolExecutions({
+      session: {},
+      speechHandle: { id: "speech_multi", _itemAdded: () => {
+      } },
+      toolCtx: { sum, upper },
+      toolCallStream,
+      controller: replyAbortController
+    });
+    await execTask.result;
+    expect(toolOutput.output.length).toBe(2);
+    const sorted = [...toolOutput.output].sort(
+      (a, b) => a.toolCall.callId.localeCompare(b.toolCall.callId)
+    );
+    expect((_a = sorted[0]) == null ? void 0 : _a.toolCall.name).toBe("sum");
+    expect((_c = (_b = sorted[0]) == null ? void 0 : _b.toolCallOutput) == null ? void 0 : _c.isError).toBe(false);
+    expect((_e = (_d = sorted[0]) == null ? void 0 : _d.toolCallOutput) == null ? void 0 : _e.output).toBe("5");
+    expect((_f = sorted[1]) == null ? void 0 : _f.toolCall.name).toBe("upper");
+    expect((_h = (_g = sorted[1]) == null ? void 0 : _g.toolCallOutput) == null ? void 0 : _h.isError).toBe(false);
+    expect((_j = (_i = sorted[1]) == null ? void 0 : _i.toolCallOutput) == null ? void 0 : _j.output).toBe('"HEY"');
+  });
+});
+//# sourceMappingURL=generation_tools.test.js.map

package/dist/voice/generation_tools.test.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/voice/generation_tools.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2025 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { ReadableStream as NodeReadableStream } from 'stream/web';\nimport { describe, expect, it } from 'vitest';\nimport { z } from 'zod';\nimport { FunctionCall, tool } from '../llm/index.js';\nimport { initializeLogger } from '../log.js';\nimport type { Task } from '../utils.js';\nimport { cancelAndWait, delay } from '../utils.js';\nimport { type _TextOut, performTextForwarding, performToolExecutions } from './generation.js';\n\nfunction createStringStream(chunks: string[], delayMs: number = 0): NodeReadableStream<string> {\n  return new NodeReadableStream<string>({\n    async start(controller) {\n      for (const c of chunks) {\n        if (delayMs > 0) {\n          await delay(delayMs);\n        }\n        controller.enqueue(c);\n      }\n      controller.close();\n    },\n  });\n}\n\nfunction createFunctionCallStream(fc: FunctionCall): NodeReadableStream<FunctionCall> {\n  return new NodeReadableStream<FunctionCall>({\n    start(controller) {\n      controller.enqueue(fc);\n      controller.close();\n    },\n  });\n}\n\nfunction createFunctionCallStreamFromArray(fcs: FunctionCall[]): NodeReadableStream<FunctionCall> {\n  return new NodeReadableStream<FunctionCall>({\n    start(controller) {\n      for (const fc of fcs) {\n        controller.enqueue(fc);\n      }\n      controller.close();\n    },\n  });\n}\n\ndescribe('Generation + Tool Execution', () => {\n  initializeLogger({ pretty: false, level: 'silent' });\n\n  it('should not abort tool when preamble forwarders are cleaned up', async () => {\n    const replyAbortController = new AbortController();\n    const forwarderController = new AbortController();\n\n    const chunks = Array.from({ length: 50 }, () => `Hi.`);\n    const fullPreambleText = chunks.join('');\n    const preamble = createStringStream(chunks, 20);\n    const [textForwardTask, textOut]: [Task<void>, _TextOut] = performTextForwarding(\n      preamble,\n      forwarderController,\n      null,\n    );\n\n    // Tool that takes > 5 seconds\n    let toolAborted = false;\n    const getWeather = tool({\n      description: 'weather',\n      parameters: z.object({ location: z.string() }),\n      execute: async ({ location }, { abortSignal }) => {\n        if (abortSignal) {\n          abortSignal.addEventListener('abort', () => {\n            toolAborted = true;\n          });\n        }\n        // 6s delay\n        await delay(6000);\n        return `Sunny in ${location}`;\n      },\n    });\n\n    const fc = FunctionCall.create({\n      callId: 'call_1',\n      name: 'getWeather',\n      args: JSON.stringify({ location: 'San Francisco' }),\n    });\n    const toolCallStream = createFunctionCallStream(fc);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_test', _itemAdded: () => {} } as any,\n      toolCtx: { getWeather } as any,\n      toolCallStream,\n      controller: replyAbortController,\n      onToolExecutionStarted: () => {},\n      onToolExecutionCompleted: () => {},\n    });\n\n    // Ensure tool has started, then cancel forwarders mid-stream (without aborting parent AbortController)\n    await toolOutput.firstToolStartedFuture.await;\n    await delay(100);\n    await cancelAndWait([textForwardTask], 5000);\n\n    await execTask.result;\n\n    expect(toolOutput.output.length).toBe(1);\n    const out = toolOutput.output[0]!;\n    expect(out.toolCallOutput?.isError).toBe(false);\n    expect(out.toolCallOutput?.output).toContain('Sunny in San Francisco');\n    // Forwarder should have been cancelled before finishing all preamble chunks\n    expect(textOut.text).not.toBe(fullPreambleText);\n    // Tool's abort signal must not have fired\n    expect(toolAborted).toBe(false);\n  }, 30_000);\n\n  it('should return basic tool execution output', async () => {\n    const replyAbortController = new AbortController();\n\n    const echo = tool({\n      description: 'echo',\n      parameters: z.object({ msg: z.string() }),\n      execute: async ({ msg }) => `echo: ${msg}`,\n    });\n\n    const fc = FunctionCall.create({\n      callId: 'call_2',\n      name: 'echo',\n      args: JSON.stringify({ msg: 'hello' }),\n    });\n    const toolCallStream = createFunctionCallStream(fc);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_test2', _itemAdded: () => {} } as any,\n      toolCtx: { echo } as any,\n      toolCallStream,\n      controller: replyAbortController,\n    });\n\n    await execTask.result;\n    expect(toolOutput.output.length).toBe(1);\n    const out = toolOutput.output[0];\n    expect(out?.toolCallOutput?.isError).toBe(false);\n    expect(out?.toolCallOutput?.output).toContain('echo: hello');\n  });\n\n  it('should abort tool when reply is aborted mid-execution', async () => {\n    const replyAbortController = new AbortController();\n\n    let aborted = false;\n    const longOp = tool({\n      description: 'longOp',\n      parameters: z.object({ ms: z.number() }),\n      execute: async ({ ms }, { abortSignal }) => {\n        if (abortSignal) {\n          abortSignal.addEventListener('abort', () => {\n            aborted = true;\n          });\n        }\n        await delay(ms);\n        return 'done';\n      },\n    });\n\n    const fc = FunctionCall.create({\n      callId: 'call_abort_1',\n      name: 'longOp',\n      args: JSON.stringify({ ms: 5000 }),\n    });\n    const toolCallStream = createFunctionCallStream(fc);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_abort', _itemAdded: () => {} } as any,\n      toolCtx: { longOp } as any,\n      toolCallStream,\n      controller: replyAbortController,\n    });\n\n    await toolOutput.firstToolStartedFuture.await;\n    replyAbortController.abort();\n    await execTask.result;\n\n    expect(aborted).toBe(true);\n    expect(toolOutput.output.length).toBe(1);\n    const out = toolOutput.output[0];\n    expect(out?.toolCallOutput?.isError).toBe(true);\n  }, 20_000);\n\n  it('should return error output on invalid tool args (zod validation failure)', async () => {\n    const replyAbortController = new AbortController();\n\n    const echo = tool({\n      description: 'echo',\n      parameters: z.object({ msg: z.string() }),\n      execute: async ({ msg }) => `echo: ${msg}`,\n    });\n\n    // invalid: msg should be string\n    const fc = FunctionCall.create({\n      callId: 'call_invalid_args',\n      name: 'echo',\n      args: JSON.stringify({ msg: 123 }),\n    });\n    const toolCallStream = createFunctionCallStream(fc);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_invalid', _itemAdded: () => {} } as any,\n      toolCtx: { echo } as any,\n      toolCallStream,\n      controller: replyAbortController,\n    });\n\n    await execTask.result;\n    expect(toolOutput.output.length).toBe(1);\n    const out = toolOutput.output[0];\n    expect(out?.toolCallOutput?.isError).toBe(true);\n  });\n\n  it('should handle multiple tool calls within a single stream', async () => {\n    const replyAbortController = new AbortController();\n\n    const sum = tool({\n      description: 'sum',\n      parameters: z.object({ a: z.number(), b: z.number() }),\n      execute: async ({ a, b }) => a + b,\n    });\n    const upper = tool({\n      description: 'upper',\n      parameters: z.object({ s: z.string() }),\n      execute: async ({ s }) => s.toUpperCase(),\n    });\n\n    const fc1 = FunctionCall.create({\n      callId: 'call_multi_1',\n      name: 'sum',\n      args: JSON.stringify({ a: 2, b: 3 }),\n    });\n    const fc2 = FunctionCall.create({\n      callId: 'call_multi_2',\n      name: 'upper',\n      args: JSON.stringify({ s: 'hey' }),\n    });\n    const toolCallStream = createFunctionCallStreamFromArray([fc1, fc2]);\n\n    const [execTask, toolOutput] = performToolExecutions({\n      session: {} as any,\n      speechHandle: { id: 'speech_multi', _itemAdded: () => {} } as any,\n      toolCtx: { sum, upper } as any,\n      toolCallStream,\n      controller: replyAbortController,\n    });\n\n    await execTask.result;\n    expect(toolOutput.output.length).toBe(2);\n\n    // sort by callId to assert deterministically\n    const sorted = [...toolOutput.output].sort((a, b) =>\n      a.toolCall.callId.localeCompare(b.toolCall.callId),\n    );\n\n    expect(sorted[0]?.toolCall.name).toBe('sum');\n    expect(sorted[0]?.toolCallOutput?.isError).toBe(false);\n    expect(sorted[0]?.toolCallOutput?.output).toBe('5');\n    expect(sorted[1]?.toolCall.name).toBe('upper');\n    expect(sorted[1]?.toolCallOutput?.isError).toBe(false);\n    expect(sorted[1]?.toolCallOutput?.output).toBe('\"HEY\"');\n  });\n});\n"],"mappings":"AAGA,SAAS,kBAAkB,0BAA0B;AACrD,SAAS,UAAU,QAAQ,UAAU;AACrC,SAAS,SAAS;AAClB,SAAS,cAAc,YAAY;AACnC,SAAS,wBAAwB;AAEjC,SAAS,eAAe,aAAa;AACrC,SAAwB,uBAAuB,6BAA6B;AAE5E,SAAS,mBAAmB,QAAkB,UAAkB,GAA+B;AAC7F,SAAO,IAAI,mBAA2B;AAAA,IACpC,MAAM,MAAM,YAAY;AACtB,iBAAW,KAAK,QAAQ;AACtB,YAAI,UAAU,GAAG;AACf,gBAAM,MAAM,OAAO;AAAA,QACrB;AACA,mBAAW,QAAQ,CAAC;AAAA,MACtB;AACA,iBAAW,MAAM;AAAA,IACnB;AAAA,EACF,CAAC;AACH;AAEA,SAAS,yBAAyB,IAAoD;AACpF,SAAO,IAAI,mBAAiC;AAAA,IAC1C,MAAM,YAAY;AAChB,iBAAW,QAAQ,EAAE;AACrB,iBAAW,MAAM;AAAA,IACnB;AAAA,EACF,CAAC;AACH;AAEA,SAAS,kCAAkC,KAAuD;AAChG,SAAO,IAAI,mBAAiC;AAAA,IAC1C,MAAM,YAAY;AAChB,iBAAW,MAAM,KAAK;AACpB,mBAAW,QAAQ,EAAE;AAAA,MACvB;AACA,iBAAW,MAAM;AAAA,IACnB;AAAA,EACF,CAAC;AACH;AAEA,SAAS,+BAA+B,MAAM;AAC5C,mBAAiB,EAAE,QAAQ,OAAO,OAAO,SAAS,CAAC;AAEnD,KAAG,iEAAiE,YAAY;AAjDlF;AAkDI,UAAM,uBAAuB,IAAI,gBAAgB;AACjD,UAAM,sBAAsB,IAAI,gBAAgB;AAEhD,UAAM,SAAS,MAAM,KAAK,EAAE,QAAQ,GAAG,GAAG,MAAM,KAAK;AACrD,UAAM,mBAAmB,OAAO,KAAK,EAAE;AACvC,UAAM,WAAW,mBAAmB,QAAQ,EAAE;AAC9C,UAAM,CAAC,iBAAiB,OAAO,IAA4B;AAAA,MACzD;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAGA,QAAI,cAAc;AAClB,UAAM,aAAa,KAAK;AAAA,MACtB,aAAa;AAAA,MACb,YAAY,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;AAAA,MAC7C,SAAS,OAAO,EAAE,SAAS,GAAG,EAAE,YAAY,MAAM;AAChD,YAAI,aAAa;AACf,sBAAY,iBAAiB,SAAS,MAAM;AAC1C,0BAAc;AAAA,UAChB,CAAC;AAAA,QACH;AAEA,cAAM,MAAM,GAAI;AAChB,eAAO,YAAY,QAAQ;AAAA,MAC7B;AAAA,IACF,CAAC;AAED,UAAM,KAAK,aAAa,OAAO;AAAA,MAC7B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,UAAU,gBAAgB,CAAC;AAAA,IACpD,CAAC;AACD,UAAM,iBAAiB,yBAAyB,EAAE;AAElD,UAAM,CAAC,UAAU,UAAU,IAAI,sBAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,eAAe,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MACxD,SAAS,EAAE,WAAW;AAAA,MACtB;AAAA,MACA,YAAY;AAAA,MACZ,wBAAwB,MAAM;AAAA,MAAC;AAAA,MAC/B,0BAA0B,MAAM;AAAA,MAAC;AAAA,IACnC,CAAC;AAGD,UAAM,WAAW,uBAAuB;AACxC,UAAM,MAAM,GAAG;AACf,UAAM,cAAc,CAAC,eAAe,GAAG,GAAI;AAE3C,UAAM,SAAS;AAEf,WAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AACvC,UAAM,MAAM,WAAW,OAAO,CAAC;AAC/B,YAAO,SAAI,mBAAJ,mBAAoB,OAAO,EAAE,KAAK,KAAK;AAC9C,YAAO,SAAI,mBAAJ,mBAAoB,MAAM,EAAE,UAAU,wBAAwB;AAErE,WAAO,QAAQ,IAAI,EAAE,IAAI,KAAK,gBAAgB;AAE9C,WAAO,WAAW,EAAE,KAAK,KAAK;AAAA,EAChC,GAAG,GAAM;AAET,KAAG,6CAA6C,YAAY;AAjH9D;AAkHI,UAAM,uBAAuB,IAAI,gBAAgB;AAEjD,UAAM,OAAO,KAAK;AAAA,MAChB,aAAa;AAAA,MACb,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;AAAA,MACxC,SAAS,OAAO,EAAE,IAAI,MAAM,SAAS,GAAG;AAAA,IAC1C,CAAC;AAED,UAAM,KAAK,aAAa,OAAO;AAAA,MAC7B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,KAAK,QAAQ,CAAC;AAAA,IACvC,CAAC;AACD,UAAM,iBAAiB,yBAAyB,EAAE;AAElD,UAAM,CAAC,UAAU,UAAU,IAAI,sBAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,gBAAgB,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MACzD,SAAS,EAAE,KAAK;AAAA,MAChB;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,UAAM,SAAS;AACf,WAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AACvC,UAAM,MAAM,WAAW,OAAO,CAAC;AAC/B,YAAO,gCAAK,mBAAL,mBAAqB,OAAO,EAAE,KAAK,KAAK;AAC/C,YAAO,gCAAK,mBAAL,mBAAqB,MAAM,EAAE,UAAU,aAAa;AAAA,EAC7D,CAAC;AAED,KAAG,yDAAyD,YAAY;AAhJ1E;AAiJI,UAAM,uBAAuB,IAAI,gBAAgB;AAEjD,QAAI,UAAU;AACd,UAAM,SAAS,KAAK;AAAA,MAClB,aAAa;AAAA,MACb,YAAY,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AAAA,MACvC,SAAS,OAAO,EAAE,GAAG,GAAG,EAAE,YAAY,MAAM;AAC1C,YAAI,aAAa;AACf,sBAAY,iBAAiB,SAAS,MAAM;AAC1C,sBAAU;AAAA,UACZ,CAAC;AAAA,QACH;AACA,cAAM,MAAM,EAAE;AACd,eAAO;AAAA,MACT;AAAA,IACF,CAAC;AAED,UAAM,KAAK,aAAa,OAAO;AAAA,MAC7B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,IAAI,IAAK,CAAC;AAAA,IACnC,CAAC;AACD,UAAM,iBAAiB,yBAAyB,EAAE;AAElD,UAAM,CAAC,UAAU,UAAU,IAAI,sBAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,gBAAgB,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MACzD,SAAS,EAAE,OAAO;AAAA,MAClB;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,UAAM,WAAW,uBAAuB;AACxC,yBAAqB,MAAM;AAC3B,UAAM,SAAS;AAEf,WAAO,OAAO,EAAE,KAAK,IAAI;AACzB,WAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AACvC,UAAM,MAAM,WAAW,OAAO,CAAC;AAC/B,YAAO,gCAAK,mBAAL,mBAAqB,OAAO,EAAE,KAAK,IAAI;AAAA,EAChD,GAAG,GAAM;AAET,KAAG,4EAA4E,YAAY;AA3L7F;AA4LI,UAAM,uBAAuB,IAAI,gBAAgB;AAEjD,UAAM,OAAO,KAAK;AAAA,MAChB,aAAa;AAAA,MACb,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;AAAA,MACxC,SAAS,OAAO,EAAE,IAAI,MAAM,SAAS,GAAG;AAAA,IAC1C,CAAC;AAGD,UAAM,KAAK,aAAa,OAAO;AAAA,MAC7B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,KAAK,IAAI,CAAC;AAAA,IACnC,CAAC;AACD,UAAM,iBAAiB,yBAAyB,EAAE;AAElD,UAAM,CAAC,UAAU,UAAU,IAAI,sBAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,kBAAkB,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MAC3D,SAAS,EAAE,KAAK;AAAA,MAChB;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,UAAM,SAAS;AACf,WAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AACvC,UAAM,MAAM,WAAW,OAAO,CAAC;AAC/B,YAAO,gCAAK,mBAAL,mBAAqB,OAAO,EAAE,KAAK,IAAI;AAAA,EAChD,CAAC;AAED,KAAG,4DAA4D,YAAY;AA1N7E;AA2NI,UAAM,uBAAuB,IAAI,gBAAgB;AAEjD,UAAM,MAAM,KAAK;AAAA,MACf,aAAa;AAAA,MACb,YAAY,EAAE,OAAO,EAAE,GAAG,EAAE,OAAO,GAAG,GAAG,EAAE,OAAO,EAAE,CAAC;AAAA,MACrD,SAAS,OAAO,EAAE,GAAG,EAAE,MAAM,IAAI;AAAA,IACnC,CAAC;AACD,UAAM,QAAQ,KAAK;AAAA,MACjB,aAAa;AAAA,MACb,YAAY,EAAE,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;AAAA,MACtC,SAAS,OAAO,EAAE,EAAE,MAAM,EAAE,YAAY;AAAA,IAC1C,CAAC;AAED,UAAM,MAAM,aAAa,OAAO;AAAA,MAC9B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,GAAG,GAAG,GAAG,EAAE,CAAC;AAAA,IACrC,CAAC;AACD,UAAM,MAAM,aAAa,OAAO;AAAA,MAC9B,QAAQ;AAAA,MACR,MAAM;AAAA,MACN,MAAM,KAAK,UAAU,EAAE,GAAG,MAAM,CAAC;AAAA,IACnC,CAAC;AACD,UAAM,iBAAiB,kCAAkC,CAAC,KAAK,GAAG,CAAC;AAEnE,UAAM,CAAC,UAAU,UAAU,IAAI,sBAAsB;AAAA,MACnD,SAAS,CAAC;AAAA,MACV,cAAc,EAAE,IAAI,gBAAgB,YAAY,MAAM;AAAA,MAAC,EAAE;AAAA,MACzD,SAAS,EAAE,KAAK,MAAM;AAAA,MACtB;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAED,UAAM,SAAS;AACf,WAAO,WAAW,OAAO,MAAM,EAAE,KAAK,CAAC;AAGvC,UAAM,SAAS,CAAC,GAAG,WAAW,MAAM,EAAE;AAAA,MAAK,CAAC,GAAG,MAC7C,EAAE,SAAS,OAAO,cAAc,EAAE,SAAS,MAAM;AAAA,IACnD;AAEA,YAAO,YAAO,CAAC,MAAR,mBAAW,SAAS,IAAI,EAAE,KAAK,KAAK;AAC3C,YAAO,kBAAO,CAAC,MAAR,mBAAW,mBAAX,mBAA2B,OAAO,EAAE,KAAK,KAAK;AACrD,YAAO,kBAAO,CAAC,MAAR,mBAAW,mBAAX,mBAA2B,MAAM,EAAE,KAAK,GAAG;AAClD,YAAO,YAAO,CAAC,MAAR,mBAAW,SAAS,IAAI,EAAE,KAAK,OAAO;AAC7C,YAAO,kBAAO,CAAC,MAAR,mBAAW,mBAAX,mBAA2B,OAAO,EAAE,KAAK,KAAK;AACrD,YAAO,kBAAO,CAAC,MAAR,mBAAW,mBAAX,mBAA2B,MAAM,EAAE,KAAK,OAAO;AAAA,EACxD,CAAC;AACH,CAAC;","names":[]}