test_prompt() — langchain Function Reference

Architecture documentation for the test_prompt() function in test_runnable.py from the langchain codebase.

Function python LangChainCore Runnables calls 2

Entity Profile

LangChainCore→ Runnables→ test_prompt() — langchain Function Reference

Dependency Diagram

graph TD
  275ec322_b94e_59cc_b503_f5f88356efb1["test_prompt()"]
  26df6ad8_0189_51d0_c3c1_6c3248893ff5["test_runnable.py"]
  275ec322_b94e_59cc_b503_f5f88356efb1 -->|defined in| 26df6ad8_0189_51d0_c3c1_6c3248893ff5
  593bb72a_4695_6c93_b95c_e277aca006ae["batch()"]
  275ec322_b94e_59cc_b503_f5f88356efb1 -->|calls| 593bb72a_4695_6c93_b95c_e277aca006ae
  a3895c50_dab4_0c36_2d24_f63121e198a0["invoke()"]
  275ec322_b94e_59cc_b503_f5f88356efb1 -->|calls| a3895c50_dab4_0c36_2d24_f63121e198a0
  style 275ec322_b94e_59cc_b503_f5f88356efb1 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_runnable.py lines 1451–1482

def test_prompt() -> None:
    prompt = ChatPromptTemplate.from_messages(
        messages=[
            SystemMessage(content="You are a nice assistant."),
            HumanMessagePromptTemplate.from_template("{question}"),
        ]
    )
    expected = ChatPromptValue(
        messages=[
            SystemMessage(content="You are a nice assistant."),
            HumanMessage(content="What is your name?"),
        ]
    )

    assert prompt.invoke({"question": "What is your name?"}) == expected

    assert prompt.batch(
        [
            {"question": "What is your name?"},
            {"question": "What is your favorite color?"},
        ]
    ) == [
        expected,
        ChatPromptValue(
            messages=[
                SystemMessage(content="You are a nice assistant."),
                HumanMessage(content="What is your favorite color?"),
            ]
        ),
    ]

    assert [*prompt.stream({"question": "What is your name?"})] == [expected]