diff --git a/ruoyi-generator/src/main/resources/vm/java/domain.java.vm b/ruoyi-generator/src/main/resources/vm/java/domain.java.vm
index 32b86fc2..2ee5c777 100644
--- a/ruoyi-generator/src/main/resources/vm/java/domain.java.vm
+++ b/ruoyi-generator/src/main/resources/vm/java/domain.java.vm
@@ -3,8 +3,9 @@ package ${packageName}.domain;
 #foreach ($import in $importList)
 import ${import};
 #end
-import com.baomidou.mybatisplus.annotation.TableField;
 import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 import com.ruoyi.common.annotation.Excel;
@@ -26,6 +27,7 @@ import com.ruoyi.common.core.domain.TreeEntity;
 #set($Entity="TreeEntity")
 #end
 @Data
+@ApiModel(value = "${functionName}")
 @EqualsAndHashCode(callSuper = true)
 @TableName(value = "${tableName}")
 public class ${ClassName} extends ${Entity}
@@ -49,6 +51,7 @@ public class ${ClassName} extends ${Entity}
     @Excel(name = "${comment}", width = 30, dateFormat = "yyyy-MM-dd")
 #else
     @Excel(name = "${comment}")
+    @ApiModelProperty(value = "${comment}")
 #end
 #end
     private $column.javaType $column.javaField;