test_serde_any_message() — langchain Function Reference

Architecture documentation for the test_serde_any_message() function in test_pydantic_serde.py from the langchain codebase.

Function python CoreAbstractions MessageSchema

Entity Profile

CoreAbstractions→ MessageSchema→ test_serde_any_message() — langchain Function Reference

Dependency Diagram

graph TD
  b92694b2_2b2f_2b45_139d_29a7fe388eab["test_serde_any_message()"]
  0d9a712e_1c82_221f_c74a_e9b7382702a5["test_pydantic_serde.py"]
  b92694b2_2b2f_2b45_139d_29a7fe388eab -->|defined in| 0d9a712e_1c82_221f_c74a_e9b7382702a5
  style b92694b2_2b2f_2b45_139d_29a7fe388eab fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/test_pydantic_serde.py lines 28–65

def test_serde_any_message() -> None:
    """Test AnyMessage() serder."""
    lc_objects = [
        HumanMessage(content="human"),
        HumanMessageChunk(content="human"),
        AIMessage(content="ai"),
        AIMessageChunk(content="ai"),
        SystemMessage(content="sys"),
        SystemMessageChunk(content="sys"),
        FunctionMessage(
            name="func",
            content="func",
        ),
        FunctionMessageChunk(
            name="func",
            content="func",
        ),
        ChatMessage(
            role="human",
            content="human",
        ),
        ChatMessageChunk(
            role="human",
            content="human",
        ),
    ]

    model = RootModel[AnyMessage]

    for lc_object in lc_objects:
        d = lc_object.model_dump()
        assert "type" in d, f"Missing key `type` for {type(lc_object)}"
        obj1 = model.model_validate(d)
        assert type(obj1.root) is type(lc_object), f"failed for {type(lc_object)}"

    with pytest.raises((TypeError, ValidationError)):
        # Make sure that specifically validation error is raised
        model.model_validate({})