@lobehub/chat/src/hooks/useModelSupportVision.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 { isDeprecatedEdition } from '@/const/version';
import { aiModelSelectors, useAiInfraStore } from '@/store/aiInfra';
import { useUserStore } from '@/store/user';
import { modelProviderSelectors } from '@/store/user/selectors';

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));
  if (isDeprecatedEdition) return oldValue;
  //

  return newValue;
};