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

Use openrouter generation endpoint for usage reporting

Browse Source
This commit is contained in:
Saoud Rizwan
2024-10-08 03:16:51 -04:00
parent e6d6ca5c56
commit 729e32514e
1 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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