chore: allow createEmbeddings to take an optional path

Author: David-Kunz

lib/embeddings.js

@@ -116,18 +116,18 @@ export async function searchEmbeddings(query, chunks) {
 }
 
 // Only to be used in scripts, not in production
-export async function createEmbeddings(id, chunks) {
+export async function createEmbeddings(id, chunks, dir = path.join(__dirname, '..', 'embeddings')) {
   const embeddings = []
 
   for (let i = 0; i < chunks.length; i++) {
     const embedding = await getEmbeddings(chunks[i])
     embeddings.push(embedding)
   }
 
-  await saveEmbeddings(id, chunks, embeddings)
+  await saveEmbeddings(id, chunks, embeddings, dir)
 }
 
-async function saveEmbeddings(id, chunks, embeddings, dir = path.join(__dirname, '..', 'embeddings')) {
+async function saveEmbeddings(id, chunks, embeddings, dir) {
   if (!chunks.length) throw new Error('No chunks to save')
   if (!embeddings || !embeddings.length) throw new Error('No embeddings to save')
   if (chunks.length !== embeddings.length) throw new Error('Chunks and embeddings length mismatch')