test_output_dict() — langchain Function Reference

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

Function python LangChainCore Runnables calls 1

Entity Profile

LangChainCore→ Runnables→ test_output_dict() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_history.py lines 390–420

def test_output_dict() -> None:
    runnable = RunnableLambda[Any, dict[str, list[AIMessage]]](
        lambda params: {
            "output": [
                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",
        output_messages_key="output",
    )
    config: RunnableConfig = {"configurable": {"session_id": "7"}}
    output = with_history.invoke({"input": "hello"}, config)
    assert output == {"output": [AIMessage(content="you said: hello")]}
    output = with_history.invoke({"input": "good bye"}, config)
    assert output == {"output": [AIMessage(content="you said: hello\ngood bye")]}