From 8129c2319fadf80ddf34a3c2ea6ef65e619a207b Mon Sep 17 00:00:00 2001
From: Louis <louis@jan.ai>
Date: Mon, 21 Oct 2024 17:11:25 +0700
Subject: [PATCH] fix: downloaded models should account for remote models

---
 web/hooks/useModels.ts | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/web/hooks/useModels.ts b/web/hooks/useModels.ts
index 1cbd970d6..b09839457 100644
--- a/web/hooks/useModels.ts
+++ b/web/hooks/useModels.ts
@@ -11,6 +11,8 @@ import {
 
 import { useSetAtom } from 'jotai'
 
+import { isLocalEngine } from '@/utils/modelEngine'
+
 import { extensionManager } from '@/extension'
 import {
   configuredModelsAtom,
@@ -28,8 +30,12 @@ const useModels = () => {
 
   const getData = useCallback(() => {
     const getDownloadedModels = async () => {
-      const models = await getModels()
-      setDownloadedModels(models)
+      const localModels = await getModels()
+      const remoteModels = ModelManager.instance()
+        .models.values()
+        .toArray()
+        .filter((e) => !isLocalEngine(e.engine))
+      setDownloadedModels([...localModels, ...remoteModels])
     }
 
     const getExtensionModels = async () => {