chore: adds accuracy test for aggregation export

Author: himanshusinghs

src/tools/mongodb/read/export.ts

@@ -18,15 +18,21 @@ export class ExportTool extends MongoDBToolBase {
             .array(
                 z.discriminatedUnion("name", [
                     z.object({
-                        name: z.literal("find"),
-                        arguments: z.object({
-                            ...FindArgs,
-                            limit: FindArgs.limit.removeDefault(),
-                        }),
+                        name: z
+                            .literal("find")
+                            .describe("The literal name 'find' to represent a find cursor as target."),
+                        arguments: z
+                            .object({
+                                ...FindArgs,
+                                limit: FindArgs.limit.removeDefault(),
+                            })
+                            .describe("The arguments for 'find' operation."),
                     }),
                     z.object({
-                        name: z.literal("aggregate"),
-                        arguments: z.object(AggregateArgs),
+                        name: z
+                            .literal("aggregate")
+                            .describe("The literal name 'aggregate' to represent an aggregation cursor as target."),
+                        arguments: z.object(AggregateArgs).describe("The arguments for 'aggregate' operation."),
                     }),
                 ])
             )

tests/accuracy/export.test.ts

@@ -95,4 +95,34 @@ describeAccuracyTests([
             },
         ],
     },
+    {
+        prompt: "Export an aggregation that groups all movie titles by the field release_year from mflix.movies",
+        expectedToolCalls: [
+            {
+                toolName: "export",
+                parameters: {
+                    database: "mflix",
+                    collection: "movies",
+                    exportTitle: Matcher.string(),
+                    exportTarget: [
+                        {
+                            name: "aggregate",
+                            arguments: {
+                                pipeline: [
+                                    {
+                                        $group: {
+                                            _id: "$release_year",
+                                            titles: {
+                                                $push: "$title",
+                                            },
+                                        },
+                                    },
+                                ],
+                            },
+                        },
+                    ],
+                },
+            },
+        ],
+    },
 ]);