fix: check queryable flag in waitUntilSearchIndexIsQueryable to prevent premature auto-embed queries

tests/integration/tools/mongodb/mongodbHelpers.ts

@@ -346,7 +346,7 @@
     return waitUntilSearchIndexIs(
         collection,
         searchIndex,
-        (index) => index.name === searchIndex && index.status === "READY",
+        (index) => index.name === searchIndex && index.status === "READY" && index.queryable,
         timeout,
         interval,
         (searchIndexes) => {
@@ -356,6 +356,25 @@
     );
 }
 
+export async function waitUntilSearchIndexHasResults(
+    collection: Collection,
+    pipeline: Document[],
+    timeout: number = SEARCH_WAIT_TIMEOUT,
+    interval: number = 1_000
+): Promise<void> {
+    await vi.waitFor(
+        async () => {
+            const results = await collection.aggregate(pipeline).toArray();
+            if (results.length === 0) {
+                throw new Error(
+                    `Vector search pipeline returned no results in ${collection.dbName}.${collection.collectionName}`
+                );
+            }
+        },
+        { timeout, interval }
+    );
+}
+
 export async function createVectorSearchIndexAndWait(
     mongoClient: MongoClient,
     database: string,

tests/integration/tools/mongodb/read/aggregate.test.ts

@@ -12,6 +12,7 @@
     describeWithMongoDB,
     getDocsFromUntrustedContent,
     validateAutoConnectBehavior,
+    waitUntilSearchIndexHasResults,
     waitUntilSearchIndexIsQueryable,
     waitUntilSearchIsReady,
 } from "../mongodbHelpers.js";
@@ -993,7 +994,7 @@
 
             // Ensure it aborted quickly, but possibly after some processing
             expect(executionTime).toBeGreaterThanOrEqual(25);
             expect(executionTime).toBeLessThan(50);
             expect(result).toBeUndefined();
             expectDefined(error);
             expect(error.message).toContain("This operation was aborted");
@@ -1052,6 +1053,25 @@
             // Auto-embed indexes take longer to build because they need to call the voyage API
             // to generate embeddings for the documents. Using a longer timeout (120s).
             await waitUntilSearchIndexIsQueryable(collection, "auto-embed-index", 120_000);
+
+            // Even after the index reports as queryable, the Voyage AI embeddings for existing
+            // documents may not be applied yet on first use (cold-start). Probe the actual
+            // $vectorSearch until it returns results before proceeding with the tests.
+            await waitUntilSearchIndexHasResults(
+                collection,
+                [
+                    {
+                        $vectorSearch: {
+                            index: "auto-embed-index",
+                            path: "plot",
+                            query: { text: "food" },
+                            limit: 1,
+                            numCandidates: 2,
+                        },
+                    },
+                ],
+                60_000
+            );
         });
 
         it("should be able to query autoEmbed text index", { timeout: 130_000 }, async () => {