Store embeddings in the DB with updated schema

praveen-palanisamy · Apr 27, 2023 · f7bc501 · f7bc501
1 parent 9accd54
commit f7bc501
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 8 deletions.
diff --git a/renderer/components/FileUploadArea.tsx b/renderer/components/FileUploadArea.tsx
@@ -89,7 +89,7 @@ function FileUploadArea(props: FileUploadAreaProps) {
                     chunks,
                     extractedText: text,
                   };
-                  console.log(fileObject);
+                  // console.log(fileObject);
                   // Store embeddings in the database
                   await storeEmbeddings(fileObject);
 

diff --git a/renderer/services/storeEmbeddings.ts b/renderer/services/storeEmbeddings.ts
@@ -4,18 +4,27 @@ import {FileLite } from "../types/file";
 
 export const storeEmbeddings = async (fileObject: FileLite) => {
 
-    console.log("storeEmbeddings:", fileObject)
     const {name, url, type, size, extractedText, embedding, chunks} = fileObject
 
-    const {error} = await supabase.from('files').insert({
+    const {data, error} = await supabase.from('File').insert({
        name,
        url,
        type,
        size,
-       extracted_text: extractedText,
-       mean_embedding: embedding,
-       chunks,
-    })
+       extractedText,
+       meanEmbedding: embedding,
+    }).select('id')
+
+    const file_id = data[0].id
+
+    console.log("File stored with file_id:", file_id)
+    console.log("error:", error)
+
+    const {error: error2} = await supabase.from('TextEmbedding').insert(chunks.map((chunk) => ({
+        text: chunk.text,
+        embedding: chunk.embedding,
+        fileId: file_id
+    })))
     console.log("Embeddings stored")
-    console.log("Error: ", error)
+    console.log("Error:", error2)
 }