@lobehub/chat 1.71.5 → 1.72.0

package/src/services/session/client.test.ts

@@ -32,7 +32,9 @@ beforeEach(async () => {
     await trx.insert(users).values([{ id: userId }, { id: '456' }]);
     await trx.insert(sessions).values([{ id: mockSessionId, userId }]);
     await trx.insert(agents).values([{ id: mockAgentId, userId }]);
-    await trx.insert(agentsToSessions).values([{ agentId: mockAgentId, sessionId: mockSessionId }]);
+    await trx
+      .insert(agentsToSessions)
+      .values([{ agentId: mockAgentId, sessionId: mockSessionId, userId }]);
     await trx.insert(sessionGroups).values([
       { id: 'group-1', name: 'group-A', sort: 2, userId },
       { id: 'group-2', name: 'group-B', sort: 1, userId },
@@ -176,7 +178,7 @@ describe('SessionService', () => {
       await clientDB.insert(agents).values({ userId, id: 'agent-1', title: 'Session Name' });
       await clientDB
         .insert(agentsToSessions)
-        .values({ agentId: 'agent-1', sessionId: mockSessionId });
+        .values({ agentId: 'agent-1', sessionId: mockSessionId, userId });
 
       // Execute
       const keyword = 'Name';
@@ -201,7 +203,7 @@ describe('SessionService', () => {
       await clientDB.insert(agents).values({ userId, id: 'agent-1' });
       await clientDB
         .insert(agentsToSessions)
-        .values({ agentId: 'agent-1', sessionId: 'duplicated-session-id' });
+        .values({ agentId: 'agent-1', sessionId: 'duplicated-session-id', userId });
 
       // Execute
       const duplicatedSessionId = await sessionService.cloneSession(mockSessionId, newTitle);

package/src/types/message/base.ts

@@ -117,3 +117,10 @@ export interface UpdateMessageParams {
   toolCalls?: MessageToolCall[];
   tools?: ChatToolPayload[] | null;
 }
+
+export interface NewMessageQueryParams {
+  embeddingsId: string;
+  messageId: string;
+  rewriteQuery: string;
+  userQuery: string;
+}

package/vitest.server.config.ts

@@ -15,7 +15,7 @@ export default defineConfig({
       reportsDirectory: './coverage/server',
     },
     environment: 'node',
-    include: ['src/database/server/**/**/*.test.ts'],
+    include: ['src/database/models/**/**/*.test.ts', 'src/database/server/**/**/*.test.ts'],
     poolOptions: {
       threads: { singleThread: true },
     },

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

@@ -1,58 +0,0 @@
-// @vitest-environment node
-import { TRPCError } from '@trpc/server';
-import { describe, expect, it, vi } from 'vitest';
-
-import { UserModel, UserNotFoundError } from '@/database/server/models/user';
-
-describe('UserNotFoundError', () => {
-  it('should extend TRPCError with correct code and message', () => {
-    const error = new UserNotFoundError();
-
-    expect(error).toBeInstanceOf(TRPCError);
-    expect(error.code).toBe('UNAUTHORIZED');
-    expect(error.message).toBe('user not found');
-  });
-});
-
-describe('UserModel', () => {
-  const mockDb = {
-    query: {
-      users: {
-        findFirst: vi.fn(),
-      },
-    },
-  };
-
-  const mockUserId = 'test-user-id';
-  const userModel = new UserModel(mockDb as any, mockUserId);
-
-  describe('getUserRegistrationDuration', () => {
-    it('should return default values when user not found', async () => {
-      mockDb.query.users.findFirst.mockResolvedValue(null);
-
-      const result = await userModel.getUserRegistrationDuration();
-
-      expect(result).toEqual({
-        createdAt: expect.any(String),
-        duration: 1,
-        updatedAt: expect.any(String),
-      });
-    });
-
-    it('should calculate duration correctly for existing user', async () => {
-      const createdAt = new Date('2024-01-01');
-      mockDb.query.users.findFirst.mockResolvedValue({
-        createdAt,
-      });
-
-      const result = await userModel.getUserRegistrationDuration();
-
-      expect(result).toEqual({
-        createdAt: '2024-01-01',
-        duration: expect.any(Number),
-        updatedAt: expect.any(String),
-      });
-      expect(result.duration).toBeGreaterThan(0);
-    });
-  });
-});