pacnpal/Roo-Code
1
0
Fork 0
mirror of https://github.com/pacnpal/Roo-Code.git synced 2025-12-20 20:31:37 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fixes

Browse Source
This commit is contained in:
Saoud Rizwan
2024-10-07 18:37:06 -04:00
parent 4ea6d17997
commit da505170d4
3 changed files with 41 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/api/providers/openrouter.ts
View File

@@ -107,6 +107,13 @@ export class OpenRouterHandler implements ApiHandler {
text: delta.content,
}
}
if (chunk.usage) {
yield {
type: "usage",
inputTokens: chunk.usage.prompt_tokens || 0,
outputTokens: chunk.usage.completion_tokens || 0,
}
}
}
try {
@@ -121,10 +128,13 @@ export class OpenRouterHandler implements ApiHandler {
console.log("OpenRouter generation details:", response.data)
yield {
type: "usage",
inputTokens: generation?.native_tokens_prompt || 0,
outputTokens: generation?.native_tokens_completion || 0,
// cacheWriteTokens: 0,
// cacheReadTokens: 0,
// openrouter generation endpoint fails often, so we'll report tokens from stream as normal
// inputTokens: generation?.native_tokens_prompt || 0,
// outputTokens: generation?.native_tokens_completion || 0,
inputTokens: 0,
outputTokens: 0,
totalCost: generation?.total_cost || 0,
}
} catch (error) {