github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 23:29:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: Add unit test for RunnableGenerator for eventstream v2 (#21990)

Browse Source 
No unit tests with runnable generator
This commit is contained in:
Eugene Yurtsev 2024-05-21 14:29:15 -04:00 committed by GitHub
parent fb6108c8f5
commit ded53297e0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 46 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
libs/core/tests/unit_tests/runnables/test_runnable_events_v2.py
View File

@ -35,6 +35,7 @@ from langchain_core.runnables import (
ConfigurableField,
Runnable,
RunnableConfig,
RunnableGenerator,
RunnableLambda,
ensure_config,
)
@ -1830,3 +1831,48 @@ async def test_astream_events_from_custom_runnable() -> None:
"tags": [],
},
]
async def test_runnable_generator() -> None:
"""Test async events from sync lambda."""
async def generator(inputs: AsyncIterator[str]) -> AsyncIterator[str]:
yield "1"
yield "2"
runnable: Runnable[str, str] = RunnableGenerator(transform=generator)
events = await _collect_events(runnable.astream_events("hello", version="v2"))
assert events == [
{
"data": {"input": "hello"},
"event": "on_chain_start",
"metadata": {},
"name": "generator",
"run_id": "",
"tags": [],
},
{
"data": {"chunk": "1"},
"event": "on_chain_stream",
"metadata": {},
"name": "generator",
"run_id": "",
"tags": [],
},
{
"data": {"chunk": "2"},
"event": "on_chain_stream",
"metadata": {},
"name": "generator",
"run_id": "",
"tags": [],
},
{
"data": {"output": "12"},
"event": "on_chain_end",
"metadata": {},
"name": "generator",
"run_id": "",
"tags": [],
},
]