How to set Runtime Context when using astream_events

After create a graph, I can set context when use invoke(), but for function astream_events, there is not params for context, is there a workaround i can follow?

Hi! You can use astream(). From LangGraph v 0.6.0, context is a parameter that you can pass inside astream() method.

You can simply put the context to first argument.

For example like this

for chunk in graph_memory.stream({"messages":input_messages},config_1, stream_mode="updates"):
 chunk["conversation"]["messages"].pretty_print()