test_output_messages() — langchain Function Reference

Architecture documentation for the test_output_messages() function in test_history.py from the langchain codebase.

Function python LangChainCore Runnables calls 1

Entity Profile

LangChainCore→ Runnables→ test_output_messages() — langchain Function Reference

Dependency Diagram

graph TD
  e95a229c_6616_58ca_5c03_a48c0650f902["test_output_messages()"]
  0af0dfde_fe33_f3e5_b5e6_a3d18260dc67["test_history.py"]
  e95a229c_6616_58ca_5c03_a48c0650f902 -->|defined in| 0af0dfde_fe33_f3e5_b5e6_a3d18260dc67
  e035a3b1_d1a1_c458_c801_af1fc0272f8b["_get_get_session_history()"]
  e95a229c_6616_58ca_5c03_a48c0650f902 -->|calls| e035a3b1_d1a1_c458_c801_af1fc0272f8b
  style e95a229c_6616_58ca_5c03_a48c0650f902 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_history.py lines 330–357

def test_output_messages() -> None:
    runnable = RunnableLambda[Any, list[AIMessage]](
        lambda params: [
            AIMessage(
                content="you said: "
                + "\n".join(
                    [
                        str(m.content)
                        for m in params["history"]
                        if isinstance(m, HumanMessage)
                    ]
                    + [params["input"]]
                )
            )
        ]
    )
    get_session_history = _get_get_session_history()
    with_history = RunnableWithMessageHistory(
        runnable,
        get_session_history,
        input_messages_key="input",
        history_messages_key="history",
    )
    config: RunnableConfig = {"configurable": {"session_id": "6"}}
    output = with_history.invoke({"input": "hello"}, config)
    assert output == [AIMessage(content="you said: hello")]
    output = with_history.invoke({"input": "good bye"}, config)
    assert output == [AIMessage(content="you said: hello\ngood bye")]