create function to fetch vectors from a namespace

Author: junefish

learn/experimental/merge-namespaces/merge-namespaces.ipynb

@@ -52,6 +52,32 @@
     "index_name = \"namespace-test\" # replace with the correct index name\n",
     "index = pc.Index(index_name)"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import random\n",
+    "\n",
+    "dimension = index.describe_index_stats()['dimension']\n",
+    "# Function to fetch all vectors from a namespace\n",
+    "def fetch_all_vectors(namespace):\n",
+    "    count = index.describe_index_stats()['namespaces'][namespace]['vector_count']\n",
+    "    random_vecs = [random.random() for _ in range(dimension)]\n",
+    "    response = index.query(\n",
+    "        namespace=namespace,\n",
+    "        vector=random_vecs,\n",
+    "        top_k=count,\n",
+    "        include_values=True\n",
+    "    )\n",
+    "    return response['matches']\n",
+    "\n",
+    "# Fetch vectors from ns1 and ns2\n",
+    "vectors_ns1 = fetch_all_vectors(\"ns1\")\n",
+    "vectors_ns2 = fetch_all_vectors(\"ns2\")"
+   ]
   }
  ],
  "metadata": {