@lobehub/chat 1.35.9 → 1.35.10

package/src/database/server/models/user.ts

@@ -2,7 +2,7 @@ import { TRPCError } from '@trpc/server';
 import { eq } from 'drizzle-orm';
 import { DeepPartial } from 'utility-types';
 
-import { serverDB } from '@/database/server/core/db';
+import { LobeChatDatabase } from '@/database/type';
 import { KeyVaultsGateKeeper } from '@/server/modules/KeyVaultsEncrypt';
 import { UserGuide, UserPreference } from '@/types/user';
 import { UserKeyVaults, UserSettings } from '@/types/user/settings';
@@ -18,38 +18,16 @@ export class UserNotFoundError extends TRPCError {
 }
 
 export class UserModel {
-  static createUser = async (params: NewUser) => {
-    // if user already exists, skip creation
-    if (params.id) {
-      const user = await serverDB.query.users.findFirst({ where: eq(users.id, params.id) });
-      if (!!user) return;
-    }
-
-    const [user] = await serverDB
-      .insert(users)
-      .values({ ...params })
-      .returning();
-
-    // Create an inbox session for the user
-    const model = new SessionModel(user.id);
-
-    await model.createInbox();
-  };
+  private userId: string;
+  private db: LobeChatDatabase;
 
-  static deleteUser = async (id: string) => {
-    return serverDB.delete(users).where(eq(users.id, id));
-  };
-
-  static findById = async (id: string) => {
-    return serverDB.query.users.findFirst({ where: eq(users.id, id) });
-  };
-
-  static findByEmail = async (email: string) => {
-    return serverDB.query.users.findFirst({ where: eq(users.email, email) });
-  };
+  constructor(db: LobeChatDatabase, userId: string) {
+    this.userId = userId;
+    this.db = db;
+  }
 
-  getUserState = async (id: string) => {
-    const result = await serverDB
+  async getUserState() {
+    const result = await this.db
       .select({
         isOnboarded: users.isOnboarded,
         preference: users.preference,
@@ -63,7 +41,7 @@ export class UserModel {
         settingsTool: userSettings.tool,
       })
       .from(users)
-      .where(eq(users.id, id))
+      .where(eq(users.id, this.userId))
       .leftJoin(userSettings, eq(users.id, userSettings.id));
 
     if (!result || !result[0]) {
@@ -82,7 +60,7 @@ export class UserModel {
         try {
           decryptKeyVaults = JSON.parse(plaintext);
         } catch (e) {
-          console.error(`Failed to parse keyVaults ,userId: ${id}. Error:`, e);
+          console.error(`Failed to parse keyVaults ,userId: ${this.userId}. Error:`, e);
         }
       }
     }
@@ -101,54 +79,22 @@ export class UserModel {
       isOnboarded: state.isOnboarded,
       preference: state.preference as UserPreference,
       settings,
-      userId: id,
+      userId: this.userId,
     };
-  };
-
-  static getUserApiKeys = async (id: string) => {
-    const result = await serverDB
-      .select({
-        settingsKeyVaults: userSettings.keyVaults,
-      })
-      .from(userSettings)
-      .where(eq(userSettings.id, id));
-
-    if (!result || !result[0]) {
-      throw new UserNotFoundError();
-    }
-
-    const state = result[0];
-
-    // Decrypt keyVaults
-    let decryptKeyVaults = {};
-    if (state.settingsKeyVaults) {
-      const gateKeeper = await KeyVaultsGateKeeper.initWithEnvKey();
-      const { wasAuthentic, plaintext } = await gateKeeper.decrypt(state.settingsKeyVaults);
-
-      if (wasAuthentic) {
-        try {
-          decryptKeyVaults = JSON.parse(plaintext);
-        } catch (e) {
-          console.error(`Failed to parse keyVaults ,userId: ${id}. Error:`, e);
-        }
-      }
-    }
-
-    return decryptKeyVaults as UserKeyVaults;
-  };
+  }
 
-  async updateUser(id: string, value: Partial<UserItem>) {
-    return serverDB
+  async updateUser(value: Partial<UserItem>) {
+    return this.db
       .update(users)
       .set({ ...value, updatedAt: new Date() })
-      .where(eq(users.id, id));
+      .where(eq(users.id, this.userId));
   }
 
-  async deleteSetting(id: string) {
-    return serverDB.delete(userSettings).where(eq(userSettings.id, id));
+  async deleteSetting() {
+    return this.db.delete(userSettings).where(eq(userSettings.id, this.userId));
   }
 
-  async updateSetting(id: string, value: Partial<UserSettings>) {
+  async updateSetting(value: Partial<UserSettings>) {
     const { keyVaults, ...res } = value;
 
     // Encrypt keyVaults
@@ -165,33 +111,99 @@ export class UserModel {
     const newValue = { ...res, keyVaults: encryptedKeyVaults };
 
     // update or create user settings
-    const settings = await serverDB.query.userSettings.findFirst({ where: eq(users.id, id) });
+    const settings = await this.db.query.userSettings.findFirst({
+      where: eq(users.id, this.userId),
+    });
     if (!settings) {
-      await serverDB.insert(userSettings).values({ id, ...newValue });
+      await this.db.insert(userSettings).values({ id: this.userId, ...newValue });
       return;
     }
 
-    return serverDB.update(userSettings).set(newValue).where(eq(userSettings.id, id));
+    return this.db.update(userSettings).set(newValue).where(eq(userSettings.id, this.userId));
   }
 
-  async updatePreference(id: string, value: Partial<UserPreference>) {
-    const user = await serverDB.query.users.findFirst({ where: eq(users.id, id) });
+  async updatePreference(value: Partial<UserPreference>) {
+    const user = await this.db.query.users.findFirst({ where: eq(users.id, this.userId) });
     if (!user) return;
 
-    return serverDB
+    return this.db
       .update(users)
       .set({ preference: merge(user.preference, value) })
-      .where(eq(users.id, id));
+      .where(eq(users.id, this.userId));
   }
 
-  async updateGuide(id: string, value: Partial<UserGuide>) {
-    const user = await serverDB.query.users.findFirst({ where: eq(users.id, id) });
+  async updateGuide(value: Partial<UserGuide>) {
+    const user = await this.db.query.users.findFirst({ where: eq(users.id, this.userId) });
     if (!user) return;
 
     const prevPreference = (user.preference || {}) as UserPreference;
-    return serverDB
+    return this.db
       .update(users)
       .set({ preference: { ...prevPreference, guide: merge(prevPreference.guide || {}, value) } })
-      .where(eq(users.id, id));
+      .where(eq(users.id, this.userId));
   }
+
+  // Static method
+
+  static createUser = async (db: LobeChatDatabase, params: NewUser) => {
+    // if user already exists, skip creation
+    if (params.id) {
+      const user = await db.query.users.findFirst({ where: eq(users.id, params.id) });
+      if (!!user) return;
+    }
+
+    const [user] = await db
+      .insert(users)
+      .values({ ...params })
+      .returning();
+
+    // Create an inbox session for the user
+    const model = new SessionModel(db, user.id);
+
+    await model.createInbox();
+  };
+
+  static deleteUser = async (db: LobeChatDatabase, id: string) => {
+    return db.delete(users).where(eq(users.id, id));
+  };
+
+  static findById = async (db: LobeChatDatabase, id: string) => {
+    return db.query.users.findFirst({ where: eq(users.id, id) });
+  };
+
+  static findByEmail = async (db: LobeChatDatabase, email: string) => {
+    return db.query.users.findFirst({ where: eq(users.email, email) });
+  };
+
+  static getUserApiKeys = async (db: LobeChatDatabase, id: string) => {
+    const result = await db
+      .select({
+        settingsKeyVaults: userSettings.keyVaults,
+      })
+      .from(userSettings)
+      .where(eq(userSettings.id, id));
+
+    if (!result || !result[0]) {
+      throw new UserNotFoundError();
+    }
+
+    const state = result[0];
+
+    // Decrypt keyVaults
+    let decryptKeyVaults = {};
+    if (state.settingsKeyVaults) {
+      const gateKeeper = await KeyVaultsGateKeeper.initWithEnvKey();
+      const { wasAuthentic, plaintext } = await gateKeeper.decrypt(state.settingsKeyVaults);
+
+      if (wasAuthentic) {
+        try {
+          decryptKeyVaults = JSON.parse(plaintext);
+        } catch (e) {
+          console.error(`Failed to parse keyVaults ,userId: ${id}. Error:`, e);
+        }
+      }
+    }
+
+    return decryptKeyVaults as UserKeyVaults;
+  };
 }

package/src/database/type.ts

@@ -0,0 +1,7 @@
+import type { NeonDatabase } from 'drizzle-orm/neon-serverless';
+
+import * as schema from './schemas';
+
+export type LobeChatDatabaseSchema = typeof schema;
+
+export type LobeChatDatabase = NeonDatabase<LobeChatDatabaseSchema>;

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

@@ -33,8 +33,6 @@ const {
  * @returns {Adapter}
  */
 export function LobeNextAuthDbAdapter(serverDB: NeonDatabase<typeof schema>): Adapter {
-  const userModel = new UserModel();
-
   return {
     async createAuthenticator(authenticator): Promise<AdapterAuthenticator> {
       const result = await serverDB
@@ -55,10 +53,10 @@ export function LobeNextAuthDbAdapter(serverDB: NeonDatabase<typeof schema>): Ad
     async createUser(user): Promise<AdapterUser> {
       const { id, name, email, emailVerified, image, providerAccountId } = user;
       // return the user if it already exists
-      let existingUser = await UserModel.findByEmail(email);
+      let existingUser = await UserModel.findByEmail(serverDB, email);
       // If the user is not found by email, try to find by providerAccountId
       if (!existingUser && providerAccountId) {
-        existingUser = await UserModel.findById(providerAccountId);
+        existingUser = await UserModel.findById(serverDB, providerAccountId);
       }
       if (existingUser) {
         const adapterUser = mapLobeUserToAdapterUser(existingUser);
@@ -66,6 +64,7 @@ export function LobeNextAuthDbAdapter(serverDB: NeonDatabase<typeof schema>): Ad
       }
       // create a new user if it does not exist
       await UserModel.createUser(
+        serverDB,
         mapAdapterUserToLobeUser({
           email,
           emailVerified,
@@ -91,10 +90,10 @@ export function LobeNextAuthDbAdapter(serverDB: NeonDatabase<typeof schema>): Ad
       return;
     },
     async deleteUser(id): Promise<AdapterUser | null | undefined> {
-      const user = await UserModel.findById(id);
+      const user = await UserModel.findById(serverDB, id);
       if (!user) throw new Error('NextAuth: Delete User not found');
 
-      await UserModel.deleteUser(id);
+      await UserModel.deleteUser(serverDB, id);
       return;
     },
 
@@ -145,7 +144,7 @@ export function LobeNextAuthDbAdapter(serverDB: NeonDatabase<typeof schema>): Ad
     },
 
     async getUser(id): Promise<AdapterUser | null> {
-      const lobeUser = await UserModel.findById(id);
+      const lobeUser = await UserModel.findById(serverDB, id);
       if (!lobeUser) return null;
       return mapLobeUserToAdapterUser(lobeUser);
     },
@@ -170,7 +169,7 @@ export function LobeNextAuthDbAdapter(serverDB: NeonDatabase<typeof schema>): Ad
     },
 
     async getUserByEmail(email): Promise<AdapterUser | null> {
-      const lobeUser = await UserModel.findByEmail(email);
+      const lobeUser = await UserModel.findByEmail(serverDB, email);
       return lobeUser ? mapLobeUserToAdapterUser(lobeUser) : null;
     },
 
@@ -228,10 +227,11 @@ export function LobeNextAuthDbAdapter(serverDB: NeonDatabase<typeof schema>): Ad
     },
 
     async updateUser(user): Promise<AdapterUser> {
-      const lobeUser = await UserModel.findById(user?.id);
+      const lobeUser = await UserModel.findById(serverDB, user?.id);
       if (!lobeUser) throw new Error('NextAuth: User not found');
+      const userModel = new UserModel(serverDB, user.id);
 
-      const updatedUser = await userModel.updateUser(user.id, {
+      const updatedUser = await userModel.updateUser({
         ...partialMapAdapterUserToLobeUser(user),
       });
       if (!updatedUser) throw new Error('NextAuth: Failed to update user');

package/src/libs/trpc/async/asyncAuth.ts

@@ -1,6 +1,7 @@
 import { TRPCError } from '@trpc/server';
 
 import { serverDBEnv } from '@/config/db';
+import { serverDB } from '@/database/server';
 import { UserModel } from '@/database/server/models/user';
 
 import { asyncTrpc } from './init';
@@ -12,7 +13,7 @@ export const asyncAuth = asyncTrpc.middleware(async (opts) => {
     throw new TRPCError({ code: 'UNAUTHORIZED' });
   }
 
-  const result = await UserModel.findById(ctx.userId);
+  const result = await UserModel.findById(serverDB, ctx.userId);
 
   if (!result) {
     throw new TRPCError({ code: 'UNAUTHORIZED', message: 'user is invalid' });

package/src/server/routers/async/file.ts

@@ -5,6 +5,7 @@ import { z } from 'zod';
 
 import { fileEnv } from '@/config/file';
 import { DEFAULT_EMBEDDING_MODEL } from '@/const/settings';
+import { serverDB } from '@/database/server';
 import { ASYNC_TASK_TIMEOUT, AsyncTaskModel } from '@/database/server/models/asyncTask';
 import { ChunkModel } from '@/database/server/models/chunk';
 import { EmbeddingModel } from '@/database/server/models/embedding';
@@ -28,11 +29,11 @@ const fileProcedure = asyncAuthedProcedure.use(async (opts) => {
 
   return opts.next({
     ctx: {
-      asyncTaskModel: new AsyncTaskModel(ctx.userId),
-      chunkModel: new ChunkModel(ctx.userId),
+      asyncTaskModel: new AsyncTaskModel(serverDB, ctx.userId),
+      chunkModel: new ChunkModel(serverDB, ctx.userId),
       chunkService: new ChunkService(ctx.userId),
-      embeddingModel: new EmbeddingModel(ctx.userId),
-      fileModel: new FileModel(ctx.userId),
+      embeddingModel: new EmbeddingModel(serverDB, ctx.userId),
+      fileModel: new FileModel(serverDB, ctx.userId),
     },
   });
 });

package/src/server/routers/async/ragEval.ts

@@ -4,6 +4,7 @@ import { z } from 'zod';
 
 import { chainAnswerWithContext } from '@/chains/answerWithContext';
 import { DEFAULT_EMBEDDING_MODEL, DEFAULT_MODEL } from '@/const/settings';
+import { serverDB } from '@/database/server';
 import { ChunkModel } from '@/database/server/models/chunk';
 import { EmbeddingModel } from '@/database/server/models/embedding';
 import { FileModel } from '@/database/server/models/file';
@@ -24,13 +25,13 @@ const ragEvalProcedure = asyncAuthedProcedure.use(async (opts) => {
 
   return opts.next({
     ctx: {
-      chunkModel: new ChunkModel(ctx.userId),
+      chunkModel: new ChunkModel(serverDB, ctx.userId),
       chunkService: new ChunkService(ctx.userId),
       datasetRecordModel: new EvalDatasetRecordModel(ctx.userId),
-      embeddingModel: new EmbeddingModel(ctx.userId),
+      embeddingModel: new EmbeddingModel(serverDB, ctx.userId),
       evalRecordModel: new EvaluationRecordModel(ctx.userId),
       evaluationModel: new EvalEvaluationModel(ctx.userId),
-      fileModel: new FileModel(ctx.userId),
+      fileModel: new FileModel(serverDB, ctx.userId),
     },
   });
 });

package/src/server/routers/lambda/_template.ts

@@ -1,5 +1,6 @@
 import { z } from 'zod';
 
+import { serverDB } from '@/database/server';
 import { SessionGroupModel } from '@/database/server/models/sessionGroup';
 import { insertSessionGroupSchema } from '@/database/schemas';
 import { authedProcedure, router } from '@/libs/trpc';
@@ -10,7 +11,7 @@ const sessionProcedure = authedProcedure.use(async (opts) => {
 
   return opts.next({
     ctx: {
-      sessionGroupModel: new SessionGroupModel(ctx.userId),
+      sessionGroupModel: new SessionGroupModel(serverDB, ctx.userId),
     },
   });
 });

package/src/server/routers/lambda/agent.ts

@@ -2,6 +2,7 @@ import { z } from 'zod';
 
 import { INBOX_SESSION_ID } from '@/const/session';
 import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
+import { serverDB } from '@/database/server';
 import { AgentModel } from '@/database/server/models/agent';
 import { FileModel } from '@/database/server/models/file';
 import { KnowledgeBaseModel } from '@/database/server/models/knowledgeBase';
@@ -16,10 +17,10 @@ const agentProcedure = authedProcedure.use(async (opts) => {
 
   return opts.next({
     ctx: {
-      agentModel: new AgentModel(ctx.userId),
-      fileModel: new FileModel(ctx.userId),
-      knowledgeBaseModel: new KnowledgeBaseModel(ctx.userId),
-      sessionModel: new SessionModel(ctx.userId),
+      agentModel: new AgentModel(serverDB, ctx.userId),
+      fileModel: new FileModel(serverDB, ctx.userId),
+      knowledgeBaseModel: new KnowledgeBaseModel(serverDB, ctx.userId),
+      sessionModel: new SessionModel(serverDB, ctx.userId),
     },
   });
 });
@@ -87,7 +88,7 @@ export const agentRouter = router({
         // if there is no session for user, create one
         if (!item) {
           // if there is no user, return default config
-          const user = await UserModel.findById(ctx.userId);
+          const user = await UserModel.findById(serverDB, ctx.userId);
           if (!user) return DEFAULT_AGENT_CONFIG;
 
           const res = await ctx.sessionModel.createInbox();

package/src/server/routers/lambda/chunk.ts

@@ -21,12 +21,12 @@ const chunkProcedure = authedProcedure.use(keyVaults).use(async (opts) => {
 
   return opts.next({
     ctx: {
-      asyncTaskModel: new AsyncTaskModel(ctx.userId),
-      chunkModel: new ChunkModel(ctx.userId),
+      asyncTaskModel: new AsyncTaskModel(serverDB, ctx.userId),
+      chunkModel: new ChunkModel(serverDB, ctx.userId),
       chunkService: new ChunkService(ctx.userId),
-      embeddingModel: new EmbeddingModel(ctx.userId),
-      fileModel: new FileModel(ctx.userId),
-      messageModel: new MessageModel(ctx.userId),
+      embeddingModel: new EmbeddingModel(serverDB, ctx.userId),
+      fileModel: new FileModel(serverDB, ctx.userId),
+      messageModel: new MessageModel(serverDB, ctx.userId),
     },
   });
 });

package/src/server/routers/lambda/file.ts

@@ -1,6 +1,7 @@
 import { TRPCError } from '@trpc/server';
 import { z } from 'zod';
 
+import { serverDB } from '@/database/server';
 import { AsyncTaskModel } from '@/database/server/models/asyncTask';
 import { ChunkModel } from '@/database/server/models/chunk';
 import { FileModel } from '@/database/server/models/file';
@@ -15,9 +16,9 @@ const fileProcedure = authedProcedure.use(async (opts) => {
 
   return opts.next({
     ctx: {
-      asyncTaskModel: new AsyncTaskModel(ctx.userId),
-      chunkModel: new ChunkModel(ctx.userId),
-      fileModel: new FileModel(ctx.userId),
+      asyncTaskModel: new AsyncTaskModel(serverDB, ctx.userId),
+      chunkModel: new ChunkModel(serverDB, ctx.userId),
+      fileModel: new FileModel(serverDB, ctx.userId),
     },
   });
 });

package/src/server/routers/lambda/knowledgeBase.ts

@@ -1,5 +1,6 @@
 import { z } from 'zod';
 
+import { serverDB } from '@/database/server';
 import { KnowledgeBaseModel } from '@/database/server/models/knowledgeBase';
 import { insertKnowledgeBasesSchema } from '@/database/schemas';
 import { authedProcedure, router } from '@/libs/trpc';
@@ -10,7 +11,7 @@ const knowledgeBaseProcedure = authedProcedure.use(async (opts) => {
 
   return opts.next({
     ctx: {
-      knowledgeBaseModel: new KnowledgeBaseModel(ctx.userId),
+      knowledgeBaseModel: new KnowledgeBaseModel(serverDB, ctx.userId),
     },
   });
 });

package/src/server/routers/lambda/message.ts

@@ -1,5 +1,6 @@
 import { z } from 'zod';
 
+import { serverDB } from '@/database/server';
 import { MessageModel } from '@/database/server/models/message';
 import { updateMessagePluginSchema } from '@/database/schemas';
 import { authedProcedure, publicProcedure, router } from '@/libs/trpc';
@@ -12,7 +13,7 @@ const messageProcedure = authedProcedure.use(async (opts) => {
   const { ctx } = opts;
 
   return opts.next({
-    ctx: { messageModel: new MessageModel(ctx.userId) },
+    ctx: { messageModel: new MessageModel(serverDB, ctx.userId) },
   });
 });
 
@@ -54,6 +55,7 @@ export const messageRouter = router({
       return ctx.messageModel.queryBySessionId(input.sessionId);
     }),
 
+  // TODO: 未来这部分方法也需要使用 authedProcedure
   getMessages: publicProcedure
     .input(
       z.object({
@@ -66,7 +68,7 @@ export const messageRouter = router({
     .query(async ({ input, ctx }) => {
       if (!ctx.userId) return [];
 
-      const messageModel = new MessageModel(ctx.userId);
+      const messageModel = new MessageModel(serverDB, ctx.userId);
 
       return messageModel.query(input);
     }),

package/src/server/routers/lambda/plugin.ts

@@ -1,5 +1,6 @@
 import { z } from 'zod';
 
+import { serverDB } from '@/database/server';
 import { PluginModel } from '@/database/server/models/plugin';
 import { authedProcedure, publicProcedure, router } from '@/libs/trpc';
 import { LobeTool } from '@/types/tool';
@@ -8,7 +9,7 @@ const pluginProcedure = authedProcedure.use(async (opts) => {
   const { ctx } = opts;
 
   return opts.next({
-    ctx: { pluginModel: new PluginModel(ctx.userId) },
+    ctx: { pluginModel: new PluginModel(serverDB, ctx.userId) },
   });
 });
 
@@ -61,10 +62,11 @@ export const pluginRouter = router({
       return data.identifier;
     }),
 
+  // TODO: 未来这部分方法也需要使用 authedProcedure
   getPlugins: publicProcedure.query(async ({ ctx }): Promise<LobeTool[]> => {
     if (!ctx.userId) return [];
 
-    const pluginModel = new PluginModel(ctx.userId);
+    const pluginModel = new PluginModel(serverDB, ctx.userId);
 
     return pluginModel.query();
   }),

package/src/server/routers/lambda/ragEval.ts

@@ -6,6 +6,7 @@ import pMap from 'p-map';
 import { z } from 'zod';
 
 import { DEFAULT_EMBEDDING_MODEL, DEFAULT_MODEL } from '@/const/settings';
+import { serverDB } from '@/database/server';
 import { FileModel } from '@/database/server/models/file';
 import {
   EvalDatasetModel,
@@ -34,7 +35,7 @@ const ragEvalProcedure = authedProcedure.use(keyVaults).use(async (opts) => {
   return opts.next({
     ctx: {
       datasetModel: new EvalDatasetModel(ctx.userId),
-      fileModel: new FileModel(ctx.userId),
+      fileModel: new FileModel(serverDB, ctx.userId),
       datasetRecordModel: new EvalDatasetRecordModel(ctx.userId),
       evaluationModel: new EvalEvaluationModel(ctx.userId),
       evaluationRecordModel: new EvaluationRecordModel(ctx.userId),

package/src/server/routers/lambda/session.ts

@@ -1,5 +1,6 @@
 import { z } from 'zod';
 
+import { serverDB } from '@/database/server';
 import { SessionModel } from '@/database/server/models/session';
 import { SessionGroupModel } from '@/database/server/models/sessionGroup';
 import { insertAgentSchema, insertSessionSchema } from '@/database/schemas';
@@ -15,8 +16,8 @@ const sessionProcedure = authedProcedure.use(async (opts) => {
 
   return opts.next({
     ctx: {
-      sessionGroupModel: new SessionGroupModel(ctx.userId),
-      sessionModel: new SessionModel(ctx.userId),
+      sessionGroupModel: new SessionGroupModel(serverDB, ctx.userId),
+      sessionModel: new SessionModel(serverDB, ctx.userId),
     },
   });
 });
@@ -84,7 +85,7 @@ export const sessionRouter = router({
         sessions: [],
       };
 
-    const sessionModel = new SessionModel(ctx.userId);
+    const sessionModel = new SessionModel(serverDB, ctx.userId);
 
     return sessionModel.queryWithGroups();
   }),

package/src/server/routers/lambda/sessionGroup.ts

@@ -1,5 +1,6 @@
 import { z } from 'zod';
 
+import { serverDB } from '@/database/server';
 import { SessionGroupModel } from '@/database/server/models/sessionGroup';
 import { insertSessionGroupSchema } from '@/database/schemas';
 import { authedProcedure, router } from '@/libs/trpc';
@@ -10,7 +11,7 @@ const sessionProcedure = authedProcedure.use(async (opts) => {
 
   return opts.next({
     ctx: {
-      sessionGroupModel: new SessionGroupModel(ctx.userId),
+      sessionGroupModel: new SessionGroupModel(serverDB, ctx.userId),
     },
   });
 });

package/src/server/routers/lambda/thread.ts

@@ -1,5 +1,6 @@
 import { z } from 'zod';
 
+import { serverDB } from '@/database/server';
 import { MessageModel } from '@/database/server/models/message';
 import { ThreadModel } from '@/database/server/models/thread';
 import { insertThreadSchema } from '@/database/schemas';
@@ -11,8 +12,8 @@ const threadProcedure = authedProcedure.use(async (opts) => {
 
   return opts.next({
     ctx: {
-      messageModel: new MessageModel(ctx.userId),
-      threadModel: new ThreadModel(ctx.userId),
+      messageModel: new MessageModel(serverDB, ctx.userId),
+      threadModel: new ThreadModel(serverDB, ctx.userId),
     },
   });
 });