chore: fix build with CJS

Author: gagik

src/common/config.ts

@@ -3,7 +3,6 @@ import type { CliOptions, ConnectionInfo } from "@mongosh/arg-parser";
 import { generateConnectionInfoFromCliArgs } from "@mongosh/arg-parser";
 import { Keychain } from "./keychain.js";
 import type { Secret } from "./keychain.js";
-
 import { z as z4 } from "zod/v4";
 import {
     commaSeparatedToArray,
@@ -14,7 +13,7 @@ import {
     validateConfigKey,
 } from "./configUtils.js";
 import { OPTIONS } from "./argsParserOptions.js";
-import { similarityEnumV4 } from "./schemas.js";
+import { similarityValues, previewFeatureValues } from "./schemas.js";
 
 export const configRegistry = z4.registry<ConfigFieldMeta>();
 
@@ -164,20 +163,19 @@ export const UserConfigSchema = z4.object({
         .number()
         .default(1024)
         .describe("Default number of dimensions for vector search embeddings."),
-    vectorSearchSimilarityFunction: similarityEnumV4
+    vectorSearchSimilarityFunction: z4
+        .enum(similarityValues)
         .default("euclidean")
         .describe("Default similarity function for vector search: 'euclidean', 'cosine', or 'dotProduct'."),
     previewFeatures: z4
         .preprocess(
             (val: string | string[] | undefined) => commaSeparatedToArray(val),
-            z4.array(z4.enum(["vectorSearch"]))
+            z4.array(z4.enum(previewFeatureValues))
         )
         .default([])
         .describe("An array of preview features that are enabled."),
 });
 
-export type Similarity = z4.infer<typeof UserConfigSchema>["vectorSearchSimilarityFunction"];
-export type PreviewFeature = z4.infer<typeof UserConfigSchema>["previewFeatures"][number];
 export type UserConfig = z4.infer<typeof UserConfigSchema> & CliOptions;
 
 export const config = setupUserConfig({

src/common/schemas.ts

@@ -1,7 +1,6 @@
-import z4 from "zod/v4";
-import z3 from "zod/v3";
+export const previewFeatureValues = ["vectorSearch"] as const;
+export type PreviewFeature = (typeof previewFeatureValues)[number];
 
-const similarityValues = ["cosine", "euclidean", "dotProduct"] as const;
+export const similarityValues = ["cosine", "euclidean", "dotProduct"] as const;
 
-export const similarityEnumV4 = z4.enum(similarityValues);
-export const similarityEnum = z3.enum(similarityValues);
+export type Similarity = (typeof similarityValues)[number];

src/common/search/vectorSearchEmbeddingsManager.ts

@@ -1,12 +1,13 @@
 import type { NodeDriverServiceProvider } from "@mongosh/service-provider-node-driver";
 import { BSON, type Document } from "bson";
-import type { Similarity, UserConfig } from "../config.js";
+import type { UserConfig } from "../config.js";
 import type { ConnectionManager } from "../connectionManager.js";
 import z from "zod";
 import { ErrorCodes, MongoDBError } from "../errors.js";
 import { getEmbeddingsProvider } from "./embeddingsProvider.js";
 import type { EmbeddingParameters, SupportedEmbeddingParameters } from "./embeddingsProvider.js";
 import { formatUntrustedData } from "../../tools/tool.js";
+import type { Similarity } from "../schemas.js";
 
 export const quantizationEnum = z.enum(["none", "scalar", "binary"]);
 export type Quantization = z.infer<typeof quantizationEnum>;

src/tools/mongodb/create/createIndex.ts

@@ -4,7 +4,7 @@ import { DbOperationArgs, MongoDBToolBase } from "../mongodbTool.js";
 import { type ToolArgs, type OperationType } from "../../tool.js";
 import type { IndexDirection } from "mongodb";
 import { quantizationEnum } from "../../../common/search/vectorSearchEmbeddingsManager.js";
-import { similarityEnum } from "../../../common/schemas.js";
+import { similarityValues, type Similarity } from "../../../common/schemas.js";
 
 export class CreateIndexTool extends MongoDBToolBase {
     private vectorSearchIndexDefinition = z.object({
@@ -39,7 +39,8 @@ export class CreateIndexTool extends MongoDBToolBase {
                                 .describe(
                                     "Number of vector dimensions that MongoDB Vector Search enforces at index-time and query-time"
                                 ),
-                            similarity: similarityEnum
+                            similarity: z
+                                .enum(similarityValues)
                                 .default(this.config.vectorSearchSimilarityFunction)
                                 .describe(
                                     "Vector similarity function to use to search for top K-nearest neighbors. You can set this field only for vector-type fields."

src/tools/tool.ts

@@ -6,9 +6,10 @@ import type { Session } from "../common/session.js";
 import { LogId } from "../common/logger.js";
 import type { Telemetry } from "../telemetry/telemetry.js";
 import { type ToolEvent } from "../telemetry/types.js";
-import type { PreviewFeature, UserConfig } from "../common/config.js";
+import type { UserConfig } from "../common/config.js";
 import type { Server } from "../server.js";
 import type { Elicitation } from "../elicitation.js";
+import type { PreviewFeature } from "../common/schemas.js";
 
 export type ToolArgs<Args extends ZodRawShape> = z.objectOutputType<Args, ZodNever>;
 export type ToolCallbackArgs<Args extends ZodRawShape> = Parameters<ToolCallback<Args>>;

tests/unit/toolBase.test.ts

@@ -3,11 +3,12 @@ import { z } from "zod";
 import { ToolBase, type OperationType, type ToolCategory, type ToolConstructorParams } from "../../src/tools/tool.js";
 import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import type { Session } from "../../src/common/session.js";
-import type { PreviewFeature, UserConfig } from "../../src/common/config.js";
+import type { UserConfig } from "../../src/common/config.js";
 import type { Telemetry } from "../../src/telemetry/telemetry.js";
 import type { Elicitation } from "../../src/elicitation.js";
 import type { CompositeLogger } from "../../src/common/logger.js";
 import type { TelemetryToolMetadata, ToolCallbackArgs } from "../../src/tools/tool.js";
+import type { PreviewFeature } from "../../src/common/schemas.js";
 
 describe("ToolBase", () => {
     let mockSession: Session;

tsconfig.cjs.json

@@ -1,7 +1,6 @@
 {
   "extends": "./tsconfig.build.json",
   "compilerOptions": {
-    "noCheck": true,
     "module": "commonjs",
     "moduleResolution": "node",
     "outDir": "./dist/cjs"