diff --git a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py index 20055ea26..df053fb90 100644 --- a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py +++ b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py @@ -891,16 +891,12 @@ def create_react_agent( if response_format is not None: workflow.add_node( - "generate_structured_response", + "respond", RunnableCallable( - generate_structured_response, - agenerate_structured_response, + respond, + arespond, ), ) - if post_model_hook is not None: - workflow.add_edge("post_model_hook", "generate_structured_response") - else: - workflow.add_edge("agent", "generate_structured_response") return workflow.compile( checkpointer=checkpointer, @@ -1096,3 +1092,4 @@ __all__ = [ +