ToolCallManager

Defined in: packages/ai/src/activities/chat/tools/tool-calls.ts:246

Manages tool call accumulation and execution for the chat() method's automatic tool execution loop.

Responsibilities:

• Accumulates streaming tool call events (ID, name, arguments)
• Validates tool calls (filters out incomplete ones)
• Executes tool execute functions with parsed arguments
• Emits TOOL_CALL_END events for client visibility
• Returns tool result messages for conversation history

This class is used internally by the AI.chat() method to handle the automatic tool execution loop. It can also be used independently for custom tool execution logic.

Example

const manager = new ToolCallManager(tools);

// During streaming, accumulate tool calls
for await (const chunk of stream) {
  if (chunk.type === 'TOOL_CALL_START') {
    manager.addToolCallStartEvent(chunk);
  } else if (chunk.type === 'TOOL_CALL_ARGS') {
    manager.addToolCallArgsEvent(chunk);
  }
}

// After stream completes, execute tools
if (manager.hasToolCalls()) {
  const toolResults = yield* manager.executeTools(finishEvent);
  messages = [...messages, ...toolResults];
  manager.clear();
}

Type Parameters

TToolsOrContext

TToolsOrContext = ReadonlyArray<AnyTool>

TContext

TContext = TToolsOrContext extends ReadonlyArray<AnyTool> ? ContextFromTools<TToolsOrContext> : TToolsOrContext

Constructors

Constructor

new ToolCallManager<TToolsOrContext, TContext>(tools): ToolCallManager<TToolsOrContext, TContext>;

Defined in: packages/ai/src/activities/chat/tools/tool-calls.ts:257

Parameters

tools

TToolsOrContext extends readonly AnyTool[] ? TToolsOrContext : readonly AnyTool[]

Returns

ToolCallManager<TToolsOrContext, TContext>

Methods

addToolCallArgsEvent()

addToolCallArgsEvent(event): void;

Defined in: packages/ai/src/activities/chat/tools/tool-calls.ts:294

Add a TOOL_CALL_ARGS event to accumulate arguments (AG-UI)

Parameters

event

ToolCallArgsEvent

Returns

void

addToolCallStartEvent()

addToolCallStartEvent(event): void;

Defined in: packages/ai/src/activities/chat/tools/tool-calls.ts:268

Add a TOOL_CALL_START event to begin tracking a tool call (AG-UI)

Parameters

event

ToolCallStartEvent

Returns

void

clear()

clear(): void;

Defined in: packages/ai/src/activities/chat/tools/tool-calls.ts:465

Clear the tool calls map for the next iteration

Returns

void

completeToolCall()

completeToolCall(event): void;

Defined in: packages/ai/src/activities/chat/tools/tool-calls.ts:312

Complete a tool call with its final input Called when TOOL_CALL_END is received

Parameters

event

ToolCallEndEvent

Returns

void

executeTools()

executeTools(finishEvent, ...contextArgs): AsyncGenerator<AdapterYieldChunk, ModelMessage<
  | string
  | ContentPart<unknown, unknown, unknown, unknown, unknown>[]
| null>[], void>;

Defined in: packages/ai/src/activities/chat/tools/tool-calls.ts:344

Execute all tool calls and return tool result messages Yields TOOL_CALL_END events for streaming

Parameters

finishEvent

RunFinishedEvent

RUN_FINISHED event from the stream

contextArgs

...ExecuteToolsContextArgs<TContext>

Returns

AsyncGenerator<AdapterYieldChunk, ModelMessage< | string | ContentPart<unknown, unknown, unknown, unknown, unknown>[] | null>[], void>

getToolCalls()

getToolCalls(): ToolCall<unknown>[];

Defined in: packages/ai/src/activities/chat/tools/tool-calls.ts:333

Get all complete tool calls (filtered for valid ID and name)

Returns

ToolCall<unknown>[]

hasToolCalls()

hasToolCalls(): boolean;

Defined in: packages/ai/src/activities/chat/tools/tool-calls.ts:326

Check if there are any complete tool calls to execute

Returns

boolean