diff --git a/.husky/pre-commit b/.husky/pre-commit index 53c4e577e..94c03b512 100644 --- a/.husky/pre-commit +++ b/.husky/pre-commit @@ -1 +1 @@ -npx oxlint@latest --fix \ No newline at end of file +yarn lint --fix --quiet \ No newline at end of file diff --git a/web/containers/ModelDropdown/index.tsx b/web/containers/ModelDropdown/index.tsx index a702d12f7..6d58b9893 100644 --- a/web/containers/ModelDropdown/index.tsx +++ b/web/containers/ModelDropdown/index.tsx @@ -35,6 +35,7 @@ import useDownloadModel from '@/hooks/useDownloadModel' import { modelDownloadStateAtom } from '@/hooks/useDownloadState' import { useGetEngines } from '@/hooks/useEngineManagement' +import { useGetModelSources } from '@/hooks/useModelSource' import useRecommendedModel from '@/hooks/useRecommendedModel' import useUpdateModelParameters from '@/hooks/useUpdateModelParameters' @@ -44,6 +45,8 @@ import { formatDownloadPercentage, toGigabytes } from '@/utils/converter' import { manualRecommendationModel } from '@/utils/model' import { getLogoEngine, getTitleByEngine } from '@/utils/modelEngine' +import { extractModelName } from '@/utils/modelSource' + import { activeAssistantAtom } from '@/helpers/atoms/Assistant.atom' import { configuredModelsAtom, @@ -84,6 +87,7 @@ const ModelDropdown = ({ const [toggle, setToggle] = useState(null) const [selectedModel, setSelectedModel] = useAtom(selectedModelAtom) const { recommendedModel, downloadedModels } = useRecommendedModel() + const { sources } = useGetModelSources() const [dropdownOptions, setDropdownOptions] = useState( null ) @@ -97,11 +101,8 @@ const ModelDropdown = ({ const configuredModels = useAtomValue(configuredModelsAtom) const { stopModel } = useActiveModel() - const featuredModels = configuredModels.filter( - (x) => - manualRecommendationModel.includes(x.id) && - x.metadata?.tags?.includes('Featured') && - x.metadata?.size < 5000000000 + const featuredModels = sources?.filter((x) => + manualRecommendationModel.includes(x.id) ) const { updateThreadMetadata } = useCreateNewThread() @@ -464,9 +465,9 @@ const ModelDropdown = ({ showModel && !searchText.length && (