chore: filter out engine download

janhq · Dec 26, 2024 · 5d708a5 · 5d708a5
1 parent 02b6809
commit 5d708a5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/extensions/inference-cortex-extension/src/index.ts b/extensions/inference-cortex-extension/src/index.ts
@@ -284,6 +284,7 @@ export default class JanInferenceCortexExtension extends LocalOAIEngine {
                   transferred: transferred,
                   total: total,
                 },
+                downloadType: data.task.type,
               }
             )
 

diff --git a/web/containers/Providers/EventListener.tsx b/web/containers/Providers/EventListener.tsx
@@ -96,7 +96,10 @@ const EventListener = () => {
   const onFileDownloadSuccess = useCallback(
     async (state: DownloadState) => {
       console.debug('onFileDownloadSuccess', state)
-      if (state.downloadType !== 'extension') {
+      if (
+        state.downloadType !== 'extension' &&
+        state.downloadType !== 'Engine'
+      ) {
         // Update model metadata accordingly
         const model = ModelManager.instance().models.get(state.modelId)
         if (model) {
-Original file line number
+Diff line change
@@ Expand Up @@
                       transferred: transferred,
                       total: total,
                     },
+                    downloadType: data.task.type,
                   }
                 )
@@ Expand Down @@