github/claude-relay-service
1
0
Fork 0
mirror of https://github.com/Wei-Shaw/claude-relay-service.git synced 2026-01-22 16:43:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

添加claude账号维度计算token费用前端修复

Browse Source
This commit is contained in:
leslie
2025-07-25 21:48:54 +08:00
parent 1cf70a627f
commit 53e0577e19
2 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/routes/openaiClaudeRoutes.js
View File

@@ -258,7 +258,8 @@ async function handleChatCompletion(req, res, apiKeyData) {
outputTokens,
cacheCreateTokens,
cacheReadTokens,
model
model,
accountId
).catch(error => {
logger.error('❌ Failed to record usage:', error);
});
@@ -327,7 +328,8 @@ async function handleChatCompletion(req, res, apiKeyData) {
usage.output_tokens || 0,
usage.cache_creation_input_tokens || 0,
usage.cache_read_input_tokens || 0,
claudeRequest.model
claudeRequest.model,
accountId
).catch(error => {
logger.error('❌ Failed to record usage:', error);
});

15
src/services/apiKeyService.js
View File

@@ -249,11 +249,16 @@ class ApiKeyService {
keyData.lastUsedAt = new Date().toISOString();
await redis.setApiKey(keyId, keyData);
// 记录账户级别的使用统计
const claudeAccountId = accountId || keyData.claudeAccountId;
if (claudeAccountId) {
await redis.incrementAccountUsage(claudeAccountId, totalTokens, inputTokens, outputTokens, cacheCreateTokens, cacheReadTokens, model);
logger.database(`📊 Recorded account usage: ${claudeAccountId} - ${totalTokens} tokens`);
// 记录账户级别的使用统计（只统计实际处理请求的账户）
if (accountId) {
await redis.incrementAccountUsage(accountId, totalTokens, inputTokens, outputTokens, cacheCreateTokens, cacheReadTokens, model);
logger.database(`📊 Recorded account usage: ${accountId} - ${totalTokens} tokens (API Key: ${keyId})`);
} else if (keyData.claudeAccountId) {
// 如果没有传入accountId，但API Key绑定了专属账户，也记录统计
await redis.incrementAccountUsage(keyData.claudeAccountId, totalTokens, inputTokens, outputTokens, cacheCreateTokens, cacheReadTokens, model);
logger.database(`📊 Recorded account usage (from API Key binding): ${keyData.claudeAccountId} - ${totalTokens} tokens (API Key: ${keyId})`);
} else {
logger.debug(`⚠️ No accountId provided and API Key not bound to account, skipping account-level statistics`);
}
}