@mhalder/qdrant-mcp-server 3.3.3 → 3.3.4

package/src/tools/code.ts

@@ -3,8 +3,8 @@
  */
 
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import logger from "../logger.js";
 import type { CodeIndexer } from "../code/indexer.js";
+import logger from "../logger.js";
 import { withToolLogging } from "./logging.js";
 import * as schemas from "./schemas.js";
 
@@ -14,10 +14,7 @@ export interface CodeToolDependencies {
   codeIndexer: CodeIndexer;
 }
 
-export function registerCodeTools(
-  server: McpServer,
-  deps: CodeToolDependencies,
-): void {
+export function registerCodeTools(server: McpServer, deps: CodeToolDependencies): void {
   const { codeIndexer } = deps;
 
   // index_codebase
@@ -39,10 +36,7 @@ export function registerCodeTools(
           path,
           { forceReindex, extensions, ignorePatterns },
           (progress) => {
-            log.debug(
-              { phase: progress.phase, percentage: progress.percentage },
-              progress.message,
-            );
+            log.debug({ phase: progress.phase, percentage: progress.percentage }, progress.message);
             if (progressToken !== undefined) {
               extra.sendNotification({
                 method: "notifications/progress",
@@ -54,7 +48,7 @@ export function registerCodeTools(
                 },
               });
             }
-          },
+          }
         );
 
         let statusMessage = `Indexed ${stats.filesIndexed}/${stats.filesScanned} files (${stats.chunksCreated} chunks) in ${(stats.durationMs / 1000).toFixed(1)}s`;
@@ -69,8 +63,8 @@ export function registerCodeTools(
           content: [{ type: "text", text: statusMessage }],
           isError: stats.status === "failed",
         };
-      },
-    ),
+      }
+    )
   );
 
   // search_code
@@ -82,48 +76,40 @@ export function registerCodeTools(
         "Search indexed codebase using natural language queries. Returns semantically relevant code chunks with file paths and line numbers.",
       inputSchema: schemas.SearchCodeSchema,
     },
-    withToolLogging(
-      "search_code",
-      async ({ path, query, limit, fileTypes, pathPattern }) => {
-        log.info(
-          { tool: "search_code", path, query: query.substring(0, 80) },
-          "Tool called",
-        );
-        const results = await codeIndexer.searchCode(path, query, {
-          limit,
-          fileTypes,
-          pathPattern,
-        });
-
-        if (results.length === 0) {
-          return {
-            content: [
-              { type: "text", text: `No results found for query: "${query}"` },
-            ],
-          };
-        }
-
-        // Format results with file references
-        const formattedResults = results
-          .map(
-            (r, idx) =>
-              `\n--- Result ${idx + 1} (score: ${r.score.toFixed(3)}) ---\n` +
-              `File: ${r.filePath}:${r.startLine}-${r.endLine}\n` +
-              `Language: ${r.language}\n\n` +
-              `${r.content}\n`,
-          )
-          .join("\n");
+    withToolLogging("search_code", async ({ path, query, limit, fileTypes, pathPattern }) => {
+      log.info({ tool: "search_code", path, query: query.substring(0, 80) }, "Tool called");
+      const results = await codeIndexer.searchCode(path, query, {
+        limit,
+        fileTypes,
+        pathPattern,
+      });
 
+      if (results.length === 0) {
         return {
-          content: [
-            {
-              type: "text",
-              text: `Found ${results.length} result(s):\n${formattedResults}`,
-            },
-          ],
+          content: [{ type: "text", text: `No results found for query: "${query}"` }],
         };
-      },
-    ),
+      }
+
+      // Format results with file references
+      const formattedResults = results
+        .map(
+          (r, idx) =>
+            `\n--- Result ${idx + 1} (score: ${r.score.toFixed(3)}) ---\n` +
+            `File: ${r.filePath}:${r.startLine}-${r.endLine}\n` +
+            `Language: ${r.language}\n\n` +
+            `${r.content}\n`
+        )
+        .join("\n");
+
+      return {
+        content: [
+          {
+            type: "text",
+            text: `Found ${results.length} result(s):\n${formattedResults}`,
+          },
+        ],
+      };
+    })
   );
 
   // reindex_changes
@@ -140,10 +126,7 @@ export function registerCodeTools(
       const progressToken = extra._meta?.progressToken;
 
       const stats = await codeIndexer.reindexChanges(path, (progress) => {
-        log.debug(
-          { phase: progress.phase, percentage: progress.percentage },
-          progress.message,
-        );
+        log.debug({ phase: progress.phase, percentage: progress.percentage }, progress.message);
         if (progressToken !== undefined) {
           extra.sendNotification({
             method: "notifications/progress",
@@ -164,18 +147,14 @@ export function registerCodeTools(
       message += `- Chunks added: ${stats.chunksAdded}\n`;
       message += `- Duration: ${(stats.durationMs / 1000).toFixed(1)}s`;
 
-      if (
-        stats.filesAdded === 0 &&
-        stats.filesModified === 0 &&
-        stats.filesDeleted === 0
-      ) {
+      if (stats.filesAdded === 0 && stats.filesModified === 0 && stats.filesDeleted === 0) {
         message = `No changes detected. Codebase is up to date.`;
       }
 
       return {
         content: [{ type: "text", text: message }],
       };
-    }),
+    })
   );
 
   // get_index_status
@@ -215,7 +194,7 @@ export function registerCodeTools(
       return {
         content: [{ type: "text", text: JSON.stringify(status, null, 2) }],
       };
-    }),
+    })
   );
 
   // clear_index
@@ -231,10 +210,8 @@ export function registerCodeTools(
       log.info({ tool: "clear_index", path }, "Tool called");
       await codeIndexer.clearIndex(path);
       return {
-        content: [
-          { type: "text", text: `Index cleared for codebase at "${path}".` },
-        ],
+        content: [{ type: "text", text: `Index cleared for codebase at "${path}".` }],
       };
-    }),
+    })
   );
 }

package/src/tools/collection.ts

@@ -3,8 +3,8 @@
  */
 
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import logger from "../logger.js";
 import type { EmbeddingProvider } from "../embeddings/base.js";
+import logger from "../logger.js";
 import type { QdrantManager } from "../qdrant/client.js";
 import { withToolLogging } from "./logging.js";
 import * as schemas from "./schemas.js";
@@ -16,10 +16,7 @@ export interface CollectionToolDependencies {
   embeddings: EmbeddingProvider;
 }
 
-export function registerCollectionTools(
-  server: McpServer,
-  deps: CollectionToolDependencies,
-): void {
+export function registerCollectionTools(server: McpServer, deps: CollectionToolDependencies): void {
   const { qdrant, embeddings } = deps;
 
   // create_collection
@@ -31,31 +28,20 @@ export function registerCollectionTools(
         "Create a new vector collection in Qdrant. The collection will be configured with the embedding provider's dimensions automatically. Set enableHybrid to true to enable hybrid search combining semantic and keyword search.",
       inputSchema: schemas.CreateCollectionSchema,
     },
-    withToolLogging(
-      "create_collection",
-      async ({ name, distance, enableHybrid }) => {
-        log.info(
-          { tool: "create_collection", collection: name },
-          "Tool called",
-        );
-        const vectorSize = embeddings.getDimensions();
-        await qdrant.createCollection(
-          name,
-          vectorSize,
-          distance,
-          enableHybrid || false,
-        );
+    withToolLogging("create_collection", async ({ name, distance, enableHybrid }) => {
+      log.info({ tool: "create_collection", collection: name }, "Tool called");
+      const vectorSize = embeddings.getDimensions();
+      await qdrant.createCollection(name, vectorSize, distance, enableHybrid || false);
 
-        let message = `Collection "${name}" created successfully with ${vectorSize} dimensions and ${distance || "Cosine"} distance metric.`;
-        if (enableHybrid) {
-          message += " Hybrid search is enabled for this collection.";
-        }
+      let message = `Collection "${name}" created successfully with ${vectorSize} dimensions and ${distance || "Cosine"} distance metric.`;
+      if (enableHybrid) {
+        message += " Hybrid search is enabled for this collection.";
+      }
 
-        return {
-          content: [{ type: "text", text: message }],
-        };
-      },
-    ),
+      return {
+        content: [{ type: "text", text: message }],
+      };
+    })
   );
 
   // list_collections
@@ -72,7 +58,7 @@ export function registerCollectionTools(
       return {
         content: [{ type: "text", text: JSON.stringify(collections, null, 2) }],
       };
-    }),
+    })
   );
 
   // get_collection_info
@@ -85,15 +71,12 @@ export function registerCollectionTools(
       inputSchema: schemas.GetCollectionInfoSchema,
     },
     withToolLogging("get_collection_info", async ({ name }) => {
-      log.info(
-        { tool: "get_collection_info", collection: name },
-        "Tool called",
-      );
+      log.info({ tool: "get_collection_info", collection: name }, "Tool called");
       const info = await qdrant.getCollectionInfo(name);
       return {
         content: [{ type: "text", text: JSON.stringify(info, null, 2) }],
       };
-    }),
+    })
   );
 
   // delete_collection
@@ -108,10 +91,8 @@ export function registerCollectionTools(
       log.info({ tool: "delete_collection", collection: name }, "Tool called");
       await qdrant.deleteCollection(name);
       return {
-        content: [
-          { type: "text", text: `Collection "${name}" deleted successfully.` },
-        ],
+        content: [{ type: "text", text: `Collection "${name}" deleted successfully.` }],
       };
-    }),
+    })
   );
 }

package/src/tools/document.ts

@@ -3,9 +3,9 @@
  */
 
 import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import logger from "../logger.js";
 import type { EmbeddingProvider } from "../embeddings/base.js";
 import { BM25SparseVectorGenerator } from "../embeddings/sparse.js";
+import logger from "../logger.js";
 import type { QdrantManager } from "../qdrant/client.js";
 import { withToolLogging } from "./logging.js";
 import * as schemas from "./schemas.js";
@@ -17,10 +17,7 @@ export interface DocumentToolDependencies {
   embeddings: EmbeddingProvider;
 }
 
-export function registerDocumentTools(
-  server: McpServer,
-  deps: DocumentToolDependencies,
-): void {
+export function registerDocumentTools(server: McpServer, deps: DocumentToolDependencies): void {
   const { qdrant, embeddings } = deps;
 
   // add_documents
@@ -33,10 +30,7 @@ export function registerDocumentTools(
       inputSchema: schemas.AddDocumentsSchema,
     },
     withToolLogging("add_documents", async ({ collection, documents }) => {
-      log.info(
-        { tool: "add_documents", collection, count: documents.length },
-        "Tool called",
-      );
+      log.info({ tool: "add_documents", collection, count: documents.length }, "Tool called");
       // Check if collection exists and get info
       const exists = await qdrant.collectionExists(collection);
       if (!exists) {
@@ -69,7 +63,7 @@ export function registerDocumentTools(
               text: string;
               metadata?: Record<string, any>;
             },
-            index: number,
+            index: number
           ) => ({
             id: doc.id,
             vector: embeddingResults[index].embedding,
@@ -78,7 +72,7 @@ export function registerDocumentTools(
               text: doc.text,
               ...doc.metadata,
             },
-          }),
+          })
         );
 
         await qdrant.addPointsWithSparse(collection, points);
@@ -91,7 +85,7 @@ export function registerDocumentTools(
               text: string;
               metadata?: Record<string, any>;
             },
-            index: number,
+            index: number
           ) => ({
             id: doc.id,
             vector: embeddingResults[index].embedding,
@@ -99,7 +93,7 @@ export function registerDocumentTools(
               text: doc.text,
               ...doc.metadata,
             },
-          }),
+          })
         );
 
         await qdrant.addPoints(collection, points);
@@ -113,7 +107,7 @@ export function registerDocumentTools(
           },
         ],
       };
-    }),
+    })
   );
 
   // delete_documents
@@ -125,10 +119,7 @@ export function registerDocumentTools(
       inputSchema: schemas.DeleteDocumentsSchema,
     },
     withToolLogging("delete_documents", async ({ collection, ids }) => {
-      log.info(
-        { tool: "delete_documents", collection, count: ids.length },
-        "Tool called",
-      );
+      log.info({ tool: "delete_documents", collection, count: ids.length }, "Tool called");
       await qdrant.deletePoints(collection, ids);
       return {
         content: [
@@ -138,6 +129,6 @@ export function registerDocumentTools(
           },
         ],
       };
-    }),
+    })
   );
 }

package/src/tools/federated.test.ts

@@ -1,12 +1,7 @@
-import { describe, it, expect, vi, beforeEach } from "vitest";
+import { beforeEach, describe, expect, it, vi } from "vitest";
 import type { CodeSearchResult } from "../code/types.js";
 import type { GitSearchResult } from "../git/types.js";
-import {
-  buildCorrelations,
-  normalizeScores,
-  calculateRRFScore,
-  pathsMatch,
-} from "./federated.js";
+import { buildCorrelations, calculateRRFScore, normalizeScores, pathsMatch } from "./federated.js";
 
 vi.mock("../logger.js", () => ({
   default: {
@@ -137,9 +132,7 @@ describe("pathsMatch", () => {
   it("should NOT match paths with different parent directories", () => {
     // This was the false positive case
     expect(pathsMatch("app/models/user.ts", "lib/user.ts")).toBe(false);
-    expect(pathsMatch("src/auth/middleware.ts", "other/middleware.ts")).toBe(
-      false,
-    );
+    expect(pathsMatch("src/auth/middleware.ts", "other/middleware.ts")).toBe(false);
   });
 
   it("should NOT match completely different filenames", () => {
@@ -162,9 +155,7 @@ describe("pathsMatch", () => {
 });
 
 describe("buildCorrelations", () => {
-  const createCodeResult = (
-    overrides: Partial<CodeSearchResult> = {},
-  ): CodeSearchResult => ({
+  const createCodeResult = (overrides: Partial<CodeSearchResult> = {}): CodeSearchResult => ({
     content: "function test() {}",
     filePath: "src/utils/helper.ts",
     startLine: 10,
@@ -175,9 +166,7 @@ describe("buildCorrelations", () => {
     ...overrides,
   });
 
-  const createGitResult = (
-    overrides: Partial<GitSearchResult> = {},
-  ): GitSearchResult => ({
+  const createGitResult = (overrides: Partial<GitSearchResult> = {}): GitSearchResult => ({
     content: "Commit content",
     commitHash: "abc123def456",
     shortHash: "abc123d",
@@ -192,9 +181,7 @@ describe("buildCorrelations", () => {
 
   it("should find correlations when file paths match", () => {
     const codeResults = [createCodeResult({ filePath: "src/utils/helper.ts" })];
-    const gitResults = [
-      createGitResult({ files: ["src/utils/helper.ts", "src/index.ts"] }),
-    ];
+    const gitResults = [createGitResult({ files: ["src/utils/helper.ts", "src/index.ts"] })];
 
     const correlations = buildCorrelations(codeResults, gitResults);
 
@@ -205,9 +192,7 @@ describe("buildCorrelations", () => {
   });
 
   it("should match partial paths (relative vs absolute)", () => {
-    const codeResults = [
-      createCodeResult({ filePath: "/home/user/project/src/utils/helper.ts" }),
-    ];
+    const codeResults = [createCodeResult({ filePath: "/home/user/project/src/utils/helper.ts" })];
     const gitResults = [createGitResult({ files: ["src/utils/helper.ts"] })];
 
     const correlations = buildCorrelations(codeResults, gitResults);
@@ -242,12 +227,8 @@ describe("buildCorrelations", () => {
 
     expect(correlations).toHaveLength(1);
     expect(correlations[0].relatedCommits).toHaveLength(2);
-    expect(correlations[0].relatedCommits.map((c) => c.shortHash)).toContain(
-      "abc123d",
-    );
-    expect(correlations[0].relatedCommits.map((c) => c.shortHash)).toContain(
-      "xyz789a",
-    );
+    expect(correlations[0].relatedCommits.map((c) => c.shortHash)).toContain("abc123d");
+    expect(correlations[0].relatedCommits.map((c) => c.shortHash)).toContain("xyz789a");
   });
 
   it("should handle multiple code results with different correlations", () => {
@@ -297,9 +278,7 @@ describe("buildCorrelations", () => {
   });
 
   it("should normalize Windows-style paths", () => {
-    const codeResults = [
-      createCodeResult({ filePath: "src\\utils\\helper.ts" }),
-    ];
+    const codeResults = [createCodeResult({ filePath: "src\\utils\\helper.ts" })];
     const gitResults = [createGitResult({ files: ["src/utils/helper.ts"] })];
 
     const correlations = buildCorrelations(codeResults, gitResults);
@@ -379,11 +358,11 @@ describe("contextual_search integration", () => {
 
     // Get the handler for contextual_search
     const contextualSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "contextual_search",
+      (call) => call[0] === "contextual_search"
     );
     expect(contextualSearchCall).toBeDefined();
 
-    const handler = contextualSearchCall![2];
+    const handler = contextualSearchCall?.[2];
     const result = await handler({
       path: "/test/repo",
       query: "test function",
@@ -392,15 +371,13 @@ describe("contextual_search integration", () => {
       correlate: true,
     });
 
-    expect(mockCodeIndexer.searchCode).toHaveBeenCalledWith(
-      "/test/repo",
-      "test function",
-      { limit: 5 },
-    );
+    expect(mockCodeIndexer.searchCode).toHaveBeenCalledWith("/test/repo", "test function", {
+      limit: 5,
+    });
     expect(mockGitHistoryIndexer.searchHistory).toHaveBeenCalledWith(
       "/test/repo",
       "test function",
-      { limit: 5 },
+      { limit: 5 }
     );
     expect(result.content[0].text).toContain("Code Results");
     expect(result.content[0].text).toContain("Git History Results");
@@ -422,9 +399,9 @@ describe("contextual_search integration", () => {
     });
 
     const contextualSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "contextual_search",
+      (call) => call[0] === "contextual_search"
     );
-    const handler = contextualSearchCall![2];
+    const handler = contextualSearchCall?.[2];
     const result = await handler({
       path: "/test/repo",
       query: "test",
@@ -449,9 +426,9 @@ describe("contextual_search integration", () => {
     });
 
     const contextualSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "contextual_search",
+      (call) => call[0] === "contextual_search"
     );
-    const handler = contextualSearchCall![2];
+    const handler = contextualSearchCall?.[2];
     const result = await handler({
       path: "/test/repo",
       query: "test",
@@ -500,9 +477,9 @@ describe("contextual_search integration", () => {
     });
 
     const contextualSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "contextual_search",
+      (call) => call[0] === "contextual_search"
     );
-    const handler = contextualSearchCall![2];
+    const handler = contextualSearchCall?.[2];
     const result = await handler({
       path: "/test/repo",
       query: "test",
@@ -574,9 +551,9 @@ describe("federated_search integration", () => {
     });
 
     const federatedSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "federated_search",
+      (call) => call[0] === "federated_search"
     );
-    const handler = federatedSearchCall![2];
+    const handler = federatedSearchCall?.[2];
     const result = await handler({
       paths: ["/repo1", "/repo2"],
       query: "test function",
@@ -608,9 +585,9 @@ describe("federated_search integration", () => {
     });
 
     const federatedSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "federated_search",
+      (call) => call[0] === "federated_search"
     );
-    const handler = federatedSearchCall![2];
+    const handler = federatedSearchCall?.[2];
     const result = await handler({
       paths: ["/repo1", "/repo2"],
       query: "test",
@@ -636,9 +613,9 @@ describe("federated_search integration", () => {
     });
 
     const federatedSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "federated_search",
+      (call) => call[0] === "federated_search"
     );
-    const handler = federatedSearchCall![2];
+    const handler = federatedSearchCall?.[2];
     await handler({
       paths: ["/repo1"],
       query: "test",
@@ -664,9 +641,9 @@ describe("federated_search integration", () => {
     });
 
     const federatedSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "federated_search",
+      (call) => call[0] === "federated_search"
     );
-    const handler = federatedSearchCall![2];
+    const handler = federatedSearchCall?.[2];
     await handler({
       paths: ["/repo1"],
       query: "test",
@@ -721,9 +698,9 @@ describe("federated_search integration", () => {
     });
 
     const federatedSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "federated_search",
+      (call) => call[0] === "federated_search"
     );
-    const handler = federatedSearchCall![2];
+    const handler = federatedSearchCall?.[2];
     const result = await handler({
       paths: ["/repo1"],
       query: "test",
@@ -751,9 +728,9 @@ describe("federated_search integration", () => {
     });
 
     const federatedSearchCall = mockServer.registerTool.mock.calls.find(
-      (call) => call[0] === "federated_search",
+      (call) => call[0] === "federated_search"
     );
-    const handler = federatedSearchCall![2];
+    const handler = federatedSearchCall?.[2];
     const result = await handler({
       paths: ["/repo1"],
       query: "nonexistent query",