Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
lgrammel committed Sep 20, 2024
1 parent ba94c43 commit 94d34f2
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions packages/openai/src/openai-chat-language-model.ts
Original file line number Diff line number Diff line change
Expand Up @@ -278,6 +278,8 @@ export class OpenAIChatLanguageModel implements LanguageModelV1 {
const { messages: rawPrompt, ...rawSettings } = args;
const choice = response.choices[0];

console.log(response.usage?.completion_tokens_details);

return {
text: choice.message.content ?? undefined,
toolCalls:
Expand Down Expand Up @@ -568,6 +570,11 @@ const openAITokenUsageSchema = z
.object({
prompt_tokens: z.number().nullish(),
completion_tokens: z.number().nullish(),
completion_tokens_details: z
.object({
reasoning_tokens: z.number().nullish(),
})
.nullish(),
})
.nullish();

Expand Down

0 comments on commit 94d34f2

Please sign in to comment.