test_chat_input_schema() — langchain Function Reference

Architecture documentation for the test_chat_input_schema() function in test_chat.py from the langchain codebase.

Function python CoreAbstractions MessageSchema

Entity Profile

CoreAbstractions→ MessageSchema→ test_chat_input_schema() — langchain Function Reference

Dependency Diagram

graph TD
  9590ffb1_3bf2_072d_a68b_d37eaad4a97e["test_chat_input_schema()"]
  bf3138ca_0068_68ea_f479_35376e264d13["test_chat.py"]
  9590ffb1_3bf2_072d_a68b_d37eaad4a97e -->|defined in| bf3138ca_0068_68ea_f479_35376e264d13
  style 9590ffb1_3bf2_072d_a68b_d37eaad4a97e fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/prompts/test_chat.py lines 1169–1192

def test_chat_input_schema(snapshot: SnapshotAssertion) -> None:
    prompt_all_required = ChatPromptTemplate(
        messages=[MessagesPlaceholder("history", optional=False), ("user", "${input}")]
    )
    assert set(prompt_all_required.input_variables) == {"input", "history"}
    assert prompt_all_required.optional_variables == []
    with pytest.raises(ValidationError):
        prompt_all_required.input_schema(input="")

    if version.parse("2.10") <= PYDANTIC_VERSION:
        assert _normalize_schema(
            prompt_all_required.get_input_jsonschema()
        ) == snapshot(name="required")
    prompt_optional = ChatPromptTemplate(
        messages=[MessagesPlaceholder("history", optional=True), ("user", "${input}")]
    )
    # input variables only lists required variables
    assert set(prompt_optional.input_variables) == {"input"}
    prompt_optional.input_schema(input="")  # won't raise error

    if version.parse("2.10") <= PYDANTIC_VERSION:
        assert _normalize_schema(prompt_optional.get_input_jsonschema()) == snapshot(
            name="partial"
        )