test_add_ai_message_chunks_usage() — langchain Function Reference

Architecture documentation for the test_add_ai_message_chunks_usage() function in test_ai.py from the langchain codebase.

Function python LangChainCore MessageInterface

Entity Profile

LangChainCore→ MessageInterface→ test_add_ai_message_chunks_usage() — langchain Function Reference

Dependency Diagram

graph TD
  202bf1f5_6c7c_3591_d57e_9a4ca1812416["test_add_ai_message_chunks_usage()"]
  f148f3ce_73bd_627d_a979_33d23ebac843["test_ai.py"]
  202bf1f5_6c7c_3591_d57e_9a4ca1812416 -->|defined in| f148f3ce_73bd_627d_a979_33d23ebac843
  style 202bf1f5_6c7c_3591_d57e_9a4ca1812416 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/messages/test_ai.py lines 171–201

def test_add_ai_message_chunks_usage() -> None:
    chunks = [
        AIMessageChunk(content="", usage_metadata=None),
        AIMessageChunk(
            content="",
            usage_metadata=UsageMetadata(
                input_tokens=2, output_tokens=3, total_tokens=5
            ),
        ),
        AIMessageChunk(
            content="",
            usage_metadata=UsageMetadata(
                input_tokens=2,
                output_tokens=3,
                total_tokens=5,
                input_token_details=InputTokenDetails(audio=1, cache_read=1),
                output_token_details=OutputTokenDetails(audio=1, reasoning=2),
            ),
        ),
    ]
    combined = add_ai_message_chunks(*chunks)
    assert combined == AIMessageChunk(
        content="",
        usage_metadata=UsageMetadata(
            input_tokens=4,
            output_tokens=6,
            total_tokens=10,
            input_token_details=InputTokenDetails(audio=1, cache_read=1),
            output_token_details=OutputTokenDetails(audio=1, reasoning=2),
        ),
    )