fix: remove all yml files when delete model (#1980)

* fix: remove all yml files when delete model

* chore: add debug

---------

Co-authored-by: sangjanai <[email protected]>

Author: vansangpfiev

docker/entrypoint.sh

@@ -5,13 +5,15 @@
 echo "apiServerHost: 0.0.0.0" > /root/.cortexrc
 echo "enableCors: true" >> /root/.cortexrc
 
+# Start the cortex server
+cortex start
+
 # Install the engine
 cortex engines install llama-cpp -s /opt/cortex.llamacpp
 
-# Start the cortex server
-cortex start
 cortex engines list
 
+
 # Keep the container running by tailing the log files
 tail -f /root/cortexcpp/logs/cortex.log &
 tail -f /root/cortexcpp/logs/cortex-cli.log &

engine/services/model_service.cc

@@ -736,8 +736,15 @@ cpp::result<void, std::string> ModelService::DeleteModel(
         fs::path(model_entry.value().path_to_model_yaml));
     yaml_handler.ModelConfigFromFile(yaml_fp.string());
     auto mc = yaml_handler.GetModelConfig();
-    // Remove yaml file
-    std::filesystem::remove(yaml_fp);
+    // 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());
+      }
+    }
+
     // Remove model files if they are not imported locally
     if (model_entry.value().branch_name != "imported" &&
         !engine_svc_->IsRemoteEngine(mc.engine)) {