abbyswag/qwen-code
1
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2025-12-20 16:57:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

update token usage with types

Browse Source
This commit is contained in:
jerop
2025-06-11 06:56:53 +00:00
committed by Jerop Kipruto
parent 9d992b32e4
commit 7ba2b13870
7 changed files with 120 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
packages/core/src/core/client.ts
View File

@@ -216,7 +216,7 @@ export class GeminiClient {
private _logApiRequest(model: string, inputTokenCount: number): void {
logApiRequest({
model,
prompt_token_count: inputTokenCount,
input_token_count: inputTokenCount,
duration_ms: 0, // Duration is not known at request time
});
}
@@ -245,6 +245,11 @@ export class GeminiClient {
attempt,
status_code: undefined,
error: responseError,
output_token_count: response.usageMetadata?.candidatesTokenCount ?? 0,
cached_content_token_count:
response.usageMetadata?.cachedContentTokenCount ?? 0,
thoughts_token_count: response.usageMetadata?.thoughtsTokenCount ?? 0,
tool_token_count: response.usageMetadata?.toolUsePromptTokenCount ?? 0,
});
}