YunxiaoXu/ea-chatbot-lg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(api): Update Chat Stream Protocol for Orchestrator Architecture

Browse Source
This commit is contained in:
Yunxiao Xu
2026-02-23 16:44:06 -08:00
parent 92c30d217e
commit 02d93120e0
2 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/tests/api/test_agent_stream.py
View File

@@ -36,24 +36,25 @@ async def test_stream_agent_events_all_features():
# Stream chunk
{
"event": "on_chat_model_stream",
"metadata": {"langgraph_node": "summarizer"},
"metadata": {"langgraph_node": "synthesizer"},
"data": {"chunk": AIMessage(content="Hello ")}
},
{
"event": "on_chat_model_stream",
"metadata": {"langgraph_node": "summarizer"},
"metadata": {"langgraph_node": "synthesizer"},
"data": {"chunk": AIMessage(content="world")}
},
# Plot event
# Plot event - with nested subgraph it might bubble up or come directly from data_analyst_worker
# Let's mock it coming from the data_analyst_worker on_chain_end
{
"event": "on_chain_end",
"name": "executor",
"name": "data_analyst_worker",
"data": {"output": {"plots": [fig]}}
},
# Final response
{
"event": "on_chain_end",
"name": "summarizer",
"name": "synthesizer",
"data": {"output": {"messages": [AIMessage(content="Hello world final")]}}
},
# Summary update
@@ -91,7 +92,7 @@ async def test_stream_agent_events_all_features():
assert any(r.get("type") == "on_chat_model_stream" for r in results)
# Verify plot was encoded
plot_event = next(r for r in results if r.get("name") == "executor")
plot_event = next(r for r in results if r.get("name") == "data_analyst_worker")
assert "encoded_plots" in plot_event["data"]
assert len(plot_event["data"]["encoded_plots"]) == 1