Check before delete

common/src/main/java/org/opensearch/ml/common/CommonValue.java

@@ -81,6 +81,7 @@ public class CommonValue {
     public static final String ML_STOP_WORDS_INDEX = ".plugins-ml-stop-words";
     public static final Set<String> stopWordsIndices = ImmutableSet.of(".plugins-ml-stop-words");
     public static final Integer ML_MEMORY_MESSAGE_INDEX_SCHEMA_VERSION = 1;
+    public static final String TOOL_MODEL_RELATED_FIELD_PREFIX = "tools.parameters.";
     public static final String USER_FIELD_MAPPING = "      \""
         + CommonValue.USER
         + "\": {\n"

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/AgentModelsSearcher.java

@@ -0,0 +1,37 @@
+package org.opensearch.ml.engine.tools;
+
+import static org.opensearch.ml.common.CommonValue.ML_AGENT_INDEX;
+import static org.opensearch.ml.common.CommonValue.TOOL_MODEL_RELATED_FIELD_PREFIX;
+
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
+
+import org.opensearch.action.search.SearchRequest;
+import org.opensearch.index.query.BoolQueryBuilder;
+import org.opensearch.index.query.QueryBuilders;
+import org.opensearch.ml.common.spi.tools.Tool;
+import org.opensearch.search.builder.SearchSourceBuilder;
+
+public class AgentModelsSearcher {
+    private final Set<String> relatedModelIdSet;
+
+    public AgentModelsSearcher(Map<String, Tool.Factory> toolFactories) {
+        relatedModelIdSet = new HashSet<>();
+        for (Map.Entry<String, Tool.Factory> entry : toolFactories.entrySet()) {
+            Tool.Factory toolFactory = entry.getValue();
+            relatedModelIdSet.addAll(toolFactory.getAllModelKeys());
+        }
+    }
+
+    public SearchRequest constructQueryRequest(String candidateModelId) {
+        SearchRequest searchRequest = new SearchRequest(ML_AGENT_INDEX);
+        BoolQueryBuilder shouldQuery = QueryBuilders.boolQuery();
+        for (String keyField : relatedModelIdSet) {
+            shouldQuery.should(QueryBuilders.termsQuery(TOOL_MODEL_RELATED_FIELD_PREFIX + keyField, candidateModelId));
+        }
+        searchRequest.source(new SearchSourceBuilder().query(shouldQuery));
+        return searchRequest;
+    }
+
+}

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/AgentTool.java

@@ -8,6 +8,7 @@
 import static org.opensearch.ml.common.utils.StringUtils.gson;
 
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 
 import org.opensearch.action.ActionRequest;
@@ -138,6 +139,11 @@ public String getDefaultType() {
         public String getDefaultVersion() {
             return null;
         }
+
+        @Override
+        public List<String> getAllModelKeys() {
+            return List.of();
+        }
     }
 
     private Map<String, String> extractInputParameters(Map<String, String> parameters) {

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/CatIndexTool.java

@@ -360,6 +360,11 @@ public String getDefaultType() {
         public String getDefaultVersion() {
             return null;
         }
+
+        @Override
+        public List<String> getAllModelKeys() {
+            return List.of();
+        }
     }
 
     private Table getTableWithHeader() {

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/ConnectorTool.java

@@ -144,5 +144,10 @@ public String getDefaultType() {
         public String getDefaultVersion() {
             return null;
         }
+
+        @Override
+        public List<String> getAllModelKeys() {
+            return List.of();
+        }
     }
 }

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/IndexMappingTool.java

@@ -206,5 +206,10 @@ public String getDefaultType() {
         public String getDefaultVersion() {
             return null;
         }
+
+        @Override
+        public List<String> getAllModelKeys() {
+            return List.of();
+        }
     }
 }

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/MLModelTool.java

@@ -172,5 +172,10 @@ public String getDefaultType() {
         public String getDefaultVersion() {
             return null;
         }
+
+        @Override
+        public List<String> getAllModelKeys() {
+            return List.of(MODEL_ID_FIELD);
+        }
     }
 }

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/SearchIndexTool.java

@@ -11,6 +11,7 @@
 import java.security.AccessController;
 import java.security.PrivilegedExceptionAction;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 import java.util.Objects;
 import java.util.Optional;
@@ -195,5 +196,10 @@ public String getDefaultType() {
         public String getDefaultVersion() {
             return null;
         }
+
+        @Override
+        public List<String> getAllModelKeys() {
+            return List.of();
+        }
     }
 }

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/VisualizationsTool.java

@@ -6,6 +6,7 @@
 package org.opensearch.ml.engine.tools;
 
 import java.util.Arrays;
+import java.util.List;
 import java.util.Locale;
 import java.util.Map;
 import java.util.Optional;
@@ -174,5 +175,10 @@ public String getDefaultType() {
         public String getDefaultVersion() {
             return null;
         }
+
+        @Override
+        public List<String> getAllModelKeys() {
+            return List.of();
+        }
     }
 }