admin-valentin/AIClient-2-API
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #421 from yin1245/fix/per-model-context-tokens

Browse source 
fix: use per-model context window size for accurate input_tokens estimation
This commit is contained in:
何夕2077 2026-03-29 19:50:30 +08:00 committed by GitHub
commit b3b78d2fa5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/providers/claude/claude-kiro.js
View file

@ -47,6 +47,21 @@ const KIRO_CONSTANTS = {
TOTAL_CONTEXT_TOKENS: 200000, // Claude Sonnet 4.5 actual context is 200K
};
// Per-model context window sizes for accurate token estimation
const MODEL_CONTEXT_TOKENS = {
"claude-opus-4-6": 1000000,
"claude-opus-4-5": 1000000,
"claude-opus-4-5-20251101": 1000000,
"claude-sonnet-4-6": 200000,
"claude-sonnet-4-5": 200000,
"claude-sonnet-4-5-20250929": 200000,
"claude-haiku-4-5": 200000,
"claude-haiku-4-5-20251001": 200000,
};
function getContextTokensForModel(model) {
return MODEL_CONTEXT_TOKENS[model] || KIRO_CONSTANTS.TOTAL_CONTEXT_TOKENS;
}
// 从 provider-models.js 获取支持的模型列表
const KIRO_MODELS = getProviderModels(MODEL_PROVIDER.KIRO_API);
@ -2631,7 +2646,8 @@ async saveCredentialsToFile(filePath, newData) {
// 总 token = TOTAL_CONTEXT_TOKENS * contextUsagePercentage / 100
// input token = 总 token - output token
if (contextUsagePercentage !== null && contextUsagePercentage > 0) {
const totalTokens = Math.round(KIRO_CONSTANTS.TOTAL_CONTEXT_TOKENS * contextUsagePercentage / 100);
const contextTokens = getContextTokensForModel(finalModel);
const totalTokens = Math.round(contextTokens * contextUsagePercentage / 100);
inputTokens = Math.max(0, totalTokens - outputTokens);
logger.info(`[Kiro] Token calculation from contextUsagePercentage: total=${totalTokens}, output=${outputTokens}, input=${inputTokens}`);
} else {