fix: cortex stop child process

engine/controllers/process_manager.cc

@@ -1,12 +1,21 @@
 #include "process_manager.h"
-#include "utils/cortex_utils.h"
-
 #include <trantor/utils/Logger.h>
 #include <cstdlib>
+#include "json/json.h"
+#include "utils/cortex_utils.h"
 
 void ProcessManager::destroy(
     const HttpRequestPtr& req,
     std::function<void(const HttpResponsePtr&)>&& callback) {
+  auto res = inference_service_->GetModels(std::make_shared<Json::Value>());
+  // Check if there are any running models
+  auto running_models = std::get<1>(res);
+  if (!running_models.get("data", Json::Value()).empty()) {
+    for (const auto& model : running_models["data"]) {
+      std::string model_id = model["id"].asString();
+      model_service_->StopModel(model_id);
+    }
+  }
 
   app().quit();
   Json::Value ret;

engine/controllers/process_manager.h

@@ -2,6 +2,8 @@
 
 #include <drogon/HttpController.h>
 #include <drogon/HttpTypes.h>
+#include "services/inference_service.h"
+#include "services/model_service.h"
 
 using namespace drogon;
 
@@ -13,4 +15,12 @@ class ProcessManager : public drogon::HttpController<ProcessManager, false> {
 
   void destroy(const HttpRequestPtr& req,
                std::function<void(const HttpResponsePtr&)>&& callback);
+
+  ProcessManager(std::shared_ptr<InferenceService> inference_service,
+                 std::shared_ptr<ModelService> model_service)
+      : inference_service_(inference_service), model_service_(model_service) {}
+
+ private:
+  std::shared_ptr<InferenceService> inference_service_;
+  std::shared_ptr<ModelService> model_service_;
 };

engine/main.cc

@@ -182,7 +182,7 @@ void RunServer(std::optional<std::string> host, std::optional<int> port,
   auto model_ctl = std::make_shared<Models>(db_service, model_service,
                                             engine_service, model_src_svc);
   auto event_ctl = std::make_shared<Events>(event_queue_ptr);
-  auto pm_ctl = std::make_shared<ProcessManager>();
+  auto pm_ctl = std::make_shared<ProcessManager>(inference_svc, model_service);
   auto hw_ctl = std::make_shared<Hardware>(engine_service, hw_service);
   auto server_ctl =
       std::make_shared<inferences::server>(inference_svc, engine_service);

engine/services/model_service.cc

@@ -582,8 +582,9 @@ ModelService::DownloadModelFromCortexsoAsync(
           pyvenv_cfg.close();
           // Add executable permission to python
 
-          set_permission_utils::SetExecutePermissionsRecursive(
-              venv_path );
+
+          set_permission_utils::SetExecutePermissionsRecursive(venv_path);
+
 
         } else {
           CTL_ERR("Failed to extract venv.zip");