@lobehub/chat 1.36.31 → 1.36.32

package/CHANGELOG.md

@@ -2,6 +2,31 @@
 
 # Changelog
 
+### [Version 1.36.32](https://github.com/lobehub/lobe-chat/compare/v1.36.31...v1.36.32)
+
+<sup>Released on **2024-12-17**</sup>
+
+#### ♻ Code Refactoring
+
+- **misc**: Refactor the drizzle code style.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Code refactoring
+
+- **misc**: Refactor the drizzle code style, closes [#5058](https://github.com/lobehub/lobe-chat/issues/5058) ([4057ad3](https://github.com/lobehub/lobe-chat/commit/4057ad3))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.36.31](https://github.com/lobehub/lobe-chat/compare/v1.36.30...v1.36.31)
 
 <sup>Released on **2024-12-17**</sup>

package/changelog/v1.json

@@ -1,4 +1,13 @@
 [
+  {
+    "children": {
+      "improvements": [
+        "Refactor the drizzle code style."
+      ]
+    },
+    "date": "2024-12-17",
+    "version": "1.36.32"
+  },
   {
     "children": {
       "improvements": [

package/docs/self-hosting/environment-variables/model-provider.mdx

@@ -288,6 +288,13 @@ If you need to use Azure OpenAI to provide model services, you can refer to the
 - Default: `https://api.Perplexity.ai`
 - Example: `https://my-Perplexity-proxy.com`
 
+### `PERPLEXITY_MODEL_LIST`
+
+- Type: Optional
+- Description: Used to control the model list, use `+` to add a model, use `-` to hide a model, use `model_name=display_name` to customize the display name of a model, separated by commas. Definition syntax rules see
+- Default: `-`
+- Example: `-all,+llama-3.1-sonar-small-128k-online,+llama-3.1-sonar-small-128k-chat`
+
 ## Minimax AI
 
 ### `MINIMAX_API_KEY`

package/docs/self-hosting/environment-variables/model-provider.zh-CN.mdx

@@ -279,6 +279,13 @@ LobeChat 在部署时提供了丰富的模型服务商相关的环境变量，
 - 默认值：-
 - 示例：`pplx-xxxxxx...xxxxxx`
 
+### `PERPLEXITY_MODEL_LIST`
+
+- 类型：可选
+- 描述：用于指定自定义 Perplexity 语言模型。模型定义语法规则见 [模型列表][model-list]
+- 默认值：-
+- 示例： `-all,+llama-3.1-sonar-small-128k-online,+llama-3.1-sonar-small-128k-chat`
+
 ### `PERPLEXITY_PROXY_URL`
 
 - 类型：可选

package/docs/self-hosting/server-database/dokploy.zh-CN.mdx

@@ -1,15 +1,15 @@
----
-title: 在 Dokploy 上部署 LobeChat 的服务端数据库版本
-description: 本文详细介绍如何在 Dokploy 中部署服务端数据库版 LobeChat，包括数据库配置、身份验证服务配置的设置步骤。
-tags:
-  - 服务端数据库
-  - Postgres
-  - Clerk
-  - Dokploy部署
-  - 数据库配置
-  - 身份验证服务
-  - 环境变量配置
----
+---
+title: 在 Dokploy 上部署 LobeChat 的服务端数据库版本
+description: 本文详细介绍如何在 Dokploy 中部署服务端数据库版 LobeChat，包括数据库配置、身份验证服务配置的设置步骤。
+tags:
+  - 服务端数据库
+  - Postgres
+  - Clerk
+  - Dokploy部署
+  - 数据库配置
+  - 身份验证服务
+  - 环境变量配置
+---
 
 # 在 Dokploy 上部署服务端数据库版
 本文将详细介绍如何在 Dokploy 中部署服务端数据库版 LobeChat。

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.36.31",
+  "version": "1.36.32",
   "description": "Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",

package/src/database/repositories/dataImporter/__tests__/index.test.ts

@@ -1,8 +1,7 @@
 // @vitest-environment node
-import { eq, inArray } from 'drizzle-orm';
+import { eq, inArray } from 'drizzle-orm/expressions';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { getTestDBInstance } from '@/database/server/core/dbForTest';
 import {
   agents,
   agentsToSessions,
@@ -12,6 +11,7 @@ import {
   topics,
   users,
 } from '@/database/schemas';
+import { getTestDBInstance } from '@/database/server/core/dbForTest';
 import { CURRENT_CONFIG_VERSION } from '@/migrations';
 import { ImporterEntryData } from '@/types/importer';
 
@@ -60,8 +60,7 @@ describe('DataImporter', () => {
     it('should skip existing session groups and return correct result', async () => {
       await serverDB
         .insert(sessionGroups)
-        .values({ clientId: 'group1', name: 'Existing Group', userId })
-        .execute();
+        .values({ clientId: 'group1', name: 'Existing Group', userId });
 
       const data: ImporterEntryData = {
         version: CURRENT_CONFIG_VERSION,
@@ -141,7 +140,7 @@ describe('DataImporter', () => {
     });
 
     it('should skip existing sessions and return correct result', async () => {
-      await serverDB.insert(sessions).values({ clientId: 'session1', userId }).execute();
+      await serverDB.insert(sessions).values({ clientId: 'session1', userId });
 
       const data: ImporterEntryData = {
         version: CURRENT_CONFIG_VERSION,
@@ -477,10 +476,7 @@ describe('DataImporter', () => {
     });
 
     it('should skip existing topics and return correct result', async () => {
-      await serverDB
-        .insert(topics)
-        .values({ clientId: 'topic1', title: 'Existing Topic', userId })
-        .execute();
+      await serverDB.insert(topics).values({ clientId: 'topic1', title: 'Existing Topic', userId });
 
       const data: ImporterEntryData = {
         version: CURRENT_CONFIG_VERSION,
@@ -616,15 +612,12 @@ describe('DataImporter', () => {
     });
 
     it('should skip existing messages and return correct result', async () => {
-      await serverDB
-        .insert(messages)
-        .values({
-          clientId: 'msg1',
-          content: 'Existing Message',
-          role: 'user',
-          userId,
-        })
-        .execute();
+      await serverDB.insert(messages).values({
+        clientId: 'msg1',
+        content: 'Existing Message',
+        role: 'user',
+        userId,
+      });
 
       const data: ImporterEntryData = {
         version: CURRENT_CONFIG_VERSION,

package/src/database/repositories/dataImporter/index.ts

@@ -1,5 +1,5 @@
-import { eq, inArray, sql } from 'drizzle-orm';
-import { and } from 'drizzle-orm/expressions';
+import { sql } from 'drizzle-orm';
+import { and, eq, inArray } from 'drizzle-orm/expressions';
 
 import {
   agents,
@@ -71,8 +71,7 @@ export class DataImporterRepos {
             set: { updatedAt: new Date() },
             target: [sessionGroups.clientId, sessionGroups.userId],
           })
-          .returning({ clientId: sessionGroups.clientId, id: sessionGroups.id })
-          .execute();
+          .returning({ clientId: sessionGroups.clientId, id: sessionGroups.id });
 
         sessionGroupResult.added = mapArray.length - query.length;
 
@@ -109,8 +108,7 @@ export class DataImporterRepos {
             set: { updatedAt: new Date() },
             target: [sessions.clientId, sessions.userId],
           })
-          .returning({ clientId: sessions.clientId, id: sessions.id })
-          .execute();
+          .returning({ clientId: sessions.clientId, id: sessions.id });
 
         // get the session client-server id map
         sessionIdMap = Object.fromEntries(mapArray.map(({ clientId, id }) => [clientId, id]));
@@ -133,18 +131,14 @@ export class DataImporterRepos {
                 userId: this.userId,
               })),
             )
-            .returning({ id: agents.id })
-            .execute();
+            .returning({ id: agents.id });
 
-          await trx
-            .insert(agentsToSessions)
-            .values(
-              shouldInsertSessionAgents.map(({ id }, index) => ({
-                agentId: agentMapArray[index].id,
-                sessionId: sessionIdMap[id],
-              })),
-            )
-            .execute();
+          await trx.insert(agentsToSessions).values(
+            shouldInsertSessionAgents.map(({ id }, index) => ({
+              agentId: agentMapArray[index].id,
+              sessionId: sessionIdMap[id],
+            })),
+          );
         }
       }
 
@@ -178,8 +172,7 @@ export class DataImporterRepos {
             set: { updatedAt: new Date() },
             target: [topics.clientId, topics.userId],
           })
-          .returning({ clientId: topics.clientId, id: topics.id })
-          .execute();
+          .returning({ clientId: topics.clientId, id: topics.id });
 
         topicIdMap = Object.fromEntries(mapArray.map(({ clientId, id }) => [clientId, id]));
 
@@ -230,7 +223,7 @@ export class DataImporterRepos {
 
           for (let i = 0; i < inertValues.length; i += BATCH_SIZE) {
             const batch = inertValues.slice(i, i + BATCH_SIZE);
-            await trx.insert(messages).values(batch).execute();
+            await trx.insert(messages).values(batch);
           }
 
           console.timeEnd('insert messages');
@@ -265,7 +258,7 @@ export class DataImporterRepos {
             .filter(Boolean);
 
           if (parentIdUpdates.length > 0) {
-            const updateQuery = trx
+            await trx
               .update(messages)
               .set({
                 parentId: sql`CASE ${sql.join(parentIdUpdates)} END`,
@@ -281,42 +274,34 @@ export class DataImporterRepos {
             // const SQL = updateQuery.toSQL();
             // console.log('sql:', SQL.sql);
             // console.log('params:', SQL.params);
-
-            await updateQuery.execute();
           }
           console.timeEnd('execute updates parentId');
 
           // 4. insert message plugins
           const pluginInserts = shouldInsertMessages.filter((msg) => msg.plugin);
           if (pluginInserts.length > 0) {
-            await trx
-              .insert(messagePlugins)
-              .values(
-                pluginInserts.map((msg) => ({
-                  apiName: msg.plugin?.apiName,
-                  arguments: msg.plugin?.arguments,
-                  id: messageIdMap[msg.id],
-                  identifier: msg.plugin?.identifier,
-                  state: msg.pluginState,
-                  toolCallId: msg.tool_call_id,
-                  type: msg.plugin?.type,
-                })),
-              )
-              .execute();
+            await trx.insert(messagePlugins).values(
+              pluginInserts.map((msg) => ({
+                apiName: msg.plugin?.apiName,
+                arguments: msg.plugin?.arguments,
+                id: messageIdMap[msg.id],
+                identifier: msg.plugin?.identifier,
+                state: msg.pluginState,
+                toolCallId: msg.tool_call_id,
+                type: msg.plugin?.type,
+              })),
+            );
           }
 
           // 5. insert message translate
           const translateInserts = shouldInsertMessages.filter((msg) => msg.extra?.translate);
           if (translateInserts.length > 0) {
-            await trx
-              .insert(messageTranslates)
-              .values(
-                translateInserts.map((msg) => ({
-                  id: messageIdMap[msg.id],
-                  ...msg.extra?.translate,
-                })),
-              )
-              .execute();
+            await trx.insert(messageTranslates).values(
+              translateInserts.map((msg) => ({
+                id: messageIdMap[msg.id],
+                ...msg.extra?.translate,
+              })),
+            );
           }
 
           // TODO: 未来需要处理 TTS 和图片的插入 （目前存在 file 的部分，不方便处理）

package/src/database/server/models/__tests__/_test_template.ts

@@ -1,5 +1,5 @@
 // @vitest-environment node
-import { eq } from 'drizzle-orm';
+import { eq } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
 import { getTestDBInstance } from '@/database/server/core/dbForTest';

package/src/database/server/models/__tests__/agent.test.ts

@@ -1,5 +1,5 @@
 // @vitest-environment node
-import { eq } from 'drizzle-orm';
+import { eq } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
 import { getTestDBInstance } from '@/database/server/core/dbForTest';

package/src/database/server/models/__tests__/asyncTask.test.ts

@@ -1,5 +1,5 @@
 // @vitest-environment node
-import { eq } from 'drizzle-orm';
+import { eq } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { getTestDBInstance } from '@/database/server/core/dbForTest';

package/src/database/server/models/__tests__/chunk.test.ts

@@ -1,5 +1,5 @@
 // @vitest-environment node
-import { eq } from 'drizzle-orm';
+import { eq } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
 import { getTestDBInstance } from '@/database/server/core/dbForTest';

package/src/database/server/models/__tests__/file.test.ts

@@ -1,5 +1,5 @@
 // @vitest-environment node
-import { eq, inArray } from 'drizzle-orm';
+import { eq, inArray } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { getTestDBInstance } from '@/database/server/core/dbForTest';

package/src/database/server/models/__tests__/knowledgeBase.test.ts

@@ -1,6 +1,5 @@
 // @vitest-environment node
-import { eq } from 'drizzle-orm';
-import { and, desc } from 'drizzle-orm/expressions';
+import { and, eq } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it } from 'vitest';
 
 import { getTestDBInstance } from '@/database/server/core/dbForTest';

package/src/database/server/models/__tests__/message.test.ts

@@ -1,4 +1,4 @@
-import { eq } from 'drizzle-orm';
+import { eq } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { getTestDBInstance } from '@/database/server/core/dbForTest';
@@ -514,11 +514,7 @@ describe('MessageModel', () => {
       await messageModel.create({ role: 'user', content: 'new message', sessionId: '1' });
 
       // 断言结果
-      const result = await serverDB
-        .select()
-        .from(messages)
-        .where(eq(messages.userId, userId))
-        .execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.userId, userId));
       expect(result).toHaveLength(1);
       expect(result[0].content).toBe('new message');
     });
@@ -549,11 +545,7 @@ describe('MessageModel', () => {
       });
 
       // 断言结果
-      const result = await serverDB
-        .select()
-        .from(messages)
-        .where(eq(messages.userId, userId))
-        .execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.userId, userId));
       expect(result[0].id).toBeDefined();
       expect(result[0].id).toHaveLength(18);
     });
@@ -582,8 +574,7 @@ describe('MessageModel', () => {
       const pluginResult = await serverDB
         .select()
         .from(messagePlugins)
-        .where(eq(messagePlugins.id, result.id))
-        .execute();
+        .where(eq(messagePlugins.id, result.id));
       expect(pluginResult).toHaveLength(1);
       expect(pluginResult[0].identifier).toBe('plugin1');
     });
@@ -650,8 +641,7 @@ describe('MessageModel', () => {
       const pluginResult = await serverDB
         .select()
         .from(messagePlugins)
-        .where(eq(messagePlugins.id, result.id))
-        .execute();
+        .where(eq(messagePlugins.id, result.id));
       expect(pluginResult).toHaveLength(1);
       expect(pluginResult[0].identifier).toBe('lobe-web-browsing');
       expect(pluginResult[0].state!).toMatchObject(state);
@@ -670,11 +660,7 @@ describe('MessageModel', () => {
       await messageModel.batchCreate(newMessages);
 
       // 断言结果
-      const result = await serverDB
-        .select()
-        .from(messages)
-        .where(eq(messages.userId, userId))
-        .execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.userId, userId));
       expect(result).toHaveLength(2);
       expect(result[0].content).toBe('message 1');
       expect(result[1].content).toBe('message 2');
@@ -692,7 +678,7 @@ describe('MessageModel', () => {
       await messageModel.update('1', { content: 'updated message' });
 
       // 断言结果
-      const result = await serverDB.select().from(messages).where(eq(messages.id, '1')).execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.id, '1'));
       expect(result[0].content).toBe('updated message');
     });
 
@@ -706,7 +692,7 @@ describe('MessageModel', () => {
       await messageModel.update('1', { content: 'updated message' });
 
       // 断言结果
-      const result = await serverDB.select().from(messages).where(eq(messages.id, '1')).execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.id, '1'));
       expect(result[0].content).toBe('message 1');
     });
 
@@ -745,7 +731,7 @@ describe('MessageModel', () => {
       });
 
       // 断言结果
-      const result = await serverDB.select().from(messages).where(eq(messages.id, '1')).execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.id, '1'));
       expect(result[0].tools[0].arguments).toBe(
         '{"query":"2024 杭州暴雨","searchEngines":["duckduckgo","google","brave"]}',
       );
@@ -763,7 +749,7 @@ describe('MessageModel', () => {
       await messageModel.deleteMessage('1');
 
       // 断言结果
-      const result = await serverDB.select().from(messages).where(eq(messages.id, '1')).execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.id, '1'));
       expect(result).toHaveLength(0);
     });
 
@@ -783,14 +769,13 @@ describe('MessageModel', () => {
       await messageModel.deleteMessage('1');
 
       // 断言结果
-      const result = await serverDB.select().from(messages).where(eq(messages.id, '1')).execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.id, '1'));
       expect(result).toHaveLength(0);
 
       const result2 = await serverDB
         .select()
         .from(messagePlugins)
-        .where(eq(messagePlugins.id, '2'))
-        .execute();
+        .where(eq(messagePlugins.id, '2'));
 
       expect(result2).toHaveLength(0);
     });
@@ -805,7 +790,7 @@ describe('MessageModel', () => {
       await messageModel.deleteMessage('1');
 
       // 断言结果
-      const result = await serverDB.select().from(messages).where(eq(messages.id, '1')).execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.id, '1'));
       expect(result).toHaveLength(1);
     });
   });
@@ -822,9 +807,9 @@ describe('MessageModel', () => {
       await messageModel.deleteMessages(['1', '2']);
 
       // 断言结果
-      const result = await serverDB.select().from(messages).where(eq(messages.id, '1')).execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.id, '1'));
       expect(result).toHaveLength(0);
-      const result2 = await serverDB.select().from(messages).where(eq(messages.id, '2')).execute();
+      const result2 = await serverDB.select().from(messages).where(eq(messages.id, '2'));
       expect(result2).toHaveLength(0);
     });
 
@@ -839,7 +824,7 @@ describe('MessageModel', () => {
       await messageModel.deleteMessages(['1', '2']);
 
       // 断言结果
-      const result = await serverDB.select().from(messages).where(eq(messages.id, '1')).execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.id, '1'));
       expect(result).toHaveLength(1);
     });
   });
@@ -857,18 +842,12 @@ describe('MessageModel', () => {
       await messageModel.deleteAllMessages();
 
       // 断言结果
-      const result = await serverDB
-        .select()
-        .from(messages)
-        .where(eq(messages.userId, userId))
-        .execute();
+      const result = await serverDB.select().from(messages).where(eq(messages.userId, userId));
+
       expect(result).toHaveLength(0);
 
-      const otherResult = await serverDB
-        .select()
-        .from(messages)
-        .where(eq(messages.userId, '456'))
-        .execute();
+      const otherResult = await serverDB.select().from(messages).where(eq(messages.userId, '456'));
+
       expect(otherResult).toHaveLength(1);
     });
   });
@@ -887,11 +866,8 @@ describe('MessageModel', () => {
       await messageModel.updatePluginState('1', { key2: 'value2' });
 
       // 断言结果
-      const result = await serverDB
-        .select()
-        .from(messagePlugins)
-        .where(eq(messagePlugins.id, '1'))
-        .execute();
+      const result = await serverDB.select().from(messagePlugins).where(eq(messagePlugins.id, '1'));
+
       expect(result[0].state).toEqual({ key1: 'value1', key2: 'value2' });
     });
 
@@ -916,11 +892,8 @@ describe('MessageModel', () => {
       await messageModel.updateMessagePlugin('1', { identifier: 'plugin2' });
 
       // 断言结果
-      const result = await serverDB
-        .select()
-        .from(messagePlugins)
-        .where(eq(messagePlugins.id, '1'))
-        .execute();
+      const result = await serverDB.select().from(messagePlugins).where(eq(messagePlugins.id, '1'));
+
       expect(result[0].identifier).toEqual('plugin2');
     });
 
@@ -950,8 +923,8 @@ describe('MessageModel', () => {
       const result = await serverDB
         .select()
         .from(messageTranslates)
-        .where(eq(messageTranslates.id, '1'))
-        .execute();
+        .where(eq(messageTranslates.id, '1'));
+
       expect(result).toHaveLength(1);
       expect(result[0].content).toBe('translated message 1');
     });
@@ -974,8 +947,8 @@ describe('MessageModel', () => {
       const result = await serverDB
         .select()
         .from(messageTranslates)
-        .where(eq(messageTranslates.id, '1'))
-        .execute();
+        .where(eq(messageTranslates.id, '1'));
+
       expect(result[0].content).toBe('updated translated message 1');
     });
   });
@@ -991,11 +964,8 @@ describe('MessageModel', () => {
       await messageModel.updateTTS('1', { contentMd5: 'md5', file: 'f1', voice: 'voice1' });
 
       // 断言结果
-      const result = await serverDB
-        .select()
-        .from(messageTTS)
-        .where(eq(messageTTS.id, '1'))
-        .execute();
+      const result = await serverDB.select().from(messageTTS).where(eq(messageTTS.id, '1'));
+
       expect(result).toHaveLength(1);
       expect(result[0].voice).toBe('voice1');
     });
@@ -1015,11 +985,8 @@ describe('MessageModel', () => {
       await messageModel.updateTTS('1', { voice: 'updated voice1' });
 
       // 断言结果
-      const result = await serverDB
-        .select()
-        .from(messageTTS)
-        .where(eq(messageTTS.id, '1'))
-        .execute();
+      const result = await serverDB.select().from(messageTTS).where(eq(messageTTS.id, '1'));
+
       expect(result[0].voice).toBe('updated voice1');
     });
   });
@@ -1037,8 +1004,8 @@ describe('MessageModel', () => {
       const result = await serverDB
         .select()
         .from(messageTranslates)
-        .where(eq(messageTranslates.id, '1'))
-        .execute();
+        .where(eq(messageTranslates.id, '1'));
+
       expect(result).toHaveLength(0);
     });
   });
@@ -1053,11 +1020,7 @@ describe('MessageModel', () => {
       await messageModel.deleteMessageTTS('1');
 
       // 断言结果
-      const result = await serverDB
-        .select()
-        .from(messageTTS)
-        .where(eq(messageTTS.id, '1'))
-        .execute();
+      const result = await serverDB.select().from(messageTTS).where(eq(messageTTS.id, '1'));
       expect(result).toHaveLength(0);
     });
   });

package/src/database/server/models/__tests__/nextauth.test.ts

@@ -5,7 +5,7 @@ import type {
   AdapterUser,
   VerificationToken,
 } from '@auth/core/adapters';
-import { eq } from 'drizzle-orm';
+import { eq } from 'drizzle-orm/expressions';
 import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import {

package/src/database/server/models/__tests__/session.test.ts

@@ -1,4 +1,4 @@
-import { and, eq, inArray } from 'drizzle-orm';
+import { and, eq, inArray } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { DEFAULT_AGENT_CONFIG } from '@/const/settings';

package/src/database/server/models/__tests__/sessionGroup.test.ts

@@ -1,6 +1,5 @@
 // @vitest-environment node
-import { eq } from 'drizzle-orm';
-import { desc } from 'drizzle-orm/expressions';
+import { eq } from 'drizzle-orm/expressions';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { getTestDBInstance } from '@/database/server/core/dbForTest';