replace supabase injection with runtime

src/agents/cj/actions/__tests__/introduce.test.ts

@@ -28,7 +28,7 @@ describe("Introduce Action", () => {
     );
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user?.id as UUID,
       userB: zeroUuid,
     });

src/agents/cj/actions/introduce.ts

@@ -85,7 +85,7 @@ const handler = async (runtime: BgentRuntime, message: Message) => {
 
   if (responseData?.userA && responseData.userB) {
     await createRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: responseData.userA,
       userB: responseData.userB,
     });

src/agents/cj/evaluators/__tests__/details.test.ts

@@ -27,7 +27,7 @@ describe("User Details", () => {
     );
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user?.id as UUID,
       userB: zeroUuid,
     });

src/agents/cj/evaluators/__tests__/profile.test.ts

@@ -61,7 +61,7 @@ describe("User Profile", () => {
     );
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user?.id as UUID,
       userB: zeroUuid,
     });

src/agents/cj/evaluators/profile.ts

@@ -137,7 +137,7 @@ const handler = async (runtime: BgentRuntime, message: Message) => {
 
     // find the room_id in 'relationships' where user_a is the agent and user_b is the user, OR vice versa
     const relationshipRecord = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA,
       userB,
     });

src/lib/__tests__/actions.test.ts

@@ -4,7 +4,6 @@ import dotenv from "dotenv";
 import { createRuntime } from "../../test/createRuntime";
 import { getRelationship } from "../relationships";
 import { type BgentRuntime } from "../runtime";
-import { populateMemories } from "../../test/populateMemories";
 
 dotenv.config();
 
@@ -13,7 +12,7 @@ const zeroUuid = "00000000-0000-0000-0000-000000000000" as UUID;
 describe("Actions", () => {
   let user: User;
   let runtime: BgentRuntime;
-  let room_id: UUID;
+  // let room_id: UUID;
 
   afterAll(async () => {
     await cleanup();
@@ -24,13 +23,13 @@ describe("Actions", () => {
     user = setup.session.user;
     runtime = setup.runtime;
 
-    const data = await getRelationship({
-      supabase: runtime.supabase,
-      userA: user?.id as UUID,
-      userB: zeroUuid,
-    });
+    // const data = await getRelationship({
+    //   runtime,
+    //   userA: user?.id as UUID,
+    //   userB: zeroUuid,
+    // });
 
-    room_id = data?.room_id;
+    // room_id = data?.room_id;
 
     await cleanup();
   });
@@ -46,9 +45,96 @@ describe("Actions", () => {
     ]);
   }
 
-  // TODO: 1. Test that actions are being loaded into context properly
+  describe("Actions", () => {
+    let user: User;
+    let runtime: BgentRuntime;
+    let room_id: UUID;
+
+    afterAll(async () => {
+      await cleanup();
+    });
+
+    beforeAll(async () => {
+      const setup = await createRuntime();
+      user = setup.session.user;
+      runtime = setup.runtime;
 
-  // TODO: 2. Test that actions are validated properply, for example we know that the continue action is always valid
+      const data = await getRelationship({
+        runtime,
+        userA: user?.id as UUID,
+        userB: zeroUuid,
+      });
+
+      room_id = data?.room_id;
+
+      await cleanup();
+    });
 
-  // TODO 3. Test that action handlers are being called properly
+    async function cleanup() {
+      await runtime.summarizationManager.removeAllMemoriesByUserIds([
+        user?.id as UUID,
+        zeroUuid,
+      ]);
+      await runtime.messageManager.removeAllMemoriesByUserIds([
+        user?.id as UUID,
+        zeroUuid,
+      ]);
+    }
+
+    // Test that actions are being loaded into context properly
+    test("Actions are loaded into context", async () => {
+      const actions = runtime.getActions();
+      expect(actions).toBeDefined();
+      expect(actions.length).toBeGreaterThan(0);
+      // Ensure the CONTINUE action is part of the loaded actions
+      const continueAction = actions.find(
+        (action) => action.name === "CONTINUE",
+      );
+      expect(continueAction).toBeDefined();
+    });
+
+    // Test that actions are validated properly
+    test("Continue action is always valid", async () => {
+      const continueAction = runtime
+        .getActions()
+        .find((action) => action.name === "CONTINUE");
+      expect(continueAction).toBeDefined();
+      if (continueAction && continueAction.validate) {
+        const isValid = await continueAction.validate(runtime, {
+          agentId: zeroUuid,
+          senderId: user.id as UUID,
+          userIds: [user.id as UUID, zeroUuid],
+          content: "Test message",
+          room_id: room_id,
+        });
+        expect(isValid).toBeTruthy();
+      } else {
+        throw new Error(
+          "Continue action or its validation function is undefined",
+        );
+      }
+    });
+
+    // Test that action handlers are being called properly
+    test("Continue action handler is called", async () => {
+      const continueAction = runtime
+        .getActions()
+        .find((action) => action.name === "CONTINUE");
+      expect(continueAction).toBeDefined();
+      if (continueAction && continueAction.handler) {
+        const mockMessage = {
+          agentId: zeroUuid,
+          senderId: user.id as UUID,
+          userIds: [user.id as UUID, zeroUuid],
+          content: "Test message for CONTINUE action",
+          room_id: room_id,
+        };
+        const response = await continueAction.handler(runtime, mockMessage);
+        expect(response).toBeDefined();
+        // Further assertions can be made based on the expected outcome of the CONTINUE action handler
+      } else {
+        throw new Error("Continue action or its handler function is undefined");
+      }
+    }, 20000);
+  });
 });

src/lib/__tests__/memory.test.ts

@@ -21,7 +21,7 @@ describe("Memory", () => {
     user = result.session.user;
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user?.id as UUID,
       userB: zeroUuid,
     });
@@ -208,7 +208,7 @@ describe("Memory - Basic tests", () => {
     user = result.session.user;
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user?.id as UUID,
       userB: zeroUuid,
     });
@@ -298,7 +298,7 @@ describe("Memory - Extended Tests", () => {
     user = result.session.user;
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user.id as UUID,
       userB: zeroUuid,
     });

src/lib/__tests__/messages.test.ts

@@ -18,14 +18,14 @@ describe("Messages Library", () => {
     runtime = setup.runtime;
     user = setup.session.user;
     actors = await getMessageActors({
-      supabase: runtime.supabase,
+      runtime,
       userIds: [user.id as UUID, "00000000-0000-0000-0000-000000000000"],
     });
   });
 
   test("getMessageActors should return actors based on given userIds", async () => {
     const result = await getMessageActors({
-      supabase: runtime.supabase,
+      runtime,
       userIds: [user.id as UUID, "00000000-0000-0000-0000-000000000000"],
     });
     expect(result.length).toBeGreaterThan(0);

src/lib/__tests__/relationships.test.ts

@@ -28,7 +28,7 @@ describe("Relationships Module", () => {
     const userB = zeroUuid;
 
     const relationship = await createRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA,
       userB,
     });
@@ -40,10 +40,10 @@ describe("Relationships Module", () => {
     const userA = user?.id as UUID;
     const userB = zeroUuid;
 
-    await createRelationship({ supabase: runtime.supabase, userA, userB });
+    await createRelationship({ runtime, userA, userB });
 
     const relationship = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA,
       userB,
     });
@@ -56,10 +56,10 @@ describe("Relationships Module", () => {
     const userA = user?.id as UUID;
     const userB = zeroUuid;
 
-    await createRelationship({ supabase: runtime.supabase, userA, userB });
+    await createRelationship({ runtime, userA, userB });
 
     const relationships = await getRelationships({
-      supabase: runtime.supabase,
+      runtime,
       userId: userA,
     });
     expect(relationships).toBeDefined();

src/lib/__tests__/runtime.test.ts

@@ -53,7 +53,7 @@ describe("Agent Runtime", () => {
     user = result.session.user;
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user?.id as UUID,
       userB: zeroUuid,
     });

src/lib/actions/__tests__/continue.test.ts

@@ -49,7 +49,7 @@ describe("User Profile", () => {
     runtime = setup.runtime;
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user.id,
       userB: zeroUuid,
     });

src/lib/actions/__tests__/ignore.test.ts

@@ -32,7 +32,7 @@ describe("Ignore action tests", () => {
     runtime = setup.runtime;
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user?.id as UUID,
       userB: zeroUuid,
     });

src/lib/actions/__tests__/wait.test.ts

@@ -32,7 +32,7 @@ describe("Wait Action Behavior", () => {
     runtime = setup.runtime;
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user?.id as UUID,
       userB: zeroUuid,
     });

src/lib/evaluators/__tests__/summarization.test.ts

@@ -34,7 +34,7 @@ describe("Factual Summarization", () => {
     runtime = setup.runtime;
 
     const data = await getRelationship({
-      supabase: runtime.supabase,
+      runtime,
       userA: user?.id as UUID,
       userB: zeroUuid,
     });

src/lib/evaluators/summarization.ts

@@ -147,8 +147,7 @@ async function handler(runtime: BgentRuntime, message: Message) {
 
   const { userIds, senderId, agentId, room_id } = state;
 
-  const actors =
-    (await getMessageActors({ supabase: runtime.supabase, userIds })) ?? [];
+  const actors = (await getMessageActors({ runtime, userIds })) ?? [];
 
   const senderName = actors?.find(
     (actor: Actor) => actor.id === senderId,