diff --git a/src/main/java/com/ai/da/model/dto/QueryLibraryPageDTO.java b/src/main/java/com/ai/da/model/dto/QueryLibraryPageDTO.java
index 3a1bc632..eb957138 100644
--- a/src/main/java/com/ai/da/model/dto/QueryLibraryPageDTO.java
+++ b/src/main/java/com/ai/da/model/dto/QueryLibraryPageDTO.java
@@ -6,6 +6,7 @@ import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
 
 import javax.validation.constraints.NotBlank;
+import java.util.List;
 
 @Data
 @ApiModel("Library分页查询")
@@ -24,5 +25,5 @@ public class QueryLibraryPageDTO extends PageQueryBaseVo {
     private String pictureName;
 
     @ApiModelProperty("分类ID")
-    private Long classificationId;
+    private List<Long> classificationIdList;
 }