@lobehub/chat 1.53.10 → 1.53.12

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 1.53.12](https://github.com/lobehub/lobe-chat/compare/v1.53.11...v1.53.12)
+
+<sup>Released on **2025-02-14**</sup>
+
+#### ♻ Code Refactoring
+
+- **misc**: Improve model fetch behavior.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Code refactoring
+
+- **misc**: Improve model fetch behavior, closes [#6055](https://github.com/lobehub/lobe-chat/issues/6055) ([4c2aaf6](https://github.com/lobehub/lobe-chat/commit/4c2aaf6))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 1.53.11](https://github.com/lobehub/lobe-chat/compare/v1.53.10...v1.53.11)
+
+<sup>Released on **2025-02-13**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix provider form api key.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Fix provider form api key, closes [#6115](https://github.com/lobehub/lobe-chat/issues/6115) ([d074238](https://github.com/lobehub/lobe-chat/commit/d074238))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.53.10](https://github.com/lobehub/lobe-chat/compare/v1.53.9...v1.53.10)
 
 <sup>Released on **2025-02-13**</sup>

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "improvements": [
+        "Improve model fetch behavior."
+      ]
+    },
+    "date": "2025-02-14",
+    "version": "1.53.12"
+  },
+  {
+    "children": {
+      "fixes": [
+        "Fix provider form api key."
+      ]
+    },
+    "date": "2025-02-13",
+    "version": "1.53.11"
+  },
   {
     "children": {
       "fixes": [

package/locales/ar/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "يرجى إدخال مفتاح API الخاص بك",
-      "required": "يرجى إدخال مفتاح API الخاص بك",
       "title": "مفتاح API"
     },
     "basicTitle": "المعلومات الأساسية",
@@ -94,7 +93,8 @@
       "title": "اسم المزود"
     },
     "proxyUrl": {
-      "placeholder": "يرجى إدخال عنوان الطلب الخاص بك، إذا لم يتم إدخاله سيتم استخدام عنوان الطلب المقابل لـ SDK",
+      "placeholder": "يرجى إدخال عنوان الطلب الخاص بك",
+      "required": "يرجى إدخال عنوان الوكيل",
       "title": "عنوان الوكيل"
     },
     "sdkType": {

package/locales/bg-BG/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Моля, въведете вашия API ключ",
-      "required": "Моля, въведете вашия API ключ",
       "title": "API ключ"
     },
     "basicTitle": "Основна информация",
@@ -94,7 +93,8 @@
       "title": "Име на доставчика"
     },
     "proxyUrl": {
-      "placeholder": "Моля, въведете адреса на вашето запитване, ако не е попълнен, ще се използва съответния адрес на SDK",
+      "placeholder": "Моля, въведете адреса на вашето запитване",
+      "required": "Моля, въведете адреса на проксито",
       "title": "Адрес на прокси"
     },
     "sdkType": {

package/locales/de-DE/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Bitte geben Sie Ihren API-Schlüssel ein",
-      "required": "Bitte geben Sie Ihren API-Schlüssel ein",
       "title": "API-Schlüssel"
     },
     "basicTitle": "Grundinformationen",
@@ -94,7 +93,8 @@
       "title": "Name des Anbieters"
     },
     "proxyUrl": {
-      "placeholder": "Bitte geben Sie Ihre Anforderungsadresse ein, wenn nicht angegeben, wird die entsprechende Anforderungsadresse des SDK verwendet",
+      "placeholder": "Bitte geben Sie Ihre Anforderungs-URL ein",
+      "required": "Bitte geben Sie die Proxy-Adresse ein",
       "title": "Proxy-Adresse"
     },
     "sdkType": {

package/locales/en-US/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Please enter your API Key",
-      "required": "Please enter your API Key",
       "title": "API Key"
     },
     "basicTitle": "Basic Information",
@@ -94,7 +93,8 @@
       "title": "Provider Name"
     },
     "proxyUrl": {
-      "placeholder": "Please enter your request URL, if left blank, the corresponding SDK request URL will be used",
+      "placeholder": "Please enter your request URL",
+      "required": "Please enter the proxy address",
       "title": "Proxy URL"
     },
     "sdkType": {

package/locales/es-ES/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Por favor, introduce tu API Key",
-      "required": "Por favor, introduce tu API Key",
       "title": "API Key"
     },
     "basicTitle": "Información básica",
@@ -94,7 +93,8 @@
       "title": "Nombre del proveedor"
     },
     "proxyUrl": {
-      "placeholder": "Por favor, introduce tu dirección de solicitud, si no se completa, se usará la dirección de solicitud correspondiente del SDK",
+      "placeholder": "Por favor, introduce tu dirección de solicitud",
+      "required": "Por favor, introduce la dirección del proxy",
       "title": "Dirección del proxy"
     },
     "sdkType": {

package/locales/fa-IR/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "لطفاً کلید API خود را وارد کنید",
-      "required": "لطفاً کلید API خود را وارد کنید",
       "title": "کلید API"
     },
     "basicTitle": "اطلاعات پایه",
@@ -94,7 +93,8 @@
       "title": "نام ارائه‌دهنده"
     },
     "proxyUrl": {
-      "placeholder": "لطفاً آدرس درخواست خود را وارد کنید، در صورت عدم وارد کردن از آدرس درخواست مربوط به SDK استفاده خواهد شد",
+      "placeholder": "لطفاً آدرس درخواست خود را وارد کنید",
+      "required": "لطفاً آدرس پروکسی را وارد کنید",
       "title": "آدرس پروکسی"
     },
     "sdkType": {

package/locales/fr-FR/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Veuillez entrer votre clé API",
-      "required": "Veuillez entrer votre clé API",
       "title": "Clé API"
     },
     "basicTitle": "Informations de base",
@@ -94,7 +93,8 @@
       "title": "Nom du fournisseur"
     },
     "proxyUrl": {
-      "placeholder": "Veuillez entrer votre adresse de requête, sinon l'adresse de requête correspondante au SDK sera utilisée",
+      "placeholder": "Veuillez saisir votre adresse de requête",
+      "required": "Veuillez remplir l'adresse du proxy",
       "title": "Adresse du proxy"
     },
     "sdkType": {

package/locales/it-IT/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Inserisci la tua API Key",
-      "required": "Inserisci la tua API Key",
       "title": "API Key"
     },
     "basicTitle": "Informazioni di base",
@@ -94,7 +93,8 @@
       "title": "Nome del fornitore"
     },
     "proxyUrl": {
-      "placeholder": "Inserisci il tuo indirizzo di richiesta, se non specificato verrà utilizzato l'indirizzo di richiesta corrispondente all'SDK",
+      "placeholder": "Inserisci il tuo URL di richiesta",
+      "required": "Inserisci l'indirizzo del proxy",
       "title": "Indirizzo proxy"
     },
     "sdkType": {

package/locales/ja-JP/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "あなたの API キーを入力してください",
-      "required": "あなたの API キーを入力してください",
       "title": "API キー"
     },
     "basicTitle": "基本情報",
@@ -94,7 +93,8 @@
       "title": "サービスプロバイダー名"
     },
     "proxyUrl": {
-      "placeholder": "リクエストアドレスを入力してください。未入力の場合は SDK に対応するリクエストアドレスが使用されます",
+      "placeholder": "リクエストURLを入力してください",
+      "required": "プロキシURLを入力してください",
       "title": "プロキシアドレス"
     },
     "sdkType": {

package/locales/ko-KR/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "API 키를 입력하세요",
-      "required": "API 키를 입력하세요",
       "title": "API 키"
     },
     "basicTitle": "기본 정보",
@@ -94,7 +93,8 @@
       "title": "서비스 제공자 이름"
     },
     "proxyUrl": {
-      "placeholder": "요청 주소를 입력하세요, 입력하지 않으면 SDK의 요청 주소가 사용됩니다",
+      "placeholder": "요청 주소를 입력하세요",
+      "required": "프록시 주소를 입력하세요",
       "title": "프록시 주소"
     },
     "sdkType": {

package/locales/nl-NL/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Vul je API-sleutel in",
-      "required": "Vul je API-sleutel in",
       "title": "API-sleutel"
     },
     "basicTitle": "Basisinformatie",
@@ -94,7 +93,8 @@
       "title": "Naam van de provider"
     },
     "proxyUrl": {
-      "placeholder": "Vul je aanvraagadres in, als je dit niet invult, wordt het overeenkomstige aanvraagadres van de SDK gebruikt",
+      "placeholder": "Vul je verzoekadres in",
+      "required": "Vul het proxyadres in",
       "title": "Proxy-adres"
     },
     "sdkType": {

package/locales/pl-PL/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Proszę wpisać swój klucz API",
-      "required": "Proszę wpisać swój klucz API",
       "title": "Klucz API"
     },
     "basicTitle": "Podstawowe informacje",
@@ -94,7 +93,8 @@
       "title": "Nazwa dostawcy"
     },
     "proxyUrl": {
-      "placeholder": "Proszę wpisać adres żądania, jeśli nie zostanie podany, użyty zostanie odpowiedni adres żądania SDK",
+      "placeholder": "Proszę wpisać adres żądania",
+      "required": "Proszę wpisać adres proxy",
       "title": "Adres proxy"
     },
     "sdkType": {

package/locales/pt-BR/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Por favor, insira sua API Key",
-      "required": "Por favor, insira sua API Key",
       "title": "API Key"
     },
     "basicTitle": "Informações Básicas",
@@ -94,7 +93,8 @@
       "title": "Nome do Provedor"
     },
     "proxyUrl": {
-      "placeholder": "Por favor, insira seu endereço de requisição, se não for preenchido, será usado o endereço correspondente do SDK",
+      "placeholder": "Por favor, insira o seu endereço de solicitação",
+      "required": "Por favor, insira o endereço do proxy",
       "title": "Endereço do Proxy"
     },
     "sdkType": {

package/locales/ru-RU/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Пожалуйста, введите ваш API Key",
-      "required": "Пожалуйста, введите ваш API Key",
       "title": "API Key"
     },
     "basicTitle": "Основная информация",
@@ -94,7 +93,8 @@
       "title": "Имя провайдера"
     },
     "proxyUrl": {
-      "placeholder": "Пожалуйста, введите ваш адрес запроса, если не заполните, будет использован соответствующий адрес SDK",
+      "placeholder": "Пожалуйста, введите ваш адрес запроса",
+      "required": "Пожалуйста, введите адрес прокси",
       "title": "Адрес прокси"
     },
     "sdkType": {

package/locales/tr-TR/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Lütfen API Anahtarınızı girin",
-      "required": "Lütfen API Anahtarınızı girin",
       "title": "API Anahtarı"
     },
     "basicTitle": "Temel Bilgiler",
@@ -94,7 +93,8 @@
       "title": "Hizmet Sağlayıcı Adı"
     },
     "proxyUrl": {
-      "placeholder": "Lütfen istek adresinizi girin, boş bırakılırsa SDK'nın ilgili istek adresi kullanılacaktır",
+      "placeholder": "Lütfen istek adresinizi girin",
+      "required": "Lütfen proxy adresini girin",
       "title": "Proxy Adresi"
     },
     "sdkType": {

package/locales/vi-VN/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "Vui lòng nhập API Key của bạn",
-      "required": "Vui lòng nhập API Key của bạn",
       "title": "API Key"
     },
     "basicTitle": "Thông tin cơ bản",
@@ -94,7 +93,8 @@
       "title": "Tên nhà cung cấp"
     },
     "proxyUrl": {
-      "placeholder": "Vui lòng nhập địa chỉ yêu cầu của bạn, nếu không sẽ sử dụng địa chỉ yêu cầu tương ứng của SDK",
+      "placeholder": "Vui lòng nhập địa chỉ yêu cầu của bạn",
+      "required": "Vui lòng nhập địa chỉ proxy",
       "title": "Địa chỉ proxy"
     },
     "sdkType": {

package/locales/zh-CN/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "请填写你的 API Key",
-      "required": "请填写你的 API Key",
       "title": "API Key"
     },
     "basicTitle": "基本信息",
@@ -94,7 +93,8 @@
       "title": "服务商名称"
     },
     "proxyUrl": {
-      "placeholder": "请填写你的请求地址，如果不填则会使用 SDK 对应的请求地址",
+      "placeholder": "请填写你的请求地址",
+      "required": "请填写代理地址",
       "title": "代理地址"
     },
     "sdkType": {
@@ -305,4 +305,4 @@
   "zhipu": {
     "title": "智谱"
   }
-}
+}

package/locales/zh-TW/modelProvider.json

@@ -66,7 +66,6 @@
   "createNewAiProvider": {
     "apiKey": {
       "placeholder": "請填寫你的 API Key",
-      "required": "請填寫你的 API Key",
       "title": "API Key"
     },
     "basicTitle": "基本資訊",
@@ -94,7 +93,8 @@
       "title": "服務商名稱"
     },
     "proxyUrl": {
-      "placeholder": "請填寫你的請求地址，如果不填則會使用 SDK 對應的請求地址",
+      "placeholder": "請填寫你的請求地址",
+      "required": "請填寫代理地址",
       "title": "代理地址"
     },
     "sdkType": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.53.10",
+  "version": "1.53.12",
   "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/features/CreateNewProvider/index.tsx

@@ -91,6 +91,7 @@ const CreateNewProvider = memo<CreateNewProviderProps>(({ onClose, open }) => {
           options={[
             { label: 'OpenAI', value: 'openai' },
             { label: 'Anthropic', value: 'anthropic' },
+            { label: 'Ollama', value: 'ollama' },
           ]}
         />
       ),
@@ -98,6 +99,13 @@ const CreateNewProvider = memo<CreateNewProviderProps>(({ onClose, open }) => {
       name: 'sdkType',
       rules: [{ message: t('createNewAiProvider.sdkType.required'), required: true }],
     },
+    {
+      children: <Input allowClear placeholder={'https://xxxx-proxy.com/v1'} variant={'filled'} />,
+      label: t('createNewAiProvider.proxyUrl.title'),
+      minWidth: 400,
+      name: [KeyVaultsConfigKey, LLMProviderBaseUrlKey],
+      rules: [{ message: t('createNewAiProvider.proxyUrl.required'), required: true }],
+    },
     {
       children: (
         <Input.Password
@@ -109,14 +117,6 @@ const CreateNewProvider = memo<CreateNewProviderProps>(({ onClose, open }) => {
       label: t('createNewAiProvider.apiKey.title'),
       minWidth: 400,
       name: [KeyVaultsConfigKey, LLMProviderApiTokenKey],
-      rules: [{ message: t('createNewAiProvider.apiKey.required'), required: true }],
-    },
-    {
-      children: <Input allowClear placeholder={'https://xxxx-proxy.com/v1'} variant={'filled'} />,
-      desc: t('createNewAiProvider.proxyUrl.placeholder'),
-      label: t('createNewAiProvider.proxyUrl.title'),
-      minWidth: 400,
-      name: [KeyVaultsConfigKey, LLMProviderBaseUrlKey],
     },
   ];
 

package/src/config/aiModels/spark.ts

@@ -32,6 +32,9 @@ const sparkChatModels: AIChatModelCard[] = [
     type: 'chat',
   },
   {
+    abilities: {
+      functionCall: true,
+    },
     contextWindowTokens: 8192,
     description:
       'Spark Max 为功能最为全面的版本，支持联网搜索及众多内置插件。其全面优化的核心能力以及系统角色设定和函数调用功能，使其在各种复杂应用场景中的表现极为优异和出色。',
@@ -42,6 +45,9 @@ const sparkChatModels: AIChatModelCard[] = [
     type: 'chat',
   },
   {
+    abilities: {
+      functionCall: true,
+    },
     contextWindowTokens: 32_768,
     description:
       'Spark Max 32K 配置了大上下文处理能力，更强的上下文理解和逻辑推理能力，支持32K tokens的文本输入，适用于长文档阅读、私有知识问答等场景',
@@ -52,6 +58,9 @@ const sparkChatModels: AIChatModelCard[] = [
     type: 'chat',
   },
   {
+    abilities: {
+      functionCall: true,
+    },
     contextWindowTokens: 8192,
     description:
       'Spark Ultra 是星火大模型系列中最为强大的版本，在升级联网搜索链路同时，提升对文本内容的理解和总结能力。它是用于提升办公生产力和准确响应需求的全方位解决方案，是引领行业的智能产品。',

package/src/libs/agent-runtime/ai360/index.ts

@@ -1,7 +1,7 @@
 import { ModelProvider } from '../types';
 import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
 
-import { LOBE_DEFAULT_MODEL_LIST } from '@/config/aiModels';
+import type { ChatModelCard } from '@/types/llm';
 
 export interface Ai360ModelCard {
   id: string;
@@ -22,28 +22,44 @@ export const LobeAi360AI = LobeOpenAICompatibleFactory({
   debug: {
     chatCompletion: () => process.env.DEBUG_AI360_CHAT_COMPLETION === '1',
   },
-  models: {
-    transformModel: (m) => {
-      const reasoningKeywords = [
-        '360gpt2-o1',
-        '360zhinao2-o1',
-      ];
+  models: async ({ client }) => {
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
 
-      const model = m as unknown as Ai360ModelCard;
+    const reasoningKeywords = [
+      '360gpt2-o1',
+      '360zhinao2-o1',
+    ];
 
-      return {
-        contextWindowTokens: model.total_tokens,
-        displayName: LOBE_DEFAULT_MODEL_LIST.find((m) => model.id === m.id)?.displayName ?? undefined,
-        enabled: LOBE_DEFAULT_MODEL_LIST.find((m) => model.id === m.id)?.enabled || false,
-        functionCall: model.id === '360gpt-pro',
-        id: model.id,
-        maxTokens:
-          typeof model.max_tokens === 'number'
-            ? model.max_tokens
-            : undefined,
-        reasoning: reasoningKeywords.some(keyword => model.id.toLowerCase().includes(keyword)),
-      };
-    },
+    const modelsPage = await client.models.list() as any;
+    const modelList: Ai360ModelCard[] = modelsPage.data;
+
+    return modelList
+      .map((model) => {
+        const knownModel = LOBE_DEFAULT_MODEL_LIST.find((m) => model.id.toLowerCase() === m.id.toLowerCase());
+
+        return {
+          contextWindowTokens: model.total_tokens,
+          displayName: knownModel?.displayName ?? undefined,
+          enabled: knownModel?.enabled || false,
+          functionCall:
+            model.id === '360gpt-pro'
+            || knownModel?.abilities?.functionCall
+            || false,
+          id: model.id,
+          maxTokens:
+            typeof model.max_tokens === 'number'
+              ? model.max_tokens
+              : undefined,
+          reasoning:
+            reasoningKeywords.some(keyword => model.id.toLowerCase().includes(keyword))
+            || knownModel?.abilities?.reasoning
+            || false,
+          vision:
+            knownModel?.abilities?.vision
+            || false,
+        };
+      })
+      .filter(Boolean) as ChatModelCard[];
   },
   provider: ModelProvider.Ai360,
 });

package/src/libs/agent-runtime/anthropic/index.ts

@@ -13,7 +13,6 @@ import { buildAnthropicMessages, buildAnthropicTools } from '../utils/anthropicH
 import { StreamingResponse } from '../utils/response';
 import { AnthropicStream } from '../utils/streams';
 
-import { LOBE_DEFAULT_MODEL_LIST } from '@/config/aiModels';
 import type { ChatModelCard } from '@/types/llm';
 
 export interface AnthropicModelCard {
@@ -114,6 +113,8 @@ export class LobeAnthropicAI implements LobeRuntimeAI {
   }
 
   async models() {
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+
     const url = `${this.baseURL}/v1/models`;
     const response = await fetch(url, {
       headers: {
@@ -128,13 +129,24 @@ export class LobeAnthropicAI implements LobeRuntimeAI {
   
     return modelList
       .map((model) => {
+        const knownModel = LOBE_DEFAULT_MODEL_LIST.find((m) => model.id.toLowerCase() === m.id.toLowerCase());
+
         return {
-          contextWindowTokens: LOBE_DEFAULT_MODEL_LIST.find((m) => model.id === m.id)?.contextWindowTokens ?? undefined,
+          contextWindowTokens: knownModel?.contextWindowTokens ?? undefined,
           displayName: model.display_name,
-          enabled: LOBE_DEFAULT_MODEL_LIST.find((m) => model.id === m.id)?.enabled || false,
-          functionCall: model.id.toLowerCase().includes('claude-3'),
+          enabled: knownModel?.enabled || false,
+          functionCall:
+            model.id.toLowerCase().includes('claude-3')
+            || knownModel?.abilities?.functionCall
+            || false,
           id: model.id,
-          vision: model.id.toLowerCase().includes('claude-3') && !model.id.toLowerCase().includes('claude-3-5-haiku'),
+          reasoning:
+            knownModel?.abilities?.reasoning
+            || false,
+          vision:
+            model.id.toLowerCase().includes('claude-3') && !model.id.toLowerCase().includes('claude-3-5-haiku')
+            || knownModel?.abilities?.vision
+            || false,
         };
       })
       .filter(Boolean) as ChatModelCard[];

package/src/libs/agent-runtime/baichuan/index.ts

@@ -3,7 +3,6 @@ import OpenAI from 'openai';
 import { ChatStreamPayload, ModelProvider } from '../types';
 import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
 
-import { LOBE_DEFAULT_MODEL_LIST } from '@/config/aiModels';
 import type { ChatModelCard } from '@/types/llm';
 
 export interface BaichuanModelCard {
@@ -32,18 +31,28 @@ export const LobeBaichuanAI = LobeOpenAICompatibleFactory({
     chatCompletion: () => process.env.DEBUG_BAICHUAN_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+
     const modelsPage = await client.models.list() as any;
     const modelList: BaichuanModelCard[] = modelsPage.data;
 
     return modelList
       .map((model) => {
+        const knownModel = LOBE_DEFAULT_MODEL_LIST.find((m) => model.model.toLowerCase() === m.id.toLowerCase());
+
         return {
           contextWindowTokens: model.max_input_length,
           displayName: model.model_show_name,
-          enabled: LOBE_DEFAULT_MODEL_LIST.find((m) => model.model === m.id)?.enabled || false,
+          enabled: knownModel?.enabled || false,
           functionCall: model.function_call,
           id: model.model,
           maxTokens: model.max_tokens,
+          reasoning:
+            knownModel?.abilities?.reasoning
+            || false,
+          vision:
+            knownModel?.abilities?.vision
+            || false,
         };
       })
       .filter(Boolean) as ChatModelCard[];