From 7bd1bb74819d2c9d6d33a43d7e0075dc2fa63a92 Mon Sep 17 00:00:00 2001 From: sangjanai Date: Tue, 11 Mar 2025 09:56:07 +0700 Subject: [PATCH] fix: delete model for remote engine --- engine/services/model_service.cc | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/engine/services/model_service.cc b/engine/services/model_service.cc index 7ca86ac7c..cc735d006 100644 --- a/engine/services/model_service.cc +++ b/engine/services/model_service.cc @@ -757,12 +757,17 @@ cpp::result ModelService::DeleteModel( fs::path(model_entry.value().path_to_model_yaml)); yaml_handler.ModelConfigFromFile(yaml_fp.string()); auto mc = yaml_handler.GetModelConfig(); - // Remove yaml files - for (const auto& entry : - std::filesystem::directory_iterator(yaml_fp.parent_path())) { - if (entry.is_regular_file() && (entry.path().extension() == ".yml")) { - std::filesystem::remove(entry); - CTL_INF("Removed: " << entry.path().string()); + if (engine_svc_->IsRemoteEngine(mc.engine)) { + std::filesystem::remove(yaml_fp); + CTL_INF("Removed: " << yaml_fp.string()); + } else { + // Remove yaml files + for (const auto& entry : + std::filesystem::directory_iterator(yaml_fp.parent_path())) { + if (entry.is_regular_file() && (entry.path().extension() == ".yml")) { + std::filesystem::remove(entry); + CTL_INF("Removed: " << entry.path().string()); + } } }