diff --git a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py
index d4facdfea..4bf799e23 100644
--- a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py
+++ b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py
@@ -67,7 +67,7 @@ class AgentState(TypedDict):
 
     is_last_step: IsLastStep
 
-    remaining_steps: RemainingSteps
+    remaining_steps: NotRequired[RemainingSteps]
 
 
 class AgentStatePydantic(BaseModel):
@@ -963,3 +963,4 @@ __all__ = [
     "AgentStateWithStructuredResponsePydantic",
 ]
 
+