feat: complete the supabase adapter method

src/adapters/supabase/helpers/client.ts

@@ -581,7 +581,33 @@ export const _approveLabelChange = async (changeId: number) => {
  * @param issue - The issue number
  * @param embeddings - The vector of floating point numbers
  */
-export const upsertEmbeddings = async (org: string, repo: string, issue: number, embeddings: Array<number>) => {
-  // TODO: Insert or Update embeddings data
-  return;
+export const upsertEmbeddings = async (org: string, repo: string, issue: number, embedding: number[]) => {
+  const { supabase } = getAdapters();
+  const logger = getLogger();
+  const { data } = await supabase.from("embeddings").select("*").eq("org", org).eq("repo", repo).eq("issue", issue).single();
+  if (data) {
+    // Update the existing record with the new embedding for a given set.
+    const id = data["id"] as number;
+    const { error } = await supabase.from("embeddings").upsert({
+      id: id,
+      embedding,
+      updated_at: new Date().toISOString(),
+    });
+    if (error) {
+      logger.info(`Updating embedding table failed. id: ${id}, org: ${org}, repo: ${repo}, issue: ${issue}, embedding: ${embedding.join(", ")}`);
+    }
+  } else {
+    // Insert a new record to the `embeddings` table.
+    const { error } = await supabase.from("embeddings").upsert({
+      org,
+      repo,
+      issue,
+      embedding,
+      created_at: new Date().toISOString(),
+      updated_at: new Date().toISOString(),
+    });
+    if (error) {
+      logger.info(`Inserting to embedding table failed. org: ${org}, repo: ${repo}, issue: ${issue}, embedding: ${embedding.join(", ")}`);
+    }
+  }
 };

supabase/migrations/20231030085814_create_embedding_table.sql

@@ -1,11 +1,11 @@
--- Creates the `embeddings` table to store embeddings for every issue
+-- Creates the `embeddings` table to store an embedding for every issue
 -- We've decided to use the `text-embedding-ada-002` model in the beginning which has 1536 output dimensions
 CREATE TABLE IF NOT EXISTS embeddings (
     id serial PRIMARY KEY,
     org character varying(255) NOT NULL,
     repo character varying(255) NOT NULL,
     issue integer NOT NULL,
-    embeddings vector(1536) NOT NULL, 
+    embedding vector(1536) NOT NULL, 
     created_at timestamptz NOT NULL,
     updated_at timestamptz NOT NULL,
 );