chore: fix accuracy tests and add defaults

Author: kmruiz

src/common/search/embeddingsProvider.ts

@@ -20,10 +20,16 @@ interface EmbeddingsProvider<SupportedModels extends string, SupportedEmbeddingP
     ): Promise<Embeddings[]>;
 }
 
-export const zVoyageModels = z.enum(["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"]);
+export const zVoyageModels = z
+    .enum(["voyage-3-large", "voyage-3.5", "voyage-3.5-lite", "voyage-code-3"])
+    .default("voyage-3-large");
+
 export const zVoyageEmbeddingParameters = z.object({
-    outputDimension: z.union([z.literal(256), z.literal(512), z.literal(1024), z.literal(2048), z.literal(4096)]),
-    outputDType: z.enum(["float", "int8", "uint8", "binary", "ubinary"]),
+    outputDimension: z
+        .union([z.literal(256), z.literal(512), z.literal(1024), z.literal(2048), z.literal(4096)])
+        .optional()
+        .default(1024),
+    outputDType: z.enum(["float", "int8", "uint8", "binary", "ubinary"]).optional().default("float"),
 });
 
 type VoyageModels = z.infer<typeof zVoyageModels>;

tests/accuracy/aggregate.test.ts

@@ -48,7 +48,9 @@ describeAccuracyTests([
                                 index: "titles",
                                 path: "title_embeddings",
                                 queryVector: "hammer of justice",
-                                embeddingModel: "voyage-3-large",
+                                embeddingParameters: {
+                                    model: "voyage-3-large",
+                                },
                             },
                         },
                     ],
@@ -103,7 +105,9 @@ describeAccuracyTests([
                                 index: "my-index",
                                 path: "plot_embeddings",
                                 queryVector: "sci-fy",
-                                embeddingModel: "voyage-3-large",
+                                embeddingParameters: {
+                                    model: "voyage-3-large",
+                                },
                             },
                         },
                     ],