tests issue

core/jest.config.js

@@ -2,9 +2,9 @@
 export default {
     preset: "ts-jest",
     testEnvironment: "node",
-    rootDir: "./src",
+    rootDir: "./",
     testMatch: ["**/*.test.ts"],
-    setupFilesAfterEnv: ["<rootDir>/test_resources/testSetup.ts"],
+    setupFilesAfterEnv: ["<rootDir>/src/test_resources/testSetup.ts"],
     testTimeout: 120000,
     globals: {
         __DEV__: true,
@@ -23,4 +23,4 @@ export default {
         "^(\\.{1,2}/.*)\\.js$": "$1",
     },
     extensionsToTreatAsEsm: [".ts"],
-}
+};

core/src/providers/trustScoreProvider.ts

@@ -81,6 +81,9 @@ export class TrustScoreProvider {
 
         const recommenderMetrics =
             await this.trustScoreDb.getRecommenderMetrics(recommenderId);
+        if (!recommenderMetrics) {
+            throw new Error("Recommender not found");
+        }
 
         const isRapidDump = await this.isRapidDump(tokenAddress);
         const sustainedGrowth = await this.sustainedGrowth(tokenAddress);
@@ -109,12 +112,13 @@ export class TrustScoreProvider {
                     processedData.tradeData.price_change_24h_percent,
                 volumeChange24h: processedData.tradeData.volume_24h,
                 trade_24h_change:
-                    processedData.tradeData.trade_24h_change_percent,
+                    processedData.tradeData.trade_24h_change_percent || 0,
                 liquidity:
                     processedData.dexScreenerData.pairs[0]?.liquidity.usd || 0,
                 liquidityChange24h: 0,
                 holderChange24h:
-                    processedData.tradeData.unique_wallet_24h_change_percent,
+                    processedData.tradeData.unique_wallet_24h_change_percent ||
+                    0,
                 rugPull: false, // TODO: Implement rug pull detection
                 isScam: false, // TODO: Implement scam detection
                 marketCapChange24h: 0, // TODO: Implement market cap change
@@ -147,6 +151,9 @@ export class TrustScoreProvider {
     ): Promise<void> {
         const recommenderMetrics =
             await this.trustScoreDb.getRecommenderMetrics(recommenderId);
+        if (!recommenderMetrics) {
+            throw new Error("Recommender not found");
+        }
 
         const totalRecommendations =
             recommenderMetrics.totalRecommendations + 1;
@@ -269,6 +276,13 @@ export class TrustScoreProvider {
         const processedData: ProcessedTokenData =
             await this.tokenProvider.getProcessedTokenData();
         console.log(`Fetched processed token data for token: ${tokenAddress}`);
+        if (
+            !processedData ||
+            !processedData.tradeData ||
+            !processedData.tradeData.volume_24h_change_percent
+        ) {
+            return false;
+        }
 
         return processedData.tradeData.volume_24h_change_percent > 50;
     }
@@ -278,6 +292,14 @@ export class TrustScoreProvider {
             await this.tokenProvider.getProcessedTokenData();
         console.log(`Fetched processed token data for token: ${tokenAddress}`);
 
+        if (
+            !processedData ||
+            !processedData.tradeData ||
+            !processedData.tradeData.trade_24h_change_percent
+        ) {
+            return false;
+        }
+
         return processedData.tradeData.trade_24h_change_percent < -50;
     }
 
@@ -384,6 +406,9 @@ export class TrustScoreProvider {
             recommenderId,
             isSimulation
         );
+        if (!trade) {
+            throw new Error("Trade not found");
+        }
         const buyTimeStamp = trade.buy_timeStamp;
         const marketCap =
             processedData.dexScreenerData.pairs[0]?.marketCap || 0;

core/tests/browser.test.ts

@@ -1,6 +1,7 @@
 import dotenv from "dotenv";
-import { createRuntime } from "../test_resources/createRuntime.ts";
-import { BrowserService } from "./browser.ts";
+import { createRuntime } from "../src/test_resources/createRuntime.ts";
+import { BrowserService } from "../src/services/browser.ts";
+import { IAgentRuntime } from "../src/core/types.ts";
 
 dotenv.config();
 
@@ -12,7 +13,10 @@ describe("BrowserService", () => {
             env: process.env as Record<string, string>,
             actions: [],
         });
-        browserService = BrowserService.getInstance(runtime);
+        if (runtime.llamaService === null) {
+            throw new Error("llamaService cannot be null");
+        }
+        browserService = BrowserService.getInstance(runtime as IAgentRuntime);
         await browserService.initialize();
     });
 
@@ -25,7 +29,9 @@ describe("BrowserService", () => {
             env: process.env as Record<string, string>,
             actions: [],
         });
-        const newBrowserService = BrowserService.getInstance(runtime);
+        const newBrowserService = BrowserService.getInstance(
+            runtime as IAgentRuntime
+        );
         await expect(newBrowserService.initialize()).resolves.not.toThrow();
         await expect(newBrowserService.closeBrowser()).resolves.not.toThrow();
     });

core/tests/continue.test.ts

@@ -12,8 +12,8 @@ import { getOrCreateRelationship } from "../src/test_resources/getOrCreateRelati
 import { populateMemories } from "../src/test_resources/populateMemories.ts";
 import { runAiTest } from "../src/test_resources/runAiTest.ts";
 import { type User } from "../src/test_resources/types.ts";
-import action from "../src/actions/continue.ts";
-import ignore from "../src/actions/ignore.ts";
+import { continueAction as action } from "../src/actions/continue.ts";
+import { ignore } from "../src/actions/ignore.ts";
 
 dotenv.config({ path: ".dev.vars" });
 
@@ -56,7 +56,10 @@ describe("User Profile", () => {
             actions: [action, ignore],
         });
         user = setup.session.user;
-        runtime = setup.runtime;
+        if (setup.runtime.llamaService === null) {
+            throw new Error("llamaService cannot be null");
+        }
+        runtime = setup.runtime as IAgentRuntime;
 
         const data = await getOrCreateRelationship({
             runtime,
@@ -86,6 +89,7 @@ describe("User Profile", () => {
                 userId: user.id as UUID,
                 content: { text: "Hello" },
                 roomId: roomId as UUID,
+                agentId: runtime.agentId,
             };
 
             const validate = action.validate!;
@@ -104,6 +108,7 @@ describe("User Profile", () => {
                     action: "CONTINUE",
                 },
                 roomId: roomId as UUID,
+                agentId: runtime.agentId,
             };
 
             const result2 = await validate(runtime, message2);
@@ -121,6 +126,7 @@ describe("User Profile", () => {
                     action: "CONTINUE",
                 },
                 roomId,
+                agentId: runtime.agentId,
             };
 
             const handler = action.handler!;
@@ -145,6 +151,7 @@ describe("User Profile", () => {
                         text: "Write a short story in three parts, using the CONTINUE action for each part.",
                     },
                     roomId: roomId,
+                    agentId: runtime.agentId,
                 };
 
                 const initialMessageCount =
@@ -196,6 +203,7 @@ describe("User Profile", () => {
                     text: "Tell me more about your favorite food.",
                 },
                 roomId: roomId as UUID,
+                agentId: runtime.agentId,
             };
 
             const initialMessageCount =
@@ -216,6 +224,7 @@ describe("User Profile", () => {
                 userId: user?.id as UUID,
                 content: { text: "Bye" },
                 roomId: roomId as UUID,
+                agentId: runtime.agentId,
             };
 
             const handler = action.handler!;

core/tests/evaluation.test.ts

@@ -25,7 +25,10 @@ describe("Evaluation Process", () => {
             env: process.env as Record<string, string>,
             evaluators: [TEST_EVALUATOR, TEST_EVALUATOR_FAIL],
         });
-        runtime = setup.runtime;
+        if (setup.runtime.llamaService === null) {
+            throw new Error("llamaService cannot be null");
+        }
+        runtime = setup.runtime as IAgentRuntime;
         user = setup.session.user;
 
         const data = await getOrCreateRelationship({
@@ -61,6 +64,7 @@ describe("Evaluation Process", () => {
             userId: user.id as UUID,
             content: { text: "Test message for evaluation" },
             roomId,
+            agentId: runtime.agentId,
         };
 
         const state = await runtime.composeState(message);
@@ -79,6 +83,7 @@ describe("Evaluation Process", () => {
             userId: user.id as UUID,
             content: { text: "Run TEST_EVALUATOR handler" },
             roomId,
+            agentId: runtime.agentId,
         };
 
         const result = await TEST_EVALUATOR.handler(runtime, message);
@@ -93,6 +98,7 @@ describe("Evaluation Process", () => {
                     text: "We are in testing mode. We want to make sure that the test passes by replying with the evaluator TEST_EVALUATOR in the array of evaluators that are returned. Please run the TEST_EVALUATOR",
                 },
                 roomId,
+                agentId: runtime.agentId,
             };
 
             const state = await runtime.composeState(message);
@@ -113,6 +119,7 @@ describe("Evaluation Process", () => {
             userId: user.id as UUID,
             content: { text: "Test message for evaluation" },
             roomId,
+            agentId: runtime.agentId,
         };
 
         const state = await runtime.composeState(message);

core/tests/fact.test.ts

@@ -33,7 +33,10 @@ describe("Facts Evaluator", () => {
             actions: defaultActions,
         });
         user = setup.session.user;
-        runtime = setup.runtime;
+        if (setup.runtime.llamaService === null) {
+            throw new Error("llamaService cannot be null");
+        }
+        runtime = setup.runtime as IAgentRuntime;
 
         if (!user.id) {
             throw new Error("User ID is undefined");
@@ -66,6 +69,7 @@ describe("Facts Evaluator", () => {
                 userId: user.id as UUID,
                 content: { text: "" },
                 roomId,
+                agentId: runtime.agentId,
             };
 
             const result = await evaluator.handler(runtime, message);
@@ -91,6 +95,7 @@ describe("Facts Evaluator", () => {
                     userId: user.id as UUID,
                     content: { text: "" },
                     roomId,
+                    agentId: runtime.agentId,
                 };
 
                 const result = await evaluator.handler(runtime, message);
@@ -124,6 +129,7 @@ async function addFacts(
             content: { text: fact },
             roomId: roomId,
             embedding: existingEmbedding,
+            agentId: runtime.agentId,
         });
         await runtime.factManager.createMemory(bakedMemory);
         if (!existingEmbedding) {

core/tests/goal.test.ts

@@ -32,7 +32,10 @@ describe("Goals Evaluator", () => {
             actions: defaultActions,
         });
         user = setup.session.user;
-        runtime = setup.runtime;
+        if (setup.runtime.llamaService === null) {
+            throw new Error("llamaService cannot be null");
+        }
+        runtime = setup.runtime as IAgentRuntime;
 
         const data = await getOrCreateRelationship({
             runtime,
@@ -108,6 +111,7 @@ describe("Goals Evaluator", () => {
                         text: "I've completed task 1 and task 2 for the Test Goal. Both are finished. Everything is done and I'm ready for the next goal.",
                     },
                     roomId,
+                    agentId: runtime.agentId,
                 };
 
                 // Process the message with the goal evaluator
@@ -177,6 +181,7 @@ describe("Goals Evaluator", () => {
                     userId: user.id as UUID,
                     content: { text: "I've decided to mark Goal Y as failed." },
                     roomId,
+                    agentId: runtime.agentId,
                 };
 
                 await evaluator.handler(runtime, message, {} as State, {

core/tests/goals.test.ts

@@ -18,7 +18,11 @@ describe("Goals", () => {
         const result = await createRuntime({
             env: process.env as Record<string, string>,
         });
-        runtime = result.runtime;
+
+        if (result.runtime.llamaService === null) {
+            throw new Error("llamaService cannot be null");
+        }
+        runtime = result.runtime as IAgentRuntime;
         user = result.session.user;
         await runtime.databaseAdapter.removeAllGoals(zeroUuid);
     });

core/tests/ignore.test.ts

@@ -21,7 +21,7 @@ import { populateMemories } from "../src/test_resources/populateMemories.ts";
 import { runAiTest } from "../src/test_resources/runAiTest.ts";
 import { messageHandlerTemplate } from "../src/test_resources/templates.ts";
 import { type User } from "../src/test_resources/types.ts";
-import action from "../src/actions/ignore.ts";
+import { ignore as action } from "../src/actions/ignore.ts";
 import { generateMessageResponse } from "../src/core/generation.ts";
 
 async function handleMessage(
@@ -42,6 +42,7 @@ async function handleMessage(
                 },
                 roomId,
                 embedding: embeddingZeroVector,
+                agentId: runtime.agentId,
             });
         }
     };
@@ -76,6 +77,7 @@ async function handleMessage(
         content: response,
         roomId,
         embedding: embeddingZeroVector,
+        agentId: runtime.agentId,
     };
 
     if (responseMessage.content.text?.trim()) {
@@ -107,7 +109,10 @@ describe("Ignore action tests", () => {
             actions: [action],
         });
         user = setup.session.user;
-        runtime = setup.runtime;
+        if (setup.runtime.llamaService === null) {
+            throw new Error("llamaService cannot be null");
+        }
+        runtime = setup.runtime as IAgentRuntime;
 
         const data = await getOrCreateRelationship({
             runtime,
@@ -139,6 +144,7 @@ describe("Ignore action tests", () => {
                 userId: user?.id as UUID,
                 content: { text: "Never talk to me again" },
                 roomId: roomId as UUID,
+                agentId: runtime.agentId,
             };
 
             await populateMemories(runtime, user, roomId, [
@@ -159,6 +165,7 @@ describe("Ignore action tests", () => {
                     userId: user.id as UUID,
                     content: { text: "", action: "IGNORE" },
                     roomId: roomId as UUID,
+                    agentId: runtime.agentId,
                 };
 
                 await populateMemories(runtime, user, roomId, [
@@ -184,6 +191,7 @@ describe("Ignore action tests", () => {
                     userId: user.id as UUID,
                     content: { text: "", action: "IGNORE" },
                     roomId: roomId as UUID,
+                    agentId: runtime.agentId,
                 };
 
                 await populateMemories(runtime, user, roomId, [
@@ -207,6 +215,7 @@ describe("Ignore action tests", () => {
                 userId: user.id as UUID,
                 content: { text: "Bye" },
                 roomId: roomId as UUID,
+                agentId: runtime.agentId,
             };
 
             await populateMemories(runtime, user, roomId, [Goodbye1]);