BIGTOP-4308: Add some info tools to chatbot

bigtop-manager-common/src/main/java/org/apache/bigtop/manager/common/utils/JsonUtils.java

@@ -23,6 +23,7 @@
 import com.fasterxml.jackson.databind.DeserializationFeature;
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.SerializationFeature;
 import lombok.extern.slf4j.Slf4j;
 
 import java.io.File;
@@ -31,11 +32,15 @@
 public class JsonUtils {
 
     public static final ObjectMapper OBJECTMAPPER;
+    public static final ObjectMapper INDENT_MAPPER;
 
     static {
         OBJECTMAPPER = new ObjectMapper();
         OBJECTMAPPER.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
         OBJECTMAPPER.setSerializationInclusion(JsonInclude.Include.NON_NULL);
+
+        INDENT_MAPPER = OBJECTMAPPER.copy();
+        INDENT_MAPPER.enable(SerializationFeature.INDENT_OUTPUT);
     }
 
     public static <T> void writeToFile(String fileName, T obj) {
@@ -129,4 +134,15 @@ public static <T> String writeAsString(T obj) {
             throw new RuntimeException(e);
         }
     }
+
+    public static <T> String indentWriteAsString(T obj) {
+        if (obj == null) {
+            return null;
+        }
+        try {
+            return INDENT_MAPPER.writeValueAsString(obj);
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
 }

bigtop-manager-dao/src/main/resources/mapper/postgresql/HostMapper.xml

@@ -58,7 +58,7 @@
                 and h.status = #{query.status}
             </if>
         </where>
-        group by h.id, c.name
+        group by h.id, c.name, c.display_name
     </select>
 
     <select id="findDetailsById" resultType="org.apache.bigtop.manager.dao.po.HostPO">
@@ -70,7 +70,7 @@
         left join cluster c on h.cluster_id = c.id
         left join component comp on h.id = comp.host_id
         where h.id = #{id}
-        group by h.id, c.name
+        group by h.id, c.name, c.display_name
         limit 1
     </select>
 

bigtop-manager-server/src/main/java/org/apache/bigtop/manager/server/service/impl/ChatbotServiceImpl.java

@@ -47,7 +47,7 @@
 import org.apache.bigtop.manager.server.model.vo.ChatThreadVO;
 import org.apache.bigtop.manager.server.model.vo.TalkVO;
 import org.apache.bigtop.manager.server.service.ChatbotService;
-import org.apache.bigtop.manager.server.tools.AiServiceToolsProvider;
+import org.apache.bigtop.manager.server.tools.provider.AiServiceToolsProvider;
 
 import org.springframework.context.i18n.LocaleContextHolder;
 import org.springframework.stereotype.Service;
@@ -77,6 +77,9 @@ public class ChatbotServiceImpl implements ChatbotService {
     @Resource
     private ChatMessageDao chatMessageDao;
 
+    @Resource
+    private AiServiceToolsProvider aiServiceToolsProvider;
+
     private AIAssistantFactory aiAssistantFactory;
 
     private static final int CHAT_THREAD_NAME_LENGTH = 100;
@@ -131,7 +134,7 @@ private AIAssistant buildAIAssistant(
                             getPlatformType(platformName),
                             getAIAssistantConfig(model, credentials),
                             threadId,
-                            new AiServiceToolsProvider(command));
+                            aiServiceToolsProvider.getToolsProvide(command));
         }
     }
 

bigtop-manager-server/src/main/java/org/apache/bigtop/manager/server/tools/functions/ClusterFunctions.java

@@ -0,0 +1,101 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.bigtop.manager.server.tools.functions;
+
+import org.apache.bigtop.manager.common.utils.JsonUtils;
+import org.apache.bigtop.manager.server.model.vo.ClusterVO;
+import org.apache.bigtop.manager.server.service.ClusterService;
+
+import org.springframework.stereotype.Component;
+
+import dev.langchain4j.agent.tool.JsonSchemaProperty;
+import dev.langchain4j.agent.tool.ToolSpecification;
+import dev.langchain4j.service.tool.ToolExecutor;
+import lombok.extern.slf4j.Slf4j;
+
+import jakarta.annotation.Resource;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+@Component
+@Slf4j
+public class ClusterFunctions {
+    @Resource
+    private ClusterService clusterService;
+
+    public Map<ToolSpecification, ToolExecutor> listCluster() {
+        ToolSpecification toolSpecification = ToolSpecification.builder()
+                .name("listCluster")
+                .description("Get cluster list")
+                .build();
+        ToolExecutor toolExecutor =
+                (toolExecutionRequest, memoryId) -> JsonUtils.indentWriteAsString(clusterService.list());
+
+        return Map.of(toolSpecification, toolExecutor);
+    }
+
+    public Map<ToolSpecification, ToolExecutor> getClusterById() {
+        ToolSpecification toolSpecification = ToolSpecification.builder()
+                .name("getClusterById")
+                .description("Get cluster information based on ID")
+                .addParameter("clusterId", JsonSchemaProperty.NUMBER, JsonSchemaProperty.description("cluster id"))
+                .build();
+        ToolExecutor toolExecutor = (toolExecutionRequest, memoryId) -> {
+            Map<String, Object> arguments = JsonUtils.readFromString(toolExecutionRequest.arguments());
+            Long clusterId = Long.valueOf(arguments.get("clusterId").toString());
+            ClusterVO clusterVO = clusterService.get(clusterId);
+            if (clusterVO == null) {
+                return "Cluster not found";
+            }
+            return JsonUtils.indentWriteAsString(clusterVO);
+        };
+
+        return Map.of(toolSpecification, toolExecutor);
+    }
+
+    public Map<ToolSpecification, ToolExecutor> getClusterByName() {
+        ToolSpecification toolSpecification = ToolSpecification.builder()
+                .name("getClusterByName")
+                .description("Get cluster information based on cluster name")
+                .addParameter("clusterName", JsonSchemaProperty.STRING, JsonSchemaProperty.description("cluster name"))
+                .build();
+        ToolExecutor toolExecutor = (toolExecutionRequest, memoryId) -> {
+            Map<String, Object> arguments = JsonUtils.readFromString(toolExecutionRequest.arguments());
+            String clusterName = arguments.get("clusterName").toString();
+            List<ClusterVO> clusterVOS = clusterService.list();
+            for (ClusterVO clusterVO : clusterVOS) {
+                if (clusterVO.getName().equals(clusterName)) {
+                    return JsonUtils.indentWriteAsString(clusterVO);
+                }
+            }
+            return "Cluster not found";
+        };
+
+        return Map.of(toolSpecification, toolExecutor);
+    }
+
+    public Map<ToolSpecification, ToolExecutor> getAllFunctions() {
+        Map<ToolSpecification, ToolExecutor> functions = new HashMap<>();
+        functions.putAll(listCluster());
+        functions.putAll(getClusterById());
+        functions.putAll(getClusterByName());
+        return functions;
+    }
+}

bigtop-manager-server/src/main/java/org/apache/bigtop/manager/server/tools/functions/HostFunctions.java

@@ -0,0 +1,87 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *    https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.bigtop.manager.server.tools.functions;
+
+import org.apache.bigtop.manager.common.utils.JsonUtils;
+import org.apache.bigtop.manager.dao.query.HostQuery;
+import org.apache.bigtop.manager.server.model.vo.HostVO;
+import org.apache.bigtop.manager.server.model.vo.PageVO;
+import org.apache.bigtop.manager.server.service.HostService;
+
+import org.springframework.stereotype.Component;
+
+import dev.langchain4j.agent.tool.JsonSchemaProperty;
+import dev.langchain4j.agent.tool.ToolSpecification;
+import dev.langchain4j.service.tool.ToolExecutor;
+import lombok.extern.slf4j.Slf4j;
+
+import jakarta.annotation.Resource;
+import java.util.HashMap;
+import java.util.Map;
+
+@Component
+@Slf4j
+public class HostFunctions {
+    @Resource
+    private HostService hostService;
+
+    public Map<ToolSpecification, ToolExecutor> getHostById() {
+        ToolSpecification toolSpecification = ToolSpecification.builder()
+                .name("getHostById")
+                .description("Get host information based on ID")
+                .addParameter("hostId", JsonSchemaProperty.NUMBER, JsonSchemaProperty.description("host id"))
+                .build();
+        ToolExecutor toolExecutor = (toolExecutionRequest, memoryId) -> {
+            Map<String, Object> arguments = JsonUtils.readFromString(toolExecutionRequest.arguments());
+            Long hostId = Long.valueOf(arguments.get("hostId").toString());
+            HostVO hostVO = hostService.get(hostId);
+            if (hostVO == null) {
+                return "Host not found";
+            }
+            return JsonUtils.indentWriteAsString(hostVO);
+        };
+
+        return Map.of(toolSpecification, toolExecutor);
+    }
+
+    public Map<ToolSpecification, ToolExecutor> getHostByName() {
+        ToolSpecification toolSpecification = ToolSpecification.builder()
+                .name("getHostByName")
+                .description("Get host information based on cluster name")
+                .addParameter("hostName", JsonSchemaProperty.STRING, JsonSchemaProperty.description("host name"))
+                .build();
+        ToolExecutor toolExecutor = (toolExecutionRequest, memoryId) -> {
+            Map<String, Object> arguments = JsonUtils.readFromString(toolExecutionRequest.arguments());
+            String hostName = arguments.get("hostName").toString();
+            HostQuery hostQuery = new HostQuery();
+            hostQuery.setHostname(hostName);
+            PageVO<HostVO> hostVO = hostService.list(hostQuery);
+            return JsonUtils.indentWriteAsString(hostVO);
+        };
+
+        return Map.of(toolSpecification, toolExecutor);
+    }
+
+    public Map<ToolSpecification, ToolExecutor> getAllFunctions() {
+        Map<ToolSpecification, ToolExecutor> functions = new HashMap<>();
+        functions.putAll(getHostById());
+        functions.putAll(getHostByName());
+        return functions;
+    }
+}