fix tests

Author: arvinxx

src/libs/next-auth/adapter/index.ts

@@ -81,7 +81,7 @@ export function LobeNextAuthDbAdapter(serverDB: NeonDatabase<typeof schema>): Ad
       );
 
       // 3. Create an inbox session for the user
-      const agentService = new AgentService(id);
+      const agentService = new AgentService(serverDB, id);
       await agentService.createInbox();
 
       return { ...user, id: providerAccountId ?? id };

src/server/routers/lambda/agent.ts

@@ -19,7 +19,7 @@ const agentProcedure = authedProcedure.use(async (opts) => {
   return opts.next({
     ctx: {
       agentModel: new AgentModel(serverDB, ctx.userId),
-      agentService: new AgentService(ctx.userId),
+      agentService: new AgentService(serverDB, ctx.userId),
       fileModel: new FileModel(serverDB, ctx.userId),
       knowledgeBaseModel: new KnowledgeBaseModel(serverDB, ctx.userId),
       sessionModel: new SessionModel(serverDB, ctx.userId),

src/server/services/agent/index.ts

@@ -1,17 +1,19 @@
 import { appEnv } from '@/config/app';
-import { serverDB } from '@/database/server';
 import { SessionModel } from '@/database/server/models/session';
+import { LobeChatDatabase } from '@/database/type';
 import { parseAgentConfig } from '@/server/globalConfig/parseDefaultAgent';
 
 export class AgentService {
   private readonly userId: string;
+  private readonly db: LobeChatDatabase;
 
-  constructor(userId: string) {
+  constructor(db: LobeChatDatabase, userId: string) {
     this.userId = userId;
+    this.db = db;
   }
 
   async createInbox() {
-    const sessionModel = new SessionModel(serverDB, this.userId);
+    const sessionModel = new SessionModel(this.db, this.userId);
 
     const defaultAgentConfig = parseAgentConfig(appEnv.DEFAULT_AGENT_CONFIG) || {};
 

src/server/services/user/index.ts

@@ -43,7 +43,7 @@ export class UserService {
     });
 
     // 3. Create an inbox session for the user
-    const agentService = new AgentService(id);
+    const agentService = new AgentService(serverDB, id);
     await agentService.createInbox();
 
     /* ↓ cloud slot ↓ */