From 337ccc058fdeaa8a9dbd56c0b47344952badeb33 Mon Sep 17 00:00:00 2001
From: Niklas Martin <55481472+niklasmartin@users.noreply.github.com>
Date: Thu, 29 Aug 2024 15:47:05 +0200
Subject: [PATCH] [docs]: fix type hint for state (#1528)

---
 examples/reflection/reflection.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/reflection/reflection.ipynb b/examples/reflection/reflection.ipynb
index 253dfd075..494bb4aca 100644
--- a/examples/reflection/reflection.ipynb
+++ b/examples/reflection/reflection.ipynb
@@ -44,7 +44,7 @@
    "metadata": {},
    "outputs": [
     {
-     "name": "stdin",
+     "name": "stdout",
      "output_type": "stream",
      "text": [
       "LANGCHAIN_API_KEY ········\n",
@@ -348,7 +348,7 @@
     "builder.add_edge(START, \"generate\")\n",
     "\n",
     "\n",
-    "def should_continue(state: List[BaseMessage]):\n",
+    "def should_continue(state: State):\n",
     "    if len(state[\"messages\"]) > 6:\n",
     "        # End after 3 iterations\n",
     "        return END\n",