From 2a0d87a393d3d1e1167ea9c6bcbafc80c939ea70 Mon Sep 17 00:00:00 2001
From: Louis <louis@jan.ai>
Date: Wed, 30 Oct 2024 16:05:13 +0700
Subject: [PATCH] fix: inconsistent models from dropdown and hub

---
 web/containers/ModelDropdown/index.tsx              | 7 ++++++-
 web/screens/Settings/MyModels/MyModelList/index.tsx | 4 ++--
 web/screens/Settings/MyModels/index.tsx             | 7 ++++++-
 3 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/web/containers/ModelDropdown/index.tsx b/web/containers/ModelDropdown/index.tsx
index 4ff8edac5..a58febabf 100644
--- a/web/containers/ModelDropdown/index.tsx
+++ b/web/containers/ModelDropdown/index.tsx
@@ -509,7 +509,12 @@ const ModelDropdown = ({
 
                     <ul className="pb-2">
                       {filteredDownloadedModels
-                        .filter((x) => x.engine === engine)
+                        .filter(
+                          (x) =>
+                            x.engine === engine ||
+                            (x.engine === InferenceEngine.nitro &&
+                              engine === InferenceEngine.cortex_llamacpp)
+                        )
                         .filter((y) => {
                           if (isLocalEngine(y.engine) && !searchText.length) {
                             return downloadedModels.find((c) => c.id === y.id)
diff --git a/web/screens/Settings/MyModels/MyModelList/index.tsx b/web/screens/Settings/MyModels/MyModelList/index.tsx
index 756520107..26dd26b6c 100644
--- a/web/screens/Settings/MyModels/MyModelList/index.tsx
+++ b/web/screens/Settings/MyModels/MyModelList/index.tsx
@@ -54,14 +54,14 @@ const MyModelList = ({ model }: Props) => {
             <h6
               className={twMerge(
                 'font-medium lg:line-clamp-1 lg:min-w-[280px] lg:max-w-[280px]',
-                model.engine !== InferenceEngine.nitro &&
+                !isLocalEngine(model.engine) &&
                   'max-w-none text-[hsla(var(--text-secondary))]'
               )}
               title={model.name}
             >
               {model.name}
             </h6>
-            {model.engine === InferenceEngine.nitro && (
+            {!isLocalEngine(model.engine) && (
               <div className="flex gap-x-8">
                 <p
                   className="line-clamp-1 text-[hsla(var(--text-secondary))] lg:min-w-[160px] lg:max-w-[160px] xl:max-w-none"
diff --git a/web/screens/Settings/MyModels/index.tsx b/web/screens/Settings/MyModels/index.tsx
index ba42d12c2..218f8cb62 100644
--- a/web/screens/Settings/MyModels/index.tsx
+++ b/web/screens/Settings/MyModels/index.tsx
@@ -250,7 +250,12 @@ const MyModels = () => {
                     <div className="mt-2">
                       {filteredDownloadedModels
                         ? filteredDownloadedModels
-                            .filter((x) => x.engine === engine)
+                            .filter(
+                              (x) =>
+                                x.engine === engine ||
+                                (x.engine === InferenceEngine.nitro &&
+                                  engine === InferenceEngine.cortex_llamacpp)
+                            )
                             .map((model) => {
                               if (!showModel) return null
                               return (