Skip to content

Commit 27830f2

Browse files
committed
fix: update Task.ts to use new calculateApiCost signature
The cost calculation functions now return a number directly instead of an object with totalCost, totalInputTokens, and totalOutputTokens properties.
1 parent b69e2b5 commit 27830f2

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/core/task/Task.ts

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2003,11 +2003,11 @@ export class Task extends EventEmitter<TaskEvents> implements TaskLike {
20032003

20042004
this.clineMessages[lastApiReqIndex].text = JSON.stringify({
20052005
...existingData,
2006-
tokensIn: costResult.totalInputTokens,
2007-
tokensOut: costResult.totalOutputTokens,
2006+
tokensIn: inputTokens,
2007+
tokensOut: outputTokens,
20082008
cacheWrites: cacheWriteTokens,
20092009
cacheReads: cacheReadTokens,
2010-
cost: totalCost ?? costResult.totalCost,
2010+
cost: totalCost ?? costResult,
20112011
cancelReason,
20122012
streamingFailedMessage,
20132013
} satisfies ClineApiReqInfo)
@@ -2234,11 +2234,11 @@ export class Task extends EventEmitter<TaskEvents> implements TaskLike {
22342234
)
22352235

22362236
TelemetryService.instance.captureLlmCompletion(this.taskId, {
2237-
inputTokens: costResult.totalInputTokens,
2238-
outputTokens: costResult.totalOutputTokens,
2237+
inputTokens: tokens.input,
2238+
outputTokens: tokens.output,
22392239
cacheWriteTokens: tokens.cacheWrite,
22402240
cacheReadTokens: tokens.cacheRead,
2241-
cost: tokens.total ?? costResult.totalCost,
2241+
cost: tokens.total ?? costResult,
22422242
})
22432243
}
22442244
}

0 commit comments

Comments
 (0)