@lobehub/chat 1.53.8 → 1.53.10

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 1.53.10](https://github.com/lobehub/lobe-chat/compare/v1.53.9...v1.53.10)
+
+<sup>Released on **2025-02-13**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix api key input issue.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Fix api key input issue, closes [#6112](https://github.com/lobehub/lobe-chat/issues/6112) ([48e3b85](https://github.com/lobehub/lobe-chat/commit/48e3b85))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 1.53.9](https://github.com/lobehub/lobe-chat/compare/v1.53.8...v1.53.9)
+
+<sup>Released on **2025-02-13**</sup>
+
+#### 💄 Styles
+
+- **misc**: Support select check models.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Styles
+
+- **misc**: Support select check models, closes [#6106](https://github.com/lobehub/lobe-chat/issues/6106) ([2243bbb](https://github.com/lobehub/lobe-chat/commit/2243bbb))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.53.8](https://github.com/lobehub/lobe-chat/compare/v1.53.7...v1.53.8)
 
 <sup>Released on **2025-02-13**</sup>

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "fixes": [
+        "Fix api key input issue."
+      ]
+    },
+    "date": "2025-02-13",
+    "version": "1.53.10"
+  },
+  {
+    "children": {
+      "improvements": [
+        "Support select check models."
+      ]
+    },
+    "date": "2025-02-13",
+    "version": "1.53.9"
+  },
   {
     "children": {
       "fixes": [

package/docs/self-hosting/advanced/model-list.mdx

@@ -17,7 +17,7 @@ LobeChat supports customizing the model list during deployment. This configurati
 You can use `+` to add a model, `-` to hide a model, and use `model name=display name<extension configuration>` to customize the display name of a model, separated by English commas. The basic syntax is as follows:
 
 ```text
-id=displayName<maxToken:vision:fc:file>,model2,model3
+id=displayName<maxToken:vision:reasoning:fc:file>,model2,model3
 ```
 
 For example: `+qwen-7b-chat,+glm-6b,-gpt-3.5-turbo,gpt-4-0125-preview=gpt-4-turbo`
@@ -29,7 +29,7 @@ In the above example, it adds `qwen-7b-chat` and `glm-6b` to the model list, rem
 Considering the diversity of model capabilities, we started to add extension configuration in version `0.147.8`, with the following rules:
 
 ```shell
-id=displayName<maxToken:vision:fc:file>
+id=displayName<maxToken:vision:reasoning:fc:file>
 ```
 
 The first value in angle brackets is designated as the `maxToken` for this model. The second value and beyond are the model's extension capabilities, separated by colons `:`, and the order is not important.
@@ -39,12 +39,14 @@ Examples are as follows:
 - `chatglm-6b=ChatGLM 6B<4096>`: ChatGLM 6B, maximum context of 4k, no advanced capabilities;
 - `spark-v3.5=讯飞星火 v3.5<8192:fc>`: Xunfei Spark 3.5 model, maximum context of 8k, supports Function Call;
 - `gemini-1.5-flash-latest=Gemini 1.5 Flash<16000:vision>`: Google Vision model, maximum context of 16k, supports image recognition;
+- `o3-mini=OpenAI o3-mini<200000:reasoning:fc>`: OpenAI o3-mini model, maximum context of 200k, supports reasoning and Function Call;
 - `gpt-4-all=ChatGPT Plus<128000:fc:vision:file>`, hacked version of ChatGPT Plus web, context of 128k, supports image recognition, Function Call, file upload.
 
 Currently supported extension capabilities are:
 
-| ---      | Description                                              |
-| -------- | -------------------------------------------------------- |
-| `fc`     | Function Calling                                         |
-| `vision` | Image Recognition                                        |
-| `file`   | File Upload (a bit hacky, not recommended for daily use) |
+| ---         | Description                                              |
+| ----------- | -------------------------------------------------------- |
+| `fc`        | Function Calling                                         |
+| `vision`    | Image Recognition                                        |
+| `reasoning` | Support Reasoning                                        |
+| `file`      | File Upload (a bit hacky, not recommended for daily use) |

package/docs/self-hosting/advanced/model-list.zh-CN.mdx

@@ -16,7 +16,7 @@ LobeChat 支持在部署时自定义模型列表，详情请参考 [模型提供
 你可以使用 `+` 增加一个模型，使用 `-` 来隐藏一个模型，使用 `模型名=展示名<扩展配置>` 来自定义模型的展示名，用英文逗号隔开。通过 `<>` 来添加扩展配置。基本语法如下：
 
 ```text
-id=displayName<maxToken:vision:fc:file>,model2,model3
+id=displayName<maxToken:vision:reasoning:fc:file>,model2,model3
 ```
 
 例如： `+qwen-7b-chat,+glm-6b,-gpt-3.5-turbo,gpt-4-0125-preview=gpt-4-turbo`
@@ -28,7 +28,7 @@ id=displayName<maxToken:vision:fc:file>,model2,model3
 考虑到模型的能力多样性，我们在 `0.147.8` 版本开始增加扩展性配置，它的规则如下：
 
 ```shell
-id=displayName<maxToken:vision:fc:file>
+id=displayName<maxToken:vision:reasoning:fc:file>
 ```
 
 尖括号第一个值约定为这个模型的 `maxToken` 。第二个及以后作为模型的扩展能力，能力与能力之间用冒号 `:` 作为分隔符，顺序不重要。
@@ -38,12 +38,14 @@ id=displayName<maxToken:vision:fc:file>
 - `chatglm-6b=ChatGLM 6B<4096>`：ChatGLM 6B，最大上下文 4k，没有高阶能力；
 - `spark-v3.5=讯飞星火 v3.5<8192:fc>`：讯飞星火 3.5 模型，最大上下文 8k，支持 Function Call；
 - `gemini-1.5-flash-latest=Gemini 1.5 Flash<16000:vision>`：Google 视觉模型，最大上下文 16k，支持图像识别；
+- `o3-mini=OpenAI o3-mini<200000:reasoning:fc>`：OpenAI o3-mini 模型，最大上下文 200k，支持推理及 Function Call；
 - `gpt-4-all=ChatGPT Plus<128000:fc:vision:file>`，hack 的 ChatGPT Plus 网页版，上下 128k ，支持图像识别、Function Call、文件上传
 
 目前支持的扩展能力有：
 
-| ---      | 描述                     |
-| -------- | ---------------------- |
-| `fc`     | 函数调用（function calling） |
-| `vision` | 视觉识别                   |
-| `file`   | 文件上传（比较 hack，不建议日常使用）  |
+| ---         | 描述                                  |
+| ----------- | ------------------------------------- |
+| `fc`        | 函数调用（function calling）          |
+| `vision`    | 视觉识别                              |
+| `reasoning` | 支持推理                              |
+| `file`      | 文件上传（比较 hack，不建议日常使用） |

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.53.8",
+  "version": "1.53.10",
   "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)/settings/provider/(detail)/[id]/page.tsx

@@ -45,5 +45,3 @@ const Page = async (props: PagePropsWithId) => {
 };
 
 export default Page;
-
-export const dynamic = 'auto';

package/src/app/[variants]/(main)/settings/provider/(detail)/azure/page.tsx

@@ -77,15 +77,11 @@ const useProviderCard = (): ProviderItem => {
         ) : (
           <AutoComplete
             options={[
+              '2024-10-21',
               '2024-06-01',
-              '2024-02-01',
-              '2024-05-01-preview',
-              '2024-04-01-preview',
-              '2024-03-01-preview',
-              '2024-02-15-preview',
-              '2023-10-01-preview',
-              '2023-06-01-preview',
-              '2023-05-15',
+              '2025-01-01-preview',
+              '2024-09-01-preview',
+              '2024-10-01-preview',
             ].map((i) => ({ label: i, value: i }))}
             placeholder={'20XX-XX-XX'}
           />

package/src/app/[variants]/(main)/settings/provider/features/ProviderConfig/Checker.tsx

@@ -1,18 +1,19 @@
 'use client';
 
 import { CheckCircleFilled } from '@ant-design/icons';
-import { Alert, Highlighter } from '@lobehub/ui';
-import { Button } from 'antd';
+import { ModelIcon } from '@lobehub/icons';
+import { Alert, Highlighter, Icon } from '@lobehub/ui';
+import { Button, Select, Space } from 'antd';
 import { useTheme } from 'antd-style';
+import { Loader2Icon } from 'lucide-react';
 import { ReactNode, memo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
 import { TraceNameMap } from '@/const/trace';
-import { useIsMobile } from '@/hooks/useIsMobile';
 import { useProviderName } from '@/hooks/useProviderName';
 import { chatService } from '@/services/chat';
-import { aiProviderSelectors, useAiInfraStore } from '@/store/aiInfra';
+import { aiModelSelectors, aiProviderSelectors, useAiInfraStore } from '@/store/aiInfra';
 import { ChatMessageError } from '@/types/message';
 
 const Error = memo<{ error: ChatMessageError }>(({ error }) => {
@@ -20,9 +21,8 @@ const Error = memo<{ error: ChatMessageError }>(({ error }) => {
   const providerName = useProviderName(error.body?.provider);
 
   return (
-    <Flexbox gap={8} style={{ maxWidth: '600px', width: '100%' }}>
+    <Flexbox gap={8} style={{ width: '100%' }}>
       <Alert
-        banner
         extra={
           <Flexbox>
             <Highlighter copyButtonSize={'small'} language={'json'} type={'pure'}>
@@ -54,10 +54,15 @@ const Checker = memo<ConnectionCheckerProps>(
   ({ model, provider, checkErrorRender: CheckErrorRender }) => {
     const { t } = useTranslation('setting');
 
-    const disabled = useAiInfraStore(aiProviderSelectors.isProviderConfigUpdating(provider));
+    const isProviderConfigUpdating = useAiInfraStore(
+      aiProviderSelectors.isProviderConfigUpdating(provider),
+    );
+    const totalModels = useAiInfraStore(aiModelSelectors.aiProviderChatModelListIds);
+    const updateAiProviderConfig = useAiInfraStore((s) => s.updateAiProviderConfig);
 
     const [loading, setLoading] = useState(false);
     const [pass, setPass] = useState(false);
+    const [checkModel, setCheckModel] = useState(model);
 
     const theme = useTheme();
     const [error, setError] = useState<ChatMessageError | undefined>();
@@ -71,6 +76,7 @@ const Checker = memo<ConnectionCheckerProps>(
           setPass(false);
           isError = true;
         },
+
         onFinish: async (value) => {
           if (!isError && value) {
             setError(undefined);
@@ -104,7 +110,6 @@ const Checker = memo<ConnectionCheckerProps>(
         },
       });
     };
-    const isMobile = useIsMobile();
 
     const defaultError = error ? <Error error={error as ChatMessageError} /> : null;
 
@@ -115,26 +120,42 @@ const Checker = memo<ConnectionCheckerProps>(
     );
 
     return (
-      <Flexbox align={isMobile ? 'flex-start' : 'flex-end'} gap={8}>
-        <Flexbox
-          align={'center'}
-          direction={isMobile ? 'horizontal-reverse' : 'horizontal'}
-          gap={12}
-        >
-          {pass && (
-            <Flexbox gap={4} horizontal>
-              <CheckCircleFilled
-                style={{
-                  color: theme.colorSuccess,
-                }}
-              />
-              {t('llm.checker.pass')}
-            </Flexbox>
-          )}
-          <Button disabled={disabled} loading={loading} onClick={checkConnection}>
+      <Flexbox gap={8}>
+        <Space.Compact block>
+          <Select
+            listItemHeight={36}
+            onSelect={async (value) => {
+              setCheckModel(value);
+              await updateAiProviderConfig(provider, { checkModel: value });
+            }}
+            optionRender={({ value }) => {
+              return (
+                <Flexbox align={'center'} gap={6} horizontal>
+                  <ModelIcon model={value as string} size={20} />
+                  {value}
+                </Flexbox>
+              );
+            }}
+            options={totalModels.map((id) => ({ label: id, value: id }))}
+            suffixIcon={isProviderConfigUpdating && <Icon icon={Loader2Icon} spin />}
+            value={checkModel}
+            virtual
+          />
+          <Button disabled={isProviderConfigUpdating} loading={loading} onClick={checkConnection}>
             {t('llm.checker.button')}
           </Button>
-        </Flexbox>
+        </Space.Compact>
+
+        {pass && (
+          <Flexbox gap={4} horizontal>
+            <CheckCircleFilled
+              style={{
+                color: theme.colorSuccess,
+              }}
+            />
+            {t('llm.checker.pass')}
+          </Flexbox>
+        )}
         {error && errorContent}
       </Flexbox>
     );

package/src/app/[variants]/(main)/settings/provider/features/ProviderConfig/index.tsx

@@ -275,7 +275,11 @@ const ProviderConfig = memo<ProviderConfigProps>(
             children: isLoading ? (
               <Skeleton.Button active />
             ) : (
-              <Checker checkErrorRender={checkErrorRender} model={checkModel!} provider={id} />
+              <Checker
+                checkErrorRender={checkErrorRender}
+                model={data?.checkModel || checkModel!}
+                provider={id}
+              />
             ),
             desc: t('providerModels.config.checker.desc'),
             label: t('providerModels.config.checker.title'),

package/src/features/Conversation/Error/APIKeyForm/LoadingContext.ts

@@ -0,0 +1,11 @@
+import { createContext } from 'react';
+
+interface LoadingContextValue {
+  loading: boolean;
+  setLoading: (loading: boolean) => void;
+}
+
+export const LoadingContext = createContext<LoadingContextValue>({
+  loading: false,
+  setLoading: () => {},
+});

package/src/features/Conversation/Error/APIKeyForm/ProviderApiKeyForm.tsx

@@ -1,9 +1,11 @@
 import { Icon } from '@lobehub/ui';
-import { Button, Input } from 'antd';
-import { Network } from 'lucide-react';
-import { ReactNode, memo, useState } from 'react';
+import { Button } from 'antd';
+import { Loader2Icon, Network } from 'lucide-react';
+import { ReactNode, memo, useContext, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 
+import { FormInput, FormPassword } from '@/components/FormInput';
+import { LoadingContext } from '@/features/Conversation/Error/APIKeyForm/LoadingContext';
 import { useProviderName } from '@/hooks/useProviderName';
 import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
 import { GlobalLLMProviderKey } from '@/types/user/settings';
@@ -27,6 +29,7 @@ const ProviderApiKeyForm = memo<ProviderApiKeyFormProps>(
     const { apiKey, baseURL, setConfig } = useApiKey(provider);
     const { showOpenAIProxyUrl } = useServerConfigStore(featureFlagsSelectors);
     const providerName = useProviderName(provider);
+    const { loading } = useContext(LoadingContext);
 
     return (
       <FormAction
@@ -34,25 +37,25 @@ const ProviderApiKeyForm = memo<ProviderApiKeyFormProps>(
         description={t(`unlock.apiKey.description`, { name: providerName, ns: 'error' })}
         title={t(`unlock.apiKey.title`, { name: providerName, ns: 'error' })}
       >
-        <Input.Password
+        <FormPassword
           autoComplete={'new-password'}
-          onChange={(e) => {
-            setConfig(provider, { apiKey: e.target.value });
+          onChange={(value) => {
+            setConfig(provider, { apiKey: value });
           }}
           placeholder={apiKeyPlaceholder || 'sk-***********************'}
-          type={'block'}
+          suffix={<div>{loading && <Icon icon={Loader2Icon} spin />}</div>}
           value={apiKey}
         />
 
         {showEndpoint &&
           showOpenAIProxyUrl &&
           (showProxy ? (
-            <Input
-              onChange={(e) => {
-                setConfig(provider, { baseURL: e.target.value });
+            <FormInput
+              onChange={(value) => {
+                setConfig(provider, { baseURL: value });
               }}
               placeholder={'https://api.openai.com/v1'}
-              type={'block'}
+              suffix={<div>{loading && <Icon icon={Loader2Icon} spin />}</div>}
               value={baseURL}
             />
           ) : (

package/src/features/Conversation/Error/APIKeyForm/index.tsx

@@ -1,6 +1,6 @@
 import { ProviderIcon } from '@lobehub/icons';
 import { Button } from 'antd';
-import { memo, useMemo } from 'react';
+import { memo, useMemo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Center, Flexbox } from 'react-layout-kit';
 
@@ -9,6 +9,7 @@ import { useChatStore } from '@/store/chat';
 import { GlobalLLMProviderKey } from '@/types/user/settings';
 
 import BedrockForm from './Bedrock';
+import { LoadingContext } from './LoadingContext';
 import ProviderApiKeyForm from './ProviderApiKeyForm';
 
 interface APIKeyFormProps {
@@ -18,6 +19,7 @@ interface APIKeyFormProps {
 
 const APIKeyForm = memo<APIKeyFormProps>(({ id, provider }) => {
   const { t } = useTranslation('error');
+  const [loading, setLoading] = useState(false);
 
   const [resend, deleteMessage] = useChatStore((s) => [s.regenerateMessage, s.deleteMessage]);
 
@@ -62,38 +64,41 @@ const APIKeyForm = memo<APIKeyFormProps>(({ id, provider }) => {
   }, [provider]);
 
   return (
-    <Center gap={16} style={{ maxWidth: 300 }}>
-      {provider === ModelProvider.Bedrock ? (
-        <BedrockForm />
-      ) : (
-        <ProviderApiKeyForm
-          apiKeyPlaceholder={apiKeyPlaceholder}
-          avatar={<ProviderIcon provider={provider} size={80} type={'avatar'} />}
-          provider={provider as GlobalLLMProviderKey}
-          showEndpoint={provider === ModelProvider.OpenAI}
-        />
-      )}
-      <Flexbox gap={12} width={'100%'}>
-        <Button
-          block
-          onClick={() => {
-            resend(id);
-            deleteMessage(id);
-          }}
-          style={{ marginTop: 8 }}
-          type={'primary'}
-        >
-          {t('unlock.confirm')}
-        </Button>
-        <Button
-          onClick={() => {
-            deleteMessage(id);
-          }}
-        >
-          {t('unlock.closeMessage')}
-        </Button>
-      </Flexbox>
-    </Center>
+    <LoadingContext value={{ loading, setLoading }}>
+      <Center gap={16} style={{ maxWidth: 300 }}>
+        {provider === ModelProvider.Bedrock ? (
+          <BedrockForm />
+        ) : (
+          <ProviderApiKeyForm
+            apiKeyPlaceholder={apiKeyPlaceholder}
+            avatar={<ProviderIcon provider={provider} size={80} type={'avatar'} />}
+            provider={provider as GlobalLLMProviderKey}
+            showEndpoint={provider === ModelProvider.OpenAI}
+          />
+        )}
+        <Flexbox gap={12} width={'100%'}>
+          <Button
+            block
+            disabled={loading}
+            onClick={() => {
+              resend(id);
+              deleteMessage(id);
+            }}
+            style={{ marginTop: 8 }}
+            type={'primary'}
+          >
+            {t('unlock.confirm')}
+          </Button>
+          <Button
+            onClick={() => {
+              deleteMessage(id);
+            }}
+          >
+            {t('unlock.closeMessage')}
+          </Button>
+        </Flexbox>
+      </Center>
+    </LoadingContext>
   );
 });
 

package/src/features/Conversation/Error/APIKeyForm/useApiKey.ts

@@ -1,6 +1,8 @@
 import isEqual from 'fast-deep-equal';
+import { useContext } from 'react';
 
 import { isDeprecatedEdition } from '@/const/version';
+import { LoadingContext } from '@/features/Conversation/Error/APIKeyForm/LoadingContext';
 import { aiProviderSelectors, useAiInfraStore } from '@/store/aiInfra';
 import { useUserStore } from '@/store/user';
 import { keyVaultsConfigSelectors } from '@/store/user/selectors';
@@ -11,7 +13,7 @@ export const useApiKey = (provider: string) => {
     keyVaultsConfigSelectors.getVaultByProvider(provider as any)(s)?.baseURL,
     s.updateKeyVaultConfig,
   ]);
-
+  const { setLoading } = useContext(LoadingContext);
   const updateAiProviderConfig = useAiInfraStore((s) => s.updateAiProviderConfig);
   const data = useAiInfraStore(aiProviderSelectors.providerConfigById(provider), isEqual);
 
@@ -23,12 +25,14 @@ export const useApiKey = (provider: string) => {
     apiKey: data?.keyVaults.apiKey,
     baseURL: data?.keyVaults?.baseURL,
     setConfig: async (id: string, params: Record<string, string>) => {
+      const next = { ...data?.keyVaults, ...params };
+      if (isEqual(data?.keyVaults, next)) return;
+
+      setLoading(true);
       await updateAiProviderConfig(id, {
-        keyVaults: {
-          ...data?.keyVaults,
-          ...params,
-        },
+        keyVaults: { ...data?.keyVaults, ...params },
       });
+      setLoading(false);
     },
   };
 };

package/src/services/aiModel/server.test.ts

@@ -0,0 +1,122 @@
+import { describe, expect, it, vi } from 'vitest';
+
+import { lambdaClient } from '@/libs/trpc/client';
+import { AiProviderModelListItem } from '@/types/aiModel';
+
+import { ServerService } from './server';
+
+vi.mock('@/libs/trpc/client', () => ({
+  lambdaClient: {
+    aiModel: {
+      createAiModel: { mutate: vi.fn() },
+      getAiProviderModelList: { query: vi.fn() },
+      getAiModelById: { query: vi.fn() },
+      toggleModelEnabled: { mutate: vi.fn() },
+      updateAiModel: { mutate: vi.fn() },
+      batchUpdateAiModels: { mutate: vi.fn() },
+      batchToggleAiModels: { mutate: vi.fn() },
+      clearModelsByProvider: { mutate: vi.fn() },
+      clearRemoteModels: { mutate: vi.fn() },
+      updateAiModelOrder: { mutate: vi.fn() },
+      removeAiModel: { mutate: vi.fn() },
+    },
+  },
+}));
+
+describe('ServerService', () => {
+  const service = new ServerService();
+
+  it('should create AI model', async () => {
+    const params = {
+      id: 'test-id',
+      providerId: 'test-provider',
+      displayName: 'Test Model',
+    };
+    await service.createAiModel(params);
+    expect(vi.mocked(lambdaClient.aiModel.createAiModel.mutate)).toHaveBeenCalledWith(params);
+  });
+
+  it('should get AI provider model list', async () => {
+    await service.getAiProviderModelList('123');
+    expect(vi.mocked(lambdaClient.aiModel.getAiProviderModelList.query)).toHaveBeenCalledWith({
+      id: '123',
+    });
+  });
+
+  it('should get AI model by id', async () => {
+    await service.getAiModelById('123');
+    expect(vi.mocked(lambdaClient.aiModel.getAiModelById.query)).toHaveBeenCalledWith({
+      id: '123',
+    });
+  });
+
+  it('should toggle model enabled', async () => {
+    const params = { id: '123', providerId: 'test', enabled: true };
+    await service.toggleModelEnabled(params);
+    expect(vi.mocked(lambdaClient.aiModel.toggleModelEnabled.mutate)).toHaveBeenCalledWith(params);
+  });
+
+  it('should update AI model', async () => {
+    const value = { contextWindowTokens: 4000, displayName: 'Updated Model' };
+    await service.updateAiModel('123', 'openai', value);
+    expect(vi.mocked(lambdaClient.aiModel.updateAiModel.mutate)).toHaveBeenCalledWith({
+      id: '123',
+      providerId: 'openai',
+      value,
+    });
+  });
+
+  it('should batch update AI models', async () => {
+    const models: AiProviderModelListItem[] = [
+      {
+        id: '123',
+        enabled: true,
+        type: 'chat',
+      },
+    ];
+    await service.batchUpdateAiModels('provider1', models);
+    expect(vi.mocked(lambdaClient.aiModel.batchUpdateAiModels.mutate)).toHaveBeenCalledWith({
+      id: 'provider1',
+      models,
+    });
+  });
+
+  it('should batch toggle AI models', async () => {
+    const models = ['123', '456'];
+    await service.batchToggleAiModels('provider1', models, true);
+    expect(vi.mocked(lambdaClient.aiModel.batchToggleAiModels.mutate)).toHaveBeenCalledWith({
+      id: 'provider1',
+      models,
+      enabled: true,
+    });
+  });
+
+  it('should clear models by provider', async () => {
+    await service.clearModelsByProvider('provider1');
+    expect(vi.mocked(lambdaClient.aiModel.clearModelsByProvider.mutate)).toHaveBeenCalledWith({
+      providerId: 'provider1',
+    });
+  });
+
+  it('should clear remote models', async () => {
+    await service.clearRemoteModels('provider1');
+    expect(vi.mocked(lambdaClient.aiModel.clearRemoteModels.mutate)).toHaveBeenCalledWith({
+      providerId: 'provider1',
+    });
+  });
+
+  it('should update AI model order', async () => {
+    const items = [{ id: '123', sort: 1 }];
+    await service.updateAiModelOrder('provider1', items);
+    expect(vi.mocked(lambdaClient.aiModel.updateAiModelOrder.mutate)).toHaveBeenCalledWith({
+      providerId: 'provider1',
+      sortMap: items,
+    });
+  });
+
+  it('should delete AI model', async () => {
+    const params = { id: '123', providerId: 'openai' };
+    await service.deleteAiModel(params);
+    expect(vi.mocked(lambdaClient.aiModel.removeAiModel.mutate)).toHaveBeenCalledWith(params);
+  });
+});

package/src/store/aiInfra/slices/aiModel/selectors.ts

@@ -1,6 +1,8 @@
 import { AIProviderStoreState } from '@/store/aiInfra/initialState';
 import { AiModelSourceEnum } from '@/types/aiModel';
 
+const aiProviderChatModelListIds = (s: AIProviderStoreState) =>
+  s.aiProviderModelList.filter((item) => item.type === 'chat').map((item) => item.id);
 // List
 const enabledAiProviderModelList = (s: AIProviderStoreState) =>
   s.aiProviderModelList.filter((item) => item.enabled);
@@ -68,6 +70,7 @@ const modelContextWindowTokens = (id: string, provider: string) => (s: AIProvide
 };
 
 export const aiModelSelectors = {
+  aiProviderChatModelListIds,
   disabledAiProviderModelList,
   enabledAiProviderModelList,
   filteredAiProviderModelList,