docs: use google.colab module to read env var

pingcap · Mar 3, 2025 · d337daf · d337daf
1 parent eee2d29
commit d337daf
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/core/tests/quickstart.ipynb b/core/tests/quickstart.ipynb
@@ -9,15 +9,15 @@
    ]
   },
   {
-   "metadata": {},
    "cell_type": "code",
-   "outputs": [],
    "execution_count": null,
+   "id": "592aef6b523d1535",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "%pip install autoflow-ai==0.0.1.dev7\n",
     "%pip install dotenv sqlalchemy ipywidgets pymysql"
-   ],
-   "id": "592aef6b523d1535"
+   ]
   },
   {
    "cell_type": "markdown",
@@ -79,15 +79,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "id": "84f43a00",
    "metadata": {},
    "outputs": [],
    "source": [
     "from sqlalchemy import create_engine\n",
     "from autoflow import Autoflow\n",
+    "from google.colab import userdata\n",
     "\n",
-    "db_engine = create_engine(os.getenv(\"DATABASE_URL\"))\n",
+    "# db_engine = create_engine(os.getenv(\"DATABASE_URL\"))\n",
+    "db_engine = create_engine(userdata.get('DATABASE_URL'))\n",
     "af = Autoflow(db_engine=db_engine)"
    ]
   },