From 7f803df5869c9bbafa2914862ccb8548b05b1ac8 Mon Sep 17 00:00:00 2001
From: Alexey Bondarenko <alexey.bond.94.55@gmail.com>
Date: Wed, 12 Mar 2025 23:54:18 +0600
Subject: [PATCH] Add state schemas to __all__ in chat_agent_executor.py
 (#3798)

---
 libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py
index ceed3ef47..0fead9c1e 100644
--- a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py
+++ b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py
@@ -859,4 +859,7 @@ __all__ = [
     "create_react_agent",
     "create_tool_calling_executor",
     "AgentState",
+    "AgentStatePydantic",
+    "AgentStateWithStructuredResponse",
+    "AgentStateWithStructuredResponsePydantic",
 ]