scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-08-28 05:27:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Yields the usage tokens

Browse source
This commit is contained in:
Vignesh Subbiah 2025-01-28 21:11:16 +05:30
parent 4008a1a53e
commit 5329c712de
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/api/providers/unbound.ts
View file

@ -96,6 +96,7 @@ export class UnboundHandler implements ApiHandler, SingleCompletionHandler {
for await (const chunk of completion) {
const delta = chunk.choices[0]?.delta
const usage = chunk.usage
if (delta?.content) {
yield {
@ -103,6 +104,14 @@ export class UnboundHandler implements ApiHandler, SingleCompletionHandler {
text: delta.content,
}
}
if (usage) {
yield {
type: "usage",
inputTokens: usage?.prompt_tokens || 0,
outputTokens: usage?.completion_tokens || 0,
}
}
}
}