chore: add accuracy test

Author: kmruiz

src/tools/mongodb/read/aggregate.ts

@@ -26,7 +26,7 @@ const VectorSearchStage = z.object({
                 .describe(
                     "When true, uses an ENN algorithm, otherwise uses ANN. Using ENN is not compatible with numCandidates, in that case, numCandidates must be left empty."
                 ),
-            index: z.string().optional().default("default"),
+            index: z.string().describe("Name of the index, as retrieved from the collection-indexes tool."),
             path: z
                 .string()
                 .describe(

tests/accuracy/aggregate.test.ts

@@ -1,5 +1,6 @@
 import { describeAccuracyTests } from "./sdk/describeAccuracyTests.js";
 import { Matcher } from "./sdk/matcher.js";
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 
 describeAccuracyTests([
     {
@@ -24,4 +25,59 @@ describeAccuracyTests([
             },
         ],
     },
+    {
+        prompt: "Run a vectorSearch query on mflix.movies on path 'plot_embeddings' with the model voyage-3-large to find all 'sci-fy' movies.",
+        expectedToolCalls: [
+            {
+                toolName: "collection-indexes",
+                parameters: {
+                    database: "mflix",
+                    collection: "movies",
+                },
+            },
+            {
+                toolName: "aggregate",
+                parameters: {
+                    database: "mflix",
+                    collection: "movies",
+                    pipeline: [
+                        {
+                            $vectorSearch: {
+                                exact: Matcher.anyOf(Matcher.undefined, Matcher.boolean(true)),
+                                index: "my-index",
+                                path: "plot_embeddings",
+                                queryVector: "sci-fy",
+                                embeddingModel: "voyage-3-large",
+                            },
+                        },
+                    ],
+                    responseBytesLimit: Matcher.anyOf(Matcher.number(), Matcher.undefined),
+                },
+            },
+        ],
+        mockedTools: {
+            "collection-indexes": (): CallToolResult => {
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: JSON.stringify({
+                                name: "my-index",
+                                type: "vectorSearch",
+                                status: "READY",
+                                queryable: true,
+                                latestDefinition: {
+                                    type: "vector",
+                                    path: "plot_embeddings",
+                                    numDimensions: 1024,
+                                    quantization: "none",
+                                    similarity: "euclidean",
+                                },
+                            }),
+                        },
+                    ],
+                };
+            },
+        },
+    },
 ]);