Update LangGraph notebook agent factory

Author: dex-the-ai

langgraph/couchbase_persistence_langgraph.ipynb

@@ -61,7 +61,7 @@
     "\n",
     "Set `OPENAI_API_KEY` before running. For local Couchbase runs, this notebook also reads `CB_CONN_STR`, `CB_USER`, `CB_PASS`, `CB_BUCKET_NAME`, and `CB_SCOPE_NAME` from the environment.\n",
     "\n",
-    "For the local example below, create bucket `test`, scope `langgraph`, and the `checkpoints` / `checkpoint_writes` collections first, or provide equivalent names through the environment variables.\n"
+    "For the local example below, create bucket `test`, scope `langgraph`, and the `checkpoints` / `checkpoint_writes` collections first, or provide equivalent names through the environment variables. The default `CB_*` values shown later are for local development only.\n"
    ]
   },
   {
@@ -134,7 +134,7 @@
     "from typing import Literal\n",
     "from langchain_core.tools import tool\n",
     "from langchain_openai import ChatOpenAI\n",
-    "from langgraph.prebuilt import create_react_agent\n",
+    "from langchain.agents import create_agent\n",
     "\n",
     "\n",
     "@tool\n",
@@ -188,7 +188,7 @@
     "    bucket_name=CB_BUCKET_NAME,\n",
     "    scope_name=CB_SCOPE_NAME,\n",
     ") as checkpointer:\n",
-    "    graph = create_react_agent(model, tools=tools, checkpointer=checkpointer)\n",
+    "    graph = create_agent(model, tools=tools, checkpointer=checkpointer)\n",
     "    config = {\"configurable\": {\"thread_id\": \"1\"}}\n",
     "    res = graph.invoke({\"messages\": [(\"human\", \"what's the weather in sf\")]}, config)\n",
     "    \n",
@@ -319,7 +319,7 @@
     "    bucket_name=CB_BUCKET_NAME,\n",
     "    scope_name=CB_SCOPE_NAME,\n",
     ") as checkpointer:\n",
-    "    graph = create_react_agent(model, tools=tools, checkpointer=checkpointer)\n",
+    "    graph = create_agent(model, tools=tools, checkpointer=checkpointer)\n",
     "    config = {\"configurable\": {\"thread_id\": \"2\"}}\n",
     "    res = await graph.ainvoke(\n",
     "        {\"messages\": [(\"human\", \"what's the weather in nyc\")]}, config\n",