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

14
packages/core/src/telemetry/loggers.ts
View File

@@ -25,7 +25,7 @@ import {
} from './types.js';
import {
recordApiErrorMetrics,
recordApiRequestMetrics,
recordTokenUsageMetrics,
recordApiResponseMetrics,
recordToolCallMetrics,
} from './metrics.js';
@@ -120,11 +120,11 @@ export function logApiRequest(
};
const logger = logs.getLogger(SERVICE_NAME);
const logRecord: LogRecord = {
body: `API request to ${event.model}. Tokens: ${event.prompt_token_count}.`,
body: `API request to ${event.model}. Tokens: ${event.input_token_count}.`,
attributes,
};
logger.emit(logRecord);
recordApiRequestMetrics(event.model, event.prompt_token_count);
recordTokenUsageMetrics(event.model, event.input_token_count, 'input');
}
export function logApiError(
@@ -188,4 +188,12 @@ export function logApiResponse(
event.status_code,
event.error,
);
recordTokenUsageMetrics(event.model, event.output_token_count, 'output');
recordTokenUsageMetrics(
event.model,
event.cached_content_token_count,
'cache',
);
recordTokenUsageMetrics(event.model, event.thoughts_token_count, 'thought');
recordTokenUsageMetrics(event.model, event.tool_token_count, 'tool');
}