feat: integrate fuzzy search into model dropdown (#5197)

* feat: integrate fuzzy search into model dropdown

- Replace DropdownMenu with Popover for better search UX
- Include search input with clear functionality
- Reorganize layout with capabilities at end of row
- Maintain provider grouping and model selection functionality

Improves model discovery and selection with instant search across
model names, providers, and capabilities.

* chore: enhance input search style

* feat: enhance model dropdown with search highlighting and fixed positioning

- Add FZF search highlighting with text-accent color for matched characters
- Fix dropdown to only appear below (prevent upward positioning)
- Import highlightFzfMatch utility for search result highlighting
- Update SearchableModel interface to include highlightedId property
- Modify FZF selector to target model.id for more accurate highlighting
- Use dangerouslySetInnerHTML to render highlighted search matches
- Add avoidCollisions=false to PopoverContent for consistent positioning

---------

Co-authored-by: Faisal Amir <urmauur@gmail.com>

web-app/src/containers/DropdownModelProvider.tsx

@@ -1,26 +1,33 @@
+import { useEffect, useState, useRef, useMemo, useCallback } from 'react'
 import {
-  DropdownMenu,
+  Popover,
-  DropdownMenuContent,
+  PopoverContent,
-  DropdownMenuGroup,
+  PopoverTrigger,
-  DropdownMenuItem,
+} from '@/components/ui/popover'
-  DropdownMenuLabel,
-  DropdownMenuTrigger,
-} from '@/components/ui/dropdown-menu'
 import { useModelProvider } from '@/hooks/useModelProvider'
 import { cn, getProviderTitle } from '@/lib/utils'
-import { useEffect, useState } from 'react'
+import { highlightFzfMatch } from '@/utils/highlight'
 import Capabilities from './Capabilities'
-import { IconSettings } from '@tabler/icons-react'
+import { IconSettings, IconX, IconCheck } from '@tabler/icons-react'
 import { useNavigate } from '@tanstack/react-router'
 import { route } from '@/constants/routes'
 import { useThreads } from '@/hooks/useThreads'
 import { ModelSetting } from '@/containers/ModelSetting'
 import ProvidersAvatar from '@/containers/ProvidersAvatar'
+import { Fzf } from 'fzf'
 
 type DropdownModelProviderProps = {
   model?: ThreadModel
 }
 
+interface SearchableModel {
+  provider: ModelProvider
+  model: Model
+  searchStr: string
+  value: string
+  highlightedId?: string
+}
+
 const DropdownModelProvider = ({ model }: DropdownModelProviderProps) => {
   const {
     providers,
@@ -34,6 +41,11 @@ const DropdownModelProvider = ({ model }: DropdownModelProviderProps) => {
   const { updateCurrentThreadModel } = useThreads()
   const navigate = useNavigate()
 
+  // Search state
+  const [open, setOpen] = useState(false)
+  const [searchValue, setSearchValue] = useState('')
+  const searchInputRef = useRef<HTMLInputElement>(null)
+
   // Initialize model provider only once
   useEffect(() => {
     // Auto select model when existing thread is passed
@@ -43,7 +55,7 @@ const DropdownModelProvider = ({ model }: DropdownModelProviderProps) => {
       // default model, we should add from setting
       selectModelProvider('llama.cpp', 'llama3.2:3b')
     }
-  }, [model, selectModelProvider, updateCurrentThreadModel]) // Only run when threadData changes
+  }, [model, selectModelProvider, updateCurrentThreadModel])
 
   // Update display model when selection changes
   useEffect(() => {
@@ -54,6 +66,116 @@ const DropdownModelProvider = ({ model }: DropdownModelProviderProps) => {
     }
   }, [selectedProvider, selectedModel])
 
+  // Reset search value when dropdown closes
+  const onOpenChange = useCallback((open: boolean) => {
+    setOpen(open)
+    if (!open) {
+      requestAnimationFrame(() => setSearchValue(''))
+    } else {
+      // Focus search input when opening
+      setTimeout(() => {
+        searchInputRef.current?.focus()
+      }, 100)
+    }
+  }, [])
+
+  // Clear search and focus input
+  const onClearSearch = useCallback(() => {
+    setSearchValue('')
+    searchInputRef.current?.focus()
+  }, [])
+
+  // Create searchable items from all models
+  const searchableItems = useMemo(() => {
+    const items: SearchableModel[] = []
+
+    providers.forEach((provider) => {
+      if (!provider.active) return
+
+      provider.models.forEach((modelItem) => {
+        // Skip models that require API key but don't have one (except llama.cpp)
+        if (provider.provider !== 'llama.cpp' && !provider.api_key?.length) {
+          return
+        }
+
+        const capabilities = modelItem.capabilities || []
+        const capabilitiesString = capabilities.join(' ')
+        const providerTitle = getProviderTitle(provider.provider)
+
+        // Create search string with model id, provider, and capabilities
+        const searchStr =
+          `${modelItem.id} ${providerTitle} ${provider.provider} ${capabilitiesString}`.toLowerCase()
+
+        items.push({
+          provider,
+          model: modelItem,
+          searchStr,
+          value: `${provider.provider}:${modelItem.id}`,
+        })
+      })
+    })
+
+    return items
+  }, [providers])
+
+  // Create Fzf instance for fuzzy search
+  const fzfInstance = useMemo(() => {
+    return new Fzf(searchableItems, {
+      selector: (item) => item.model.id,
+    })
+  }, [searchableItems])
+
+  // Filter models based on search value
+  const filteredItems = useMemo(() => {
+    if (!searchValue) return searchableItems
+
+    return fzfInstance.find(searchValue).map((result) => {
+      const item = result.item
+      const positions = Array.from(result.positions) || []
+      const highlightedId = highlightFzfMatch(
+        item.model.id,
+        positions,
+        'text-accent'
+      )
+
+      return {
+        ...item,
+        highlightedId,
+      }
+    })
+  }, [searchableItems, searchValue, fzfInstance])
+
+  // Group filtered items by provider
+  const groupedItems = useMemo(() => {
+    const groups: Record<string, SearchableModel[]> = {}
+
+    filteredItems.forEach((item) => {
+      const providerKey = item.provider.provider
+      if (!groups[providerKey]) {
+        groups[providerKey] = []
+      }
+      groups[providerKey].push(item)
+    })
+
+    return groups
+  }, [filteredItems])
+
+  const handleSelect = useCallback(
+    (searchableModel: SearchableModel) => {
+      selectModelProvider(
+        searchableModel.provider.provider,
+        searchableModel.model.id
+      )
+      updateCurrentThreadModel({
+        id: searchableModel.model.id,
+        provider: searchableModel.provider.provider,
+      })
+      setSearchValue('')
+      setOpen(false)
+    },
+    [selectModelProvider, updateCurrentThreadModel]
+  )
+
   const currentModel = selectedModel?.id
     ? getModelBy(selectedModel?.id)
     : undefined
@@ -63,10 +185,9 @@ const DropdownModelProvider = ({ model }: DropdownModelProviderProps) => {
   const provider = getProviderByName(selectedProvider)
 
   return (
-    <>
+    <Popover open={open} onOpenChange={onOpenChange}>
-      <DropdownMenu>
       <div className="bg-main-view-fg/5 hover:bg-main-view-fg/8 px-2 py-1 flex items-center gap-1.5 rounded-sm max-h-[32px]">
-          <DropdownMenuTrigger asChild>
+        <PopoverTrigger asChild>
           <button
             title={displayModel}
             className="font-medium cursor-pointer flex items-center gap-1.5 relative z-20 max-w-38"
@@ -85,92 +206,138 @@ const DropdownModelProvider = ({ model }: DropdownModelProviderProps) => {
               {displayModel}
             </span>
           </button>
-          </DropdownMenuTrigger>
+        </PopoverTrigger>
         {currentModel?.settings && provider && (
           <ModelSetting model={currentModel as Model} provider={provider} />
         )}
       </div>
-        <DropdownMenuContent
+
-          className="w-60 max-h-[320px]"
+      <PopoverContent
+        className="w-80 p-0 max-h-[400px] overflow-hidden"
         side="bottom"
         align="start"
         sideOffset={10}
         alignOffset={-8}
+        avoidCollisions={false}
       >
-          <DropdownMenuGroup>
+        <div className="flex flex-col w-full">
-            {providers.map((provider, index) => {
+          {/* Search input */}
-              // Only show active providers
+          <div className="relative px-2 py-1.5 border-b border-main-view-fg/10">
-              if (!provider.active) return null
+            <input
+              ref={searchInputRef}
+              value={searchValue}
+              onChange={(e) => setSearchValue(e.target.value)}
+              placeholder="Search models..."
+              className="text-sm font-normal outline-0"
+            />
+            {searchValue.length > 0 && (
+              <div className="absolute right-2 top-0 bottom-0 flex items-center justify-center">
+                <IconX
+                  size={16}
+                  className="text-main-view-fg/50 hover:text-main-view-fg cursor-pointer"
+                  onClick={onClearSearch}
+                />
+              </div>
+            )}
+          </div>
+
+          {/* Model list */}
+          <div className="max-h-[320px] overflow-y-auto">
+            {Object.keys(groupedItems).length === 0 && searchValue ? (
+              <div className="py-3 px-4 text-sm text-main-view-fg/60">
+                No models found for "{searchValue}"
+              </div>
+            ) : (
+              <div className="py-1">
+                {Object.entries(groupedItems).map(([providerKey, models]) => {
+                  const providerInfo = providers.find(
+                    (p) => p.provider === providerKey
+                  )
+                  if (!providerInfo) return null
 
                   return (
                     <div
-                  className={cn(
+                      key={providerKey}
-                    'bg-main-view-fg/4 first:mt-0 rounded-sm my-1.5 first:mb-0 '
+                      className="bg-main-view-fg/4 first:mt-0 rounded-sm my-1.5 mx-1.5 first:mb-0"
-                  )}
-                  key={`provider-${index}`}
                     >
-                  <div className="flex items-center justify-between">
+                      {/* Provider header */}
-                    <DropdownMenuLabel className="flex items-center gap-1.5">
+                      <div className="flex items-center justify-between px-2 py-1">
-                      <ProvidersAvatar provider={provider} />
+                        <div className="flex items-center gap-1.5">
-                      <span className="capitalize truncate text-sm">
+                          <ProvidersAvatar provider={providerInfo} />
-                        {getProviderTitle(provider.provider)}
+                          <span className="capitalize truncate text-sm font-medium text-main-view-fg/80">
+                            {getProviderTitle(providerInfo.provider)}
                           </span>
-                    </DropdownMenuLabel>
+                        </div>
                         <div
-                      className="size-6 cursor-pointer flex items-center justify-center rounded hover:bg-main-view-fg/10 transition-all duration-200 ease-in-out mr-2"
+                          className="size-6 cursor-pointer flex items-center justify-center rounded hover:bg-main-view-fg/10 transition-all duration-200 ease-in-out"
-                      onClick={() =>
+                          onClick={(e) => {
+                            e.stopPropagation()
                             navigate({
                               to: route.settings.providers,
-                          params: { providerName: provider.provider },
+                              params: { providerName: providerInfo.provider },
                             })
-                      }
+                            setOpen(false)
+                          }}
                         >
                           <IconSettings
-                        size={18}
+                            size={16}
                             className="text-main-view-fg/50"
                           />
                         </div>
                       </div>
 
-                  {provider.models.map((model, modelIndex) => {
+                      {/* Models for this provider */}
-                    const capabilities = model.capabilities || []
+                      {models.map((searchableModel) => {
+                        const isSelected =
+                          selectedModel?.id === searchableModel.model.id &&
+                          selectedProvider === searchableModel.provider.provider
+                        const capabilities =
+                          searchableModel.model.capabilities || []
 
                         return (
-                      <DropdownMenuItem
+                          <div
+                            key={searchableModel.value}
+                            onClick={() => handleSelect(searchableModel)}
                             className={cn(
-                          'h-8 mx-1',
+                              'mx-1 mb-1 px-2 py-1.5 rounded cursor-pointer flex items-center gap-2 transition-all duration-200',
-                          provider.provider !== 'llama.cpp' &&
+                              'hover:bg-main-view-fg/10',
-                            !provider.api_key?.length &&
+                              isSelected && 'bg-main-view-fg/15'
-                            'hidden'
                             )}
-                        title={model.id}
-                        key={`model-${modelIndex}`}
-                        onClick={() => {
-                          selectModelProvider(provider.provider, model.id)
-                          updateCurrentThreadModel({
-                            id: model.id,
-                            provider: provider.provider,
-                          })
-                        }}
                           >
-                        <div className="flex items-center gap-1.5 w-full">
+                            <div className="flex items-center gap-2 flex-1 min-w-0">
-                          <span className="truncate text-main-view-fg/70">
+                              <span
-                            {model.id}
+                                className="truncate text-main-view-fg/80 text-sm"
-                          </span>
+                                dangerouslySetInnerHTML={{
-                          <div className="-mr-1.5">
+                                  __html:
+                                    searchableModel.highlightedId ||
+                                    searchableModel.model.id,
+                                }}
+                              />
+                              {isSelected && (
+                                <IconCheck
+                                  size={16}
+                                  className="text-accent shrink-0"
+                                />
+                              )}
+                              <div className="flex-1"></div>
+                              {capabilities.length > 0 && (
+                                <div className="flex-shrink-0">
                                   <Capabilities capabilities={capabilities} />
                                 </div>
+                              )}
+                            </div>
                           </div>
-                      </DropdownMenuItem>
                         )
                       })}
                     </div>
                   )
                 })}
-          </DropdownMenuGroup>
+              </div>
-        </DropdownMenuContent>
+            )}
-      </DropdownMenu>
+          </div>
-    </>
+        </div>
+      </PopoverContent>
+    </Popover>
   )
 }