chore: update model's size on import

janhq · Nov 4, 2024 · 10ce225 · 10ce225
1 parent 55bbe0d
commit 10ce225
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/engine/controllers/models.cc b/engine/controllers/models.cc
@@ -1,5 +1,6 @@
 #include "database/models.h"
 #include <drogon/HttpTypes.h>
+#include <cstdint>
 #include <filesystem>
 #include <optional>
 #include "config/gguf_parser.h"
@@ -351,8 +352,12 @@ void Models::ImportModel(
           modelPath, file_path,
           std::filesystem::copy_options::update_existing);
       model_config.files.push_back(file_path.string());
+      auto size = std::filesystem::file_size(file_path);
+      model_config.size = size;
     } else {
       model_config.files.push_back(modelPath);
+      auto size = std::filesystem::file_size(modelPath);
+      model_config.size = size;
     }
     model_config.model = modelHandle;
     model_config.name = modelName.empty() ? model_config.name : modelName;

diff --git a/engine/services/download_service.cc b/engine/services/download_service.cc
@@ -1,4 +1,3 @@
-#include "download_service.h"
 #include <curl/curl.h>
 #include <httplib.h>
 #include <stdio.h>