test_schema_with_itemgetter() — langchain Function Reference

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

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_schema_with_itemgetter() — langchain Function Reference

Dependency Diagram

graph TD
  cb65bf9b_721b_a0a9_a048_5949e6782e09["test_schema_with_itemgetter()"]
  26df6ad8_0189_51d0_c3c1_6c3248893ff5["test_runnable.py"]
  cb65bf9b_721b_a0a9_a048_5949e6782e09 -->|defined in| 26df6ad8_0189_51d0_c3c1_6c3248893ff5
  style cb65bf9b_721b_a0a9_a048_5949e6782e09 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_runnable.py lines 666–682

def test_schema_with_itemgetter() -> None:
    """Test runnable with itemgetter."""
    foo = RunnableLambda(itemgetter("hello"))
    assert _schema(foo.input_schema) == {
        "properties": {"hello": {"title": "Hello"}},
        "required": ["hello"],
        "title": "RunnableLambdaInput",
        "type": "object",
    }
    prompt = ChatPromptTemplate.from_template("what is {language}?")
    chain: Runnable = {"language": itemgetter("language")} | prompt
    assert _schema(chain.input_schema) == {
        "properties": {"language": {"title": "Language"}},
        "required": ["language"],
        "title": "RunnableParallel<language>Input",
        "type": "object",
    }