|
17 | 17 | #include "utils/cortex_utils.h" |
18 | 18 | #include "utils/logging_utils.h" |
19 | 19 |
|
20 | | -CommandLineParser::CommandLineParser() : app_("Cortex.cpp CLI") {} |
| 20 | +CommandLineParser::CommandLineParser() |
| 21 | + : app_("Cortex.cpp CLI"), engine_service_{EngineService()} {} |
21 | 22 |
|
22 | 23 | bool CommandLineParser::SetupCommand(int argc, char** argv) { |
23 | 24 | std::string model_id; |
@@ -127,9 +128,10 @@ bool CommandLineParser::SetupCommand(int argc, char** argv) { |
127 | 128 | command.Exec(); |
128 | 129 | }); |
129 | 130 |
|
130 | | - EngineManagement(engines_cmd, "cortex.llamacpp", version); |
131 | | - EngineManagement(engines_cmd, "cortex.onnx", version); |
132 | | - EngineManagement(engines_cmd, "cortex.tensorrt-llm", version); |
| 131 | + for (auto& engine : engine_service_.kSupportEngines) { |
| 132 | + std::string engine_name{engine}; |
| 133 | + EngineManagement(engines_cmd, engine_name, version); |
| 134 | + } |
133 | 135 |
|
134 | 136 | EngineGet(engines_cmd); |
135 | 137 | } |
@@ -186,9 +188,8 @@ void CommandLineParser::EngineManagement(CLI::App* parent, |
186 | 188 |
|
187 | 189 | void CommandLineParser::EngineGet(CLI::App* parent) { |
188 | 190 | auto get_cmd = parent->add_subcommand("get", "Get an engine info"); |
189 | | - auto engine_service = EngineService(); |
190 | 191 |
|
191 | | - for (auto& engine : engine_service.kSupportEngines) { |
| 192 | + for (auto& engine : engine_service_.kSupportEngines) { |
192 | 193 | std::string engine_name{engine}; |
193 | 194 | std::string desc = "Get " + engine_name + " status"; |
194 | 195 |
|
|
0 commit comments