@nanocollective/nanocoder/dist/usage/calculator.js

A local-first CLI coding agent that brings the power of agentic coding tools like Claude Code and Gemini CLI to local models or controlled APIs like OpenRouter

/**
 * Usage calculator
 * Calculates token breakdown by category
 */
import { TOKENS_PER_TOOL_ESTIMATE, USAGE_ERROR_THRESHOLD_PERCENT, USAGE_SUCCESS_THRESHOLD_PERCENT, } from '../constants.js';
/**
 * Calculate token breakdown from messages
 * @param messages - Messages to calculate breakdown for
 * @param tokenizer - Tokenizer instance (used as fallback if getTokens not provided)
 * @param getTokens - Optional cached token counting function for performance
 */
export function calculateTokenBreakdown(messages, tokenizer, getTokens) {
    const breakdown = {
        system: 0,
        userMessages: 0,
        assistantMessages: 0,
        toolDefinitions: 0,
        toolResults: 0,
        total: 0,
    };
    for (const message of messages) {
        const tokens = getTokens
            ? getTokens(message)
            : tokenizer.countTokens(message);
        switch (message.role) {
            case 'system':
                breakdown.system += tokens;
                break;
            case 'user':
                breakdown.userMessages += tokens;
                break;
            case 'assistant':
                breakdown.assistantMessages += tokens;
                break;
            case 'tool':
                breakdown.toolResults += tokens;
                break;
            default:
                // Unknown roles go to assistant messages
                breakdown.assistantMessages += tokens;
        }
    }
    // Calculate total
    breakdown.total =
        breakdown.system +
            breakdown.userMessages +
            breakdown.assistantMessages +
            breakdown.toolDefinitions +
            breakdown.toolResults;
    return breakdown;
}
/**
 * Calculate tool definitions token count
 * This estimates the tokens used by tool definitions sent to the model
 */
export function calculateToolDefinitionsTokens(toolCount) {
    // Rough estimate: each tool definition is about TOKENS_PER_TOOL_ESTIMATE tokens
    // This includes name, description, parameters schema
    return toolCount * TOKENS_PER_TOOL_ESTIMATE;
}
/**
 * Get status color based on percentage used
 */
export function getUsageStatusColor(percentUsed) {
    if (percentUsed < USAGE_SUCCESS_THRESHOLD_PERCENT) {
        return 'success';
    }
    else if (percentUsed < USAGE_ERROR_THRESHOLD_PERCENT) {
        return 'warning';
    }
    else {
        return 'error';
    }
}
/**
 * Format token count with thousands separator
 */
export function formatTokenCount(tokens) {
    return tokens.toLocaleString();
}
//# sourceMappingURL=calculator.js.map