@lobehub/chat 1.36.8 → 1.36.10

package/src/libs/agent-runtime/openai/__snapshots__/index.test.ts.snap

@@ -20,14 +20,7 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "id": "gpt-3.5-turbo-16k-0613",
   },
   {
-    "description": "最新的 GPT-4 Turbo 模型具备视觉功能。现在，视觉请求可以使用 JSON 模式和函数调用。 GPT-4 Turbo 是一个增强版本，为多模态任务提供成本效益高的支持。它在准确性和效率之间找到平衡，适合需要进行实时交互的应用程序场景。",
     "id": "gpt-4-1106-vision-preview",
-    "pricing": {
-      "input": 10,
-      "output": 30,
-    },
-    "tokens": 128000,
-    "vision": true,
   },
   {
     "id": "gpt-3.5-turbo-instruct-0914",
@@ -93,13 +86,10 @@ exports[`LobeOpenAI > models > should get models 1`] = `
     "tokens": 128000,
   },
   {
-    "description": "最新的 GPT-4 Turbo 模型具备视觉功能。现在，视觉请求可以使用 JSON 模式和函数调用。 GPT-4 Turbo 是一个增强版本，为多模态任务提供成本效益高的支持。它在准确性和效率之间找到平衡，适合需要进行实时交互的应用程序场景。",
-    "displayName": "GPT-4 Turbo Vision Preview",
+    "deploymentName": "gpt-4-vision",
+    "description": "GPT-4 视觉预览版，专为图像分析和处理任务设计。",
+    "displayName": "GPT 4 Turbo with Vision Preview",
     "id": "gpt-4-vision-preview",
-    "pricing": {
-      "input": 10,
-      "output": 30,
-    },
     "tokens": 128000,
     "vision": true,
   },

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

@@ -53,7 +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 = email.trim() ? await UserModel.findByEmail(serverDB, email) : undefined;
+      let existingUser =
+        email && typeof email === 'string' && email.trim()
+          ? await UserModel.findByEmail(serverDB, email)
+          : undefined;
       // If the user is not found by email, try to find by providerAccountId
       if (!existingUser && providerAccountId) {
         existingUser = await UserModel.findById(serverDB, providerAccountId);
@@ -169,7 +172,10 @@ export function LobeNextAuthDbAdapter(serverDB: NeonDatabase<typeof schema>): Ad
     },
 
     async getUserByEmail(email): Promise<AdapterUser | null> {
-      const lobeUser = email.trim() ? await UserModel.findByEmail(serverDB, email) : undefined;
+      const lobeUser =
+        email && typeof email === 'string' && email.trim()
+          ? await UserModel.findByEmail(serverDB, email)
+          : undefined;
       return lobeUser ? mapLobeUserToAdapterUser(lobeUser) : null;
     },
 

package/src/server/routers/edge/config/__snapshots__/index.test.ts.snap

@@ -98,16 +98,9 @@ exports[`configRouter > getGlobalConfig > Model Provider env > OPENAI_MODEL_LIST
     "tokens": 128000,
   },
   {
-    "description": "最新的 GPT-4 Turbo 模型具备视觉功能。现在，视觉请求可以使用 JSON 模式和函数调用。 GPT-4 Turbo 是一个增强版本，为多模态任务提供成本效益高的支持。它在准确性和效率之间找到平衡，适合需要进行实时交互的应用程序场景。",
-    "displayName": "GPT-4 Turbo Vision Preview",
+    "displayName": "gpt-4-vision",
     "enabled": true,
-    "id": "gpt-4-vision-preview",
-    "pricing": {
-      "input": 10,
-      "output": 30,
-    },
-    "tokens": 128000,
-    "vision": true,
+    "id": "gpt-4-vision",
   },
 ]
 `;

package/src/server/routers/edge/config/index.test.ts

@@ -45,7 +45,7 @@ describe('configRouter', () => {
 
         it('should work correct with gpt-4', async () => {
           process.env.OPENAI_MODEL_LIST =
-            '-all,+gpt-3.5-turbo-1106,+gpt-3.5-turbo,+gpt-4,+gpt-4-32k,+gpt-4-1106-preview,+gpt-4-vision-preview';
+            '-all,+gpt-3.5-turbo-1106,+gpt-3.5-turbo,+gpt-4,+gpt-4-32k,+gpt-4-1106-preview,+gpt-4-vision';
 
           const response = await router.getGlobalConfig();
 

package/src/server/services/user/index.test.ts

@@ -0,0 +1,200 @@
+import { UserJSON } from '@clerk/backend';
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+import { UserItem } from '@/database/schemas';
+import { UserModel } from '@/database/server/models/user';
+import { pino } from '@/libs/logger';
+
+import { UserService } from './index';
+
+// Mock dependencies
+vi.mock('@/database/server/models/user', () => {
+  const MockUserModel = vi.fn();
+  // @ts-ignore
+  MockUserModel.findById = vi.fn();
+  // @ts-ignore
+  MockUserModel.createUser = vi.fn();
+  // @ts-ignore
+  MockUserModel.deleteUser = vi.fn();
+
+  // Mock instance methods
+  MockUserModel.prototype.updateUser = vi.fn();
+
+  return { UserModel: MockUserModel };
+});
+
+vi.mock('@/libs/logger', () => ({
+  pino: {
+    info: vi.fn(),
+  },
+}));
+
+let service: UserService;
+const mockUserId = 'test-user-id';
+
+// Mock user data
+const mockUserJSON: UserJSON = {
+  id: mockUserId,
+  email_addresses: [{ id: 'email-1', email_address: 'test@example.com' }],
+  phone_numbers: [{ id: 'phone-1', phone_number: '+1234567890' }],
+  primary_email_address_id: 'email-1',
+  primary_phone_number_id: 'phone-1',
+  image_url: 'https://example.com/avatar.jpg',
+  first_name: 'Test',
+  last_name: 'User',
+  username: 'testuser',
+  created_at: '2023-01-01T00:00:00Z',
+} as unknown as UserJSON;
+
+beforeEach(() => {
+  service = new UserService();
+  vi.clearAllMocks();
+});
+
+describe('UserService', () => {
+  describe('createUser', () => {
+    it('should create a new user when user does not exist', async () => {
+      // Mock user not found
+      vi.mocked(UserModel.findById).mockResolvedValue(null as any);
+
+      await service.createUser(mockUserId, mockUserJSON);
+
+      expect(UserModel.findById).toHaveBeenCalledWith(expect.anything(), mockUserId);
+      expect(UserModel.createUser).toHaveBeenCalledWith(
+        expect.anything(),
+        expect.objectContaining({
+          id: mockUserId,
+          email: 'test@example.com',
+          phone: '+1234567890',
+          firstName: 'Test',
+          lastName: 'User',
+          username: 'testuser',
+          avatar: 'https://example.com/avatar.jpg',
+          clerkCreatedAt: new Date('2023-01-01T00:00:00Z'),
+        }),
+      );
+    });
+
+    it('should not create user if already exists', async () => {
+      // Mock user found
+      vi.mocked(UserModel.findById).mockResolvedValue({ id: mockUserId } as UserItem);
+
+      const result = await service.createUser(mockUserId, mockUserJSON);
+
+      expect(UserModel.findById).toHaveBeenCalledWith(expect.anything(), mockUserId);
+      expect(UserModel.createUser).not.toHaveBeenCalled();
+      expect(result).toEqual({
+        message: 'user not created due to user already existing in the database',
+        success: false,
+      });
+    });
+
+    it('should handle user without primary phone number', async () => {
+      vi.mocked(UserModel.findById).mockResolvedValue(null as any);
+
+      const userWithoutPrimaryPhone = {
+        ...mockUserJSON,
+        primary_phone_number_id: null,
+        phone_numbers: [{ id: 'phone-1', phone_number: '+1234567890' }],
+      } as UserJSON;
+
+      await service.createUser(mockUserId, userWithoutPrimaryPhone);
+
+      expect(UserModel.createUser).toHaveBeenCalledWith(
+        expect.anything(),
+        expect.objectContaining({
+          phone: '+1234567890', // Should use first phone number
+        }),
+      );
+    });
+  });
+
+  describe('deleteUser', () => {
+    it('should delete user', async () => {
+      await service.deleteUser(mockUserId);
+
+      expect(UserModel.deleteUser).toHaveBeenCalledWith(expect.anything(), mockUserId);
+    });
+
+    it('should throw error if deletion fails', async () => {
+      const error = new Error('Deletion failed');
+      vi.mocked(UserModel.deleteUser).mockRejectedValue(error);
+
+      await expect(service.deleteUser(mockUserId)).rejects.toThrow('Deletion failed');
+    });
+  });
+
+  describe('updateUser', () => {
+    it('should update user when user exists', async () => {
+      // Mock user found
+      vi.mocked(UserModel.findById).mockResolvedValue({ id: mockUserId } as UserItem);
+      const mockUpdateUser = vi.mocked(UserModel.prototype.updateUser);
+
+      const result = await service.updateUser(mockUserId, mockUserJSON);
+
+      expect(UserModel.findById).toHaveBeenCalledWith(expect.anything(), mockUserId);
+      expect(pino.info).toHaveBeenCalledWith('updating user due to clerk webhook');
+      expect(mockUpdateUser).toHaveBeenCalledWith(
+        expect.objectContaining({
+          id: mockUserId,
+          email: 'test@example.com',
+          phone: '+1234567890',
+          firstName: 'Test',
+          lastName: 'User',
+          username: 'testuser',
+          avatar: 'https://example.com/avatar.jpg',
+        }),
+      );
+      expect(result).toEqual({
+        message: 'user updated',
+        success: true,
+      });
+    });
+
+    it('should not update user when user does not exist', async () => {
+      // Mock user not found
+      vi.mocked(UserModel.findById).mockResolvedValue(null as any);
+      const mockUpdateUser = vi.mocked(UserModel.prototype.updateUser);
+
+      const result = await service.updateUser(mockUserId, mockUserJSON);
+
+      expect(UserModel.findById).toHaveBeenCalledWith(expect.anything(), mockUserId);
+      expect(mockUpdateUser).not.toHaveBeenCalled();
+      expect(result).toEqual({
+        message: "user not updated due to the user don't existing in the database",
+        success: false,
+      });
+    });
+
+    it('should handle user without primary email and phone', async () => {
+      vi.mocked(UserModel.findById).mockResolvedValue({ id: mockUserId } as UserItem);
+      const mockUpdateUser = vi.mocked(UserModel.prototype.updateUser);
+
+      const userWithoutPrimaryContacts = {
+        ...mockUserJSON,
+        primary_email_address_id: null,
+        primary_phone_number_id: null,
+        email_addresses: [{ id: 'email-1', email_address: 'test@example.com' }],
+        phone_numbers: [{ id: 'phone-1', phone_number: '+1234567890' }],
+      } as UserJSON;
+
+      await service.updateUser(mockUserId, userWithoutPrimaryContacts);
+
+      // Verify that the first email and phone are used when primary is not specified
+      expect(mockUpdateUser).toHaveBeenCalledWith(
+        expect.objectContaining({
+          phone: '+1234567890',
+        }),
+      );
+    });
+
+    it('should handle update failure', async () => {
+      vi.mocked(UserModel.findById).mockResolvedValue({ id: mockUserId } as UserItem);
+      const mockUpdateUser = vi.mocked(UserModel.prototype.updateUser);
+      const error = new Error('Update failed');
+      mockUpdateUser.mockRejectedValue(error);
+
+      await expect(service.updateUser(mockUserId, mockUserJSON)).rejects.toThrow('Update failed');
+    });
+  });
+});

package/src/server/services/user/index.ts

@@ -1,5 +1,4 @@
 import { UserJSON } from '@clerk/backend';
-import { NextResponse } from 'next/server';
 
 import { serverDB } from '@/database/server';
 import { UserModel } from '@/database/server/models/user';
@@ -12,16 +11,18 @@ export class UserService {
 
     // If user already exists, skip creating a new user
     if (res)
-      return NextResponse.json(
-        {
-          message: 'user not created due to user already existing in the database',
-          success: false,
-        },
-        { status: 200 },
-      );
+      return {
+        message: 'user not created due to user already existing in the database',
+        success: false,
+      };
 
     const email = params.email_addresses.find((e) => e.id === params.primary_email_address_id);
-    const phone = params.phone_numbers.find((e) => e.id === params.primary_phone_number_id);
+
+    const phone = params.phone_numbers.find((e, index) => {
+      if (!!params.primary_phone_number_id) return e.id === params.primary_phone_number_id;
+
+      return index === 0;
+    });
 
     /* ↓ cloud slot ↓ */
 
@@ -43,25 +44,14 @@ export class UserService {
 
     /* ↑ cloud slot ↑ */
 
-    return NextResponse.json({ message: 'user created', success: true }, { status: 200 });
+    return { message: 'user created', success: true };
   };
 
-  deleteUser = async (id?: string) => {
-    if (id) {
-      pino.info('delete user due to clerk webhook');
-
-      await UserModel.deleteUser(serverDB, id);
-
-      return NextResponse.json({ message: 'user deleted' }, { status: 200 });
-    } else {
-      pino.warn('clerk sent a delete user request, but no user ID was included in the payload');
-      return NextResponse.json({ message: 'ok' }, { status: 200 });
-    }
+  deleteUser = async (id: string) => {
+    await UserModel.deleteUser(serverDB, id);
   };
 
   updateUser = async (id: string, params: UserJSON) => {
-    pino.info('updating user due to clerk webhook');
-
     const userModel = new UserModel(serverDB, id);
 
     // Check if user already exists
@@ -69,16 +59,18 @@ export class UserService {
 
     // If user not exists, skip update the user
     if (!res)
-      return NextResponse.json(
-        {
-          message: "user not updated due to the user don't existing in the database",
-          success: false,
-        },
-        { status: 200 },
-      );
+      return {
+        message: "user not updated due to the user don't existing in the database",
+        success: false,
+      };
+
+    pino.info('updating user due to clerk webhook');
 
     const email = params.email_addresses.find((e) => e.id === params.primary_email_address_id);
-    const phone = params.phone_numbers.find((e) => e.id === params.primary_phone_number_id);
+    const phone = params.phone_numbers.find((e, index) => {
+      if (params.primary_phone_number_id) return e.id === params.primary_phone_number_id;
+      return index === 0;
+    });
 
     await userModel.updateUser({
       avatar: params.image_url,
@@ -90,6 +82,6 @@ export class UserService {
       username: params.username,
     });
 
-    return NextResponse.json({ message: 'user updated', success: true }, { status: 200 });
+    return { message: 'user updated', success: true };
   };
 }

package/src/store/chat/slices/aiChat/actions/generateAIChat.ts

@@ -411,16 +411,6 @@ export const generateAIChat: StateCreator<
       ? agentConfig.params.max_tokens
       : undefined;
 
-    // 5. handle config for the vision model
-    // Due to the gpt-4-vision-preview model's default max_tokens is very small
-    // we need to set the max_tokens a larger one.
-    if (agentConfig.model === 'gpt-4-vision-preview') {
-      /* eslint-disable unicorn/no-lonely-if */
-      if (!agentConfig.params.max_tokens)
-        // refs: https://github.com/lobehub/lobe-chat/issues/837
-        agentConfig.params.max_tokens = 2048;
-    }
-
     let isFunctionCall = false;
     let msgTraceId: string | undefined;
     let output = '';

package/vitest.config.ts

@@ -31,7 +31,7 @@ export default defineConfig({
       '**/dist/**',
       '**/build/**',
       'src/database/server/**/**',
-      'src/server/services/!(discover)/**/**',
+      'src/server/services/dataImporter/**/**',
     ],
     globals: true,
     server: {