be precise for search index

Author: edgarw19

src/tools/mongodb/create/createSearchIndex.ts

@@ -2,13 +2,14 @@ import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { DbOperationArgs, MongoDBToolBase, SearchIndexArgs } from "../mongodbTool.js";
 import { OperationType, ToolArgs } from "../../tool.js";
 
+const SEARCH_INDEX_TYPE = "search";
+
 export class CreateSearchIndexTool extends MongoDBToolBase {
     protected name = "create-search-index";
     protected description = "Create an Atlas Search index for a collection";
     protected argsShape = {
         ...DbOperationArgs,
         name: SearchIndexArgs.name,
-        type: SearchIndexArgs.type,
         analyzer: SearchIndexArgs.analyzer,
         mappings: SearchIndexArgs.mappings,
     };
@@ -19,15 +20,14 @@ export class CreateSearchIndexTool extends MongoDBToolBase {
         database,
         collection,
         name,
-        type,
         analyzer,
         mappings,
     }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> {
         const provider = await this.ensureConnected();
         const indexes = await provider.createSearchIndexes(database, collection, [
             {
                 name,
-                type,
+                type: SEARCH_INDEX_TYPE,
                 definition: {
                     analyzer,
                     mappings,
@@ -38,7 +38,7 @@ export class CreateSearchIndexTool extends MongoDBToolBase {
         return {
             content: [
                 {
-                    text: `Created the index "${indexes[0]}" on collection "${collection}" in database "${database}"`,
+                    text: `Created the search index "${indexes[0]}" on collection "${collection}" in database "${database}"`,
                     type: "text",
                 },
             ],

src/tools/mongodb/mongodbTool.ts

@@ -12,7 +12,6 @@ export const DbOperationArgs = {
 
 export const SearchIndexArgs = {
     name: z.string().describe("The name of the index"),
-    type: z.enum(["search", "vectorSearch"]).optional().default("search").describe("The type of the index"),
     analyzer: z
         .string()
         .optional()
@@ -42,7 +41,6 @@ export const SearchIndexArgs = {
                                     "document",
                                     "embeddedDocuments",
                                     "geo",
-                                    "knnVector",
                                     "number",
                                     "objectId",
                                     "string",
@@ -61,7 +59,7 @@ export const SearchIndexArgs = {
                 .describe("The field mapping definitions. If `dynamic` is set to false, this is required."),
         })
         .describe(
-            "Document describing the index to create. The definition syntax depends on whether you create a standard search index or a Vector Search index."
+            "Document describing the index to create."
         ),
 };
 

src/tools/mongodb/update/updateSearchIndex.ts

@@ -4,7 +4,7 @@ import { OperationType, ToolArgs } from "../../tool.js";
 
 export class UpdateSearchIndexTool extends MongoDBToolBase {
     protected name = "update-search-index";
-    protected description = "Updates a search index for a collection";
+    protected description = "Updates an Atlas Search index for a collection";
     protected argsShape = {
         ...DbOperationArgs,
         name: SearchIndexArgs.name,
@@ -32,7 +32,7 @@ export class UpdateSearchIndexTool extends MongoDBToolBase {
         return {
             content: [
                 {
-                    text: `Successfully updated index "${name}" on collection "${collection}" in database "${database}"`,
+                    text: `Successfully updated search index "${name}" on collection "${collection}" in database "${database}"`,
                     type: "text",
                 },
             ],