test_deep_stream() — langchain Function Reference

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

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_deep_stream() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_runnable.py lines 3527–3548

def test_deep_stream() -> None:
    prompt = (
        SystemMessagePromptTemplate.from_template("You are a nice assistant.")
        + "{question}"
    )
    llm = FakeStreamingListLLM(responses=["foo-lish"])

    chain = prompt | llm | StrOutputParser()

    stream = chain.stream({"question": "What up"})

    chunks = list(stream)

    assert len(chunks) == len("foo-lish")
    assert "".join(chunks) == "foo-lish"

    chunks = []
    for chunk in (chain | RunnablePassthrough()).stream({"question": "What up"}):
        chunks.append(chunk)

    assert len(chunks) == len("foo-lish")
    assert "".join(chunks) == "foo-lish"