docs: update async connection pool example in postgres how-to (#1318)

langchain-ai · Aug 12, 2024 · 6a22b7c · 6a22b7c
1 parent 4e643ba
commit 6a22b7c
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/examples/persistence_postgres.ipynb b/examples/persistence_postgres.ipynb
@@ -364,14 +364,12 @@
    "source": [
     "from psycopg_pool import AsyncConnectionPool\n",
     "\n",
-    "pool = AsyncConnectionPool(\n",
+    "async with AsyncConnectionPool(\n",
     "    # Example configuration\n",
     "    conninfo=DB_URI,\n",
     "    max_size=20,\n",
     "    kwargs=connection_kwargs\n",
-    ")\n",
-    "\n",
-    "async with pool.connection() as conn:\n",
+    ") as pool, pool.connection() as conn:\n",
     "    checkpointer = AsyncPostgresSaver(conn)\n",
     "\n",
     "    # NOTE: you need to call .setup() the first time you're using your checkpointer\n",