Use JSON5 for embeddings (#719)

huggingface · Jan 22, 2024 · 6e0b0ea · 6e0b0ea
1 parent 1e4a456
commit 6e0b0ea
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/lib/server/embeddingModels.ts b/src/lib/server/embeddingModels.ts
@@ -9,6 +9,8 @@ import {
 } from "$lib/server/embeddingEndpoints/embeddingEndpoints";
 import { embeddingEndpointTransformersJS } from "$lib/server/embeddingEndpoints/transformersjs/embeddingEndpoints";
 
+import JSON5 from "json5";
+
 const modelConfig = z.object({
 	/** Used as an identifier in DB */
 	id: z.string().optional(),
@@ -37,7 +39,7 @@ const rawEmbeddingModelJSON =
 	}
 ]`;
 
-const embeddingModelsRaw = z.array(modelConfig).parse(JSON.parse(rawEmbeddingModelJSON));
+const embeddingModelsRaw = z.array(modelConfig).parse(JSON5.parse(rawEmbeddingModelJSON));
 
 const processEmbeddingModel = async (m: z.infer<typeof modelConfig>) => ({
 	...m,