chatbot_agentic: introduced memory

notebooks/chatbot_agentic.ipynb

@@ -19,6 +19,7 @@
     "import json\n",
     "from typing import Annotated, List\n",
     "from typing_extensions import TypedDict\n",
+    "from langgraph.checkpoint.memory import MemorySaver\n",
     "from langchain_core.messages import ToolMessage\n",
     "from langgraph.graph import StateGraph, START, END\n",
     "from langgraph.graph.message import add_messages\n",
@@ -187,7 +188,8 @@
     "# Any time a tool is called, we return to the chatbot to decide the next step\n",
     "graph_builder.add_edge(\"tools\", \"chatbot\")\n",
     "\n",
-    "graph = graph_builder.compile()"
+    "memory = MemorySaver()\n",
+    "graph = graph_builder.compile(checkpointer=memory)"
    ]
   },
   {
@@ -211,10 +213,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "config = {\"configurable\": {\"thread_id\": \"1\"}}\n",
+    "\n",
     "def stream_graph_updates(user_input: str):\n",
-    "    for event in graph.stream({\"messages\": [(\"user\", user_input)]}):\n",
-    "        for value in event.values():\n",
-    "            print(\"Assistant:\", value[\"messages\"][-1].content)\n",
+    "    events= graph.stream(\n",
+    "        {\"messages\": [(\"user\", user_input)]},\n",
+    "        config,\n",
+    "        stream_mode=\"values\"\n",
+    "        )\n",
+    "    for event in events:\n",
+    "        event[\"messages\"][-1].pretty_print()\n",
+    "        #for value in event.values():\n",
+    "        #    print(\"Assistant:\", value[\"messages\"][-1].content)\n",
     "\n",
     "\n",
     "while True:\n",
@@ -223,7 +233,8 @@
     "        if user_input.lower() in [\"quit\", \"exit\", \"q\"]:\n",
     "            print(\"Goodbye!\")\n",
     "            break\n",
-    "\n",
+    "        snapshot = graph.get_state(config)\n",
+    "        print(f'Current state: {snapshot}')\n",
     "        stream_graph_updates(user_input)\n",
     "    except Exception as e:\n",
     "        # fallback if input() is not available\n",