update

Signed-off-by: James <namnh0122@gmail.com>

Author: namchuai

engine/commands/engine_get_cmd.cc

@@ -6,22 +6,17 @@
 
 namespace commands {
 
-void EngineGetCmd::Exec() const {
-  CTL_INF("[EngineGetCmd] engine: " << engine_);
-
-  auto engine_service = EngineService();
-  try {
-    auto status = engine_service.GetEngineInfo(engine_);
-    tabulate::Table table;
-    table.add_row({"Name", "Supported Formats", "Version", "Status"});
-    table.add_row(
-        {status.product_name, status.format, status.version, status.status});
-    std::cout << table << std::endl;
-  } catch (const std::runtime_error& e) {
-    std::cerr << "Engine " << engine_ << " is not supported!" << "\n";
-  } catch (const std::exception& e) {
-    std::cerr << "Failed to get engine info for " << engine_ << ": " << e.what()
-              << "\n";
+void EngineGetCmd::Exec(const std::string& engine_name) const {
+  auto engine = engine_service_.GetEngineInfo(engine_name);
+  if (engine == std::nullopt) {
+    CLI_LOG("Engine " + engine_name + " is not supported!");
+    return;
   }
+
+  tabulate::Table table;
+  table.add_row({"Name", "Supported Formats", "Version", "Status"});
+  table.add_row(
+      {engine->product_name, engine->format, engine->version, engine->status});
+  std::cout << table << std::endl;
 }
 };  // namespace commands

engine/commands/engine_get_cmd.h

@@ -1,15 +1,15 @@
 #pragma once
 
-#include <string>
+#include "services/engine_service.h"
 
 namespace commands {
 class EngineGetCmd {
  public:
-  EngineGetCmd(const std::string& engine) : engine_{engine} {};
+  explicit EngineGetCmd() : engine_service_{EngineService()} {};
 
-  void Exec() const;
+  void Exec(const std::string& engineName) const;
 
  private:
-  std::string engine_;
+  EngineService engine_service_;
 };
 }  // namespace commands

engine/commands/engine_init_cmd.cc

@@ -0,0 +1,17 @@
+#include "engine_install_cmd.h"
+// clang-format off
+#include "utils/cortexso_parser.h" 
+#include "utils/archive_utils.h"
+// clang-format on
+#include "utils/cuda_toolkit_utils.h"
+#include "utils/engine_matcher_utils.h"
+#include "utils/logging_utils.h"
+
+namespace commands {
+
+void EngineInstallCmd::Exec(const std::string& engine,
+                            const std::string& version) {
+  engine_service_.InstallEngine(engine, version);
+  CLI_LOG("Engine " << engine << " installed successfully!");
+}
+};  // namespace commands

engine/commands/engine_init_cmd.h

@@ -0,0 +1,17 @@
+#pragma once
+
+#include <string>
+#include "services/engine_service.h"
+
+namespace commands {
+
+class EngineInstallCmd {
+ public:
+  explicit EngineInstallCmd() : engine_service_{EngineService()} {};
+
+  void Exec(const std::string& engine, const std::string& version = "latest");
+
+ private:
+  EngineService engine_service_;
+};
+}  // namespace commands

engine/commands/engine_install_cmd.cc

@@ -4,18 +4,8 @@
 
 namespace commands {
 
-EngineUninstallCmd::EngineUninstallCmd(std::string engine)
-    : engine_{std::move(engine)} {}
-
-void EngineUninstallCmd::Exec() const {
-  CTL_INF("Uninstall engine " + engine_);
-  auto engine_service = EngineService();
-
-  try {
-    engine_service.UninstallEngine(engine_);
-    CLI_LOG("Engine " << engine_ << " uninstalled successfully!")
-  } catch (const std::exception& e) {
-    CLI_LOG("Failed to uninstall engine " << engine_ << ": " << e.what());
-  }
+void EngineUninstallCmd::Exec(const std::string& engine) {
+  engine_service_.UninstallEngine(engine);
+  CLI_LOG("Engine " << engine << " uninstalled successfully!");
 }
 };  // namespace commands

engine/commands/engine_install_cmd.h

@@ -1,15 +1,16 @@
 #pragma once
 
 #include <string>
+#include "services/engine_service.h"
 
 namespace commands {
 class EngineUninstallCmd {
  public:
-  EngineUninstallCmd(std::string engine);
+  explicit EngineUninstallCmd() : engine_service_{EngineService()} {};
 
-  void Exec() const;
+  void Exec(const std::string& engine);
 
  private:
-  std::string engine_;
+  EngineService engine_service_;
 };
 }  // namespace commands

engine/commands/engine_uninstall_cmd.cc

@@ -1,9 +1,8 @@
 #include "model_pull_cmd.h"
-#include <iostream>
 #include "utils/cortexso_parser.h"
 
 namespace commands {
-bool ModelPullCmd::Exec(const std::string& input) {
-  return model_service_.DownloadModel(input);
+void ModelPullCmd::Exec(const std::string& input) {
+  model_service_.DownloadModel(input);
 }
 };  // namespace commands

engine/commands/engine_uninstall_cmd.h

@@ -6,8 +6,8 @@ namespace commands {
 
 class ModelPullCmd {
  public:
-  ModelPullCmd() : model_service_{ModelService()} {};
-  bool Exec(const std::string& input);
+  explicit ModelPullCmd() : model_service_{ModelService()} {};
+  void Exec(const std::string& input);
 
  private:
   ModelService model_service_;