chore: schema def mods for better LLM responses

In the previous commit, our schema description was not clear enough for
LLM to correctly provide the name of command and it was hallucinating by
providing listDatabases for example instead of list-databases.
This commit modifies the descriptions a bit to achieve 100% accuracy for
the same prompts.

Author: himanshusinghs

src/tools/mongodb/ddl/ddl.ts

@@ -10,34 +10,40 @@ export class MongoDBDDLTool extends MongoDBToolBase {
     protected description =
         "List databases, collections, indexes and describe the schema of a collection in a MongoDB database";
     protected argsShape = {
-        command: z.discriminatedUnion("name", [
-            z
-                .object({
-                    name: z.literal("list-databases"),
-                    parameters: z.object({}),
-                })
-                .describe("List all databases for a MongoDB connection"),
-            z
-                .object({
-                    name: z.literal("list-collections"),
-                    parameters: z.object({
-                        database: DbOperationArgs.database,
-                    }),
-                })
-                .describe("List all collections for a given database"),
-            z
-                .object({
-                    name: z.literal("collection-indexes"),
-                    parameters: z.object(DbOperationArgs),
-                })
-                .describe("Describe the indexes for a collection"),
-            z
-                .object({
-                    name: z.literal("collection-schema"),
-                    parameters: z.object(DbOperationArgs),
-                })
-                .describe("Describe the schema for a collection"),
-        ]),
+        command: z
+            .discriminatedUnion("name", [
+                z
+                    .object({
+                        name: z.literal("list-databases"),
+                        parameters: z.object({}),
+                    })
+                    .describe(
+                        "The shape of 'list-databases' command to list all the databases for a MongoDB connection."
+                    ),
+                z
+                    .object({
+                        name: z.literal("list-collections"),
+                        parameters: z.object({
+                            database: DbOperationArgs.database,
+                        }),
+                    })
+                    .describe(
+                        "The shape of 'list-collections' command to list all the collections for a given database."
+                    ),
+                z
+                    .object({
+                        name: z.literal("collection-indexes"),
+                        parameters: z.object(DbOperationArgs),
+                    })
+                    .describe("The shape of 'collection-indexes' command to describe the indexes for a collection."),
+                z
+                    .object({
+                        name: z.literal("collection-schema"),
+                        parameters: z.object(DbOperationArgs),
+                    })
+                    .describe("The shape of 'collection-schema' command to describe the schema for a collection."),
+            ])
+            .describe("The command to be provided to the tool."),
     };
     protected operationType: OperationType = "read";
 

tests/accuracy/collection-indexes.test.ts

@@ -7,10 +7,15 @@ function callsCollectionIndexes(prompt: string): AccuracyTestConfig {
         prompt: prompt,
         expectedToolCalls: [
             {
-                toolName: "collection-indexes",
+                toolName: "mongodb-ddl",
                 parameters: {
-                    database: "mflix",
-                    collection: "movies",
+                    command: {
+                        name: "collection-indexes",
+                        parameters: {
+                            database: "mflix",
+                            collection: "movies",
+                        },
+                    },
                 },
             },
         ],

tests/accuracy/collection-schema.test.ts

@@ -7,10 +7,15 @@ function callsCollectionSchema(prompt: string): AccuracyTestConfig {
         prompt: prompt,
         expectedToolCalls: [
             {
-                toolName: "collection-schema",
+                toolName: "mongodb-ddl",
                 parameters: {
-                    database: "db1",
-                    collection: "coll1",
+                    command: {
+                        name: "collection-schema",
+                        parameters: {
+                            database: "db1",
+                            collection: "coll1",
+                        },
+                    },
                 },
             },
         ],

tests/accuracy/list-databases.test.ts

@@ -7,8 +7,13 @@ function callsListDatabases(prompt: string): AccuracyTestConfig {
         prompt: prompt,
         expectedToolCalls: [
             {
-                toolName: "list-databases",
-                parameters: {},
+                toolName: "mongodb-ddl",
+                parameters: {
+                    command: {
+                        name: "list-databases",
+                        parameters: {},
+                    },
+                },
             },
         ],
     };