package com.ai.da.model.vo;

import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;

import java.util.List;

@Data
@Schema(description = "generate响应vo")
public class GenerateCollectionVO {

    @Schema(description = "generateId")
    private Long generateId;

    @Schema(description = "collection")
    private Long collectionId;

    @Schema(description = "生成的图片信息")
    private List<GenerateCollectionItemVO> generatedCollectionItems;

    @Schema(description = "在当前队列中的排序")
    private Long rankPosition;

    public GenerateCollectionVO(Long rankPosition) {
        this.rankPosition = rankPosition;
    }

    public GenerateCollectionVO(Long generateId, Long collectionId, List<GenerateCollectionItemVO> generatedCollectionItems) {
        this.generateId = generateId;
        this.collectionId = collectionId;
        this.generatedCollectionItems = generatedCollectionItems;
    }

    public GenerateCollectionVO() {
    }
}
