[feature] support ai api with zhipu ai

common/src/main/java/org/apache/hertzbeat/common/constants/AiConstants.java

@@ -0,0 +1,54 @@
+/*
+ * 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
+ *
+ *     http://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.hertzbeat.common.constants;
+
+/**
+ * AI constants
+ */
+public interface AiConstants {
+
+    /**
+     * zhiPu constants
+     */
+    interface ZhiPuConstants {
+
+        /**
+         * zhiPu request url
+         */
+        String URL = "https://open.bigmodel.cn/api/paas/v4/chat/completions";
+
+        /**
+         * request role param
+         */
+        String REQUEST_ROLE = "user";
+
+        /**
+         * The model outputs the maximum tokens, with a maximum output of 8192 and a default value of 1024
+         */
+        Integer MAX_TOKENS = 1024;
+
+        /**
+         * The sampling temperature, which controls the randomness of the output, must be positive
+         * The value ranges from 0.0 to 1.0, and cannot be equal to 0. The default value is 0.95. The larger the value,
+         * the more random and creative the output will be. The smaller the value, the more stable or certain the output will be
+         * You are advised to adjust top_p or temperature parameters based on application scenarios, but do not adjust the two parameters at the same time
+         */
+        double TEMPERATURE = 0.95;
+
+    }
+}

manager/pom.xml

@@ -106,6 +106,13 @@
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-thymeleaf</artifactId>
         </dependency>
+
+        <!--webflux-->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-webflux</artifactId>
+        </dependency>
+
         <!--   freemarker     -->
         <dependency>
             <groupId>org.springframework.boot</groupId>

manager/src/main/java/org/apache/hertzbeat/manager/controller/AiController.java

@@ -0,0 +1,64 @@
+/*
+ * 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
+ *
+ *     http://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.hertzbeat.manager.controller;
+
+import static org.springframework.http.MediaType.TEXT_EVENT_STREAM_VALUE;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import org.apache.hertzbeat.manager.service.AiService;
+import org.apache.hertzbeat.manager.service.impl.AiServiceFactoryImpl;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.codec.ServerSentEvent;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+import reactor.core.publisher.Flux;
+
+
+/**
+ * AI Management API
+ */
+@Tag(name = "AI Manage API")
+@RestController
+@RequestMapping(value = "/api/ai")
+public class AiController {
+
+    /**
+     * AI beanFactory
+     */
+    @Autowired
+    private AiServiceFactoryImpl aiServiceFactory;
+
+    @Value("${aiConfig.type:0}")
+    private String type;
+
+    /**
+     * request AI
+     * @param text                  request text
+     * @param currentlyDisabledType Currently disabled, later released
+     * @return                      AI response
+     */
+    @GetMapping(path = "/get", produces = {TEXT_EVENT_STREAM_VALUE})
+    public Flux<ServerSentEvent<String>> requestAi(@RequestParam("text") String text,
+                                                     @RequestParam(value = "type", required = false) String currentlyDisabledType) {
+        AiService aiServiceImplBean = aiServiceFactory.getAiServiceImplBean(type);
+
+        return aiServiceImplBean.requestAi(text);
+    }
+}

manager/src/main/java/org/apache/hertzbeat/manager/controller/MonitorsController.java

@@ -152,6 +152,7 @@ public ResponseEntity<Message<List<Monitor>>> getAppMonitors(
         return ResponseEntity.ok(message);
     }
 
+
     @DeleteMapping
     @Operation(summary = "Delete monitoring items in batches according to the monitoring ID list",
             description = "Delete monitoring items in batches according to the monitoring ID list")

manager/src/main/java/org/apache/hertzbeat/manager/pojo/dto/AiMessage.java

@@ -0,0 +1,41 @@
+/*
+ * 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
+ *
+ *     http://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.hertzbeat.manager.pojo.dto;
+
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+/**
+ * ai message
+ */
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+public class AiMessage {
+    /**
+     * role
+     */
+    private String role;
+
+    /**
+     * content
+     */
+    private String content;
+}

manager/src/main/java/org/apache/hertzbeat/manager/pojo/dto/ZhiPuAiResponse.java

@@ -0,0 +1,93 @@
+/*
+ * 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
+ *
+ *     http://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.hertzbeat.manager.pojo.dto;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import java.util.List;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+/**
+ * ZhiPuAiResponse
+ */
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+public class ZhiPuAiResponse {
+
+    /**
+     * Task order number generated by the AI open platform. Use this order number when invoking the request result interface
+     */
+    private String id;
+
+    /**
+     * The request creation time is a Unix timestamp in seconds
+     */
+    private Long created;
+
+    /**
+     * response message
+     */
+    private List<Choice> choices;
+
+    /**
+     * Returns the number of tokens invoked by the model at the end.
+     */
+    private Tokens usage;
+
+    /**
+     * Choice
+     */
+    @Data
+    @AllArgsConstructor
+    @NoArgsConstructor
+    public class Choice {
+        private int index;
+        private AiMessage delta;
+    }
+
+    /**
+     * Tokens
+     */
+    @Data
+    @AllArgsConstructor
+    @NoArgsConstructor
+    public class Tokens {
+
+        /**
+         * The number of tokens entered by users
+         */
+        @JsonProperty("prompt_tokens")
+        private Integer promptTokens;
+
+        /**
+         * The number of tokens that the model outputs
+         */
+        @JsonProperty("completion_tokens")
+        private Integer completionTokens;
+
+        /**
+         * Total number of tokens
+         */
+        @JsonProperty("total_tokens")
+        private Integer totalTokens;
+    }
+
+}
+

manager/src/main/java/org/apache/hertzbeat/manager/pojo/dto/ZhiPuRequestParamDTO.java

@@ -0,0 +1,70 @@
+/*
+ * 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
+ *
+ *     http://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.hertzbeat.manager.pojo.dto;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import java.util.List;
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+
+/**
+ * zhiPu Request param
+ */
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+@Builder
+public class ZhiPuRequestParamDTO {
+
+
+    /**
+     * ai version
+     */
+    private String model;
+
+    /**
+     * request message
+     */
+    private List<AiMessage> messages;
+
+    /**
+     * The sampling temperature, which controls the randomness of the output, must be positive
+     * The value ranges from 0.0 to 1.0, and cannot be equal to 0. The default value is 0.95.
+     * The larger the value, the more random and creative the output will be. The smaller the value, the more stable or certain the output will be
+     * You are advised to adjust top_p or temperature parameters based on application scenarios, but do not adjust the two parameters at the same time
+     */
+    private double temperature;
+
+    /**
+     * The model outputs the maximum tokens, with a maximum output of 8192 and a default value of 1024
+     */
+    @JsonProperty("max_tokens")
+    private Integer maxTokens;
+
+    /**
+     * stream response
+     */
+    private Boolean stream = Boolean.FALSE;
+
+}
+
+
+

manager/src/main/java/org/apache/hertzbeat/manager/service/AiService.java

@@ -0,0 +1,43 @@
+/*
+ * 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
+ *
+ *     http://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.hertzbeat.manager.service;
+
+
+import org.springframework.http.codec.ServerSentEvent;
+import reactor.core.publisher.Flux;
+
+
+/**
+ * AI Service
+ */
+public interface AiService {
+
+    /**
+     * get AI type
+     * @return     type
+     */
+    String getType();
+
+    /**
+     * AI response
+     * @param text  text
+     * @return      AI response
+     */
+    Flux<ServerSentEvent<String>> requestAi(String text);
+
+}