@lobehub/chat/src/hooks/useAgentEnableSearch.ts

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.

import { useAgentStore } from '@/store/agent';
import { agentChatConfigSelectors, agentSelectors } from '@/store/agent/selectors';
import { aiModelSelectors, useAiInfraStore } from '@/store/aiInfra';

export const useAgentEnableSearch = () => {
  const [model, provider, agentSearchMode] = useAgentStore((s) => [
    agentSelectors.currentAgentModel(s),
    agentSelectors.currentAgentModelProvider(s),
    agentChatConfigSelectors.agentSearchMode(s),
  ]);

  const isModelSupportToolUse = useAiInfraStore(
    aiModelSelectors.isModelSupportToolUse(model, provider),
  );
  const searchImpl = useAiInfraStore(aiModelSelectors.modelBuiltinSearchImpl(model, provider));

  // 只要是内置的搜索实现，一定可以联网搜索
  if (searchImpl === 'internal') return true;

  // 如果是关闭状态，一定不能联网搜索
  if (agentSearchMode === 'off') return false;

  // 如果是智能模式，根据是否支持 Tool Calling 判断
  if (agentSearchMode === 'auto') {
    return isModelSupportToolUse;
  }
};