chore: more accuracy test tweaks

src/tools/mongodb/metadata/collectionSchema.ts

@@ -21,7 +21,7 @@ export class CollectionSchemaTool extends MongoDBToolBase {
             .optional()
             .default(ONE_MB)
             .describe(
-                `The maximum number of bytes to return in the response. This value is capped by the server’s configured maxBytesPerQuery and cannot be exceeded.`
+                `The maximum number of bytes to return in the response. This value is capped by the server's configured maxBytesPerQuery and cannot be exceeded.`
             ),
     };
 

src/tools/mongodb/read/aggregate.ts

@@ -17,7 +17,7 @@ import { LogId } from "../../../common/logger.js";
 export const AggregateArgs = {
     pipeline: z.array(zEJSON()).describe("An array of aggregation stages to execute"),
     responseBytesLimit: z.number().optional().default(ONE_MB).describe(`\
-The maximum number of bytes to return in the response. This value is capped by the server’s configured maxBytesPerQuery and cannot be exceeded. \
+The maximum number of bytes to return in the response. This value is capped by the server's configured maxBytesPerQuery and cannot be exceeded. \
 Note to LLM: If the entire aggregation result is required, use the "export" tool instead of increasing this limit.\
 `),
 };

src/tools/mongodb/read/find.ts

@@ -30,7 +30,7 @@ export const FindArgs = {
             "A document, describing the sort order, matching the syntax of the sort argument of cursor.sort(). The keys of the object are the fields to sort on, while the values are the sort directions (1 for ascending, -1 for descending)."
         ),
     responseBytesLimit: z.number().optional().default(ONE_MB).describe(`\
-The maximum number of bytes to return in the response. This value is capped by the server’s configured maxBytesPerQuery and cannot be exceeded. \
+The maximum number of bytes to return in the response. This value is capped by the server's configured maxBytesPerQuery and cannot be exceeded. \
 Note to LLM: If the entire query result is required, use the "export" tool instead of increasing this limit.\
 `),
 };

tests/accuracy/collectionSchema.test.ts

@@ -1,26 +1,41 @@
 import { describeAccuracyTests } from "./sdk/describeAccuracyTests.js";
+import { Matcher } from "./sdk/matcher.js";
+
+const listCollectionsOptionalCall = {
+    toolName: "list-collections",
+    parameters: {
+        database: "mflix",
+    },
+    optional: true,
+};
 
 describeAccuracyTests([
     {
         prompt: "Is there a title field in 'mflix.movies' namespace?",
         expectedToolCalls: [
+            listCollectionsOptionalCall,
             {
                 toolName: "collection-schema",
                 parameters: {
                     database: "mflix",
                     collection: "movies",
+                    sampleSize: Matcher.anyOf(Matcher.undefined, Matcher.number()),
+                    responseBytesLimit: Matcher.anyOf(Matcher.undefined, Matcher.number()),
                 },
             },
         ],
     },
     {
         prompt: "What is the type of value stored in title field in movies collection in mflix database?",
         expectedToolCalls: [
+            listCollectionsOptionalCall,
             {
                 toolName: "collection-schema",
                 parameters: {
                     database: "mflix",
                     collection: "movies",
+                    sampleSize: Matcher.anyOf(Matcher.undefined, Matcher.number()),
+                    responseBytesLimit: Matcher.anyOf(Matcher.undefined, Matcher.number()),
                 },
             },
         ],

tests/accuracy/explain.test.ts

@@ -45,6 +45,7 @@ describeAccuracyTests([
                                         $match: { release_year: 2020 },
                                     },
                                 ],
+                                responseBytesLimit: Matcher.anyOf(Matcher.undefined, Matcher.number()),
                             },
                         },
                     ],

tests/accuracy/export.test.ts

@@ -14,7 +14,9 @@ describeAccuracyTests([
                     exportTarget: [
                         {
                             name: "find",
-                            arguments: {},
+                            arguments: {
+                                filter: Matcher.emptyObjectOrUndefined,
+                            },
                         },
                     ],
                     jsonExportFormat: Matcher.anyValue,

tests/accuracy/find.test.ts

@@ -1,10 +1,27 @@
+import type { ExpectedToolCall } from "./sdk/accuracyResultStorage/resultStorage.js";
 import { describeAccuracyTests } from "./sdk/describeAccuracyTests.js";
 import { Matcher } from "./sdk/matcher.js";
 
+const optionalListCalls: (database: string) => ExpectedToolCall[] = (database) => [
+    {
+        toolName: "list-databases",
+        parameters: {},
+        optional: true,
+    },
+    {
+        toolName: "list-collections",
+        parameters: {
+            database,
+        },
+        optional: true,
+    },
+];
+
 describeAccuracyTests([
     {
         prompt: "List all the movies in 'mflix.movies' namespace.",
         expectedToolCalls: [
+            ...optionalListCalls("mflix"),
             {
                 toolName: "find",
                 parameters: {
@@ -18,6 +35,7 @@ describeAccuracyTests([
     {
         prompt: "List all the documents in 'comics.books' namespace.",
         expectedToolCalls: [
+            ...optionalListCalls("comics"),
             {
                 toolName: "find",
                 parameters: {
@@ -31,6 +49,7 @@ describeAccuracyTests([
     {
         prompt: "Find all the movies in 'mflix.movies' namespace with runtime less than 100.",
         expectedToolCalls: [
+            ...optionalListCalls("mflix"),
             {
                 toolName: "find",
                 parameters: {
@@ -46,6 +65,7 @@ describeAccuracyTests([
     {
         prompt: "Find all movies in 'mflix.movies' collection where director is 'Christina Collins'",
         expectedToolCalls: [
+            ...optionalListCalls("mflix"),
             {
                 toolName: "find",
                 parameters: {
@@ -61,6 +81,7 @@ describeAccuracyTests([
     {
         prompt: "Give me all the movie titles available in 'mflix.movies' namespace",
         expectedToolCalls: [
+            ...optionalListCalls("mflix"),
             {
                 toolName: "find",
                 parameters: {
@@ -81,6 +102,7 @@ describeAccuracyTests([
     {
         prompt: "Use 'mflix.movies' namespace to answer who were casted in the movie 'Certain Fish'",
         expectedToolCalls: [
+            ...optionalListCalls("mflix"),
             {
                 toolName: "find",
                 parameters: {
@@ -99,6 +121,7 @@ describeAccuracyTests([
     {
         prompt: "From the mflix.movies namespace, give me first 2 movies of Horror genre sorted ascending by their runtime",
         expectedToolCalls: [
+            ...optionalListCalls("mflix"),
             {
                 toolName: "find",
                 parameters: {
@@ -112,8 +135,9 @@ describeAccuracyTests([
         ],
     },
     {
-        prompt: "I want a COMPLETE list of all the movies ONLY from 'mflix.movies' namespace.",
+        prompt: "I want an exported COMPLETE list of all the movies ONLY from 'mflix.movies' namespace.",
         expectedToolCalls: [
+            ...optionalListCalls("mflix"),
             {
                 toolName: "find",
                 parameters: {

tests/accuracy/getPerformanceAdvisor.test.ts

@@ -1,5 +1,6 @@
 import { describeAccuracyTests } from "./sdk/describeAccuracyTests.js";
 import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { Matcher } from "./sdk/matcher.js";
 
 // Shared mock tool implementations
 const mockedTools = {
@@ -127,6 +128,15 @@ describeAccuracyTests([
                 parameters: {
                     projectId: "mflix",
                     clusterName: "mflix-cluster",
+                    operations: Matcher.anyOf(
+                        Matcher.undefined,
+                        Matcher.value([
+                            "suggestedIndexes",
+                            "dropIndexSuggestions",
+                            "slowQueryLogs",
+                            "schemaSuggestions",
+                        ])
+                    ),
                 },
             },
         ],

tests/accuracy/logs.test.ts

@@ -9,6 +9,7 @@ describeAccuracyTests([
                 toolName: "mongodb-logs",
                 parameters: {
                     type: "startupWarnings",
+                    limit: Matcher.anyOf(Matcher.undefined, Matcher.number()),
                 },
             },
         ],

tests/integration/tools/mongodb/metadata/collectionSchema.test.ts

@@ -26,7 +26,7 @@ describeWithMongoDB("collectionSchema tool", (integration) => {
         {
             name: "responseBytesLimit",
             type: "number",
-            description: `The maximum number of bytes to return in the response. This value is capped by the server’s configured maxBytesPerQuery and cannot be exceeded.`,
+            description: `The maximum number of bytes to return in the response. This value is capped by the server's configured maxBytesPerQuery and cannot be exceeded.`,
             required: false,
         },
     ]);

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

@@ -26,8 +26,7 @@ describeWithMongoDB("aggregate tool", (integration) => {
         },
         {
             name: "responseBytesLimit",
-            description:
-                'The maximum number of bytes to return in the response. This value is capped by the server’s configured maxBytesPerQuery and cannot be exceeded. Note to LLM: If the entire aggregation result is required, use the "export" tool instead of increasing this limit.',
+            description: `The maximum number of bytes to return in the response. This value is capped by the server's configured maxBytesPerQuery and cannot be exceeded. Note to LLM: If the entire aggregation result is required, use the "export" tool instead of increasing this limit.`,
             type: "number",
             required: false,
         },

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

@@ -56,8 +56,7 @@ describeWithMongoDB("find tool with default configuration", (integration) => {
         },
         {
             name: "responseBytesLimit",
-            description:
-                'The maximum number of bytes to return in the response. This value is capped by the server’s configured maxBytesPerQuery and cannot be exceeded. Note to LLM: If the entire query result is required, use the "export" tool instead of increasing this limit.',
+            description: `The maximum number of bytes to return in the response. This value is capped by the server's configured maxBytesPerQuery and cannot be exceeded. Note to LLM: If the entire query result is required, use the "export" tool instead of increasing this limit.`,
             type: "number",
             required: false,
         },