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

Requesty: Only report final usage (#3542)

Browse source
This commit is contained in:
Daniel Trugman 2025-05-13 12:24:28 +01:00 • committed by GitHub
parent f6cae4df05
commit a14b655590
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
.changeset/twelve-ants-scream.md Normal file
View file

@ -0,0 +1,5 @@
---
"roo-cline": patch
---
Requesty: Fix token usage counts

8
src/api/providers/requesty.ts
View file

@ -101,6 +101,8 @@ export class RequestyHandler extends BaseProvider implements SingleCompletionHan
const stream = await this.client.chat.completions.create(completionParams)
let lastUsage: any = undefined
for await (const chunk of stream) {
const delta = chunk.choices[0]?.delta
if (delta?.content) {
@ -118,9 +120,13 @@ export class RequestyHandler extends BaseProvider implements SingleCompletionHan
}
if (chunk.usage) {
yield this.processUsageMetrics(chunk.usage, model.info)
lastUsage = chunk.usage
}
}
if (lastUsage) {
yield this.processUsageMetrics(lastUsage, model.info)
}
}
async completePrompt(prompt: string): Promise<string> {