test_runnable_branch_stream() — langchain Function Reference

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

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_runnable_branch_stream() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_runnable.py lines 4669–4681

def test_runnable_branch_stream() -> None:
    """Verify that stream works for RunnableBranch."""
    llm_res = "i'm a textbot"
    # sleep to better simulate a real stream
    llm = FakeStreamingListLLM(responses=[llm_res], sleep=0.01)

    branch = RunnableBranch[str, Any](
        (lambda x: x == "hello", llm),
        lambda x: x,
    )

    assert list(branch.stream("hello")) == list(llm_res)
    assert list(branch.stream("bye")) == ["bye"]