@librechat/agents 3.2.32 → 3.2.33

package/dist/esm/responses/index.mjs

@@ -1,440 +1,352 @@
-import { GraphEvents } from '../common/enum.mjs';
-
+import "../common/enum.mjs";
+import "../common/index.mjs";
+//#region src/responses/index.ts
 let responseItemId = 0;
 function createItemId(prefix) {
-    return `${prefix}_${Date.now().toString(36)}${(responseItemId++).toString(36)}`;
+	return `${prefix}_${Date.now().toString(36)}${(responseItemId++).toString(36)}`;
 }
 function createResponseTracker() {
-    const tracker = {
-        sequenceNumber: 0,
-        items: [],
-        message: undefined,
-        reasoning: undefined,
-        functionCalls: new Map(),
-        functionCallsByStep: new Map(),
-        responseCreated: false,
-        usage: {
-            inputTokens: 0,
-            outputTokens: 0,
-        },
-        nextSequence: () => tracker.sequenceNumber++,
-    };
-    return tracker;
+	const tracker = {
+		sequenceNumber: 0,
+		items: [],
+		message: void 0,
+		reasoning: void 0,
+		functionCalls: /* @__PURE__ */ new Map(),
+		functionCallsByStep: /* @__PURE__ */ new Map(),
+		responseCreated: false,
+		usage: {
+			inputTokens: 0,
+			outputTokens: 0
+		},
+		nextSequence: () => tracker.sequenceNumber++
+	};
+	return tracker;
 }
 function getTokenCount(value) {
-    return typeof value === 'number' && Number.isFinite(value) ? value : 0;
+	return typeof value === "number" && Number.isFinite(value) ? value : 0;
 }
 function getToolCallIndex(toolCall, fallbackIndex) {
-    return typeof toolCall.index === 'number' ? toolCall.index : fallbackIndex;
+	return typeof toolCall.index === "number" ? toolCall.index : fallbackIndex;
 }
 function getToolCallPositionKey(stepId, toolCall, fallbackIndex) {
-    const index = getToolCallIndex(toolCall, fallbackIndex);
-    if (stepId !== '') {
-        return `${stepId}:${index}`;
-    }
-    return `tool:${index}`;
+	const index = getToolCallIndex(toolCall, fallbackIndex);
+	if (stepId !== "") return `${stepId}:${index}`;
+	return `tool:${index}`;
 }
 function getToolCallIdKey(toolCall) {
-    return toolCall.id != null && toolCall.id !== '' ? toolCall.id : undefined;
+	return toolCall.id != null && toolCall.id !== "" ? toolCall.id : void 0;
 }
 function getExistingFunctionCall(config, idKey, positionKey, stepId, toolCall) {
-    const keyed = idKey == null
-        ? undefined
-        : (config.tracker.functionCalls.get(idKey) ??
-            (positionKey == null
-                ? undefined
-                : config.tracker.functionCalls.get(positionKey)));
-    if (keyed != null) {
-        return keyed;
-    }
-    const positioned = positionKey == null
-        ? undefined
-        : config.tracker.functionCalls.get(positionKey);
-    if (positioned != null) {
-        return positioned;
-    }
-    if (stepId === '') {
-        return undefined;
-    }
-    return findFunctionCallByStep(config, stepId, toolCall);
+	const keyed = idKey == null ? void 0 : config.tracker.functionCalls.get(idKey) ?? (positionKey == null ? void 0 : config.tracker.functionCalls.get(positionKey));
+	if (keyed != null) return keyed;
+	const positioned = positionKey == null ? void 0 : config.tracker.functionCalls.get(positionKey);
+	if (positioned != null) return positioned;
+	if (stepId === "") return;
+	return findFunctionCallByStep(config, stepId, toolCall);
 }
 function findFunctionCallByStep(config, stepId, toolCall) {
-    const items = config.tracker.functionCallsByStep.get(stepId);
-    if (items == null) {
-        return undefined;
-    }
-    const index = typeof toolCall.index === 'number' ? toolCall.index : undefined;
-    if (index != null) {
-        return items[index];
-    }
-    const name = getToolCallName(toolCall);
-    let fallback;
-    for (const item of items) {
-        if (item == null || item.status === 'completed') {
-            continue;
-        }
-        if (name !== '' && item.name !== name) {
-            continue;
-        }
-        if (fallback != null) {
-            return undefined;
-        }
-        fallback = item;
-    }
-    return fallback;
+	const items = config.tracker.functionCallsByStep.get(stepId);
+	if (items == null) return;
+	const index = typeof toolCall.index === "number" ? toolCall.index : void 0;
+	if (index != null) return items[index];
+	const name = getToolCallName(toolCall);
+	let fallback;
+	for (const item of items) {
+		if (item == null || item.status === "completed") continue;
+		if (name !== "" && item.name !== name) continue;
+		if (fallback != null) return;
+		fallback = item;
+	}
+	return fallback;
 }
 function trackFunctionCallKeys(config, item, idKey, positionKey) {
-    config.tracker.functionCalls.set(positionKey, item);
-    if (idKey != null) {
-        config.tracker.functionCalls.set(idKey, item);
-    }
+	config.tracker.functionCalls.set(positionKey, item);
+	if (idKey != null) config.tracker.functionCalls.set(idKey, item);
 }
 function trackFunctionCallStep(config, stepId, item, fallbackIndex) {
-    if (stepId === '') {
-        return;
-    }
-    const items = config.tracker.functionCallsByStep.get(stepId) ?? [];
-    items[fallbackIndex] = item;
-    config.tracker.functionCallsByStep.set(stepId, items);
+	if (stepId === "") return;
+	const items = config.tracker.functionCallsByStep.get(stepId) ?? [];
+	items[fallbackIndex] = item;
+	config.tracker.functionCallsByStep.set(stepId, items);
 }
 function getToolCallName(toolCall) {
-    return toolCall.name ?? toolCall.function?.name ?? '';
+	return toolCall.name ?? toolCall.function?.name ?? "";
 }
 function getToolCallArguments(toolCall) {
-    const args = toolCall.args ?? toolCall.function?.arguments;
-    if (args == null) {
-        return '';
-    }
-    if (typeof args === 'string') {
-        return args;
-    }
-    return JSON.stringify(args);
+	const args = toolCall.args ?? toolCall.function?.arguments;
+	if (args == null) return "";
+	if (typeof args === "string") return args;
+	return JSON.stringify(args);
 }
 async function ensureFunctionCall(config, stepId, toolCall, fallbackIndex) {
-    await ensureResponseCreated(config);
-    const positionKey = getToolCallPositionKey(stepId, toolCall, fallbackIndex);
-    const idKey = getToolCallIdKey(toolCall);
-    const existing = getExistingFunctionCall(config, idKey, positionKey, stepId, toolCall);
-    const toolCallIndex = getToolCallIndex(toolCall, fallbackIndex);
-    const name = getToolCallName(toolCall);
-    if (existing) {
-        trackFunctionCallKeys(config, existing, idKey, positionKey);
-        trackFunctionCallStep(config, stepId, existing, toolCallIndex);
-        if (idKey != null) {
-            existing.call_id = idKey;
-        }
-        if (name !== '') {
-            existing.name = name;
-        }
-        return existing;
-    }
-    const item = {
-        type: 'function_call',
-        id: createItemId('fc'),
-        call_id: idKey ?? positionKey,
-        name,
-        arguments: '',
-        status: 'in_progress',
-    };
-    trackFunctionCallKeys(config, item, idKey, positionKey);
-    trackFunctionCallStep(config, stepId, item, toolCallIndex);
-    config.tracker.items.push(item);
-    await writeResponseEvent(config.writer, {
-        type: 'response.output_item.added',
-        sequence_number: config.tracker.nextSequence(),
-        output_index: config.tracker.items.length - 1,
-        item,
-    });
-    return item;
+	await ensureResponseCreated(config);
+	const positionKey = getToolCallPositionKey(stepId, toolCall, fallbackIndex);
+	const idKey = getToolCallIdKey(toolCall);
+	const existing = getExistingFunctionCall(config, idKey, positionKey, stepId, toolCall);
+	const toolCallIndex = getToolCallIndex(toolCall, fallbackIndex);
+	const name = getToolCallName(toolCall);
+	if (existing) {
+		trackFunctionCallKeys(config, existing, idKey, positionKey);
+		trackFunctionCallStep(config, stepId, existing, toolCallIndex);
+		if (idKey != null) existing.call_id = idKey;
+		if (name !== "") existing.name = name;
+		return existing;
+	}
+	const item = {
+		type: "function_call",
+		id: createItemId("fc"),
+		call_id: idKey ?? positionKey,
+		name,
+		arguments: "",
+		status: "in_progress"
+	};
+	trackFunctionCallKeys(config, item, idKey, positionKey);
+	trackFunctionCallStep(config, stepId, item, toolCallIndex);
+	config.tracker.items.push(item);
+	await writeResponseEvent(config.writer, {
+		type: "response.output_item.added",
+		sequence_number: config.tracker.nextSequence(),
+		output_index: config.tracker.items.length - 1,
+		item
+	});
+	return item;
 }
 async function emitFunctionCallArgumentsDelta(params) {
-    if (params.delta === '') {
-        return;
-    }
-    params.item.arguments += params.delta;
-    await writeResponseEvent(params.config.writer, {
-        type: 'response.function_call_arguments.delta',
-        sequence_number: params.config.tracker.nextSequence(),
-        item_id: params.item.id,
-        output_index: params.config.tracker.items.indexOf(params.item),
-        call_id: params.item.call_id,
-        delta: params.delta,
-    });
+	if (params.delta === "") return;
+	params.item.arguments += params.delta;
+	await writeResponseEvent(params.config.writer, {
+		type: "response.function_call_arguments.delta",
+		sequence_number: params.config.tracker.nextSequence(),
+		item_id: params.item.id,
+		output_index: params.config.tracker.items.indexOf(params.item),
+		call_id: params.item.call_id,
+		delta: params.delta
+	});
 }
 async function emitFunctionCallArgumentsDone(config, item) {
-    await writeResponseEvent(config.writer, {
-        type: 'response.function_call_arguments.done',
-        sequence_number: config.tracker.nextSequence(),
-        item_id: item.id,
-        output_index: config.tracker.items.indexOf(item),
-        call_id: item.call_id,
-        name: item.name,
-        arguments: item.arguments,
-    });
+	await writeResponseEvent(config.writer, {
+		type: "response.function_call_arguments.done",
+		sequence_number: config.tracker.nextSequence(),
+		item_id: item.id,
+		output_index: config.tracker.items.indexOf(item),
+		call_id: item.call_id,
+		name: item.name,
+		arguments: item.arguments
+	});
 }
 async function completeFunctionCall(config, stepId, toolCall) {
-    const fallbackIndex = typeof toolCall.index === 'number' ? toolCall.index : undefined;
-    const positionKey = fallbackIndex == null
-        ? undefined
-        : getToolCallPositionKey(stepId, toolCall, fallbackIndex);
-    const item = getExistingFunctionCall(config, getToolCallIdKey(toolCall), positionKey, stepId, toolCall) ??
-        (await ensureFunctionCall(config, stepId, toolCall, fallbackIndex ?? 0));
-    if (item.status === 'completed') {
-        return;
-    }
-    const finalArguments = getToolCallArguments(toolCall);
-    if (finalArguments !== '' &&
-        finalArguments !== item.arguments &&
-        finalArguments.startsWith(item.arguments)) {
-        await emitFunctionCallArgumentsDelta({
-            config,
-            item,
-            delta: finalArguments.slice(item.arguments.length),
-        });
-    }
-    else if (finalArguments !== '') {
-        item.arguments = finalArguments;
-    }
-    await emitFunctionCallArgumentsDone(config, item);
-    item.status = 'completed';
-    await writeResponseEvent(config.writer, {
-        type: 'response.output_item.done',
-        sequence_number: config.tracker.nextSequence(),
-        output_index: config.tracker.items.indexOf(item),
-        item,
-    });
+	const fallbackIndex = typeof toolCall.index === "number" ? toolCall.index : void 0;
+	const positionKey = fallbackIndex == null ? void 0 : getToolCallPositionKey(stepId, toolCall, fallbackIndex);
+	const item = getExistingFunctionCall(config, getToolCallIdKey(toolCall), positionKey, stepId, toolCall) ?? await ensureFunctionCall(config, stepId, toolCall, fallbackIndex ?? 0);
+	if (item.status === "completed") return;
+	const finalArguments = getToolCallArguments(toolCall);
+	if (finalArguments !== "" && finalArguments !== item.arguments && finalArguments.startsWith(item.arguments)) await emitFunctionCallArgumentsDelta({
+		config,
+		item,
+		delta: finalArguments.slice(item.arguments.length)
+	});
+	else if (finalArguments !== "") item.arguments = finalArguments;
+	await emitFunctionCallArgumentsDone(config, item);
+	item.status = "completed";
+	await writeResponseEvent(config.writer, {
+		type: "response.output_item.done",
+		sequence_number: config.tracker.nextSequence(),
+		output_index: config.tracker.items.indexOf(item),
+		item
+	});
 }
-function buildResponse(context, tracker, status = 'in_progress') {
-    const completed = status === 'completed';
-    return {
-        id: context.responseId,
-        object: 'response',
-        created_at: context.createdAt,
-        completed_at: completed ? Math.floor(Date.now() / 1000) : null,
-        status,
-        model: context.model,
-        previous_response_id: context.previousResponseId ?? null,
-        instructions: context.instructions ?? null,
-        output: tracker.items,
-        error: null,
-        usage: completed
-            ? {
-                input_tokens: tracker.usage.inputTokens,
-                output_tokens: tracker.usage.outputTokens,
-                total_tokens: tracker.usage.inputTokens + tracker.usage.outputTokens,
-            }
-            : null,
-    };
+function buildResponse(context, tracker, status = "in_progress") {
+	const completed = status === "completed";
+	return {
+		id: context.responseId,
+		object: "response",
+		created_at: context.createdAt,
+		completed_at: completed ? Math.floor(Date.now() / 1e3) : null,
+		status,
+		model: context.model,
+		previous_response_id: context.previousResponseId ?? null,
+		instructions: context.instructions ?? null,
+		output: tracker.items,
+		error: null,
+		usage: completed ? {
+			input_tokens: tracker.usage.inputTokens,
+			output_tokens: tracker.usage.outputTokens,
+			total_tokens: tracker.usage.inputTokens + tracker.usage.outputTokens
+		} : null
+	};
 }
 async function writeResponseEvent(writer, event) {
-    await writer.write(`event: ${event.type}\n`);
-    await writer.write(`data: ${JSON.stringify(event)}\n\n`);
+	await writer.write(`event: ${event.type}\n`);
+	await writer.write(`data: ${JSON.stringify(event)}\n\n`);
 }
 async function writeResponsesDone(writer) {
-    await writer.write('data: [DONE]\n\n');
+	await writer.write("data: [DONE]\n\n");
 }
 async function ensureResponseCreated(config) {
-    if (config.tracker.responseCreated) {
-        return;
-    }
-    config.tracker.responseCreated = true;
-    await writeResponseEvent(config.writer, {
-        type: 'response.created',
-        sequence_number: config.tracker.nextSequence(),
-        response: buildResponse(config.context, config.tracker),
-    });
+	if (config.tracker.responseCreated) return;
+	config.tracker.responseCreated = true;
+	await writeResponseEvent(config.writer, {
+		type: "response.created",
+		sequence_number: config.tracker.nextSequence(),
+		response: buildResponse(config.context, config.tracker)
+	});
 }
 async function emitOutputContentDone(config, item) {
-    const outputIndex = config.tracker.items.indexOf(item);
-    if (item.type === 'message') {
-        await writeResponseEvent(config.writer, {
-            type: 'response.output_text.done',
-            sequence_number: config.tracker.nextSequence(),
-            item_id: item.id,
-            output_index: outputIndex,
-            content_index: 0,
-            text: item.content[0].text,
-        });
-        return;
-    }
-    await writeResponseEvent(config.writer, {
-        type: 'response.reasoning_text.done',
-        sequence_number: config.tracker.nextSequence(),
-        item_id: item.id,
-        output_index: outputIndex,
-        content_index: 0,
-        text: item.content[0].text,
-    });
+	const outputIndex = config.tracker.items.indexOf(item);
+	if (item.type === "message") {
+		await writeResponseEvent(config.writer, {
+			type: "response.output_text.done",
+			sequence_number: config.tracker.nextSequence(),
+			item_id: item.id,
+			output_index: outputIndex,
+			content_index: 0,
+			text: item.content[0].text
+		});
+		return;
+	}
+	await writeResponseEvent(config.writer, {
+		type: "response.reasoning_text.done",
+		sequence_number: config.tracker.nextSequence(),
+		item_id: item.id,
+		output_index: outputIndex,
+		content_index: 0,
+		text: item.content[0].text
+	});
 }
 async function ensureMessage(config) {
-    await ensureResponseCreated(config);
-    if (config.tracker.message) {
-        return config.tracker.message;
-    }
-    const item = {
-        type: 'message',
-        id: createItemId('msg'),
-        role: 'assistant',
-        status: 'in_progress',
-        content: [{ type: 'output_text', text: '', annotations: [], logprobs: [] }],
-    };
-    config.tracker.message = item;
-    config.tracker.items.push(item);
-    await writeResponseEvent(config.writer, {
-        type: 'response.output_item.added',
-        sequence_number: config.tracker.nextSequence(),
-        output_index: config.tracker.items.length - 1,
-        item,
-    });
-    return item;
+	await ensureResponseCreated(config);
+	if (config.tracker.message) return config.tracker.message;
+	const item = {
+		type: "message",
+		id: createItemId("msg"),
+		role: "assistant",
+		status: "in_progress",
+		content: [{
+			type: "output_text",
+			text: "",
+			annotations: [],
+			logprobs: []
+		}]
+	};
+	config.tracker.message = item;
+	config.tracker.items.push(item);
+	await writeResponseEvent(config.writer, {
+		type: "response.output_item.added",
+		sequence_number: config.tracker.nextSequence(),
+		output_index: config.tracker.items.length - 1,
+		item
+	});
+	return item;
 }
 async function ensureReasoning(config) {
-    await ensureResponseCreated(config);
-    if (config.tracker.reasoning) {
-        return config.tracker.reasoning;
-    }
-    const item = {
-        type: 'reasoning',
-        id: createItemId('reason'),
-        status: 'in_progress',
-        content: [{ type: 'reasoning_text', text: '' }],
-        summary: [],
-    };
-    config.tracker.reasoning = item;
-    config.tracker.items.push(item);
-    await writeResponseEvent(config.writer, {
-        type: 'response.output_item.added',
-        sequence_number: config.tracker.nextSequence(),
-        output_index: config.tracker.items.length - 1,
-        item,
-    });
-    return item;
+	await ensureResponseCreated(config);
+	if (config.tracker.reasoning) return config.tracker.reasoning;
+	const item = {
+		type: "reasoning",
+		id: createItemId("reason"),
+		status: "in_progress",
+		content: [{
+			type: "reasoning_text",
+			text: ""
+		}],
+		summary: []
+	};
+	config.tracker.reasoning = item;
+	config.tracker.items.push(item);
+	await writeResponseEvent(config.writer, {
+		type: "response.output_item.added",
+		sequence_number: config.tracker.nextSequence(),
+		output_index: config.tracker.items.length - 1,
+		item
+	});
+	return item;
 }
 function createResponsesEventHandlers(config) {
-    return {
-        [GraphEvents.ON_MESSAGE_DELTA]: {
-            handle: async (_event, data) => {
-                const item = await ensureMessage(config);
-                for (const part of data.delta.content ?? []) {
-                    if (!('text' in part) || typeof part.text !== 'string') {
-                        continue;
-                    }
-                    item.content[0].text += part.text;
-                    await writeResponseEvent(config.writer, {
-                        type: 'response.output_text.delta',
-                        sequence_number: config.tracker.nextSequence(),
-                        item_id: item.id,
-                        output_index: config.tracker.items.indexOf(item),
-                        content_index: 0,
-                        delta: part.text,
-                    });
-                }
-            },
-        },
-        [GraphEvents.ON_REASONING_DELTA]: {
-            handle: async (_event, data) => {
-                const item = await ensureReasoning(config);
-                for (const part of data.delta.content ??
-                    []) {
-                    let text;
-                    if ('think' in part && typeof part.think === 'string') {
-                        text = part.think;
-                    }
-                    else if ('text' in part && typeof part.text === 'string') {
-                        text = part.text;
-                    }
-                    if (typeof text !== 'string') {
-                        continue;
-                    }
-                    item.content[0].text += text;
-                    await writeResponseEvent(config.writer, {
-                        type: 'response.reasoning_text.delta',
-                        sequence_number: config.tracker.nextSequence(),
-                        item_id: item.id,
-                        output_index: config.tracker.items.indexOf(item),
-                        content_index: 0,
-                        delta: text,
-                    });
-                }
-            },
-        },
-        [GraphEvents.ON_RUN_STEP]: {
-            handle: async (_event, data) => {
-                const runStep = data;
-                if (runStep.stepDetails.type !== 'tool_calls') {
-                    return;
-                }
-                const toolCalls = runStep.stepDetails.tool_calls ?? [];
-                for (let index = 0; index < toolCalls.length; index++) {
-                    await ensureFunctionCall(config, runStep.id, toolCalls[index], index);
-                }
-            },
-        },
-        [GraphEvents.ON_RUN_STEP_DELTA]: {
-            handle: async (_event, data) => {
-                const runStepDelta = data;
-                if (runStepDelta.delta.type !== 'tool_calls') {
-                    return;
-                }
-                const toolCalls = runStepDelta.delta.tool_calls ?? [];
-                for (let index = 0; index < toolCalls.length; index++) {
-                    const item = await ensureFunctionCall(config, runStepDelta.id, toolCalls[index], index);
-                    await emitFunctionCallArgumentsDelta({
-                        config,
-                        item,
-                        delta: getToolCallArguments(toolCalls[index]),
-                    });
-                }
-            },
-        },
-        [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-            handle: async (_event, data) => {
-                const completed = data;
-                if (!completed.result?.tool_call) {
-                    return;
-                }
-                await completeFunctionCall(config, completed.result.id ?? '', completed.result.tool_call);
-            },
-        },
-        [GraphEvents.CHAT_MODEL_END]: {
-            handle: (_event, data) => {
-                const usage = data?.output?.usage_metadata;
-                if (!usage) {
-                    return;
-                }
-                config.tracker.usage.inputTokens += getTokenCount(usage.input_tokens);
-                config.tracker.usage.outputTokens += getTokenCount(usage.output_tokens);
-            },
-        },
-    };
+	return {
+		["on_message_delta"]: { handle: async (_event, data) => {
+			const item = await ensureMessage(config);
+			for (const part of data.delta.content ?? []) {
+				if (!("text" in part) || typeof part.text !== "string") continue;
+				item.content[0].text += part.text;
+				await writeResponseEvent(config.writer, {
+					type: "response.output_text.delta",
+					sequence_number: config.tracker.nextSequence(),
+					item_id: item.id,
+					output_index: config.tracker.items.indexOf(item),
+					content_index: 0,
+					delta: part.text
+				});
+			}
+		} },
+		["on_reasoning_delta"]: { handle: async (_event, data) => {
+			const item = await ensureReasoning(config);
+			for (const part of data.delta.content ?? []) {
+				let text;
+				if ("think" in part && typeof part.think === "string") text = part.think;
+				else if ("text" in part && typeof part.text === "string") text = part.text;
+				if (typeof text !== "string") continue;
+				item.content[0].text += text;
+				await writeResponseEvent(config.writer, {
+					type: "response.reasoning_text.delta",
+					sequence_number: config.tracker.nextSequence(),
+					item_id: item.id,
+					output_index: config.tracker.items.indexOf(item),
+					content_index: 0,
+					delta: text
+				});
+			}
+		} },
+		["on_run_step"]: { handle: async (_event, data) => {
+			const runStep = data;
+			if (runStep.stepDetails.type !== "tool_calls") return;
+			const toolCalls = runStep.stepDetails.tool_calls ?? [];
+			for (let index = 0; index < toolCalls.length; index++) await ensureFunctionCall(config, runStep.id, toolCalls[index], index);
+		} },
+		["on_run_step_delta"]: { handle: async (_event, data) => {
+			const runStepDelta = data;
+			if (runStepDelta.delta.type !== "tool_calls") return;
+			const toolCalls = runStepDelta.delta.tool_calls ?? [];
+			for (let index = 0; index < toolCalls.length; index++) await emitFunctionCallArgumentsDelta({
+				config,
+				item: await ensureFunctionCall(config, runStepDelta.id, toolCalls[index], index),
+				delta: getToolCallArguments(toolCalls[index])
+			});
+		} },
+		["on_run_step_completed"]: { handle: async (_event, data) => {
+			const completed = data;
+			if (!completed.result?.tool_call) return;
+			await completeFunctionCall(config, completed.result.id ?? "", completed.result.tool_call);
+		} },
+		["on_chat_model_end"]: { handle: (_event, data) => {
+			const usage = data?.output?.usage_metadata;
+			if (!usage) return;
+			config.tracker.usage.inputTokens += getTokenCount(usage.input_tokens);
+			config.tracker.usage.outputTokens += getTokenCount(usage.output_tokens);
+		} }
+	};
 }
 async function emitResponseCompleted(config) {
-    await ensureResponseCreated(config);
-    for (const item of config.tracker.items) {
-        if (item.status === 'completed') {
-            continue;
-        }
-        if (item.type === 'message' || item.type === 'reasoning') {
-            await emitOutputContentDone(config, item);
-        }
-        else {
-            await emitFunctionCallArgumentsDone(config, item);
-        }
-        item.status = 'completed';
-        await writeResponseEvent(config.writer, {
-            type: 'response.output_item.done',
-            sequence_number: config.tracker.nextSequence(),
-            output_index: config.tracker.items.indexOf(item),
-            item,
-        });
-    }
-    await writeResponseEvent(config.writer, {
-        type: 'response.completed',
-        sequence_number: config.tracker.nextSequence(),
-        response: buildResponse(config.context, config.tracker, 'completed'),
-    });
-    await writeResponsesDone(config.writer);
+	await ensureResponseCreated(config);
+	for (const item of config.tracker.items) {
+		if (item.status === "completed") continue;
+		if (item.type === "message" || item.type === "reasoning") await emitOutputContentDone(config, item);
+		else await emitFunctionCallArgumentsDone(config, item);
+		item.status = "completed";
+		await writeResponseEvent(config.writer, {
+			type: "response.output_item.done",
+			sequence_number: config.tracker.nextSequence(),
+			output_index: config.tracker.items.indexOf(item),
+			item
+		});
+	}
+	await writeResponseEvent(config.writer, {
+		type: "response.completed",
+		sequence_number: config.tracker.nextSequence(),
+		response: buildResponse(config.context, config.tracker, "completed")
+	});
+	await writeResponsesDone(config.writer);
 }
-
+//#endregion
 export { buildResponse, createResponseTracker, createResponsesEventHandlers, emitResponseCompleted, ensureResponseCreated, writeResponseEvent, writeResponsesDone };
-//# sourceMappingURL=index.mjs.map
+
+//# sourceMappingURL=index.mjs.map