From 222a6fae1a8c0cc9906d5354baf8e01910add57e Mon Sep 17 00:00:00 2001
From: "github-actions[bot]" <github-actions[bot]@users.noreply.github.com>
Date: Fri, 29 Dec 2023 04:28:40 +0000
Subject: [PATCH] Add tools to ToolPlugin (#81)

Signed-off-by: Mingshi Liu <mingshl@amazon.com>
(cherry picked from commit 018cd039bb98480eed623ff59d77a69018d7dd1c)
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
---
 .../java/org/opensearch/agent/ToolPlugin.java  | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/src/main/java/org/opensearch/agent/ToolPlugin.java b/src/main/java/org/opensearch/agent/ToolPlugin.java
index a411acbb..e78b2550 100644
--- a/src/main/java/org/opensearch/agent/ToolPlugin.java
+++ b/src/main/java/org/opensearch/agent/ToolPlugin.java
@@ -12,7 +12,12 @@
 
 import org.opensearch.agent.tools.NeuralSparseSearchTool;
 import org.opensearch.agent.tools.PPLTool;
+import org.opensearch.agent.tools.RAGTool;
+import org.opensearch.agent.tools.SearchAlertsTool;
+import org.opensearch.agent.tools.SearchAnomalyDetectorsTool;
+import org.opensearch.agent.tools.SearchAnomalyResultsTool;
 import org.opensearch.agent.tools.SearchIndexTool;
+import org.opensearch.agent.tools.SearchMonitorsTool;
 import org.opensearch.agent.tools.VectorDBTool;
 import org.opensearch.agent.tools.VisualizationsTool;
 import org.opensearch.client.Client;
@@ -62,6 +67,11 @@ public Collection<Object> createComponents(
         NeuralSparseSearchTool.Factory.getInstance().init(client, xContentRegistry);
         VectorDBTool.Factory.getInstance().init(client, xContentRegistry);
         SearchIndexTool.Factory.getInstance().init(client, xContentRegistry);
+        RAGTool.Factory.getInstance().init(client, xContentRegistry);
+        SearchAlertsTool.Factory.getInstance().init(client);
+        SearchAnomalyDetectorsTool.Factory.getInstance().init(client);
+        SearchAnomalyResultsTool.Factory.getInstance().init(client);
+        SearchMonitorsTool.Factory.getInstance().init(client);
         return Collections.emptyList();
     }
 
@@ -72,7 +82,13 @@ public List<Tool.Factory<? extends Tool>> getToolFactories() {
                 PPLTool.Factory.getInstance(),
                 NeuralSparseSearchTool.Factory.getInstance(),
                 VectorDBTool.Factory.getInstance(),
-                VisualizationsTool.Factory.getInstance()
+                VisualizationsTool.Factory.getInstance(),
+                SearchIndexTool.Factory.getInstance(),
+                RAGTool.Factory.getInstance(),
+                SearchAlertsTool.Factory.getInstance(),
+                SearchAnomalyDetectorsTool.Factory.getInstance(),
+                SearchAnomalyResultsTool.Factory.getInstance(),
+                SearchMonitorsTool.Factory.getInstance()
             );
     }
 }