fix docs

Author: abrookins

langgraph/docs/create-react-agent-memory.ipynb

@@ -58,7 +58,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 1,
    "id": "a213e11a-5c62-4ddb-a707-490d91add383",
    "metadata": {},
    "outputs": [],
@@ -69,10 +69,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 2,
    "id": "23a1885c-04ab-4750-aefa-105891fddf3e",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdin",
+     "output_type": "stream",
+     "text": [
+      "OPENAI_API_KEY:  ········\n"
+     ]
+    }
+   ],
    "source": [
     "import getpass\n",
     "import os\n",
@@ -109,24 +117,35 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 3,
    "id": "7a154152-973e-4b5d-aa13-48c617744a4c",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "22:31:51 redisvl.index.index INFO   Index already exists, not overwriting.\n",
+      "22:31:51 redisvl.index.index INFO   Index already exists, not overwriting.\n",
+      "22:31:51 redisvl.index.index INFO   Index already exists, not overwriting.\n"
+     ]
+    }
+   ],
    "source": [
+    "from typing import Literal\n",
+    "\n",
+    "from langchain_core.tools import tool\n",
+    "\n",
     "# First we initialize the model we want to use.\n",
     "from langchain_openai import ChatOpenAI\n",
     "\n",
+    "from langgraph.checkpoint.redis import RedisSaver\n",
+    "from langgraph.prebuilt import create_react_agent\n",
+    "\n",
     "model = ChatOpenAI(model=\"gpt-4o\", temperature=0)\n",
     "\n",
     "\n",
     "# For this tutorial we will use custom tool that returns pre-defined values for weather in two cities (NYC & SF)\n",
-    "\n",
-    "from typing import Literal\n",
-    "\n",
-    "from langchain_core.tools import tool\n",
-    "\n",
-    "\n",
     "@tool\n",
     "def get_weather(city: Literal[\"nyc\", \"sf\"]):\n",
     "    \"\"\"Use this to get weather information.\"\"\"\n",
@@ -140,19 +159,18 @@
     "\n",
     "tools = [get_weather]\n",
     "\n",
-    "# We can add \"chat memory\" to the graph with LangGraph's Redi checkpointer\n",
+    "# We can add \"chat memory\" to the graph with LangGraph's Redis checkpointer\n",
     "# to retain the chat context between interactions\n",
-    "from langgraph.checkpoint.redis import RedisSaver\n",
+    "\n",
     "\n",
     "REDIS_URI = \"redis://redis:6379\"\n",
     "memory = None\n",
     "with RedisSaver.from_conn_string(REDIS_URI) as cp:\n",
-    "  cp.setup()\n",
-    "  memory = cp\n",
+    "    cp.setup()\n",
+    "    memory = cp\n",
     "\n",
     "# Define the graph\n",
     "\n",
-    "from langgraph.prebuilt import create_react_agent\n",
     "\n",
     "graph = create_react_agent(model, tools=tools, checkpointer=memory)"
    ]
@@ -169,7 +187,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 4,
    "id": "16636975-5f2d-4dc7-ab8e-d0bea0830a28",
    "metadata": {},
    "outputs": [],
@@ -185,7 +203,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 5,
    "id": "9ffff6c3-a4f5-47c9-b51d-97caaee85cd6",
    "metadata": {},
    "outputs": [
@@ -198,8 +216,8 @@
       "What's the weather in NYC?\n",
       "==================================\u001b[1m Ai Message \u001b[0m==================================\n",
       "Tool Calls:\n",
-      "  get_weather (call_RHv6T6OBCn7eKOlm5qEpLK4P)\n",
-      " Call ID: call_RHv6T6OBCn7eKOlm5qEpLK4P\n",
+      "  get_weather (call_Edwfw0WiyIJ7vt9xzU9xvyeg)\n",
+      " Call ID: call_Edwfw0WiyIJ7vt9xzU9xvyeg\n",
       "  Args:\n",
       "    city: nyc\n",
       "=================================\u001b[1m Tool Message \u001b[0m=================================\n",
@@ -229,7 +247,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 6,
    "id": "187479f9-32fa-4611-9487-cf816ba2e147",
    "metadata": {},
    "outputs": [
@@ -242,7 +260,7 @@
       "What's it known for?\n",
       "==================================\u001b[1m Ai Message \u001b[0m==================================\n",
       "\n",
-      "Could you please specify what \"it\" refers to? Are you asking about a specific place, person, event, or something else?\n"
+      "Could you please specify what \"it\" refers to? Are you asking about a specific city, person, event, or something else?\n"
      ]
     }
    ],
@@ -274,3 +292,4 @@
  "nbformat": 4,
  "nbformat_minor": 5
 }
+