From 5c185d2740a8baa91cc732f0822b9ca31b2b95e6 Mon Sep 17 00:00:00 2001
From: NamH <NamNh0122@gmail.com>
Date: Thu, 29 Feb 2024 14:35:05 +0700
Subject: [PATCH] fix: download model error does not reset state in model hub
 (#2199)

Signed-off-by: James <james@jan.ai>
Co-authored-by: James <james@jan.ai>
---
 core/src/node/api/processors/download.ts | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/core/src/node/api/processors/download.ts b/core/src/node/api/processors/download.ts
index bff6f47f0..4ddeff160 100644
--- a/core/src/node/api/processors/download.ts
+++ b/core/src/node/api/processors/download.ts
@@ -42,6 +42,24 @@ export class Downloader implements Processor {
     // Downloading file to a temp file first
     const downloadingTempFile = `${destination}.download`
 
+    // adding initial download state
+    const initialDownloadState: DownloadState = {
+      modelId,
+      fileName,
+      time: {
+        elapsed: 0,
+        remaining: 0,
+      },
+      speed: 0,
+      percent: 0,
+      size: {
+        total: 0,
+        transferred: 0,
+      },
+      downloadState: 'downloading',
+    }
+    DownloadManager.instance.downloadProgressMap[modelId] = initialDownloadState
+
     progress(rq, {})
       .on('progress', (state: any) => {
         const downloadState: DownloadState = {