|
|
|
@ -396,14 +396,6 @@ async def test_event_stream_with_simple_chain() -> None:
|
|
|
|
|
"run_id": "",
|
|
|
|
|
"tags": ["my_chain", "my_model", "seq:step:2"],
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"data": {"chunk": AIMessageChunk(content="hello")},
|
|
|
|
|
"event": "on_chain_stream",
|
|
|
|
|
"metadata": {"foo": "bar"},
|
|
|
|
|
"name": "my_chain",
|
|
|
|
|
"run_id": "",
|
|
|
|
|
"tags": ["my_chain"],
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"data": {"chunk": AIMessageChunk(content="hello")},
|
|
|
|
|
"event": "on_chat_model_stream",
|
|
|
|
@ -413,7 +405,7 @@ async def test_event_stream_with_simple_chain() -> None:
|
|
|
|
|
"tags": ["my_chain", "my_model", "seq:step:2"],
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"data": {"chunk": AIMessageChunk(content=" ")},
|
|
|
|
|
"data": {"chunk": AIMessageChunk(content="hello")},
|
|
|
|
|
"event": "on_chain_stream",
|
|
|
|
|
"metadata": {"foo": "bar"},
|
|
|
|
|
"name": "my_chain",
|
|
|
|
@ -429,7 +421,7 @@ async def test_event_stream_with_simple_chain() -> None:
|
|
|
|
|
"tags": ["my_chain", "my_model", "seq:step:2"],
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"data": {"chunk": AIMessageChunk(content="world!")},
|
|
|
|
|
"data": {"chunk": AIMessageChunk(content=" ")},
|
|
|
|
|
"event": "on_chain_stream",
|
|
|
|
|
"metadata": {"foo": "bar"},
|
|
|
|
|
"name": "my_chain",
|
|
|
|
@ -444,6 +436,14 @@ async def test_event_stream_with_simple_chain() -> None:
|
|
|
|
|
"run_id": "",
|
|
|
|
|
"tags": ["my_chain", "my_model", "seq:step:2"],
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"data": {"chunk": AIMessageChunk(content="world!")},
|
|
|
|
|
"event": "on_chain_stream",
|
|
|
|
|
"metadata": {"foo": "bar"},
|
|
|
|
|
"name": "my_chain",
|
|
|
|
|
"run_id": "",
|
|
|
|
|
"tags": ["my_chain"],
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"data": {
|
|
|
|
|
"input": {
|
|
|
|
|