When using astream with stream_mode="messages", how do I get the final graph state after the graph is done?

[cyc00518]

I have already checked #562 , But it doesn't solve my problem.
In my situation, there are 3 steps :

1. Return incoming tool choose for user to avoid long-time waiting.
2. Only stream final anwer token by token. (So astream_events is unsuitable.)
3. After graph is done, I need to dump information in State for training my llm and log.

How can I solve this?
Many thanks for your kind help!

[vbarda]

@cyc00518 you can use mixed stream modes, and i think to achieve what you want you can do 2 things:

(1) call .astream with stream_mode=["messages", "values"]
(2) filter messages streaming chunks to keep only the ones coming from your last node using langgraph_node in the metadata streamed in the chunks.

since you would be streaming values as well, the last values chunk that you stream should have full information you might need for training.

hope this helps!

[cyc00518]

@vbarda

Thanks for your quick reply!
It really solved my problem!