streaming

[navaneethstabilix]

react_graph_memory = builder.compile(checkpointer=memory)

input_message = input()
node_to_stream = 'assistant'
async for event in react_graph_memory.astream_events({"messages": [input_message]}, config, version="v2"):
# Get chat model tokens from a particular node
if event["event"] == "on_chain_stream" and event['metadata'].get('langgraph_node','') == node_to_stream:
print(event["data"])

    instead of print use yielf function to stream in api just return the ai response