Tracking cluster creation

ajosh0504 · ajosh0504 · commit 05d00932d089 · 2025-06-17T08:10:31.000-07:00
diff --git a/notebooks/ai-agents-lab.ipynb b/notebooks/ai-agents-lab.ipynb
@@ -23,7 +23,8 @@
       "outputs": [],
       "source": [
         "import os\n",
-        "from pymongo import MongoClient"
+        "from pymongo import MongoClient\n",
+        "from utils import track_progress"
       ]
     },
     {
@@ -40,6 +41,16 @@
         "mongodb_client.admin.command(\"ping\")"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# Track progress of key steps-- DO NOT CHANGE\n",
+        "track_progress(\"cluster_creation\", \"ai_agents_lab\")"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {},
@@ -164,7 +175,7 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "from utils import create_index, check_index_ready, track_progress"
+        "from utils import create_index, check_index_ready"
       ]
     },
     {
@@ -227,7 +238,7 @@
       "outputs": [],
       "source": [
         "# Track progress of key steps-- DO NOT CHANGE\n",
-        "track_progress(\"vs_index_creation\", DB_NAME)"
+        "track_progress(\"vs_index_creation\", \"ai_agents_lab\")"
       ]
     },
     {
diff --git a/notebooks/ai-rag-lab.ipynb b/notebooks/ai-rag-lab.ipynb
@@ -21,7 +21,8 @@
       "outputs": [],
       "source": [
         "import os\n",
-        "from pymongo import MongoClient"
+        "from pymongo import MongoClient\n",
+        "from utils import track_progress"
       ]
     },
     {
@@ -38,6 +39,16 @@
         "mongodb_client.admin.command(\"ping\")"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# Track progress of key steps-- DO NOT CHANGE\n",
+        "track_progress(\"cluster_creation\", \"ai_rag_lab\")"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -379,7 +390,7 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "from utils import create_index, check_index_ready, track_progress"
+        "from utils import create_index, check_index_ready"
       ]
     },
     {
@@ -442,7 +453,7 @@
       "outputs": [],
       "source": [
         "# Track progress of key steps-- DO NOT CHANGE\n",
-        "track_progress(\"vs_index_creation\", DB_NAME)"
+        "track_progress(\"vs_index_creation\", \"ai_rag_lab\")"
       ]
     },
     {
diff --git a/notebooks/vector-search-lab.ipynb b/notebooks/vector-search-lab.ipynb
@@ -21,7 +21,8 @@
       "outputs": [],
       "source": [
         "import os\n",
-        "from pymongo import MongoClient"
+        "from pymongo import MongoClient\n",
+        "from utils import track_progress"
       ]
     },
     {
@@ -38,6 +39,16 @@
         "mongodb_client.admin.command(\"ping\")"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# Track progress of key steps-- DO NOT CHANGE\n",
+        "track_progress(\"cluster_creation\", \"ai_vs_lab\")"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {},
@@ -306,7 +317,7 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "from utils import create_index, check_index_ready, track_progress"
+        "from utils import create_index, check_index_ready"
       ]
     },
     {
@@ -369,7 +380,7 @@
       "outputs": [],
       "source": [
         "# Track progress of key steps-- DO NOT CHANGE\n",
-        "track_progress(\"vs_index_creation\", DB_NAME)"
+        "track_progress(\"vs_index_creation\", \"ai_vs_lab\")"
       ]
     },
     {

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,8 @@`
`23`	`23`	`"outputs": [],`
`24`	`24`	`"source": [`
`25`	`25`	`"import os\n",`
`26`		`- "from pymongo import MongoClient"`
	`26`	`+ "from pymongo import MongoClient\n",`
	`27`	`+ "from utils import track_progress"`
`27`	`28`	`]`
`28`	`29`	`},`
`29`	`30`	`{`
`@@ -40,6 +41,16 @@`
`40`	`41`	`"mongodb_client.admin.command(\"ping\")"`
`41`	`42`	`]`
`42`	`43`	`},`
	`44`	`+ {`
	`45`	`+ "cell_type": "code",`
	`46`	`+ "execution_count": null,`
	`47`	`+ "metadata": {},`
	`48`	`+ "outputs": [],`
	`49`	`+ "source": [`
	`50`	`+ "# Track progress of key steps-- DO NOT CHANGE\n",`
	`51`	`+ "track_progress(\"cluster_creation\", \"ai_agents_lab\")"`
	`52`	`+ ]`
	`53`	`+ },`
`43`	`54`	`{`
`44`	`55`	`"cell_type": "markdown",`
`45`	`56`	`"metadata": {},`
`@@ -164,7 +175,7 @@`
`164`	`175`	`"metadata": {},`
`165`	`176`	`"outputs": [],`
`166`	`177`	`"source": [`
`167`		`- "from utils import create_index, check_index_ready, track_progress"`
	`178`	`+ "from utils import create_index, check_index_ready"`
`168`	`179`	`]`
`169`	`180`	`},`
`170`	`181`	`{`
`@@ -227,7 +238,7 @@`
`227`	`238`	`"outputs": [],`
`228`	`239`	`"source": [`
`229`	`240`	`"# Track progress of key steps-- DO NOT CHANGE\n",`
`230`		`- "track_progress(\"vs_index_creation\", DB_NAME)"`
	`241`	`+ "track_progress(\"vs_index_creation\", \"ai_agents_lab\")"`
`231`	`242`	`]`
`232`	`243`	`},`
`233`	`244`	`{`
Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,8 @@`
`21`	`21`	`"outputs": [],`
`22`	`22`	`"source": [`
`23`	`23`	`"import os\n",`
`24`		`- "from pymongo import MongoClient"`
	`24`	`+ "from pymongo import MongoClient\n",`
	`25`	`+ "from utils import track_progress"`
`25`	`26`	`]`
`26`	`27`	`},`
`27`	`28`	`{`
`@@ -38,6 +39,16 @@`
`38`	`39`	`"mongodb_client.admin.command(\"ping\")"`
`39`	`40`	`]`
`40`	`41`	`},`
	`42`	`+ {`
	`43`	`+ "cell_type": "code",`
	`44`	`+ "execution_count": null,`
	`45`	`+ "metadata": {},`
	`46`	`+ "outputs": [],`
	`47`	`+ "source": [`
	`48`	`+ "# Track progress of key steps-- DO NOT CHANGE\n",`
	`49`	`+ "track_progress(\"cluster_creation\", \"ai_rag_lab\")"`
	`50`	`+ ]`
	`51`	`+ },`
`41`	`52`	`{`
`42`	`53`	`"cell_type": "code",`
`43`	`54`	`"execution_count": null,`
`@@ -379,7 +390,7 @@`
`379`	`390`	`"metadata": {},`
`380`	`391`	`"outputs": [],`
`381`	`392`	`"source": [`
`382`		`- "from utils import create_index, check_index_ready, track_progress"`
	`393`	`+ "from utils import create_index, check_index_ready"`
`383`	`394`	`]`
`384`	`395`	`},`
`385`	`396`	`{`
`@@ -442,7 +453,7 @@`
`442`	`453`	`"outputs": [],`
`443`	`454`	`"source": [`
`444`	`455`	`"# Track progress of key steps-- DO NOT CHANGE\n",`
`445`		`- "track_progress(\"vs_index_creation\", DB_NAME)"`
	`456`	`+ "track_progress(\"vs_index_creation\", \"ai_rag_lab\")"`
`446`	`457`	`]`
`447`	`458`	`},`
`448`	`459`	`{`