Updating progress tracker, adding to notebooks

Author: ajosh0504

notebooks/ai-agents-lab.ipynb

@@ -164,7 +164,7 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "from utils import create_index, check_index_ready"
+        "from utils import create_index, check_index_ready, track_progress"
       ]
     },
     {
@@ -220,6 +220,16 @@
         "check_index_ready(vs_collection, VS_INDEX_NAME)"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# Track progress of key steps-- DO NOT CHANGE\n",
+        "track_progress(\"vs_index_creation\", DB_NAME)"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {

notebooks/ai-rag-lab.ipynb

@@ -379,7 +379,7 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "from utils import create_index, check_index_ready"
+        "from utils import create_index, check_index_ready, track_progress"
       ]
     },
     {
@@ -435,6 +435,16 @@
         "check_index_ready(collection, ATLAS_VECTOR_SEARCH_INDEX_NAME)"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# Track progress of key steps-- DO NOT CHANGE\n",
+        "track_progress(\"vs_index_creation\", DB_NAME)"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {},
@@ -1054,8 +1064,14 @@
     }
   ],
   "metadata": {
+    "kernelspec": {
+      "display_name": "Python 3",
+      "language": "python",
+      "name": "python3"
+    },
     "language_info": {
-      "name": "python"
+      "name": "python",
+      "version": "3.12.1"
     },
     "widgets": {
       "application/vnd.jupyter.widget-state+json": {

notebooks/utils/__init__.py

@@ -1,3 +1,3 @@
-from .utils import create_index, check_index_ready
+from .utils import create_index, check_index_ready, track_progress
 
-__all__ = ["create_index", "check_index_ready"]
+__all__ = ["create_index", "check_index_ready", "track_progress"]

notebooks/utils/utils.py

@@ -73,13 +73,14 @@ def check_index_ready(collection: Collection, index_name: str) -> None:
         time.sleep(SLEEP_TIMER)
 
 
-def track_progress(task: str) -> None:
+def track_progress(task: str, workshop_id: str) -> None:
     """
     Track progress of a task
 
     Args:
         task (str): Task name
+        workshop (str): Workshop name
     """
     print(f"Tracking progress for task {task}")
-    payload = {"task": task, "id": CODESPACE_NAME}
+    payload = {"task": task, "workshop_id": workshop_id, "sandbox_id": CODESPACE_NAME}
     requests.post(url=SERVERLESS_URL, json={"task": "track_progress", "data": payload})

notebooks/vector-search-lab.ipynb

@@ -306,7 +306,7 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "from utils import create_index, check_index_ready"
+        "from utils import create_index, check_index_ready, track_progress"
       ]
     },
     {
@@ -362,6 +362,16 @@
         "check_index_ready(collection, ATLAS_VECTOR_SEARCH_INDEX_NAME)"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# Track progress of key steps-- DO NOT CHANGE\n",
+        "track_progress(\"vs_index_creation\", DB_NAME)"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {},
@@ -924,8 +934,14 @@
     }
   ],
   "metadata": {
+    "kernelspec": {
+      "display_name": "Python 3",
+      "language": "python",
+      "name": "python3"
+    },
     "language_info": {
-      "name": "python"
+      "name": "python",
+      "version": "3.12.1"
     },
     "widgets": {
       "application/vnd.jupyter.widget-state+json": {