Skip to content

Commit 96637b5

Browse files
committed
1 parent 92cda10 commit 96637b5

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

backend/api/routers/llms.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,11 @@ async def stream_graph(
7272
config = get_config(langfuse_handler)
7373
events = dict(messages=[HumanMessage(content=query)])
7474

75-
async for event in graph.astream_events(events, config, version="v2"):
76-
if event.get("event").endswith("end"):
77-
print(event)
75+
async for event in graph.astream_events(
76+
events,
77+
config,
78+
version="v2",
79+
stream_mode="updates",
80+
):
7881
yield dict(data=event)
82+
print(event)

0 commit comments

Comments
 (0)