result = st.session_state.graph.invoke(graph_input, config=st.session_state.config)
st.session_state.agent_state = result