feat: add model health check functionality and improve model configuration (#30)

* refactor: rename artifactId and application name to 'datamate'; add model configuration and related services

* refactor: simplify package scanning by using wildcard for mapper packages

* feat: add model health check functionality and improve model configuration

Author: Dallas98

backend/pom.xml

@@ -207,6 +207,19 @@
                 <artifactId>spring-boot-maven-plugin</artifactId>
                 <version>${spring-boot.version}</version>
             </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <version>3.11.0</version>
+                <configuration>
+                    <source>${maven.compiler.source}</source>
+                    <target>${maven.compiler.target}</target>
+                    <parameters>true</parameters>
+                    <compilerArgs>
+                        <arg>-parameters</arg>
+                    </compilerArgs>
+                </configuration>
+            </plugin>
         </plugins>
     </build>
 </project>

backend/shared/domain-common/src/main/java/com/datamate/common/models/domain/entity/ModelConfig.java

@@ -5,6 +5,7 @@
 import lombok.Builder;
 import lombok.Getter;
 import lombok.Setter;
+import lombok.ToString;
 
 /**
  * 模型配置实体类
@@ -16,6 +17,7 @@
 @Setter
 @TableName("t_model_config")
 @Builder
+@ToString
 public class ModelConfig extends BaseEntity<String> {
     /**
      * 模型名称（如 qwen2）

backend/shared/domain-common/src/main/java/com/datamate/common/models/infrastructure/client/ModelClient.java

@@ -1,20 +1,21 @@
 package com.datamate.common.models.infrastructure.client;
 
+import com.datamate.common.infrastructure.exception.BusinessException;
 import com.datamate.common.models.domain.entity.ModelConfig;
-import com.datamate.common.models.domain.entity.ModelType;
+import com.datamate.common.models.infrastructure.exception.ModelsErrorCode;
 import dev.langchain4j.model.chat.ChatModel;
 import dev.langchain4j.model.embedding.EmbeddingModel;
 import dev.langchain4j.model.openai.OpenAiChatModel;
 import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
-
-import java.util.function.Consumer;
+import lombok.extern.slf4j.Slf4j;
 
 /**
  * 模型客户端接口
  *
  * @author dallas
  * @since 2025-10-27
  */
+@Slf4j
 public class ModelClient {
     public static <T> T invokeModel(ModelConfig modelConfig, Class<T> modelInterface) {
         return switch (modelConfig.getType()) {
@@ -23,15 +24,15 @@ public static <T> T invokeModel(ModelConfig modelConfig, Class<T> modelInterface
         };
     }
 
-    private static EmbeddingModel invokeEmbeddingModel(ModelConfig modelConfig) {
+    public static EmbeddingModel invokeEmbeddingModel(ModelConfig modelConfig) {
         return OpenAiEmbeddingModel.builder()
                 .baseUrl(modelConfig.getBaseUrl())
                 .apiKey(modelConfig.getApiKey())
                 .modelName(modelConfig.getModelName())
                 .build();
     }
 
-    private static ChatModel invokeChatModel(ModelConfig modelConfig) {
+    public static ChatModel invokeChatModel(ModelConfig modelConfig) {
         return OpenAiChatModel.builder()
                 .baseUrl(modelConfig.getBaseUrl())
                 .apiKey(modelConfig.getApiKey())
@@ -40,5 +41,24 @@ private static ChatModel invokeChatModel(ModelConfig modelConfig) {
     }
 
     public static void checkHealth(ModelConfig modelConfig) {
+        try {
+            switch (modelConfig.getType()) {
+                case CHAT -> checkChatModelHealth(modelConfig);
+                case EMBEDDING -> checkEmbeddingModelHealth(modelConfig);
+            }
+        } catch (Exception e) {
+            log.error("Model health check failed for modelConfig: {}", modelConfig, e);
+            throw BusinessException.of(ModelsErrorCode.MODEL_HEALTH_CHECK_FAILED);
+        }
+    }
+
+    private static void checkEmbeddingModelHealth(ModelConfig modelConfig) {
+        EmbeddingModel embeddingModel = invokeEmbeddingModel(modelConfig);
+        embeddingModel.embed("text");
+    }
+
+    private static void checkChatModelHealth(ModelConfig modelConfig) {
+        ChatModel chatModel = invokeChatModel(modelConfig);
+        chatModel.chat("hello");
     }
 }

backend/shared/domain-common/src/main/java/com/datamate/common/models/infrastructure/exception/ModelsErrorCode.java

@@ -20,7 +20,12 @@ public enum ModelsErrorCode implements ErrorCode {
     /**
      * 模型配置已存在
      */
-    MODEL_CONFIG_ALREADY_EXISTS("model.0002", "模型配置已存在");
+    MODEL_CONFIG_ALREADY_EXISTS("model.0002", "模型配置已存在"),
+
+    /**
+     * 模型健康检查失败
+     */
+    MODEL_HEALTH_CHECK_FAILED("model.0003", "模型健康检查失败");
 
     private final String code;
     private final String message;

backend/shared/domain-common/src/main/java/com/datamate/common/models/interfaces/rest/ModelConfigController.java

@@ -19,7 +19,7 @@
  * @since 2025-10-27
  */
 @RestController
-@RequestMapping("/api/models")
+@RequestMapping("/models")
 @RequiredArgsConstructor
 public class ModelConfigController {
     private final ModelConfigApplicationService modelConfigApplicationService;
@@ -40,7 +40,7 @@ public List<ModelConfig> getProviders() {
      * @return 模型列表
      */
     @GetMapping("/list")
-    public PagedResponse<ModelConfig> getModels(@RequestParam QueryModelRequest queryModelRequest) {
+    public PagedResponse<ModelConfig> getModels(QueryModelRequest queryModelRequest) {
         return modelConfigApplicationService.getModels(queryModelRequest);
     }
 
@@ -51,7 +51,7 @@ public PagedResponse<ModelConfig> getModels(@RequestParam QueryModelRequest quer
      * @return 模型详情
      */
     @GetMapping("/{modelId}")
-    public ModelConfig getModelDetail(@PathVariable String modelId) {
+    public ModelConfig getModelDetail(@PathVariable("modelId") String modelId) {
         return modelConfigApplicationService.getModelDetail(modelId);
     }
 

scripts/db/model-management-init.sql

@@ -1,6 +1,8 @@
+USE datamate;
+
 CREATE TABLE t_model_config
 (
-    id         VARCHAR(36) AUTO_INCREMENT PRIMARY KEY COMMENT '主键ID',
+    id         VARCHAR(36)  PRIMARY KEY COMMENT '主键ID',
     model_name VARCHAR(100) NOT NULL COMMENT '模型名称（如 qwen2）',
     provider   VARCHAR(50)  NOT NULL COMMENT '模型提供商（如 Ollama、OpenAI、DeepSeek）',
     base_url   VARCHAR(255) NOT NULL COMMENT 'API 基础地址',