@lobehub/chat 1.45.9 → 1.45.11

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 1.45.11](https://github.com/lobehub/lobe-chat/compare/v1.45.10...v1.45.11)
+
+<sup>Released on **2025-01-14**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Support Gemini 2.0 HarmBlockThreshold.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Support Gemini 2.0 HarmBlockThreshold, closes [#5422](https://github.com/lobehub/lobe-chat/issues/5422) ([4d1a4cd](https://github.com/lobehub/lobe-chat/commit/4d1a4cd))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 1.45.10](https://github.com/lobehub/lobe-chat/compare/v1.45.9...v1.45.10)
+
+<sup>Released on **2025-01-14**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix some providers issues.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Fix some providers issues, closes [#5429](https://github.com/lobehub/lobe-chat/issues/5429) ([b7edbd3](https://github.com/lobehub/lobe-chat/commit/b7edbd3))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.45.9](https://github.com/lobehub/lobe-chat/compare/v1.45.8...v1.45.9)
 
 <sup>Released on **2025-01-14**</sup>

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "fixes": [
+        "Support Gemini 2.0 HarmBlockThreshold."
+      ]
+    },
+    "date": "2025-01-14",
+    "version": "1.45.11"
+  },
+  {
+    "children": {
+      "fixes": [
+        "Fix some providers issues."
+      ]
+    },
+    "date": "2025-01-14",
+    "version": "1.45.10"
+  },
   {
     "children": {
       "fixes": [

package/locales/ar/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "إضافة نموذج",
       "disabled": "غير مفعل",
+      "disabledActions": {
+        "showMore": "عرض الكل"
+      },
       "empty": {
         "desc": "يرجى إنشاء نموذج مخصص أو سحب نموذج للبدء في الاستخدام",
         "title": "لا توجد نماذج متاحة"

package/locales/bg-BG/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Добавяне на модел",
       "disabled": "Неактивен",
+      "disabledActions": {
+        "showMore": "Покажи всичко"
+      },
       "empty": {
         "desc": "Моля, създайте персонализиран модел или изтеглете модел, за да започнете да го използвате",
         "title": "Няма налични модели"

package/locales/de-DE/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Modell hinzufügen",
       "disabled": "Nicht aktiviert",
+      "disabledActions": {
+        "showMore": "Alle anzeigen"
+      },
       "empty": {
         "desc": "Bitte erstellen Sie ein benutzerdefiniertes Modell oder ziehen Sie ein Modell, um zu beginnen.",
         "title": "Keine verfügbaren Modelle"

package/locales/en-US/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Add Model",
       "disabled": "Disabled",
+      "disabledActions": {
+        "showMore": "Show All"
+      },
       "empty": {
         "desc": "Please create a custom model or pull a model to get started.",
         "title": "No available models"

package/locales/es-ES/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Agregar modelo",
       "disabled": "No habilitado",
+      "disabledActions": {
+        "showMore": "Mostrar todo"
+      },
       "empty": {
         "desc": "Por favor, crea un modelo personalizado o importa un modelo para comenzar a usarlo.",
         "title": "No hay modelos disponibles"

package/locales/fa-IR/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "مدل جدید اضافه کنید",
       "disabled": "غیرفعال",
+      "disabledActions": {
+        "showMore": "نمایش همه"
+      },
       "empty": {
         "desc": "لطفاً یک مدل سفارشی ایجاد کنید یا پس از بارگذاری مدل‌ها، شروع به استفاده کنید",
         "title": "مدل قابل استفاده‌ای وجود ندارد"

package/locales/fr-FR/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Ajouter un modèle",
       "disabled": "Non activé",
+      "disabledActions": {
+        "showMore": "Afficher tout"
+      },
       "empty": {
         "desc": "Veuillez créer un modèle personnalisé ou importer un modèle pour commencer à l'utiliser.",
         "title": "Aucun modèle disponible"

package/locales/it-IT/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Aggiungi modello",
       "disabled": "Non attivato",
+      "disabledActions": {
+        "showMore": "Mostra tutto"
+      },
       "empty": {
         "desc": "Si prega di creare un modello personalizzato o di importare un modello per iniziare a utilizzarlo",
         "title": "Nessun modello disponibile"

package/locales/ja-JP/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "モデルを追加",
       "disabled": "無効",
+      "disabledActions": {
+        "showMore": "すべて表示"
+      },
       "empty": {
         "desc": "カスタムモデルを作成するか、モデルを取得してから使用を開始してください",
         "title": "利用可能なモデルはありません"

package/locales/ko-KR/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "모델 추가",
       "disabled": "비활성화",
+      "disabledActions": {
+        "showMore": "모두 보기"
+      },
       "empty": {
         "desc": "사용할 수 있는 모델이 없습니다. 사용자 정의 모델을 생성하거나 모델을 가져온 후 시작하세요.",
         "title": "사용 가능한 모델이 없습니다."

package/locales/nl-NL/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Model toevoegen",
       "disabled": "Niet ingeschakeld",
+      "disabledActions": {
+        "showMore": "Toon alles"
+      },
       "empty": {
         "desc": "Maak een aangepast model of haal een model op om te beginnen met gebruiken.",
         "title": "Geen beschikbare modellen"

package/locales/pl-PL/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Dodaj model",
       "disabled": "Nieaktywne",
+      "disabledActions": {
+        "showMore": "Pokaż więcej"
+      },
       "empty": {
         "desc": "Proszę utworzyć model niestandardowy lub pobrać model, aby rozpocząć korzystanie.",
         "title": "Brak dostępnych modeli"

package/locales/pt-BR/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Adicionar Modelo",
       "disabled": "Não habilitado",
+      "disabledActions": {
+        "showMore": "Mostrar tudo"
+      },
       "empty": {
         "desc": "Por favor, crie um modelo personalizado ou importe um modelo para começar a usar.",
         "title": "Nenhum modelo disponível"

package/locales/ru-RU/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Добавить модель",
       "disabled": "Не активирован",
+      "disabledActions": {
+        "showMore": "Показать все"
+      },
       "empty": {
         "desc": "Пожалуйста, создайте пользовательскую модель или загрузите модель, чтобы начать использовать.",
         "title": "Нет доступных моделей"

package/locales/tr-TR/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Model Ekle",
       "disabled": "Devre dışı",
+      "disabledActions": {
+        "showMore": "Hepsini Göster"
+      },
       "empty": {
         "desc": "Lütfen özel bir model oluşturun veya kullanmaya başlamadan önce bir model çekin",
         "title": "Kullanılabilir model yok"

package/locales/vi-VN/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "Thêm mô hình",
       "disabled": "Chưa được kích hoạt",
+      "disabledActions": {
+        "showMore": "Hiển thị tất cả"
+      },
       "empty": {
         "desc": "Vui lòng tạo mô hình tùy chỉnh hoặc kéo mô hình để bắt đầu sử dụng",
         "title": "Chưa có mô hình nào khả dụng"

package/locales/zh-CN/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "添加模型",
       "disabled": "未启用",
+      "disabledActions": {
+        "showMore": "显示全部"
+      },
       "empty": {
         "desc": "请创建自定义模型或拉取模型后开始使用吧",
         "title": "暂无可用模型"

package/locales/zh-TW/modelProvider.json

@@ -273,6 +273,9 @@
     "list": {
       "addNew": "新增模型",
       "disabled": "未啟用",
+      "disabledActions": {
+        "showMore": "顯示全部"
+      },
       "empty": {
         "desc": "請創建自定義模型或拉取模型後開始使用吧",
         "title": "暫無可用模型"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.45.9",
+  "version": "1.45.11",
   "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",
@@ -236,8 +236,8 @@
     "yaml": "^2.7.0",
     "yjs": "^13.6.21",
     "zod": "^3.24.1",
-    "zustand": "5.0.1",
-    "zustand-utils": "^1.3.2"
+    "zustand": "^5.0.3",
+    "zustand-utils": "^2"
   },
   "devDependencies": {
     "@commitlint/cli": "^19.6.1",

package/src/app/(main)/chat/(workspace)/_layout/Desktop/ChatHeader/Tags.tsx

@@ -12,14 +12,15 @@ import PluginTag from '../../../features/PluginTag';
 import KnowledgeTag from './KnowledgeTag';
 
 const TitleTags = memo(() => {
-  const [model, hasKnowledge] = useAgentStore((s) => [
+  const [model, provider, hasKnowledge] = useAgentStore((s) => [
     agentSelectors.currentAgentModel(s),
+    agentSelectors.currentAgentModelProvider(s),
     agentSelectors.hasKnowledge(s),
   ]);
   const plugins = useAgentStore(agentSelectors.currentAgentPlugins, isEqual);
   const enabledKnowledge = useAgentStore(agentSelectors.currentEnabledKnowledge, isEqual);
 
-  const showPlugin = useModelSupportToolUse(model);
+  const showPlugin = useModelSupportToolUse(model, provider);
 
   return (
     <Flexbox align={'center'} horizontal>

package/src/app/(main)/settings/provider/features/ModelList/DisabledModels.tsx

@@ -1,6 +1,8 @@
-import { Typography } from 'antd';
+import { Icon } from '@lobehub/ui';
+import { Button, Typography } from 'antd';
 import isEqual from 'fast-deep-equal';
-import { memo } from 'react';
+import { ChevronDown } from 'lucide-react';
+import { memo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
@@ -12,16 +14,32 @@ import ModelItem from './ModelItem';
 const DisabledModels = memo(() => {
   const { t } = useTranslation('modelProvider');
 
+  const [showMore, setShowMore] = useState(false);
   const disabledModels = useAiInfraStore(aiModelSelectors.disabledAiProviderModelList, isEqual);
 
+  const displayModels = showMore ? disabledModels : disabledModels.slice(0, 10);
   return (
     <Flexbox>
       <Typography.Text style={{ fontSize: 12, marginTop: 8 }} type={'secondary'}>
         {t('providerModels.list.disabled')}
       </Typography.Text>
-      {disabledModels.map((item) => (
+      {displayModels.map((item) => (
         <ModelItem {...item} key={item.id} />
       ))}
+      {!showMore && (
+        <Button
+          block
+          color={'default'}
+          icon={<Icon icon={ChevronDown} />}
+          onClick={() => {
+            setShowMore(true);
+          }}
+          size={'small'}
+          variant={'filled'}
+        >
+          {t('providerModels.list.disabledActions.showMore')}
+        </Button>
+      )}
     </Flexbox>
   );
 });

package/src/features/ChatInput/ActionBar/Token/TokenTag.tsx

@@ -6,6 +6,7 @@ import { memo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Center, Flexbox } from 'react-layout-kit';
 
+import { useModelContextWindowTokens } from '@/hooks/useModelContextWindowTokens';
 import { useModelSupportToolUse } from '@/hooks/useModelSupportToolUse';
 import { useTokenCount } from '@/hooks/useTokenCount';
 import { useAgentStore } from '@/store/agent';
@@ -14,8 +15,6 @@ import { useChatStore } from '@/store/chat';
 import { topicSelectors } from '@/store/chat/selectors';
 import { useToolStore } from '@/store/tool';
 import { toolSelectors } from '@/store/tool/selectors';
-import { useUserStore } from '@/store/user';
-import { modelProviderSelectors } from '@/store/user/selectors';
 
 import TokenProgress from './TokenProgress';
 
@@ -31,22 +30,23 @@ const Token = memo<TokenTagProps>(({ total: messageString }) => {
     topicSelectors.currentActiveTopicSummary(s)?.content || '',
   ]);
 
-  const [systemRole, model] = useAgentStore((s) => {
+  const [systemRole, model, provider] = useAgentStore((s) => {
     const config = agentSelectors.currentAgentChatConfig(s);
 
     return [
       agentSelectors.currentAgentSystemRole(s),
       agentSelectors.currentAgentModel(s) as string,
+      agentSelectors.currentAgentModelProvider(s) as string,
       // add these two params to enable the component to re-render
       config.historyCount,
       config.enableHistoryCount,
     ];
   });
 
-  const maxTokens = useUserStore(modelProviderSelectors.modelMaxToken(model));
+  const maxTokens = useModelContextWindowTokens(model, provider);
 
   // Tool usage token
-  const canUseTool = useModelSupportToolUse(model);
+  const canUseTool = useModelSupportToolUse(model, provider);
   const plugins = useAgentStore(agentSelectors.currentAgentPlugins);
   const toolsString = useToolStore((s) => {
     const pluginSystemRoles = toolSelectors.enabledSystemRoles(plugins)(s);

package/src/features/ChatInput/ActionBar/Token/index.tsx

@@ -1,21 +1,18 @@
 import dynamic from 'next/dynamic';
 import { PropsWithChildren, memo } from 'react';
 
-import { useAgentStore } from '@/store/agent';
-import { agentSelectors } from '@/store/agent/selectors';
+import { useModelHasContextWindowToken } from '@/hooks/useModelHasContextWindowToken';
 import { useChatStore } from '@/store/chat';
 import { chatSelectors, threadSelectors } from '@/store/chat/selectors';
-import { useUserStore } from '@/store/user';
-import { modelProviderSelectors } from '@/store/user/selectors';
 
 const LargeTokenContent = dynamic(() => import('./TokenTag'), { ssr: false });
 
 const Token = memo<PropsWithChildren>(({ children }) => {
-  const model = useAgentStore(agentSelectors.currentAgentModel);
-  const showTag = useUserStore(modelProviderSelectors.isModelHasMaxToken(model));
+  const showTag = useModelHasContextWindowToken();
 
   return showTag && children;
 });
+
 export const MainToken = memo(() => {
   const total = useChatStore(chatSelectors.mainAIChatsMessageString);
 

package/src/features/ChatInput/ActionBar/Tools/index.tsx

@@ -15,7 +15,9 @@ const Tools = memo(() => {
   const { enablePlugins } = useServerConfigStore(featureFlagsSelectors);
 
   const model = useAgentStore(agentSelectors.currentAgentModel);
-  const enableFC = useModelSupportToolUse(model);
+  const provider = useAgentStore(agentSelectors.currentAgentModelProvider);
+
+  const enableFC = useModelSupportToolUse(model, provider);
 
   return (
     enablePlugins && (

package/src/features/ChatInput/ActionBar/Upload/ServerMode.tsx

@@ -25,8 +25,9 @@ const FileUpload = memo(() => {
   const upload = useFileStore((s) => s.uploadChatFiles);
 
   const model = useAgentStore(agentSelectors.currentAgentModel);
+  const provider = useAgentStore(agentSelectors.currentAgentModelProvider);
 
-  const canUploadImage = useModelSupportVision(model);
+  const canUploadImage = useModelSupportVision(model, provider);
 
   const items: MenuProps['items'] = [
     {

package/src/features/ChatInput/Desktop/FilePreview/index.tsx

@@ -1,18 +1,18 @@
 import { memo } from 'react';
 
 import DragUpload from '@/components/DragUpload';
+import { useModelSupportVision } from '@/hooks/useModelSupportVision';
 import { useAgentStore } from '@/store/agent';
 import { agentSelectors } from '@/store/agent/slices/chat';
 import { useFileStore } from '@/store/file';
-import { useUserStore } from '@/store/user';
-import { modelProviderSelectors } from '@/store/user/selectors';
 
 import FileItemList from './FileList';
 
 const FilePreview = memo(() => {
   const model = useAgentStore(agentSelectors.currentAgentModel);
+  const provider = useAgentStore(agentSelectors.currentAgentModelProvider);
 
-  const canUploadImage = useUserStore(modelProviderSelectors.isModelEnabledUpload(model));
+  const canUploadImage = useModelSupportVision(model, provider);
 
   const [uploadFiles] = useFileStore((s) => [s.uploadChatFiles]);
 

package/src/hooks/useModelContextWindowTokens.ts

@@ -0,0 +1,15 @@
+import { isServerMode } from '@/const/version';
+import { aiModelSelectors, useAiInfraStore } from '@/store/aiInfra';
+import { useUserStore } from '@/store/user';
+import { modelProviderSelectors } from '@/store/user/selectors';
+
+export const useModelContextWindowTokens = (model: string, provider: string) => {
+  const newValue = useAiInfraStore(aiModelSelectors.modelContextWindowTokens(model, provider));
+
+  // TODO: remove this in V2.0
+  const oldValue = useUserStore(modelProviderSelectors.modelMaxToken(model));
+  if (!isServerMode) return oldValue;
+  //
+
+  return newValue as number;
+};

package/src/hooks/useModelHasContextWindowToken.ts

@@ -0,0 +1,19 @@
+import { isServerMode } from '@/const/version';
+import { useAgentStore } from '@/store/agent';
+import { agentSelectors } from '@/store/agent/slices/chat';
+import { aiModelSelectors, useAiInfraStore } from '@/store/aiInfra';
+import { useUserStore } from '@/store/user';
+import { modelProviderSelectors } from '@/store/user/selectors';
+
+export const useModelHasContextWindowToken = () => {
+  const model = useAgentStore(agentSelectors.currentAgentModel);
+  const provider = useAgentStore(agentSelectors.currentAgentModelProvider);
+  const newValue = useAiInfraStore(aiModelSelectors.isModelHasContextWindowToken(model, provider));
+
+  // TODO: remove this in V2.0
+  const oldValue = useUserStore(modelProviderSelectors.isModelHasMaxToken(model));
+  if (!isServerMode) return oldValue;
+  //
+
+  return newValue;
+};

package/src/hooks/useModelSupportToolUse.ts

@@ -3,8 +3,8 @@ import { aiModelSelectors, useAiInfraStore } from '@/store/aiInfra';
 import { useUserStore } from '@/store/user';
 import { modelProviderSelectors } from '@/store/user/selectors';
 
-export const useModelSupportToolUse = (model: string) => {
-  const newValue = useAiInfraStore(aiModelSelectors.isModelSupportToolUse(model));
+export const useModelSupportToolUse = (model: string, provider: string) => {
+  const newValue = useAiInfraStore(aiModelSelectors.isModelSupportToolUse(model, provider));
 
   // TODO: remove this in V2.0
   const oldValue = useUserStore(modelProviderSelectors.isModelEnabledFunctionCall(model));

package/src/hooks/useModelSupportVision.ts

@@ -3,8 +3,8 @@ import { aiModelSelectors, useAiInfraStore } from '@/store/aiInfra';
 import { useUserStore } from '@/store/user';
 import { modelProviderSelectors } from '@/store/user/selectors';
 
-export const useModelSupportVision = (model: string) => {
-  const newValue = useAiInfraStore(aiModelSelectors.isModelSupportVision(model));
+export const useModelSupportVision = (model: string, provider: string) => {
+  const newValue = useAiInfraStore(aiModelSelectors.isModelSupportVision(model, provider));
 
   // TODO: remove this in V2.0
   const oldValue = useUserStore(modelProviderSelectors.isModelEnabledVision(model));

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

@@ -36,6 +36,7 @@ enum HarmCategory {
 
 enum HarmBlockThreshold {
   BLOCK_NONE = 'BLOCK_NONE',
+  OFF = 'OFF', // https://discuss.ai.google.dev/t/59352
 }
 
 export class LobeGoogleAI implements LobeRuntimeAI {
@@ -70,19 +71,19 @@ export class LobeGoogleAI implements LobeRuntimeAI {
             safetySettings: [
               {
                 category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
-                threshold: HarmBlockThreshold.BLOCK_NONE,
+                threshold: model.includes('2.0') ? (HarmBlockThreshold.OFF as any) : HarmBlockThreshold.BLOCK_NONE,
               },
               {
                 category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
-                threshold: HarmBlockThreshold.BLOCK_NONE,
+                threshold: model.includes('2.0') ? (HarmBlockThreshold.OFF as any) : HarmBlockThreshold.BLOCK_NONE,
               },
               {
                 category: HarmCategory.HARM_CATEGORY_HARASSMENT,
-                threshold: HarmBlockThreshold.BLOCK_NONE,
+                threshold: model.includes('2.0') ? (HarmBlockThreshold.OFF as any) : HarmBlockThreshold.BLOCK_NONE,
               },
               {
                 category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
-                threshold: HarmBlockThreshold.BLOCK_NONE,
+                threshold: model.includes('2.0') ? (HarmBlockThreshold.OFF as any) : HarmBlockThreshold.BLOCK_NONE,
               },
             ],
           },

package/src/locales/default/modelProvider.ts

@@ -278,6 +278,7 @@ export default {
     list: {
       addNew: '添加模型',
       disabled: '未启用',
+      disabledActions: { showMore: '显示全部' },
       empty: {
         desc: '请创建自定义模型或拉取模型后开始使用吧',
         title: '暂无可用模型',

package/src/services/__tests__/chat.test.ts

@@ -797,6 +797,7 @@ describe('ChatService', () => {
         const output = chatService['processMessages']({
           messages,
           model: 'gpt-4o',
+          provider: 'openai',
         });
 
         expect(output).toEqual([

package/src/services/chat.ts

@@ -40,13 +40,13 @@ import { createTraceHeader, getTraceId } from '@/utils/trace';
 import { createHeaderWithAuth, createPayloadWithKeyVaults } from './_auth';
 import { API_ENDPOINTS } from './_url';
 
-const isCanUseFC = (model: string) => {
+const isCanUseFC = (model: string, provider: string) => {
   // TODO: remove isDeprecatedEdition condition in V2.0
   if (!isServerMode) {
     return modelProviderSelectors.isModelEnabledFunctionCall(model)(useUserStore.getState());
   }
 
-  return aiModelSelectors.isModelSupportToolUse(model)(useAiInfraStore.getState());
+  return aiModelSelectors.isModelSupportToolUse(model, provider)(useAiInfraStore.getState());
 };
 
 const findAzureDeploymentName = (model: string) => {
@@ -232,6 +232,7 @@ class ChatService {
       {
         messages,
         model: payload.model,
+        provider: payload.provider!,
         tools: enabledPlugins,
       },
       options,
@@ -242,7 +243,7 @@ class ChatService {
     const filterTools = toolSelectors.enabledSchema(enabledPlugins)(useToolStore.getState());
 
     // check this model can use function call
-    const canUseFC = isCanUseFC(payload.model);
+    const canUseFC = isCanUseFC(payload.model, payload.provider!);
 
     // the rule that model can use tools:
     // 1. tools is not empty
@@ -440,9 +441,11 @@ class ChatService {
       messages,
       tools,
       model,
+      provider,
     }: {
       messages: ChatMessage[];
       model: string;
+      provider: string;
       tools?: string[];
     },
     options?: FetchOptions,
@@ -513,7 +516,7 @@ class ChatService {
 
       // Inject Tool SystemRole
       const hasTools = tools && tools?.length > 0;
-      const hasFC = hasTools && isCanUseFC(model);
+      const hasFC = hasTools && isCanUseFC(model, provider);
       const toolsSystemRoles =
         hasFC && toolSelectors.enabledSystemRoles(tools)(useToolStore.getState());
 

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

@@ -33,21 +33,34 @@ const isModelLoading = (id: string) => (s: AIProviderStoreState) =>
 const getAiModelById = (id: string) => (s: AIProviderStoreState) =>
   s.aiProviderModelList.find((i) => i.id === id);
 
-const getEnabledModelById = (id: string) => (s: AIProviderStoreState) =>
-  s.enabledAiModels?.find((i) => i.id === id);
+const getEnabledModelById = (id: string, provider: string) => (s: AIProviderStoreState) =>
+  s.enabledAiModels?.find((i) => i.id === id && (provider ? provider === i.providerId : true));
 
-const isModelSupportToolUse = (id: string) => (s: AIProviderStoreState) => {
-  const model = getEnabledModelById(id)(s);
+const isModelSupportToolUse = (id: string, provider: string) => (s: AIProviderStoreState) => {
+  const model = getEnabledModelById(id, provider)(s);
 
   return model?.abilities?.functionCall;
 };
 
-const isModelSupportVision = (id: string) => (s: AIProviderStoreState) => {
-  const model = getEnabledModelById(id)(s);
+const isModelSupportVision = (id: string, provider: string) => (s: AIProviderStoreState) => {
+  const model = getEnabledModelById(id, provider)(s);
 
   return model?.abilities?.vision;
 };
 
+const isModelHasContextWindowToken =
+  (id: string, provider: string) => (s: AIProviderStoreState) => {
+    const model = getEnabledModelById(id, provider)(s);
+
+    return typeof model?.contextWindowTokens === 'number';
+  };
+
+const modelContextWindowTokens = (id: string, provider: string) => (s: AIProviderStoreState) => {
+  const model = getEnabledModelById(id, provider)(s);
+
+  return model?.contextWindowTokens;
+};
+
 export const aiModelSelectors = {
   disabledAiProviderModelList,
   enabledAiProviderModelList,
@@ -56,8 +69,10 @@ export const aiModelSelectors = {
   hasRemoteModels,
   isEmptyAiProviderModelList,
   isModelEnabled,
+  isModelHasContextWindowToken,
   isModelLoading,
   isModelSupportToolUse,
   isModelSupportVision,
+  modelContextWindowTokens,
   totalAiProviderModelList,
 };

package/src/types/aiModel.ts

@@ -283,8 +283,8 @@ export const UpdateAiModelSchema = z.object({
       deploymentName: z.string().optional(),
     })
     .optional(),
-  contextWindowTokens: z.number().optional(),
-  displayName: z.string().optional(),
+  contextWindowTokens: z.number().nullable().optional(),
+  displayName: z.string().nullable().optional(),
 });
 
 export type UpdateAiModelParams = z.infer<typeof UpdateAiModelSchema>;