diff --git a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py
index 8437f5d1b..f460cc9c2 100644
--- a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py
+++ b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py
@@ -392,7 +392,7 @@ def create_react_agent(
             !!! Note
                 Only available with `version="v2"`.
         state_schema: An optional state schema that defines graph state.
-            Must have `messages` and `remaining_steps` keys.
+            Must have `messages` key. May optionally have `remaining_steps` and `is_last_step` keys.
             Defaults to `AgentState` that defines those two keys.
         context_schema: An optional schema for runtime context.
         checkpointer: An optional checkpoint saver object. This is used for persisting
@@ -967,3 +967,4 @@ __all__ = [
 
 
 
+