@lobehub/chat 1.61.3 → 1.61.5

package/locales/zh-CN/models.json

@@ -1577,6 +1577,9 @@
   "sonar-reasoning": {
     "description": "由 DeepSeek 推理模型提供支持的新 API 产品。"
   },
+  "sonar-reasoning-pro": {
+    "description": "由 DeepSeek 推理模型提供支持的新 API 产品。"
+  },
   "step-1-128k": {
     "description": "平衡性能与成本，适合一般场景。"
   },

package/locales/zh-TW/auth.json

@@ -34,6 +34,15 @@
   "profile": {
     "avatar": "頭像",
     "email": "電子郵件地址",
+    "sso": {
+      "loading": "正在載入已綁定的第三方帳戶",
+      "providers": "連結的帳戶",
+      "unlink": {
+        "description": "解除綁定後，您將無法使用 {{provider}} 帳戶「{{providerAccountId}}」登入。如果您需要重新綁定 {{provider}} 帳戶到當前帳戶，請確保 {{provider}} 帳戶的電子郵件地址為 {{email}}，我們會在登入時為您自動綁定到當前登入帳戶。",
+        "forbidden": "您至少需要保留一個第三方帳戶綁定。",
+        "title": "是否解除綁定該第三方帳戶 {{provider}} ？"
+      }
+    },
     "username": "用戶名"
   },
   "signout": "登出",
@@ -84,4 +93,4 @@
     "security": "安全",
     "stats": "數據統計"
   }
-}
+}

package/locales/zh-TW/models.json

@@ -1577,6 +1577,9 @@
   "sonar-reasoning": {
     "description": "由 DeepSeek 推理模型提供支持的新 API 產品。"
   },
+  "sonar-reasoning-pro": {
+    "description": "由 DeepSeek 推理模型提供支援的新 API 產品。"
+  },
   "step-1-128k": {
     "description": "平衡性能與成本，適合一般場景。"
   },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.61.3",
+  "version": "1.61.5",
   "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/app/[variants]/(main)/profile/(home)/Client.tsx

@@ -11,6 +11,8 @@ import UserAvatar from '@/features/User/UserAvatar';
 import { useUserStore } from '@/store/user';
 import { authSelectors, userProfileSelectors } from '@/store/user/selectors';
 
+import SSOProvidersList from './features/SSOProvidersList';
+
 type SettingItemGroup = ItemGroup;
 
 const Client = memo<{ mobile?: boolean }>(() => {
@@ -42,6 +44,13 @@ const Client = memo<{ mobile?: boolean }>(() => {
         label: t('profile.email'),
         minWidth: undefined,
       },
+      {
+        children: <SSOProvidersList />,
+        hidden: !isLoginWithNextAuth,
+        label: t('profile.sso.providers'),
+        layout: 'vertical',
+        minWidth: undefined,
+      },
     ],
     title: t('tab.profile'),
   };

package/src/app/[variants]/(main)/profile/(home)/features/SSOProvidersList/AuthIcons.tsx

@@ -0,0 +1,37 @@
+import {
+  Auth0,
+  Authelia,
+  Authentik,
+  Casdoor,
+  Cloudflare,
+  Github,
+  Logto,
+  MicrosoftEntra,
+  NextAuth,
+  Zitadel,
+} from '@lobehub/ui/icons';
+import React from 'react';
+
+const iconProps = {
+  size: 32,
+};
+
+const iconComponents: { [key: string]: React.ElementType } = {
+  'auth0': Auth0,
+  'authelia': Authelia.Color,
+  'authentik': Authentik.Color,
+  'casdoor': Casdoor.Color,
+  'cloudflare': Cloudflare.Color,
+  'default': NextAuth.Color,
+  'github': Github,
+  'logto': Logto.Color,
+  'microsoft-entra-id': MicrosoftEntra.Color,
+  'zitadel': Zitadel.Color,
+};
+
+const AuthIcons = (id: string) => {
+  const IconComponent = iconComponents[id] || iconComponents.default;
+  return <IconComponent {...iconProps} />;
+};
+
+export default AuthIcons;

package/src/app/[variants]/(main)/profile/(home)/features/SSOProvidersList/index.tsx

@@ -0,0 +1,93 @@
+import { ActionIcon, CopyButton, List } from '@lobehub/ui';
+import { RotateCw, Unlink } from 'lucide-react';
+import { CSSProperties, memo, useState } from 'react';
+import { useTranslation } from 'react-i18next';
+import { Flexbox } from 'react-layout-kit';
+
+import { modal, notification } from '@/components/AntdStaticMethods';
+import { useOnlyFetchOnceSWR } from '@/libs/swr';
+import { userService } from '@/services/user';
+import { useUserStore } from '@/store/user';
+import { userProfileSelectors } from '@/store/user/selectors';
+
+import AuthIcons from './AuthIcons';
+
+const { Item } = List;
+
+const providerNameStyle: CSSProperties = {
+  textTransform: 'capitalize',
+};
+
+export const SSOProvidersList = memo(() => {
+  const [userProfile] = useUserStore((s) => [userProfileSelectors.userProfile(s)]);
+  const { t } = useTranslation('auth');
+
+  const [allowUnlink, setAllowUnlink] = useState<boolean>(false);
+  const [hoveredIndex, setHoveredIndex] = useState<number | null>(null);
+
+  const { data, isLoading, mutate } = useOnlyFetchOnceSWR('profile-sso-providers', async () => {
+    const list = await userService.getUserSSOProviders();
+    setAllowUnlink(list?.length > 1);
+    return list;
+  });
+
+  const handleUnlinkSSO = async (provider: string, providerAccountId: string) => {
+    if (data?.length === 1 || !data) {
+      // At least one SSO provider should be linked
+      notification.error({
+        message: t('profile.sso.unlink.forbidden'),
+      });
+      return;
+    }
+    modal.confirm({
+      content: t('profile.sso.unlink.description', {
+        email: userProfile?.email || 'None',
+        provider,
+        providerAccountId,
+      }),
+      okButtonProps: {
+        danger: true,
+      },
+      onOk: async () => {
+        await userService.unlinkSSOProvider(provider, providerAccountId);
+        mutate();
+      },
+      title: <span style={providerNameStyle}>{t('profile.sso.unlink.title', { provider })}</span>,
+    });
+  };
+
+  return isLoading ? (
+    <Flexbox align={'center'} gap={4} horizontal>
+      <ActionIcon icon={RotateCw} spin />
+      {t('profile.sso.loading')}
+    </Flexbox>
+  ) : (
+    <Flexbox>
+      {data?.map((item, index) => (
+        <Item
+          actions={
+            <Flexbox gap={4} horizontal>
+              <CopyButton content={item.providerAccountId} size={'small'} />
+              <ActionIcon
+                disable={!allowUnlink}
+                icon={Unlink}
+                onClick={() => handleUnlinkSSO(item.provider, item.providerAccountId)}
+                size={'small'}
+              />
+            </Flexbox>
+          }
+          avatar={AuthIcons(item.provider)}
+          date={item.expires_at}
+          description={item.providerAccountId}
+          key={index}
+          onMouseEnter={() => setHoveredIndex(index)}
+          onMouseLeave={() => setHoveredIndex(null)}
+          showAction={hoveredIndex === index}
+          title={<span style={providerNameStyle}>{item.provider}</span>}
+        />
+      ))}
+    </Flexbox>
+  );
+});
+
+export default SSOProvidersList;

package/src/config/aiModels/perplexity.ts

@@ -6,17 +6,28 @@ const perplexityChatModels: AIChatModelCard[] = [
       reasoning: true,
     },
     contextWindowTokens: 127_072,
-    description:
-      '由 DeepSeek 推理模型提供支持的新 API 产品。',
+    description: '由 DeepSeek 推理模型提供支持的新 API 产品。',
+    displayName: 'Sonar Reasoning Pro',
+    enabled: true,
+    id: 'sonar-reasoning-pro',
+    maxOutput: 8192,
+    type: 'chat',
+  },
+  {
+    abilities: {
+      reasoning: true,
+    },
+    contextWindowTokens: 127_072,
+    description: '由 DeepSeek 推理模型提供支持的新 API 产品。',
     displayName: 'Sonar Reasoning',
     enabled: true,
     id: 'sonar-reasoning',
+    maxOutput: 8192,
     type: 'chat',
   },
   {
     contextWindowTokens: 200_000,
-    description:
-      '支持搜索上下文的高级搜索产品，支持高级查询和跟进。',
+    description: '支持搜索上下文的高级搜索产品，支持高级查询和跟进。',
     displayName: 'Sonar Pro',
     enabled: true,
     id: 'sonar-pro',
@@ -24,8 +35,7 @@ const perplexityChatModels: AIChatModelCard[] = [
   },
   {
     contextWindowTokens: 127_072,
-    description:
-      '基于搜索上下文的轻量级搜索产品，比 Sonar Pro 更快、更便宜。',
+    description: '基于搜索上下文的轻量级搜索产品，比 Sonar Pro 更快、更便宜。',
     displayName: 'Sonar',
     enabled: true,
     id: 'sonar',

package/src/config/modelProviders/perplexity.ts

@@ -5,24 +5,21 @@ const Perplexity: ModelProviderCard = {
   chatModels: [
     {
       contextWindowTokens: 127_072,
-      description:
-        '由 DeepSeek 推理模型提供支持的新 API 产品。',
+      description: '由 DeepSeek 推理模型提供支持的新 API 产品。',
       displayName: 'Sonar Reasoning',
       enabled: true,
       id: 'sonar-reasoning',
     },
     {
       contextWindowTokens: 200_000,
-      description:
-        '支持搜索上下文的高级搜索产品，支持高级查询和跟进。',
+      description: '支持搜索上下文的高级搜索产品，支持高级查询和跟进。',
       displayName: 'Sonar Pro',
       enabled: true,
       id: 'sonar-pro',
     },
     {
       contextWindowTokens: 127_072,
-      description:
-        '基于搜索上下文的轻量级搜索产品，比 Sonar Pro 更快、更便宜。',
+      description: '基于搜索上下文的轻量级搜索产品，比 Sonar Pro 更快、更便宜。',
       displayName: 'Sonar',
       enabled: true,
       id: 'sonar',
@@ -60,10 +57,16 @@ const Perplexity: ModelProviderCard = {
     placeholder: 'https://api.perplexity.ai',
   },
   settings: {
+    // perplexity doesn't support CORS
+    disableBrowserRequest: true,
     proxyUrl: {
       placeholder: 'https://api.perplexity.ai',
     },
     sdkType: 'openai',
+    smoothing: {
+      speed: 2,
+      text: true,
+    },
   },
   url: 'https://www.perplexity.ai',
 };

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

@@ -1,6 +1,7 @@
 import { TRPCError } from '@trpc/server';
 import dayjs from 'dayjs';
 import { eq } from 'drizzle-orm/expressions';
+import type { AdapterAccount } from 'next-auth/adapters';
 import { DeepPartial } from 'utility-types';
 
 import { LobeChatDatabase } from '@/database/type';
@@ -9,7 +10,14 @@ import { UserKeyVaults, UserSettings } from '@/types/user/settings';
 import { merge } from '@/utils/merge';
 import { today } from '@/utils/time';
 
-import { NewUser, UserItem, UserSettingsItem, userSettings, users } from '../../schemas';
+import {
+  NewUser,
+  UserItem,
+  UserSettingsItem,
+  nextauthAccounts,
+  userSettings,
+  users,
+} from '../../schemas';
 
 type DecryptUserKeyVaults = (
   encryptKeyVaultsStr: string | null,
@@ -96,6 +104,21 @@ export class UserModel {
     };
   };
 
+  getUserSSOProviders = async () => {
+    const result = await this.db
+      .select({
+        expiresAt: nextauthAccounts.expires_at,
+        provider: nextauthAccounts.provider,
+        providerAccountId: nextauthAccounts.providerAccountId,
+        scope: nextauthAccounts.scope,
+        type: nextauthAccounts.type,
+        userId: nextauthAccounts.userId,
+      })
+      .from(nextauthAccounts)
+      .where(eq(nextauthAccounts.userId, this.userId));
+    return result as unknown as AdapterAccount[];
+  };
+
   getUserSettings = async () => {
     return this.db.query.userSettings.findFirst({ where: eq(userSettings.id, this.userId) });
   };

package/src/libs/agent-runtime/perplexity/index.test.ts

@@ -1,16 +1,18 @@
 // @vitest-environment node
-import OpenAI from 'openai';
-import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { ChatStreamCallbacks, LobeOpenAICompatibleRuntime } from '@/libs/agent-runtime';
+import { LobeOpenAICompatibleRuntime, ModelProvider } from '@/libs/agent-runtime';
+import { testProvider } from '@/libs/agent-runtime/providerTestUtils';
 
-import * as debugStreamModule from '../utils/debugStream';
 import { LobePerplexityAI } from './index';
 
-const provider = 'perplexity';
-const defaultBaseURL = 'https://api.perplexity.ai';
-const bizErrorType = 'ProviderBizError';
-const invalidErrorType = 'InvalidProviderAPIKey';
+testProvider({
+  Runtime: LobePerplexityAI,
+  provider: ModelProvider.Perplexity,
+  defaultBaseURL: 'https://api.perplexity.ai',
+  chatDebugEnv: 'DEBUG_PERPLEXITY_CHAT_COMPLETION',
+  chatModel: 'sonar',
+});
 
 // Mock the console.error to avoid polluting test output
 vi.spyOn(console, 'error').mockImplementation(() => {});
@@ -31,221 +33,7 @@ afterEach(() => {
 });
 
 describe('LobePerplexityAI', () => {
-  describe('init', () => {
-    it('should correctly initialize with an API key', async () => {
-      const instance = new LobePerplexityAI({ apiKey: 'test_api_key' });
-      expect(instance).toBeInstanceOf(LobePerplexityAI);
-      expect(instance.baseURL).toEqual(defaultBaseURL);
-    });
-  });
-
   describe('chat', () => {
-    describe('Error', () => {
-      it('should return OpenAIBizError with an openai error response when OpenAI.APIError is thrown', async () => {
-        // Arrange
-        const apiError = new OpenAI.APIError(
-          400,
-          {
-            status: 400,
-            error: {
-              message: 'Bad Request',
-            },
-          },
-          'Error message',
-          {},
-        );
-
-        vi.spyOn(instance['client'].chat.completions, 'create').mockRejectedValue(apiError);
-
-        // Act
-        try {
-          await instance.chat({
-            messages: [{ content: 'Hello', role: 'user' }],
-            model: 'text-davinci-003',
-            temperature: 0,
-          });
-        } catch (e) {
-          expect(e).toEqual({
-            endpoint: defaultBaseURL,
-            error: {
-              error: { message: 'Bad Request' },
-              status: 400,
-            },
-            errorType: bizErrorType,
-            provider,
-          });
-        }
-      });
-
-      it('should throw AgentRuntimeError with NoOpenAIAPIKey if no apiKey is provided', async () => {
-        try {
-          new LobePerplexityAI({});
-        } catch (e) {
-          expect(e).toEqual({ errorType: invalidErrorType });
-        }
-      });
-
-      it('should return OpenAIBizError with the cause when OpenAI.APIError is thrown with cause', async () => {
-        // Arrange
-        const errorInfo = {
-          stack: 'abc',
-          cause: {
-            message: 'api is undefined',
-          },
-        };
-        const apiError = new OpenAI.APIError(400, errorInfo, 'module error', {});
-
-        vi.spyOn(instance['client'].chat.completions, 'create').mockRejectedValue(apiError);
-
-        // Act
-        try {
-          await instance.chat({
-            messages: [{ content: 'Hello', role: 'user' }],
-            model: 'text-davinci-003',
-            temperature: 0,
-          });
-        } catch (e) {
-          expect(e).toEqual({
-            endpoint: defaultBaseURL,
-            error: {
-              cause: { message: 'api is undefined' },
-              stack: 'abc',
-            },
-            errorType: bizErrorType,
-            provider,
-          });
-        }
-      });
-
-      it('should return OpenAIBizError with an cause response with desensitize Url', async () => {
-        // Arrange
-        const errorInfo = {
-          stack: 'abc',
-          cause: { message: 'api is undefined' },
-        };
-        const apiError = new OpenAI.APIError(400, errorInfo, 'module error', {});
-
-        instance = new LobePerplexityAI({
-          apiKey: 'test',
-
-          baseURL: 'https://api.abc.com/v1',
-        });
-
-        vi.spyOn(instance['client'].chat.completions, 'create').mockRejectedValue(apiError);
-
-        // Act
-        try {
-          await instance.chat({
-            messages: [{ content: 'Hello', role: 'user' }],
-            model: 'gpt-3.5-turbo',
-            temperature: 0,
-          });
-        } catch (e) {
-          expect(e).toEqual({
-            endpoint: 'https://api.***.com/v1',
-            error: {
-              cause: { message: 'api is undefined' },
-              stack: 'abc',
-            },
-            errorType: bizErrorType,
-            provider,
-          });
-        }
-      });
-
-      it('should throw an InvalidMoonshotAPIKey error type on 401 status code', async () => {
-        // Mock the API call to simulate a 401 error
-        const error = new Error('Unauthorized') as any;
-        error.status = 401;
-        vi.mocked(instance['client'].chat.completions.create).mockRejectedValue(error);
-
-        try {
-          await instance.chat({
-            messages: [{ content: 'Hello', role: 'user' }],
-            model: 'gpt-3.5-turbo',
-            temperature: 0,
-          });
-        } catch (e) {
-          // Expect the chat method to throw an error with InvalidMoonshotAPIKey
-          expect(e).toEqual({
-            endpoint: defaultBaseURL,
-            error: new Error('Unauthorized'),
-            errorType: invalidErrorType,
-            provider,
-          });
-        }
-      });
-
-      it('should return AgentRuntimeError for non-OpenAI errors', async () => {
-        // Arrange
-        const genericError = new Error('Generic Error');
-
-        vi.spyOn(instance['client'].chat.completions, 'create').mockRejectedValue(genericError);
-
-        // Act
-        try {
-          await instance.chat({
-            messages: [{ content: 'Hello', role: 'user' }],
-            model: 'text-davinci-003',
-            temperature: 0,
-          });
-        } catch (e) {
-          expect(e).toEqual({
-            endpoint: defaultBaseURL,
-            errorType: 'AgentRuntimeError',
-            provider,
-            error: {
-              name: genericError.name,
-              cause: genericError.cause,
-              message: genericError.message,
-              stack: genericError.stack,
-            },
-          });
-        }
-      });
-    });
-
-    describe('DEBUG', () => {
-      it('should call debugStream and return StreamingTextResponse when DEBUG_PERPLEXITY_CHAT_COMPLETION is 1', async () => {
-        // Arrange
-        const mockProdStream = new ReadableStream() as any; // 模拟的 prod 流
-        const mockDebugStream = new ReadableStream({
-          start(controller) {
-            controller.enqueue('Debug stream content');
-            controller.close();
-          },
-        }) as any;
-        mockDebugStream.toReadableStream = () => mockDebugStream; // 添加 toReadableStream 方法
-
-        // 模拟 chat.completions.create 返回值，包括模拟的 tee 方法
-        (instance['client'].chat.completions.create as Mock).mockResolvedValue({
-          tee: () => [mockProdStream, { toReadableStream: () => mockDebugStream }],
-        });
-
-        // 保存原始环境变量值
-        const originalDebugValue = process.env.DEBUG_PERPLEXITY_CHAT_COMPLETION;
-
-        // 模拟环境变量
-        process.env.DEBUG_PERPLEXITY_CHAT_COMPLETION = '1';
-        vi.spyOn(debugStreamModule, 'debugStream').mockImplementation(() => Promise.resolve());
-
-        // 执行测试
-        // 运行你的测试函数，确保它会在条件满足时调用 debugStream
-        // 假设的测试函数调用，你可能需要根据实际情况调整
-        await instance.chat({
-          messages: [{ content: 'Hello', role: 'user' }],
-          model: 'text-davinci-003',
-          temperature: 0,
-        });
-
-        // 验证 debugStream 被调用
-        expect(debugStreamModule.debugStream).toHaveBeenCalled();
-
-        // 恢复原始环境变量值
-        process.env.DEBUG_PERPLEXITY_CHAT_COMPLETION = originalDebugValue;
-      });
-    });
-
     it('should call chat method with temperature', async () => {
       vi.spyOn(instance['client'].chat.completions, 'create').mockResolvedValue(
         new ReadableStream() as any,

package/src/locales/default/auth.ts

@@ -34,6 +34,16 @@ export default {
   profile: {
     avatar: '头像',
     email: '电子邮件地址',
+    sso: {
+      loading: '正在加载已绑定的第三方账户',
+      providers: '连接的帐户',
+      unlink: {
+        description:
+          '解绑后，您将无法使用 {{provider}} 账户“{{providerAccountId}}”登录。如果您需要重新绑定 {{provider}} 账户到当前账户，请确保 {{provider}} 账户的邮件地址为 {{email}} ，我们会在登陆时为你自动绑定到当前登录账户。',
+        forbidden: '您至少需要保留一个第三方账户绑定。',
+        title: '是否解绑该第三方账户 {{provider}} ？',
+      },
+    },
     username: '用户名',
   },
   signout: '退出登录',

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

@@ -7,15 +7,24 @@ import { serverDB } from '@/database/server';
 import { MessageModel } from '@/database/server/models/message';
 import { SessionModel } from '@/database/server/models/session';
 import { UserModel, UserNotFoundError } from '@/database/server/models/user';
+import { LobeNextAuthDbAdapter } from '@/libs/next-auth/adapter';
 import { authedProcedure, router } from '@/libs/trpc';
 import { KeyVaultsGateKeeper } from '@/server/modules/KeyVaultsEncrypt';
 import { UserService } from '@/server/services/user';
-import { UserGuideSchema, UserInitializationState, UserPreference } from '@/types/user';
+import {
+  NextAuthAccountSchame,
+  UserGuideSchema,
+  UserInitializationState,
+  UserPreference,
+} from '@/types/user';
 import { UserSettings } from '@/types/user/settings';
 
 const userProcedure = authedProcedure.use(async (opts) => {
   return opts.next({
-    ctx: { userModel: new UserModel(serverDB, opts.ctx.userId) },
+    ctx: {
+      nextAuthDbAdapter: LobeNextAuthDbAdapter(serverDB),
+      userModel: new UserModel(serverDB, opts.ctx.userId),
+    },
   });
 });
 
@@ -24,6 +33,10 @@ export const userRouter = router({
     return ctx.userModel.getUserRegistrationDuration();
   }),
 
+  getUserSSOProviders: userProcedure.query(async ({ ctx }) => {
+    return ctx.userModel.getUserSSOProviders();
+  }),
+
   getUserState: userProcedure.query(async ({ ctx }): Promise<UserInitializationState> => {
     let state: Awaited<ReturnType<UserModel['getUserState']>> | undefined;
 
@@ -92,6 +105,23 @@ export const userRouter = router({
     return ctx.userModel.deleteSetting();
   }),
 
+  unlinkSSOProvider: userProcedure.input(NextAuthAccountSchame).mutation(async ({ ctx, input }) => {
+    const { provider, providerAccountId } = input;
+    if (
+      ctx.nextAuthDbAdapter?.unlinkAccount &&
+      typeof ctx.nextAuthDbAdapter.unlinkAccount === 'function' &&
+      ctx.nextAuthDbAdapter?.getAccount &&
+      typeof ctx.nextAuthDbAdapter.getAccount === 'function'
+    ) {
+      const account = await ctx.nextAuthDbAdapter.getAccount(providerAccountId, provider);
+      // The userId can either get from ctx.nextAuth?.id or ctx.userId
+      if (!account || account.userId !== ctx.userId) throw new Error('The account does not exist');
+      await ctx.nextAuthDbAdapter.unlinkAccount({ provider, providerAccountId });
+    } else {
+      throw new Error('The method in LobeNextAuthDbAdapter `unlinkAccount` is not implemented');
+    }
+  }),
+
   updateGuide: userProcedure.input(UserGuideSchema).mutation(async ({ ctx, input }) => {
     return ctx.userModel.updateGuide(input);
   }),