test_get_output_schema() — langchain Function Reference

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

Function python LangChainCore Runnables calls 1

Entity Profile

LangChainCore→ Runnables→ test_get_output_schema() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_history.py lines 490–523

def test_get_output_schema() -> None:
    """Test get output schema."""
    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",
    )
    output_type = with_history.get_output_schema()

    expected_schema: dict[str, Any] = {
        "title": "RunnableWithChatHistoryOutput",
        "type": "object",
    }
    assert _schema(output_type) == expected_schema