fix more tests

Author: nirinchev

src/tools/mongodb/metadata/explain.ts

@@ -16,7 +16,7 @@ export class ExplainTool extends MongoDBToolBase {
         ...DbOperationArgs,
         method: z
             .array(
-                z.union([
+                z.discriminatedUnion("name", [
                     z.object({
                         name: z.literal("aggregate"),
                         arguments: z.object(AggregateArgs),

src/tools/mongodb/read/aggregate.ts

@@ -6,7 +6,7 @@ import { EJSON } from "bson";
 import { checkIndexUsage } from "../../../helpers/indexCheck.js";
 
 export const AggregateArgs = {
-    pipeline: z.array(z.record(z.string(), z.unknown())).describe("An array of aggregation stages to execute"),
+    pipeline: z.array(z.object({}).passthrough()).describe("An array of aggregation stages to execute"),
 };
 
 export class AggregateTool extends MongoDBToolBase {

tests/accuracy/explain.test.ts

@@ -27,7 +27,7 @@ describeAccuracyTests([
         ],
     },
     {
-        prompt: `Will fetching documents, where release_year is 2020, from 'mflix.movies' namespace perform a collection scan?`,
+        prompt: `Will aggregating documents, where release_year is 2020, from 'mflix.movies' namespace perform a collection scan?`,
         expectedToolCalls: [
             {
                 toolName: "explain",
@@ -51,7 +51,7 @@ describeAccuracyTests([
         ],
     },
     {
-        prompt: `Will fetching documents, where release_year is 2020, from 'mflix.movies' namespace perform a collection scan?`,
+        prompt: `Will counting documents, where release_year is 2020, from 'mflix.movies' namespace perform a collection scan?`,
         expectedToolCalls: [
             {
                 toolName: "explain",