diff --git a/packages/openai/src/chat.ts b/packages/openai/src/chat.ts index 5ee4e96ff..b32ca8650 100644 --- a/packages/openai/src/chat.ts +++ b/packages/openai/src/chat.ts @@ -203,6 +203,13 @@ export class OpenAIChatModel implements IChatModel { if (!(completion instanceof Stream)) { message = completion.choices[0].message; + if (completion.usage) { + this._log.debug('Token usage', { + prompt_tokens: completion.usage.prompt_tokens, + completion_tokens: completion.usage.completion_tokens, + total_tokens: completion.usage.total_tokens, + }); + } } else { for await (const chunk of completion) { if (!chunk.choices.length) continue;