Quellcode durchsuchen

Fixing anthropic model caching in roo-code

kinandan vor 11 Monaten
Ursprung
Commit
26369ca48b
1 geänderte Dateien mit 14 neuen und 4 gelöschten Zeilen
  1. 14 4
      src/api/providers/unbound.ts

+ 14 - 4
src/api/providers/unbound.ts

@@ -106,10 +106,20 @@ export class UnboundHandler implements ApiHandler, SingleCompletionHandler {
 			}
 
 			if (usage) {
-				yield {
-					type: "usage",
-					inputTokens: usage?.prompt_tokens || 0,
-					outputTokens: usage?.completion_tokens || 0,
+				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,
+					}
 				}
 			}
 		}