fix rls and worker agent issues

JoinTheAlliance · Feb 9, 2024 · 82895b7 · 82895b7
1 parent 7fd9da6
commit 82895b7
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 9 deletions.
diff --git a/packages/agent/src/agent/agent.ts b/packages/agent/src/agent/agent.ts
@@ -144,8 +144,6 @@ export const onMessage = async (message: any, runtime: AgentRuntime) => {
   addCustomActions(runtime);
   addCustomEvaluators(runtime);
 
-  console.log('message is', message)
-
   let {
     content: senderContent,
     senderId,

diff --git a/packages/agent/src/lib/memory.ts b/packages/agent/src/lib/memory.ts
@@ -91,7 +91,7 @@ export class MemoryManager {
       count = defaultMatchCount,
       userIds = [],
     } = opts
-    console.log('embedding', embedding)
+
     const result = await this.runtime.supabase.rpc('search_memories', {
       query_table_name: this.schema.tableName,
       query_user_ids: userIds,

diff --git a/packages/app/src/Hooks/rooms/useListenToMessagesChanges.tsx b/packages/app/src/Hooks/rooms/useListenToMessagesChanges.tsx
@@ -1,4 +1,4 @@
-import { useSupabaseClient } from "@supabase/auth-helpers-react";
+import { useSession, useSupabaseClient } from "@supabase/auth-helpers-react";
 import { useEffect } from "react";
 import { showNotification } from "@mantine/notifications";
 import { Database } from "../../../types/database.types";
@@ -19,6 +19,7 @@ const useListenToMessagesChanges = ({ getRoomData }: Props) => {
     user: { uid },
     currentRoom: { roomData },
   } = useGlobalStore();
+  const session = useSession();
 
   useEffect(() => {
     const channel = supabase
@@ -108,6 +109,8 @@ const useListenToMessagesChanges = ({ getRoomData }: Props) => {
       )
       .subscribe();
 
+      supabase.realtime.accessToken = session?.access_token as any; // THIS IS REQUIRED FOR RLS!!!
+
     return () => {
       channel.unsubscribe();
     };

diff --git a/packages/app/src/Hooks/rooms/useListenToUnreadMessages.tsx b/packages/app/src/Hooks/rooms/useListenToUnreadMessages.tsx
@@ -26,6 +26,8 @@ const useListenToUnreadMessagesChanges = () => {
       )
       .subscribe();
 
+    supabase.realtime.accessToken = session?.access_token; // THIS IS REQUIRED FOR RLS!!!
+
     // eslint-disable-next-line consistent-return
     return () => {
       channel.unsubscribe();

diff --git a/packages/shell/src/index.ts b/packages/shell/src/index.ts
@@ -104,6 +104,7 @@ async function startApplication() {
 
   // get the room_id from the data
   const room_id = data[0].id
+  supabase.realtime.accessToken = session?.access_token; // THIS IS REQUIRED FOR RLS!!!
 
   // Listen to the 'messages' table for new messages in the specific room
   const channel = supabase

diff --git a/packages/worker/src/index.js b/packages/worker/src/index.js
@@ -131,9 +131,7 @@ server.registerHandler({
     if (req.method === 'OPTIONS') {
       return
     }
-    console.log('responding')
-    console.log('userId', userId)
-
+
     // parse the body from the request
     const body = await req.json();
 
@@ -155,14 +153,14 @@ server.registerHandler({
     if(!body.userIds) {
       body.userIds = [body.senderId, body.agentId];
     }
-    console.log('calling onMessage')
+
     try {
       await onMessage(body, runtime);
     } catch (error ){
       console.error('error', error)
       return new Response(error, { status: 500 });
     }
-    console.log('done')
+
     return new Response('ok', { status: 200 });
   }
 });