test_chat_message_chunks() — langchain Function Reference

Architecture documentation for the test_chat_message_chunks() function in test_messages.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_chat_message_chunks() — langchain Function Reference

Dependency Diagram

graph TD
  7752786c_1168_a89e_d1fa_ee9bf52610d9["test_chat_message_chunks()"]
  1c821c4f_25a2_9606_a15f_b3df054105ac["test_messages.py"]
  7752786c_1168_a89e_d1fa_ee9bf52610d9 -->|defined in| 1c821c4f_25a2_9606_a15f_b3df054105ac
  style 7752786c_1168_a89e_d1fa_ee9bf52610d9 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/test_messages.py lines 200–226

def test_chat_message_chunks() -> None:
    assert ChatMessageChunk(role="User", content="I am", id="ai4") + ChatMessageChunk(
        role="User", content=" indeed."
    ) == ChatMessageChunk(id="ai4", role="User", content="I am indeed."), (
        "ChatMessageChunk + ChatMessageChunk should be a ChatMessageChunk"
    )

    with pytest.raises(
        ValueError, match="Cannot concatenate ChatMessageChunks with different roles"
    ):
        ChatMessageChunk(role="User", content="I am") + ChatMessageChunk(
            role="Assistant", content=" indeed."
        )

    assert ChatMessageChunk(role="User", content="I am") + AIMessageChunk(
        content=" indeed."
    ) == ChatMessageChunk(role="User", content="I am indeed."), (
        "ChatMessageChunk + other MessageChunk should be a ChatMessageChunk "
        "with the left side's role"
    )

    assert AIMessageChunk(content="I am") + ChatMessageChunk(
        role="User", content=" indeed."
    ) == AIMessageChunk(content="I am indeed."), (
        "Other MessageChunk + ChatMessageChunk should be a MessageChunk "
        "as the left side"
    )