chore: fix other relevant tests

Author: himanshusinghs

src/tools/mongodb/mongodbSchemas.ts

@@ -6,6 +6,9 @@ export const zVoyageModels = z
     .default("voyage-3-large");
 
 export const zVoyageEmbeddingParameters = z.object({
+    // OpenAPI JSON Schema supports enum only as string so the public facing
+    // parameters that are fed to LLM providers should expect the dimensions as
+    // stringified numbers which are then transformed to actual numbers.
     outputDimension: z
         .union([z.literal("256"), z.literal("512"), z.literal("1024"), z.literal("2048"), z.literal("4096")])
         .transform((value): number => Number.parseInt(value))
@@ -16,6 +19,14 @@ export const zVoyageEmbeddingParameters = z.object({
 
 export const zVoyageAPIParameters = zVoyageEmbeddingParameters
     .extend({
+        // Unlike, public facing parameters, `zVoyageEmbeddingParameters`, the
+        // api parameters need to be correct number and because we do an
+        // additional parsing before calling the API, we override the
+        // outputDimension schema to expect a union of numbers.
+        outputDimension: z
+            .union([z.literal(256), z.literal(512), z.literal(1024), z.literal(2048), z.literal(4096)])
+            .optional()
+            .default(1024),
         inputType: z.enum(["query", "document"]),
     })
     .strip();

tests/integration/tools/mongodb/create/insertMany.test.ts

@@ -203,7 +203,7 @@ describeWithMongoDB(
                     // embeddingParameters so that we can simulate the idea
                     // of unknown or mismatched quantization.
 
-                    // embeddingParameters: { outputDimension: 256,
+                    // embeddingParameters: { outputDimension: "256",
                     // outputDtype: "float", model: "voyage-3-large", input:
                     // [
                     //         {
@@ -558,7 +558,7 @@ describeWithMongoDB(
                         documents: [{ title: "The Matrix" }],
                         embeddingParameters: {
                             model: "voyage-3.5-lite",
-                            outputDimension: 256,
+                            outputDimension: "256",
                             input: [{ titleEmbeddings: "The Matrix" }],
                         },
                     },

tests/integration/tools/mongodb/read/aggregate.test.ts

@@ -444,7 +444,7 @@ If the user requests additional filtering, include filters in \`$vectorSearch.fi
                                 limit: 10,
                                 embeddingParameters: {
                                     model: "voyage-3-large",
-                                    outputDimension: 256,
+                                    outputDimension: "256",
                                 },
                             },
                         },
@@ -508,7 +508,7 @@ If the user requests additional filtering, include filters in \`$vectorSearch.fi
                                             limit: 10,
                                             embeddingParameters: {
                                                 model: "voyage-3-large",
-                                                outputDimension: 256,
+                                                outputDimension: "256",
                                                 outputDType: dataType,
                                             },
                                         },
@@ -575,7 +575,7 @@ If the user requests additional filtering, include filters in \`$vectorSearch.fi
                                             limit: 10,
                                             embeddingParameters: {
                                                 model: "voyage-3-large",
-                                                outputDimension: 256,
+                                                outputDimension: "256",
                                                 outputDType: dataType,
                                             },
                                         },
@@ -642,7 +642,7 @@ If the user requests additional filtering, include filters in \`$vectorSearch.fi
                                             limit: 10,
                                             embeddingParameters: {
                                                 model: "voyage-3-large",
-                                                outputDimension: 256,
+                                                outputDimension: "256",
                                                 outputDType: dataType,
                                             },
                                         },
@@ -709,7 +709,7 @@ If the user requests additional filtering, include filters in \`$vectorSearch.fi
                                             limit: 10,
                                             embeddingParameters: {
                                                 model: "voyage-3-large",
-                                                outputDimension: 256,
+                                                outputDimension: "256",
                                                 outputDType: dataType,
                                             },
                                         },
@@ -774,7 +774,7 @@ If the user requests additional filtering, include filters in \`$vectorSearch.fi
                                     limit: 10,
                                     embeddingParameters: {
                                         model: "voyage-3-large",
-                                        outputDimension: 256,
+                                        outputDimension: "256",
                                         outputDType: "float",
                                     },
                                     filter: { name: 10 },
@@ -837,7 +837,7 @@ If the user requests additional filtering, include filters in \`$vectorSearch.fi
                                     limit: 10,
                                     embeddingParameters: {
                                         model: "voyage-3-large",
-                                        outputDimension: 256,
+                                        outputDimension: "256",
                                         outputDType: "float",
                                     },
                                     filter: { name: 10 },
@@ -900,7 +900,7 @@ If the user requests additional filtering, include filters in \`$vectorSearch.fi
                                     limit: 10,
                                     embeddingParameters: {
                                         model: "voyage-3-large",
-                                        outputDimension: 256,
+                                        outputDimension: "256",
                                         outputDType: "float",
                                     },
                                     filter: { name: 10 },