replace with constants

kruize · Feb 13, 2025 · 81cc3f2 · 81cc3f2
1 parent 7654c5b
commit 81cc3f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/com/autotune/analyzer/recommendations/engine/RecommendationEngine.java b/src/main/java/com/autotune/analyzer/recommendations/engine/RecommendationEngine.java
@@ -149,11 +149,11 @@ private void loadCustomRecommendationModels(List<String> modelName) throws Inval
         // Add new models
         recommendationModels = new ArrayList<>();
         for (String model : modelName) {
-            if ("cost".equalsIgnoreCase(model)) {
+            if (KruizeConstants.JSONKeys.COST.equalsIgnoreCase(model)) {
                 // Create Cost based model
                 CostBasedRecommendationModel costBasedRecommendationModel = new CostBasedRecommendationModel();
                 registerModel(costBasedRecommendationModel);
-            } else if ("performance".equalsIgnoreCase(model)) {
+            } else if (KruizeConstants.JSONKeys.PERFORMANCE.equalsIgnoreCase(model)) {
                 // Create Performance based model
                 PerformanceBasedRecommendationModel performanceBasedRecommendationModel = new PerformanceBasedRecommendationModel();
                 registerModel(performanceBasedRecommendationModel);