test_event_stream_with_retriever_and_formatter() — langchain Function Reference
Architecture documentation for the test_event_stream_with_retriever_and_formatter() function in test_runnable_events_v2.py from the langchain codebase.
Entity Profile
Dependency Diagram
graph TD 783f5e81_685a_3bc9_4b43_195563a40885["test_event_stream_with_retriever_and_formatter()"] 33c02978_2077_5819_7048_bc2a81e80625["test_runnable_events_v2.py"] 783f5e81_685a_3bc9_4b43_195563a40885 -->|defined in| 33c02978_2077_5819_7048_bc2a81e80625 716d2a5e_dc8e_3cae_e044_b56b06bee655["_collect_events()"] 783f5e81_685a_3bc9_4b43_195563a40885 -->|calls| 716d2a5e_dc8e_3cae_e044_b56b06bee655 style 783f5e81_685a_3bc9_4b43_195563a40885 fill:#6366f1,stroke:#818cf8,color:#fff
Relationship Graph
Source Code
libs/core/tests/unit_tests/runnables/test_runnable_events_v2.py lines 1274–1387
async def test_event_stream_with_retriever_and_formatter() -> None:
"""Test the event stream with a retriever."""
retriever = HardCodedRetriever(
documents=[
Document(
page_content="hello world!",
metadata={"foo": "bar"},
),
Document(
page_content="goodbye world!",
metadata={"food": "spare"},
),
]
)
def format_docs(docs: list[Document]) -> str:
"""Format the docs."""
return ", ".join([doc.page_content for doc in docs])
chain = retriever | format_docs
events = await _collect_events(chain.astream_events("hello", version="v2"))
_assert_events_equal_allow_superset_metadata(
events,
[
{
"data": {"input": "hello"},
"event": "on_chain_start",
"metadata": {},
"name": "RunnableSequence",
"run_id": "",
"parent_ids": [],
"tags": [],
},
{
"data": {"input": {"query": "hello"}},
"event": "on_retriever_start",
"metadata": {},
"name": "HardCodedRetriever",
"run_id": "",
"parent_ids": [],
"tags": ["seq:step:1"],
},
{
"data": {
"input": {"query": "hello"},
"output": [
Document(page_content="hello world!", metadata={"foo": "bar"}),
Document(
page_content="goodbye world!", metadata={"food": "spare"}
),
],
},
"event": "on_retriever_end",
"metadata": {},
"name": "HardCodedRetriever",
"run_id": "",
"parent_ids": [],
"tags": ["seq:step:1"],
},
{
"data": {},
"event": "on_chain_start",
"metadata": {},
"name": "format_docs",
"run_id": "",
"parent_ids": [],
"tags": ["seq:step:2"],
},
{
"data": {"chunk": "hello world!, goodbye world!"},
"event": "on_chain_stream",
"metadata": {},
"name": "format_docs",
"run_id": "",
"parent_ids": [],
"tags": ["seq:step:2"],
},
{
"data": {"chunk": "hello world!, goodbye world!"},
"event": "on_chain_stream",
"metadata": {},
Domain
Subdomains
Calls
Source
Frequently Asked Questions
What does test_event_stream_with_retriever_and_formatter() do?
test_event_stream_with_retriever_and_formatter() is a function in the langchain codebase, defined in libs/core/tests/unit_tests/runnables/test_runnable_events_v2.py.
Where is test_event_stream_with_retriever_and_formatter() defined?
test_event_stream_with_retriever_and_formatter() is defined in libs/core/tests/unit_tests/runnables/test_runnable_events_v2.py at line 1274.
What does test_event_stream_with_retriever_and_formatter() call?
test_event_stream_with_retriever_and_formatter() calls 1 function(s): _collect_events.
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free