_update_token_usage() — langchain Function Reference

Architecture documentation for the _update_token_usage() function in base.py from the langchain codebase.

Function python LangChainCore LanguageModelBase called by 2

Entity Profile

LangChainCore→ LanguageModelBase→ _update_token_usage() — langchain Function Reference

Dependency Diagram

graph TD
  7817fc40_57c0_6468_1c16_3ded8817a9b8["_update_token_usage()"]
  bcd5af42_a82b_f85e_ed42_ee1d1f435ba0["base.py"]
  7817fc40_57c0_6468_1c16_3ded8817a9b8 -->|defined in| bcd5af42_a82b_f85e_ed42_ee1d1f435ba0
  f5892a19_915a_df0a_fbd3_f7a5a0993b64["_generate()"]
  f5892a19_915a_df0a_fbd3_f7a5a0993b64 -->|calls| 7817fc40_57c0_6468_1c16_3ded8817a9b8
  2d321de2_cda4_2af7_7550_f3f179b1dddb["_agenerate()"]
  2d321de2_cda4_2af7_7550_f3f179b1dddb -->|calls| 7817fc40_57c0_6468_1c16_3ded8817a9b8
  style 7817fc40_57c0_6468_1c16_3ded8817a9b8 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/openai/langchain_openai/llms/base.py lines 26–35

def _update_token_usage(
    keys: set[str], response: dict[str, Any], token_usage: dict[str, Any]
) -> None:
    """Update token usage."""
    _keys_to_use = keys.intersection(response["usage"])
    for _key in _keys_to_use:
        if _key not in token_usage:
            token_usage[_key] = response["usage"][_key]
        else:
            token_usage[_key] += response["usage"][_key]