@lobehub/chat 1.128.0 → 1.128.2

package/src/services/chat/contextEngineering.ts

@@ -0,0 +1,123 @@
+import { INBOX_SESSION_ID, isDesktop, isServerMode } from '@lobechat/const';
+import {
+  type AgentState,
+  ContextEngine,
+  HistorySummaryProvider,
+  HistoryTruncateProcessor,
+  InboxGuideProvider,
+  InputTemplateProcessor,
+  MessageCleanupProcessor,
+  MessageContentProcessor,
+  PlaceholderVariablesProcessor,
+  SystemRoleInjector,
+  ToolCallProcessor,
+  ToolMessageReorder,
+  ToolSystemRoleProvider,
+} from '@lobechat/context-engine';
+import { historySummaryPrompt } from '@lobechat/prompts';
+import { ChatMessage, OpenAIChatMessage } from '@lobechat/types';
+
+import { INBOX_GUIDE_SYSTEMROLE } from '@/const/guide';
+import { getToolStoreState } from '@/store/tool';
+import { toolSelectors } from '@/store/tool/selectors';
+import { VARIABLE_GENERATORS } from '@/utils/client/parserPlaceholder';
+import { genToolCallingName } from '@/utils/toolCall';
+
+import { isCanUseFC, isCanUseVision } from './helper';
+import { FetchOptions } from './types';
+
+export const contextEngineering = async (
+  {
+    messages = [],
+    tools,
+    model,
+    provider,
+    systemRole,
+    inputTemplate,
+    enableHistoryCount,
+    historyCount,
+  }: {
+    enableHistoryCount?: boolean;
+    historyCount?: number;
+    inputTemplate?: string;
+    messages: ChatMessage[];
+    model: string;
+    provider: string;
+    systemRole?: string;
+    tools?: string[];
+  },
+  options?: FetchOptions,
+): Promise<OpenAIChatMessage[]> => {
+  const pipeline = new ContextEngine({
+    pipeline: [
+      // 1. History truncation (MUST be first, before any message injection)
+      new HistoryTruncateProcessor({ enableHistoryCount, historyCount }),
+
+      // --------- Create system role injection providers
+
+      // 2. System role injection (agent's system role)
+      new SystemRoleInjector({ systemRole }),
+
+      // 3. Inbox guide system role injection
+      new InboxGuideProvider({
+        inboxGuideSystemRole: INBOX_GUIDE_SYSTEMROLE,
+        inboxSessionId: INBOX_SESSION_ID,
+        isWelcomeQuestion: options?.isWelcomeQuestion,
+        sessionId: options?.trace?.sessionId,
+      }),
+
+      // 4. Tool system role injection
+      new ToolSystemRoleProvider({
+        getToolSystemRoles: (tools) => toolSelectors.enabledSystemRoles(tools)(getToolStoreState()),
+        isCanUseFC,
+        model,
+        provider,
+        tools,
+      }),
+
+      // 5. History summary injection
+      new HistorySummaryProvider({
+        formatHistorySummary: historySummaryPrompt,
+        historySummary: options?.historySummary,
+      }),
+
+      // Create message processing processors
+
+      // 6. Input template processing
+      new InputTemplateProcessor({
+        inputTemplate,
+      }),
+
+      // 7. Placeholder variables processing
+      new PlaceholderVariablesProcessor({ variableGenerators: VARIABLE_GENERATORS }),
+
+      // 8. Message content processing
+      new MessageContentProcessor({
+        fileContext: { enabled: isServerMode, includeFileUrl: !isDesktop },
+        isCanUseVision,
+        model,
+        provider,
+      }),
+
+      // 9. Tool call processing
+      new ToolCallProcessor({ genToolCallingName, isCanUseFC, model, provider }),
+
+      // 10. Tool message reordering
+      new ToolMessageReorder(),
+
+      // 11. Message cleanup (final step, keep only necessary fields)
+      new MessageCleanupProcessor(),
+    ],
+  });
+
+  const initialState: AgentState = { messages, model, provider, systemRole, tools };
+
+  const result = await pipeline.process({
+    initialState,
+    maxTokens: 10_000_000,
+    messages,
+    model,
+  });
+
+  return result.messages;
+};

package/src/services/chat/helper.ts

@@ -0,0 +1,61 @@
+import { isDeprecatedEdition } from '@lobechat/const';
+import { ModelProvider } from '@lobechat/model-runtime';
+
+import { getAiInfraStoreState } from '@/store/aiInfra';
+import { aiModelSelectors, aiProviderSelectors } from '@/store/aiInfra/selectors';
+import { getUserStoreState, useUserStore } from '@/store/user';
+import { modelConfigSelectors, modelProviderSelectors } from '@/store/user/selectors';
+
+export const isCanUseFC = (model: string, provider: string): boolean => {
+  // TODO: remove isDeprecatedEdition condition in V2.0
+  if (isDeprecatedEdition) {
+    return modelProviderSelectors.isModelEnabledFunctionCall(model)(getUserStoreState());
+  }
+
+  return aiModelSelectors.isModelSupportToolUse(model, provider)(getAiInfraStoreState()) || false;
+};
+
+export const isCanUseVision = (model: string, provider: string): boolean => {
+  // TODO: remove isDeprecatedEdition condition in V2.0
+  if (isDeprecatedEdition) {
+    return modelProviderSelectors.isModelEnabledVision(model)(getUserStoreState());
+  }
+  return aiModelSelectors.isModelSupportVision(model, provider)(getAiInfraStoreState());
+};
+
+/**
+ * TODO: we need to update this function to auto find deploymentName with provider setting config
+ */
+export const findDeploymentName = (model: string, provider: string) => {
+  let deploymentId = model;
+
+  // TODO: remove isDeprecatedEdition condition in V2.0
+  if (isDeprecatedEdition) {
+    const chatModelCards = modelProviderSelectors.getModelCardsById(ModelProvider.Azure)(
+      useUserStore.getState(),
+    );
+
+    const deploymentName = chatModelCards.find((i) => i.id === model)?.deploymentName;
+    if (deploymentName) deploymentId = deploymentName;
+  } else {
+    // find the model by id
+    const modelItem = getAiInfraStoreState().enabledAiModels?.find(
+      (i) => i.id === model && i.providerId === provider,
+    );
+
+    if (modelItem && modelItem.config?.deploymentName) {
+      deploymentId = modelItem.config?.deploymentName;
+    }
+  }
+
+  return deploymentId;
+};
+
+export const isEnableFetchOnClient = (provider: string) => {
+  // TODO: remove this condition in V2.0
+  if (isDeprecatedEdition) {
+    return modelConfigSelectors.isProviderFetchOnClient(provider)(useUserStore.getState());
+  } else {
+    return aiProviderSelectors.isProviderFetchOnClient(provider)(getAiInfraStoreState());
+  }
+};

package/src/services/{chat.ts → chat/index.ts}

@@ -2,22 +2,16 @@ import {
   AgentRuntimeError,
   ChatCompletionErrorPayload,
   ModelProvider,
-  ModelRuntime,
-  parseDataUri,
 } from '@lobechat/model-runtime';
-import { BuiltinSystemRolePrompts, filesPrompts } from '@lobechat/prompts';
 import { ChatErrorType, TracePayload, TraceTagMap } from '@lobechat/types';
 import { PluginRequestPayload, createHeadersWithPluginSettings } from '@lobehub/chat-plugin-sdk';
-import { produce } from 'immer';
 import { merge } from 'lodash-es';
 
 import { enableAuth } from '@/const/auth';
-import { INBOX_GUIDE_SYSTEMROLE } from '@/const/guide';
-import { INBOX_SESSION_ID } from '@/const/session';
 import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
-import { isDeprecatedEdition, isDesktop, isServerMode } from '@/const/version';
+import { isDeprecatedEdition, isDesktop } from '@/const/version';
 import { getAgentStoreState } from '@/store/agent';
-import { agentChatConfigSelectors } from '@/store/agent/selectors';
+import { agentChatConfigSelectors, agentSelectors } from '@/store/agent/selectors';
 import { aiModelSelectors, aiProviderSelectors, getAiInfraStoreState } from '@/store/aiInfra';
 import { getSessionStoreState } from '@/store/session';
 import { sessionMetaSelectors } from '@/store/session/selectors';
@@ -25,18 +19,14 @@ import { getToolStoreState } from '@/store/tool';
 import { pluginSelectors, toolSelectors } from '@/store/tool/selectors';
 import { getUserStoreState, useUserStore } from '@/store/user';
 import {
-  modelConfigSelectors,
-  modelProviderSelectors,
   preferenceSelectors,
   userGeneralSettingsSelectors,
   userProfileSelectors,
 } from '@/store/user/selectors';
 import { WebBrowsingManifest } from '@/tools/web-browsing';
 import { WorkingModel } from '@/types/agent';
-import { ChatImageItem, ChatMessage, MessageToolCall } from '@/types/message';
-import type { ChatStreamPayload, OpenAIChatMessage } from '@/types/openai/chat';
-import { UserMessageContentPart } from '@/types/openai/chat';
-import { parsePlaceholderVariablesMessages } from '@/utils/client/parserPlaceholder';
+import { ChatMessage } from '@/types/message';
+import type { ChatStreamPayload } from '@/types/openai/chat';
 import { fetchWithInvokeStream } from '@/utils/electron/desktopRemoteRPCFetch';
 import { createErrorResponse } from '@/utils/errorResponse';
 import {
@@ -45,74 +35,14 @@ import {
   getMessageError,
   standardizeAnimationStyle,
 } from '@/utils/fetch';
-import { imageUrlToBase64 } from '@/utils/imageToBase64';
-import { genToolCallingName } from '@/utils/toolCall';
 import { createTraceHeader, getTraceId } from '@/utils/trace';
-import { isLocalUrl } from '@/utils/url';
-
-import { createHeaderWithAuth, createPayloadWithKeyVaults } from './_auth';
-import { API_ENDPOINTS } from './_url';
-
-const isCanUseFC = (model: string, provider: string) => {
-  // TODO: remove isDeprecatedEdition condition in V2.0
-  if (isDeprecatedEdition) {
-    return modelProviderSelectors.isModelEnabledFunctionCall(model)(getUserStoreState());
-  }
-
-  return aiModelSelectors.isModelSupportToolUse(model, provider)(getAiInfraStoreState());
-};
-
-const isCanUseVision = (model: string, provider: string) => {
-  // TODO: remove isDeprecatedEdition condition in V2.0
-  if (isDeprecatedEdition) {
-    return modelProviderSelectors.isModelEnabledVision(model)(getUserStoreState());
-  }
-  return aiModelSelectors.isModelSupportVision(model, provider)(getAiInfraStoreState());
-};
-
-/**
- * TODO: we need to update this function to auto find deploymentName with provider setting config
- */
-const findDeploymentName = (model: string, provider: string) => {
-  let deploymentId = model;
-
-  // TODO: remove isDeprecatedEdition condition in V2.0
-  if (isDeprecatedEdition) {
-    const chatModelCards = modelProviderSelectors.getModelCardsById(ModelProvider.Azure)(
-      useUserStore.getState(),
-    );
-
-    const deploymentName = chatModelCards.find((i) => i.id === model)?.deploymentName;
-    if (deploymentName) deploymentId = deploymentName;
-  } else {
-    // find the model by id
-    const modelItem = getAiInfraStoreState().enabledAiModels?.find(
-      (i) => i.id === model && i.providerId === provider,
-    );
-
-    if (modelItem && modelItem.config?.deploymentName) {
-      deploymentId = modelItem.config?.deploymentName;
-    }
-  }
 
-  return deploymentId;
-};
-
-const isEnableFetchOnClient = (provider: string) => {
-  // TODO: remove this condition in V2.0
-  if (isDeprecatedEdition) {
-    return modelConfigSelectors.isProviderFetchOnClient(provider)(useUserStore.getState());
-  } else {
-    return aiProviderSelectors.isProviderFetchOnClient(provider)(getAiInfraStoreState());
-  }
-};
-
-interface FetchOptions extends FetchSSEOptions {
-  historySummary?: string;
-  isWelcomeQuestion?: boolean;
-  signal?: AbortSignal | undefined;
-  trace?: TracePayload;
-}
+import { createHeaderWithAuth } from '../_auth';
+import { API_ENDPOINTS } from '../_url';
+import { initializeWithClientStore } from './clientModelRuntime';
+import { contextEngineering } from './contextEngineering';
+import { findDeploymentName, isCanUseFC, isEnableFetchOnClient } from './helper';
+import { FetchOptions } from './types';
 
 interface GetChatCompletionPayload extends Partial<Omit<ChatStreamPayload, 'messages'>> {
   messages: ChatMessage[];
@@ -141,37 +71,6 @@ interface CreateAssistantMessageStream extends FetchSSEOptions {
   trace?: TracePayload;
 }
 
-/**
- * Initializes the AgentRuntime with the client store.
- * @param provider - The provider name.
- * @param payload - Init options
- * @returns The initialized AgentRuntime instance
- *
- * **Note**: if you try to fetch directly, use `fetchOnClient` instead.
- */
-export function initializeWithClientStore(provider: string, payload?: any) {
-  /**
-   * Since #5267, we map parameters for client-fetch in function `getProviderAuthPayload`
-   * which called by `createPayloadWithKeyVaults` below.
-   * @see https://github.com/lobehub/lobe-chat/pull/5267
-   * @file src/services/_auth.ts
-   */
-  const providerAuthPayload = { ...payload, ...createPayloadWithKeyVaults(provider) };
-  const commonOptions = {
-    // Allow OpenAI SDK and Anthropic SDK run on browser
-    dangerouslyAllowBrowser: true,
-  };
-  /**
-   * Configuration override order:
-   * payload -> providerAuthPayload -> commonOptions
-   */
-  return ModelRuntime.initializeWithProvider(provider, {
-    ...commonOptions,
-    ...providerAuthPayload,
-    ...payload,
-  });
-}
-
 class ChatService {
   createAssistantMessage = async (
     { plugins: enabledPlugins, messages, ...params }: GetChatCompletionPayload,
@@ -209,29 +108,35 @@ class ChatService {
       pluginIds.push(WebBrowsingManifest.identifier);
     }
 
-    // ============  1. preprocess placeholder variables   ============ //
-    const parsedMessages = parsePlaceholderVariablesMessages(messages);
+    // ============  1. preprocess messages   ============ //
 
-    // ============  2. preprocess messages   ============ //
+    const agentStoreState = getAgentStoreState();
+    const agentConfig = agentSelectors.currentAgentConfig(agentStoreState);
 
-    const oaiMessages = await this.processMessages(
+    // Apply context engineering with preprocessing configuration
+    const oaiMessages = await contextEngineering(
       {
-        messages: parsedMessages,
+        enableHistoryCount: agentChatConfigSelectors.enableHistoryCount(agentStoreState),
+        // include user messages
+        historyCount: agentChatConfigSelectors.historyCount(agentStoreState) + 2,
+        inputTemplate: chatConfig.inputTemplate,
+        messages,
         model: payload.model,
         provider: payload.provider!,
+        systemRole: agentConfig.systemRole,
         tools: pluginIds,
       },
       options,
     );
 
-    // ============  3. preprocess tools   ============ //
+    // ============  2. preprocess tools   ============ //
 
     const tools = this.prepareTools(pluginIds, {
       model: payload.model,
       provider: payload.provider!,
     });
 
-    // ============  4. process extend params   ============ //
+    // ============  3. process extend params   ============ //
 
     let extendParams: Record<string, any> = {};
 
@@ -513,7 +418,7 @@ class ChatService {
     onLoadingChange?.(true);
 
     try {
-      const oaiMessages = await this.processMessages({
+      const oaiMessages = await contextEngineering({
         messages: params.messages as any,
         model: params.model!,
         provider: params.provider!,
@@ -545,191 +450,6 @@ class ChatService {
     }
   };
 
-  private processMessages = async (
-    {
-      messages = [],
-      tools,
-      model,
-      provider,
-    }: {
-      messages: ChatMessage[];
-      model: string;
-      provider: string;
-      tools?: string[];
-    },
-    options?: FetchOptions,
-  ): Promise<OpenAIChatMessage[]> => {
-    // handle content type for vision model
-    // for the models with visual ability, add image url to content
-    // refs: https://platform.openai.com/docs/guides/vision/quick-start
-    const getUserContent = async (m: ChatMessage) => {
-      // only if message doesn't have images and files, then return the plain content
-      if ((!m.imageList || m.imageList.length === 0) && (!m.fileList || m.fileList.length === 0))
-        return m.content;
-
-      const imageList = m.imageList || [];
-      const imageContentParts = await this.processImageList({ imageList, model, provider });
-
-      const filesContext = isServerMode
-        ? filesPrompts({ addUrl: !isDesktop, fileList: m.fileList, imageList })
-        : '';
-      return [
-        { text: (m.content + '\n\n' + filesContext).trim(), type: 'text' },
-        ...imageContentParts,
-      ] as UserMessageContentPart[];
-    };
-
-    const getAssistantContent = async (m: ChatMessage) => {
-      // signature is a signal of anthropic thinking mode
-      const shouldIncludeThinking = m.reasoning && !!m.reasoning?.signature;
-
-      if (shouldIncludeThinking) {
-        return [
-          {
-            signature: m.reasoning!.signature,
-            thinking: m.reasoning!.content,
-            type: 'thinking',
-          },
-          { text: m.content, type: 'text' },
-        ] as UserMessageContentPart[];
-      }
-      // only if message doesn't have images and files, then return the plain content
-
-      if (m.imageList && m.imageList.length > 0) {
-        const imageContentParts = await this.processImageList({
-          imageList: m.imageList,
-          model,
-          provider,
-        });
-        return [
-          !!m.content ? { text: m.content, type: 'text' } : undefined,
-          ...imageContentParts,
-        ].filter(Boolean) as UserMessageContentPart[];
-      }
-
-      return m.content;
-    };
-
-    let postMessages = await Promise.all(
-      messages.map(async (m): Promise<OpenAIChatMessage> => {
-        const supportTools = isCanUseFC(model, provider);
-        switch (m.role) {
-          case 'user': {
-            return { content: await getUserContent(m), role: m.role };
-          }
-
-          case 'assistant': {
-            const content = await getAssistantContent(m);
-
-            if (!supportTools) {
-              return { content, role: m.role };
-            }
-
-            return {
-              content,
-              role: m.role,
-              tool_calls: m.tools?.map(
-                (tool): MessageToolCall => ({
-                  function: {
-                    arguments: tool.arguments,
-                    name: genToolCallingName(tool.identifier, tool.apiName, tool.type),
-                  },
-                  id: tool.id,
-                  type: 'function',
-                }),
-              ),
-            };
-          }
-
-          case 'tool': {
-            if (!supportTools) {
-              return { content: m.content, role: 'user' };
-            }
-
-            return {
-              content: m.content,
-              name: genToolCallingName(m.plugin!.identifier, m.plugin!.apiName, m.plugin?.type),
-              role: m.role,
-              tool_call_id: m.tool_call_id,
-            };
-          }
-
-          default: {
-            return { content: m.content, role: m.role as any };
-          }
-        }
-      }),
-    );
-
-    postMessages = produce(postMessages, (draft) => {
-      // if it's a welcome question, inject InboxGuide SystemRole
-      const inboxGuideSystemRole =
-        options?.isWelcomeQuestion &&
-        options?.trace?.sessionId === INBOX_SESSION_ID &&
-        INBOX_GUIDE_SYSTEMROLE;
-
-      // Inject Tool SystemRole
-      const hasTools = tools && tools?.length > 0;
-      const hasFC = hasTools && isCanUseFC(model, provider);
-      const toolsSystemRoles =
-        hasFC && toolSelectors.enabledSystemRoles(tools)(getToolStoreState());
-
-      const injectSystemRoles = BuiltinSystemRolePrompts({
-        historySummary: options?.historySummary,
-        plugins: toolsSystemRoles as string,
-        welcome: inboxGuideSystemRole as string,
-      });
-
-      if (!injectSystemRoles) return;
-
-      const systemMessage = draft.find((i) => i.role === 'system');
-
-      if (systemMessage) {
-        systemMessage.content = [systemMessage.content, injectSystemRoles]
-          .filter(Boolean)
-          .join('\n\n');
-      } else {
-        draft.unshift({
-          content: injectSystemRoles,
-          role: 'system',
-        });
-      }
-    });
-
-    return this.reorderToolMessages(postMessages);
-  };
-
-  /**
-   * Process imageList: convert local URLs to base64 and format as UserMessageContentPart
-   */
-  private processImageList = async ({
-    model,
-    provider,
-    imageList,
-  }: {
-    imageList: ChatImageItem[];
-    model: string;
-    provider: string;
-  }) => {
-    if (!isCanUseVision(model, provider)) {
-      return [];
-    }
-
-    return Promise.all(
-      imageList.map(async (image) => {
-        const { type } = parseDataUri(image.url);
-
-        let processedUrl = image.url;
-        if (type === 'url' && isLocalUrl(image.url)) {
-          const { base64, mimeType } = await imageUrlToBase64(image.url);
-          processedUrl = `data:${mimeType};base64,${base64}`;
-        }
-
-        return { image_url: { detail: 'auto', url: processedUrl }, type: 'image_url' } as const;
-      }),
-    );
-  };
-
   private mapTrace = (trace?: TracePayload, tag?: TraceTagMap): TracePayload => {
     const tags = sessionMetaSelectors.currentAgentMeta(getSessionStoreState()).tags || [];
 
@@ -768,68 +488,6 @@ class ChatService {
     return agentRuntime.chat(data, { signal: params.signal });
   };
 
-  /**
-   * Reorder tool messages to ensure that tool messages are displayed in the correct order.
-   * see https://github.com/lobehub/lobe-chat/pull/3155
-   */
-  private reorderToolMessages = (messages: OpenAIChatMessage[]): OpenAIChatMessage[] => {
-    // 1. 先收集所有 assistant 消息中的有效 tool_call_id
-    const validToolCallIds = new Set<string>();
-    messages.forEach((message) => {
-      if (message.role === 'assistant' && message.tool_calls) {
-        message.tool_calls.forEach((toolCall) => {
-          validToolCallIds.add(toolCall.id);
-        });
-      }
-    });
-
-    // 2. 收集所有有效的 tool 消息
-    const toolMessages: Record<string, OpenAIChatMessage> = {};
-    messages.forEach((message) => {
-      if (
-        message.role === 'tool' &&
-        message.tool_call_id &&
-        validToolCallIds.has(message.tool_call_id)
-      ) {
-        toolMessages[message.tool_call_id] = message;
-      }
-    });
-
-    // 3. 重新排序消息
-    const reorderedMessages: OpenAIChatMessage[] = [];
-    messages.forEach((message) => {
-      // 跳过无效的 tool 消息
-      if (
-        message.role === 'tool' &&
-        (!message.tool_call_id || !validToolCallIds.has(message.tool_call_id))
-      ) {
-        return;
-      }
-
-      // 检查是否已经添加过该 tool 消息
-      const hasPushed = reorderedMessages.some(
-        (m) => !!message.tool_call_id && m.tool_call_id === message.tool_call_id,
-      );
-
-      if (hasPushed) return;
-
-      reorderedMessages.push(message);
-
-      // 如果是 assistant 消息且有 tool_calls，添加对应的 tool 消息
-      if (message.role === 'assistant' && message.tool_calls) {
-        message.tool_calls.forEach((toolCall) => {
-          const correspondingToolMessage = toolMessages[toolCall.id];
-          if (correspondingToolMessage) {
-            reorderedMessages.push(correspondingToolMessage);
-            delete toolMessages[toolCall.id];
-          }
-        });
-      }
-    });
-
-    return reorderedMessages;
-  };
-
   private prepareTools = (pluginIds: string[], { model, provider }: WorkingModel) => {
     let filterTools = toolSelectors.enabledSchema(pluginIds)(getToolStoreState());
 

package/src/services/chat/types.ts

@@ -0,0 +1,9 @@
+import { TracePayload } from '@lobechat/types';
+import { FetchSSEOptions } from '@/utils/fetch';
+
+export interface FetchOptions extends FetchSSEOptions {
+  historySummary?: string;
+  isWelcomeQuestion?: boolean;
+  signal?: AbortSignal | undefined;
+  trace?: TracePayload;
+}

package/src/services/models.ts

@@ -7,7 +7,7 @@ import { ChatModelCard } from '@/types/llm';
 import { getMessageError } from '@/utils/fetch';
 
 import { API_ENDPOINTS } from './_url';
-import { initializeWithClientStore } from './chat';
+import { initializeWithClientStore } from './chat/clientModelRuntime';
 
 const isEnableFetchOnClient = (provider: string) => {
   // TODO: remove this condition in V2.0

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

@@ -47,7 +47,7 @@ const getEnabledModelById = (id: string, provider: string) => (s: AIProviderStor
 const isModelSupportToolUse = (id: string, provider: string) => (s: AIProviderStoreState) => {
   const model = getEnabledModelById(id, provider)(s);
 
-  return model?.abilities?.functionCall;
+  return model?.abilities?.functionCall || false;
 };
 
 const isModelSupportFiles = (id: string, provider: string) => (s: AIProviderStoreState) => {
@@ -59,7 +59,7 @@ const isModelSupportFiles = (id: string, provider: string) => (s: AIProviderStor
 const isModelSupportVision = (id: string, provider: string) => (s: AIProviderStoreState) => {
   const model = getEnabledModelById(id, provider)(s);
 
-  return model?.abilities?.vision;
+  return model?.abilities?.vision || false;
 };
 
 const isModelSupportReasoning = (id: string, provider: string) => (s: AIProviderStoreState) => {