TestSerializableTopLevel Class — langchain Architecture

Source Code

libs/core/tests/unit_tests/load/test_secret_injection.py lines 65–128

class TestSerializableTopLevel:
    """Tests with `Serializable` objects at the top level."""

    def test_human_message_with_secret_in_content(self) -> None:
        """`HumanMessage` with secret-like dict in `content`."""
        msg = HumanMessage(
            content=[
                {"type": "text", "text": "Hello"},
                {"type": "text", "text": MALICIOUS_SECRET_DICT},
            ]
        )
        _assert_no_secret_leak(msg)

    def test_human_message_with_secret_in_additional_kwargs(self) -> None:
        """`HumanMessage` with secret-like dict in `additional_kwargs`."""
        msg = HumanMessage(
            content="Hello",
            additional_kwargs={"data": MALICIOUS_SECRET_DICT},
        )
        _assert_no_secret_leak(msg)

    def test_human_message_with_secret_in_nested_additional_kwargs(self) -> None:
        """`HumanMessage` with secret-like dict nested in `additional_kwargs`."""
        msg = HumanMessage(
            content="Hello",
            additional_kwargs={"nested": {"deep": MALICIOUS_SECRET_DICT}},
        )
        _assert_no_secret_leak(msg)

    def test_human_message_with_secret_in_list_in_additional_kwargs(self) -> None:
        """`HumanMessage` with secret-like dict in a list in `additional_kwargs`."""
        msg = HumanMessage(
            content="Hello",
            additional_kwargs={"items": [MALICIOUS_SECRET_DICT]},
        )
        _assert_no_secret_leak(msg)

    def test_ai_message_with_secret_in_response_metadata(self) -> None:
        """`AIMessage` with secret-like dict in respo`nse_metadata."""
        msg = AIMessage(
            content="Hello",
            response_metadata={"data": MALICIOUS_SECRET_DICT},
        )
        _assert_no_secret_leak(msg)

    def test_document_with_secret_in_metadata(self) -> None:
        """Document with secret-like dict in `metadata`."""
        doc = Document(
            page_content="Hello",
            metadata={"data": MALICIOUS_SECRET_DICT},
        )
        _assert_no_secret_leak(doc)

    def test_nested_serializable_with_secret(self) -> None:
        """`AIMessage` containing `dumpd(HumanMessage)` with secret in kwargs."""
        inner = HumanMessage(
            content="Hello",
            additional_kwargs={"secret": MALICIOUS_SECRET_DICT},
        )
        outer = AIMessage(
            content="Outer",
            additional_kwargs={"nested": [dumpd(inner)]},
        )
        _assert_no_secret_leak(outer)