Function format_token_count

libs/code/deepagents_code/_session_stats.py:139–152

Format a token count into a human-readable short string. Args: count: Number of tokens. Returns: Formatted string like `'12.5K'`, `'1.2M'`, or `'500'`.

(count: int)

def format_token_count(count: int) -> str:
    """Format a token count into a human-readable short string.

    Args:
        count: Number of tokens.

    Returns:
        Formatted string like `'12.5K'`, `'1.2M'`, or `'500'`.
    """
    if count >= 1_000_000:  # noqa: PLR2004
        return f"{count / 1_000_000:.1f}M"
    if count >= 1000:  # noqa: PLR2004
        return f"{count / 1000:.1f}K"
    return str(count)