test_output_message() — langchain Function Reference

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

Function python LangChainCore Runnables calls 1

Entity Profile

LangChainCore→ Runnables→ test_output_message() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_history.py lines 225–250

def test_output_message() -> None:
    runnable = RunnableLambda[Any, 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": "4"}}
    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")