Function calculate_usage_stats

api_utils/utils_ext/tokens.py:20–41

(
    messages: List[Dict[str, Any]],
    response_content: str,
    reasoning_content: Optional[str] = None,
)

def calculate_usage_stats(
    messages: List[Dict[str, Any]],
    response_content: str,
    reasoning_content: Optional[str] = None,
) -> Dict[str, int]:
    prompt_text = ""
    for message in messages:
        role = message.get("role", "")
        content = message.get("content", "")
        prompt_text += f"{role}: {content}\n"
    prompt_tokens = estimate_tokens(prompt_text)

    completion_text = response_content or ""
    if reasoning_content:
        completion_text += reasoning_content
    completion_tokens = estimate_tokens(completion_text)
    total_tokens = prompt_tokens + completion_tokens
    return {
        "prompt_tokens": prompt_tokens,
        "completion_tokens": completion_tokens,
        "total_tokens": total_tokens,
    }