Fixing anthropic model caching in roo-code

This commit is contained in:
kinandan
2025-01-29 17:35:31 +05:30
committed by Vignesh Subbiah
parent 117c4abfee
commit 26369ca48b

View File

@@ -106,10 +106,20 @@ export class UnboundHandler implements ApiHandler, SingleCompletionHandler {
}
if (usage) {
if (this.getModel().id.startsWith("anthropic/")) {
yield {
type: "usage",
inputTokens: usage?.prompt_tokens || 0,
outputTokens: usage?.completion_tokens || 0,
cacheWriteTokens: (usage as any)?.cache_creation_input_tokens || 0,
cacheReadTokens: (usage as any)?.cache_read_input_tokens || 0,
}
} else {
yield {
type: "usage",
inputTokens: usage?.prompt_tokens || 0,
outputTokens: usage?.completion_tokens || 0,
}
}
}
}