From b9d62805116b13245f32540c88f9e83fd69ec099 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Thor=20=E9=9B=B7=E7=A5=9E=20Schaeff?=
 <5748289+thorwebdev@users.noreply.github.com>
Date: Tue, 16 Apr 2024 15:32:00 +0800
Subject: [PATCH] chore: import edge runtime types. (#22722)

* chore: import edge runtime types.

* chore: remove versions.

---------

Co-authored-by: Bobbie Soedirgo <bobbie@soedirgo.dev>
---
 .../supabase/functions/generate-embedding/index.ts        | 8 ++++++--
 .../ai/edge-functions/supabase/functions/search/index.ts  | 6 ++++--
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/examples/ai/edge-functions/supabase/functions/generate-embedding/index.ts b/examples/ai/edge-functions/supabase/functions/generate-embedding/index.ts
index 4befba0338..5efecefe1b 100644
--- a/examples/ai/edge-functions/supabase/functions/generate-embedding/index.ts
+++ b/examples/ai/edge-functions/supabase/functions/generate-embedding/index.ts
@@ -1,4 +1,5 @@
-/// <reference types="https://esm.sh/@supabase/functions-js@2.3.1/src/edge-runtime.d.ts" />
+/// <reference types="https://esm.sh/@supabase/functions-js/src/edge-runtime.d.ts" />
+
 import { createClient } from "npm:@supabase/supabase-js@2.42.0";
 import { Database, Tables } from "../_shared/database.types.ts";
 
@@ -15,6 +16,7 @@ const supabase = createClient<Database>(
   Deno.env.get("SUPABASE_URL")!,
   Deno.env.get("SUPABASE_SERVICE_ROLE_KEY")!,
 );
+
 const model = new Supabase.ai.Session("gte-small");
 
 Deno.serve(async (req) => {
@@ -33,7 +35,9 @@ Deno.serve(async (req) => {
   });
 
   // Store in DB
-  const { error } = await supabase.from("embeddings").update({ embedding }).eq(
+  const { error } = await supabase.from("embeddings").update({
+    embedding: JSON.stringify(embedding),
+  }).eq(
     "id",
     id,
   );
diff --git a/examples/ai/edge-functions/supabase/functions/search/index.ts b/examples/ai/edge-functions/supabase/functions/search/index.ts
index a4551d25a3..7054154a9d 100644
--- a/examples/ai/edge-functions/supabase/functions/search/index.ts
+++ b/examples/ai/edge-functions/supabase/functions/search/index.ts
@@ -1,4 +1,5 @@
-/// <reference types="https://esm.sh/@supabase/functions-js@2.3.1/src/edge-runtime.d.ts" />
+/// <reference types="https://esm.sh/@supabase/functions-js/src/edge-runtime.d.ts" />
+
 import { createClient } from "npm:@supabase/supabase-js@2.42.0";
 import { Database } from "../_shared/database.types.ts";
 
@@ -6,6 +7,7 @@ const supabase = createClient<Database>(
   Deno.env.get("SUPABASE_URL")!,
   Deno.env.get("SUPABASE_SERVICE_ROLE_KEY")!,
 );
+
 const model = new Supabase.ai.Session("gte-small");
 
 Deno.serve(async (req) => {
@@ -20,7 +22,7 @@ Deno.serve(async (req) => {
   // Query embeddings.
   const { data: result, error } = await supabase
     .rpc("query_embeddings", {
-      embedding,
+      embedding: JSON.stringify(embedding),
       match_threshold: 0.8,
     })
     .select("content")