Merge pull request #24 from mongodb-developer/update-sample-queries

Update sample queries for the vector search lab

Author: ajosh0504

.devcontainer/docker-compose.yml

@@ -14,7 +14,7 @@ services:
     network_mode: service:mongodb
 
   mongodb:
-    image: mongodb/mongodb-atlas-local:8.0.3-20250506T093411Z
+    image: mongodb/mongodb-atlas-local:8.2.0
     restart: unless-stopped
     volumes:
       - mongodb_data:/data/db

data/images/barn.png

@@ -277,9 +277,12 @@
     "    Returns:\n",
     "        List[float]: Embedding of the content as a list.\n",
     "    \"\"\"\n",
-    "    # If an image URL is provided, first load the image\n",
+    "    # If the input is an image, first load the image content\n",
     "    if mode == \"image\":\n",
-    "        content = Image.open(requests.get(content, stream=True).raw)\n",
+    "        if content.startswith(\"http\"):\n",
+    "            content = Image.open(requests.get(content, stream=True).raw)\n",
+    "        else:\n",
+    "            content = Image.open(content)\n",
     "    return vo.multimodal_embed(inputs=[[content]], model=\"voyage-multimodal-3\", input_type=input_type).embeddings[0]"
    ]
   },
@@ -458,14 +461,13 @@
    "outputs": [],
    "source": [
     "# Test the vector search with a text query\n",
-    "vector_search(\"A peaceful and uplifting atmosphere\", \"text\")\n",
+    "vector_search(\"A man wearing a golden crown\", \"text\")\n",
     "\n",
     "# Also try these text queries:\n",
-    "# - Healthy living and fitness\n",
-    "# - A man wearing a golden crown\n",
-    "# - Adorable fluffy creatures\n",
-    "# - Winning business agreement\n",
-    "# - A Winter's Tale on a Spanish Isle"
+    "# - A rainbow of lively colors\n",
+    "# - Creatures wondrous or familiar\n",
+    "# - A boy and the ocean\n",
+    "# - Houses"
    ]
   },
   {
@@ -478,11 +480,9 @@
     "vector_search(\"https://images.isbndb.com/covers/10835953482746.jpg\", \"image\")\n",
     "\n",
     "# Also try these image queries:\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/1.png\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/2.png\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/3.png\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/4.png\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/5.png"
+    "# - ../data/images/salad.jpg\n",
+    "# - ../data/images/kitten.png\n",
+    "# - ../data/images/barn.png"
    ]
   },
   {
@@ -548,7 +548,7 @@
     "filter = <CODE_BLOCK_11>\n",
     "# Pass the `filter` as an argument to the `vector_search` function.\n",
     "# Notice how this filter is incorporated in the `pipeline` in the `vector_search` function.\n",
-    "vector_search(\"A peaceful and uplifting atmosphere\", \"text\", filter)"
+    "vector_search(\"A boy and the ocean\", \"text\", filter)"
    ]
   },
   {
@@ -608,7 +608,7 @@
     "filter = <CODE_BLOCK_13>\n",
     "# Pass the `filter` as an argument to the `vector_search` function.\n",
     "# Notice how this filter is incorporated in the `pipeline` in the `vector_search` function.\n",
-    "vector_search(\"A peaceful and uplifting atmosphere\", \"text\", filter)"
+    "vector_search(\"A boy and the ocean\", \"text\", filter)"
    ]
   },
   {
@@ -889,7 +889,7 @@
    "source": [
     "# Test the hybrid search query with a weight of 1.0 for vector search and 0.0 for full-text search\n",
     "hybrid_search(\n",
-    "    user_query=\"A Winter's Tale on a Spanish Isle\",\n",
+    "    user_query=\"My Favorite Summer\",\n",
     "    vector_weight=1.0,\n",
     "    full_text_weight=0.0,\n",
     ")"
@@ -901,11 +901,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Test the hybrid search query with a weight of 0.5 for vector search and 0.5 for full-text search\n",
+    "# Test the hybrid search query with a weight of 0.3 for vector search and 0.7 for full-text search\n",
     "hybrid_search(\n",
-    "    user_query=\"A Winter's Tale on a Spanish Isle\",\n",
-    "    vector_weight=0.5,\n",
-    "    full_text_weight=0.5,\n",
+    "    user_query=\"My Favorite Summer\",\n",
+    "    vector_weight=0.3,\n",
+    "    full_text_weight=0.7,\n",
     ")"
    ]
   }

data/images/kitten.png

@@ -277,9 +277,12 @@
     "    Returns:\n",
     "        List[float]: Embedding of the content as a list.\n",
     "    \"\"\"\n",
-    "    # If an image URL is provided, first load the image\n",
+    "    # If the input is an image, first load the image content\n",
     "    if mode == \"image\":\n",
-    "        content = Image.open(requests.get(content, stream=True).raw)\n",
+    "        if content.startswith(\"http\"):\n",
+    "            content = Image.open(requests.get(content, stream=True).raw)\n",
+    "        else:\n",
+    "            content = Image.open(content)\n",
     "    return vo.multimodal_embed(inputs=[[content]], model=\"voyage-multimodal-3\", input_type=input_type).embeddings[0]"
    ]
   },
@@ -470,14 +473,13 @@
    "outputs": [],
    "source": [
     "# Test the vector search with a text query\n",
-    "vector_search(\"A peaceful and uplifting atmosphere\", \"text\")\n",
+    "vector_search(\"A man wearing a golden crown\", \"text\")\n",
     "\n",
     "# Also try these text queries:\n",
-    "# - Healthy living and fitness\n",
-    "# - A man wearing a golden crown\n",
-    "# - Adorable fluffy creatures\n",
-    "# - Winning business agreement\n",
-    "# - A Winter's Tale on a Spanish Isle"
+    "# - A rainbow of lively colors\n",
+    "# - Creatures wondrous or familiar\n",
+    "# - A boy and the ocean\n",
+    "# - Houses"
    ]
   },
   {
@@ -490,11 +492,9 @@
     "vector_search(\"https://images.isbndb.com/covers/10835953482746.jpg\", \"image\")\n",
     "\n",
     "# Also try these image queries:\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/1.png\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/2.png\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/3.png\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/4.png\n",
-    "# - https://mongodb-developer.github.io/vector-search-lab/img/query-sample/5.png"
+    "# - ../data/images/salad.jpg\n",
+    "# - ../data/images/kitten.png\n",
+    "# - ../data/images/barn.png"
    ]
   },
   {
@@ -574,7 +574,7 @@
     "filter = {\"year\": {\"$gte\": 2002}}\n",
     "# Pass the `filter` as an argument to the `vector_search` function.\n",
     "# Notice how this filter is incorporated in the `pipeline` in the `vector_search` function.\n",
-    "vector_search(\"A peaceful and uplifting atmosphere\", \"text\", filter)"
+    "vector_search(\"A boy and the ocean\", \"text\", filter)"
    ]
   },
   {
@@ -649,7 +649,7 @@
     "filter = {\"$and\": [{\"year\": {\"$gte\": 2002}}, {\"pages\": {\"$lte\": 250}}]}\n",
     "# Pass the `filter` as an argument to the `vector_search` function.\n",
     "# Notice how this filter is incorporated in the `pipeline` in the `vector_search` function.\n",
-    "vector_search(\"A peaceful and uplifting atmosphere\", \"text\", filter)"
+    "vector_search(\"A boy and the ocean\", \"text\", filter)"
    ]
   },
   {
@@ -944,7 +944,7 @@
    "source": [
     "# Test the hybrid search query with a weight of 1.0 for vector search and 0.0 for full-text search\n",
     "hybrid_search(\n",
-    "    user_query=\"A Winter's Tale on a Spanish Isle\",\n",
+    "    user_query=\"My Favorite Summer\",\n",
     "    vector_weight=1.0,\n",
     "    full_text_weight=0.0,\n",
     ")"
@@ -956,18 +956,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Test the hybrid search query with a weight of 0.5 for vector search and 0.5 for full-text search\n",
+    "# Test the hybrid search query with a weight of 0.3 for vector search and 0.7 for full-text search\n",
     "hybrid_search(\n",
-    "    user_query=\"A Winter's Tale on a Spanish Isle\",\n",
-    "    vector_weight=0.5,\n",
-    "    full_text_weight=0.5,\n",
+    "    user_query=\"My Favorite Summer\",\n",
+    "    vector_weight=0.3,\n",
+    "    full_text_weight=0.7,\n",
     ")"
    ]
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "base",
    "language": "python",
    "name": "python3"
   },
@@ -981,7 +981,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.6"
+   "version": "3.11.6"
   },
   "widgets": {
    "application/vnd.jupyter.widget-state+json": {