From e12549b9b8617bc46b5788ee73320d6b353f3763 Mon Sep 17 00:00:00 2001
From: Faisal Amir <urmauur@gmail.com>
Date: Fri, 12 Jul 2024 09:06:01 +0700
Subject: [PATCH] fix: model dropdown search result always show downloaded
 models first (#3147)

---
 web/containers/ModelDropdown/index.tsx | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/web/containers/ModelDropdown/index.tsx b/web/containers/ModelDropdown/index.tsx
index 3d72df615..c19fb64bd 100644
--- a/web/containers/ModelDropdown/index.tsx
+++ b/web/containers/ModelDropdown/index.tsx
@@ -104,8 +104,23 @@ const ModelDropdown = ({
             )
           }
         })
-        .sort((a, b) => a.name.localeCompare(b.name)),
-    [configuredModels, searchText, searchFilter]
+        .sort((a, b) => a.name.localeCompare(b.name))
+        .sort((a, b) => {
+          const aInDownloadedModels = downloadedModels.some(
+            (item) => item.id === a.id
+          )
+          const bInDownloadedModels = downloadedModels.some(
+            (item) => item.id === b.id
+          )
+          if (aInDownloadedModels && !bInDownloadedModels) {
+            return -1
+          } else if (!aInDownloadedModels && bInDownloadedModels) {
+            return 1
+          } else {
+            return 0
+          }
+        }),
+    [configuredModels, searchText, searchFilter, downloadedModels]
   )
 
   useEffect(() => {