Updated using Colab

Author: vinay0410

alloydb/notebooks/columnar_engine_accelerated_hnsw_vector_search_benchmark.ipynb

@@ -52,17 +52,6 @@
         "id": "markdown-header"
       }
     },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "ce84cca6"
-      },
-      "source": [
-        "## Resources\n",
-        "\n",
-        "*   [Get Started with AlloyDB](https://cloud.google.com/alloydb/docs/quickstart)"
-      ]
-    },
     {
       "cell_type": "code",
       "source": [
@@ -71,19 +60,25 @@
         "# 1. CONFIGURATION\n",
         "# ==========================================\n",
         "# @markdown ### **⚙️ 1. Cluster Configuration**\n",
-        "project_id = \"your-project-id\" # @param {type:\"string\"}\n",
-        "region = \"your-region-id\" # @param {type:\"string\"}\n",
-        "cluster_id = \"your-cluster-id\" # @param {type:\"string\"}\n",
-        "instance_id = \"your-instance-id\" # @param {type:\"string\"}\n",
+        "project_id = \"\" # @param {type:\"string\", placeholder:\"Project ID\"}\n",
+        "region = \"\" # @param {type:\"string\", placeholder: \"Region (e.g. us-central1)\"}\n",
+        "cluster_name = \"\" # @param {type:\"string\", placeholder: \"Cluster name\"}\n",
+        "instance_name = \"\" # @param {type:\"string\", placeholder: \"Instance Name\"}\n",
         "\n",
         "# @markdown ### **🔐 2. Database Credentials**\n",
-        "db_user = \"your-database-user\" # @param {type:\"string\"}\n",
-        "db_name = \"your-database-name\" # @param {type:\"string\"}\n",
+        "# @markdown *Note: Password will be requested securely when the benchmark is executed.*\n",
+        "db_user = \"\" # @param {type:\"string\", placeholder: \"DB User\"}\n",
+        "db_name = \"\" # @param {type:\"string\", placeholder: \"DB Name\"}\n",
         "\n",
         "# @markdown ### **⚙️ 3. Benchmark Settings**\n",
         "# @markdown *Note: Increasing these values will provide more intensive testing, but will slow down the benchmark runtime.*\n",
-        "test_queries = 1000 # @param {type:\"integer\"}\n",
-        "search_limit = 10 # @param {type:\"integer\"}\n",
+        "# @markdown ### **⚙️ 3. Benchmark Settings**\n",
+        "# @markdown *Note: Increasing these values will provide more intensive testing, but will slow down the benchmark runtime.*\n",
+        "# @markdown * `test_queries`: The total number of test queries to run during the benchmark. Increasing this value provides a more stable average but slows down the overall runtime.\n",
+        "# @markdown * `search_limit`: The number of nearest neighbors to retrieve per query (i.e., the `LIMIT` on the vector similarity search, equivalent to top-k).\n",
+        "\n",
+        "test_queries = 1000 # @param [100, 1000, 10000] {type:\"raw\"}\n",
+        "search_limit = 10 # @param [10, 100] {type:\"raw\"}\n",
         "vector_dim = 100 # Fixed for GloVe-100 dataset\n",
         "\n",
         "import getpass\n",
@@ -228,7 +223,7 @@
         "async def monitor_index_progress(connector):\n",
         "    try:\n",
         "        poll_conn = await connector.connect(\n",
-        "            f\"projects/{project_id}/locations/{region}/clusters/{cluster_id}/instances/{instance_id}\",\n",
+        "            f\"projects/{project_id}/locations/{region}/clusters/{cluster_name}/instances/{instance_name}\",\n",
         "            \"asyncpg\", user=db_user, password=db_pass, db=db_name, ip_type=\"PUBLIC\"\n",
         "        )\n",
         "    except Exception:\n",
@@ -400,7 +395,7 @@
         "        console.print(\"\\n[bold blue]Step 1/9:[/bold blue] 🔌 Initializing AlloyDB AsyncConnector...\")\n",
         "        connector = AsyncConnector()\n",
         "        conn = await connector.connect(\n",
-        "            f\"projects/{project_id}/locations/{region}/clusters/{cluster_id}/instances/{instance_id}\",\n",
+        "            f\"projects/{project_id}/locations/{region}/clusters/{cluster_name}/instances/{instance_name}\",\n",
         "            \"asyncpg\", user=db_user, password=db_pass, db=db_name, ip_type=\"PUBLIC\"\n",
         "        )\n",
         "\n",