fix: search recommended model

TJX2014 · Dec 19, 2023 · e71070a · e71070a
1 parent c1fef08
commit e71070a
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/web/screens/ExploreModels/index.tsx b/web/screens/ExploreModels/index.tsx
@@ -3,10 +3,6 @@ import { useState } from 'react'
 import {
   Input,
   ScrollArea,
-  Tooltip,
-  TooltipContent,
-  TooltipTrigger,
-  TooltipArrow,
   Select,
   SelectTrigger,
   SelectValue,
@@ -17,10 +13,6 @@ import {
 
 import { SearchIcon } from 'lucide-react'
 
-import { Code2Icon, UserIcon } from 'lucide-react'
-
-import { twMerge } from 'tailwind-merge'
-
 import Loader from '@/containers/Loader'
 
 import { useGetConfiguredModels } from '@/hooks/useGetConfiguredModels'
@@ -32,7 +24,6 @@ import ExploreModelList from './ExploreModelList'
 const ExploreModelsScreen = () => {
   const { loading, models } = useGetConfiguredModels()
   const [searchValue, setsearchValue] = useState('')
-  const [tabActive, setTabActive] = useState('Model')
   const { downloadedModels } = useGetDownloadedModels()
   const [sortSelected, setSortSelected] = useState('All Models')
   const sortMenu = ['All Models', 'Recommended', 'Downloaded']
@@ -45,8 +36,8 @@ const ExploreModelsScreen = () => {
       )
     } else if (sortSelected === 'Recommended') {
       return (
-        x.metadata.tags.includes('Featured') ||
-        x.metadata.tags.includes('Recommended')
+        x.name.toLowerCase().includes(searchValue.toLowerCase()) &&
+        x.metadata.tags.includes('Featured')
       )
     } else {
       return x.name.toLowerCase().includes(searchValue.toLowerCase())