TestDumpdInKwargs Class — langchain Architecture

Source Code

libs/core/tests/unit_tests/load/test_secret_injection.py lines 243–269

class TestDumpdInKwargs:
    """Tests for the specific pattern of `dumpd()` result stored in kwargs."""

    def test_dumpd_human_message_in_ai_message_kwargs(self) -> None:
        """`AIMessage` with `dumpd(HumanMessage)` in `additional_kwargs`."""
        h = HumanMessage("Hello")
        a = AIMessage("foo", additional_kwargs={"bar": [dumpd(h)]})
        _assert_no_secret_leak(a)

    def test_dumpd_human_message_with_secret_in_ai_message_kwargs(self) -> None:
        """`AIMessage` with `dumpd(HumanMessage w/ secret)` in `additional_kwargs`."""
        h = HumanMessage(
            "Hello",
            additional_kwargs={"secret": MALICIOUS_SECRET_DICT},
        )
        a = AIMessage("foo", additional_kwargs={"bar": [dumpd(h)]})
        _assert_no_secret_leak(a)

    def test_double_dumpd_nesting(self) -> None:
        """Double nesting: `dumpd(AIMessage(dumpd(HumanMessage)))`."""
        h = HumanMessage(
            "Hello",
            additional_kwargs={"secret": MALICIOUS_SECRET_DICT},
        )
        a = AIMessage("foo", additional_kwargs={"bar": [dumpd(h)]})
        outer = AIMessage("outer", additional_kwargs={"nested": [dumpd(a)]})
        _assert_no_secret_leak(outer)